fix: pytest errors
This commit is contained in:
parent
996cb4a775
commit
53e4573332
2 changed files with 127 additions and 3 deletions
|
|
@ -439,6 +439,13 @@ class MethodAi(MethodBase):
|
||||||
default="",
|
default="",
|
||||||
description="Additional context from upstream steps.",
|
description="Additional context from upstream steps.",
|
||||||
),
|
),
|
||||||
|
"folderId": WorkflowActionParameter(
|
||||||
|
name="folderId",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.USER_FILE_FOLDER,
|
||||||
|
required=False,
|
||||||
|
description="Target folder in My Files when persisting workflow output",
|
||||||
|
),
|
||||||
},
|
},
|
||||||
execute=generateCode.__get__(self, self.__class__)
|
execute=generateCode.__get__(self, self.__class__)
|
||||||
),
|
),
|
||||||
|
|
|
||||||
|
|
@ -71,6 +71,123 @@ class MethodContext(MethodBase):
|
||||||
required=True,
|
required=True,
|
||||||
description="Document reference(s) to extract content from",
|
description="Document reference(s) to extract content from",
|
||||||
),
|
),
|
||||||
|
"contentFilter": WorkflowActionParameter(
|
||||||
|
name="contentFilter",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["all", "textOnly", "imagesOnly", "noImages"],
|
||||||
|
required=False,
|
||||||
|
default="all",
|
||||||
|
description="Which extracted parts to keep (text, images, etc.).",
|
||||||
|
),
|
||||||
|
"outputMode": WorkflowActionParameter(
|
||||||
|
name="outputMode",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["blob", "lines", "pages", "chunks", "structured"],
|
||||||
|
required=False,
|
||||||
|
default="lines",
|
||||||
|
description="How results are structured under presentation files.",
|
||||||
|
),
|
||||||
|
"splitBy": WorkflowActionParameter(
|
||||||
|
name="splitBy",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["newline", "paragraph", "sentence"],
|
||||||
|
required=False,
|
||||||
|
default="newline",
|
||||||
|
description="Delimiter for lines/chunks output modes.",
|
||||||
|
),
|
||||||
|
"chunkSizeUnit": WorkflowActionParameter(
|
||||||
|
name="chunkSizeUnit",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["tokens", "characters", "words"],
|
||||||
|
required=False,
|
||||||
|
default="tokens",
|
||||||
|
description="Unit for chunkSize and chunkOverlap when outputMode is chunks.",
|
||||||
|
),
|
||||||
|
"chunkSize": WorkflowActionParameter(
|
||||||
|
name="chunkSize",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["256", "500", "1000", "2000", "4000"],
|
||||||
|
required=False,
|
||||||
|
default="500",
|
||||||
|
description="Target chunk size when outputMode is chunks.",
|
||||||
|
),
|
||||||
|
"chunkOverlap": WorkflowActionParameter(
|
||||||
|
name="chunkOverlap",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["0", "25", "50", "100", "200"],
|
||||||
|
required=False,
|
||||||
|
default="0",
|
||||||
|
description="Overlap between consecutive chunks.",
|
||||||
|
),
|
||||||
|
"filterEmptyLines": WorkflowActionParameter(
|
||||||
|
name="filterEmptyLines",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["true", "false"],
|
||||||
|
required=False,
|
||||||
|
default="true",
|
||||||
|
description="Remove empty segments for lines/chunks modes.",
|
||||||
|
),
|
||||||
|
"trimWhitespace": WorkflowActionParameter(
|
||||||
|
name="trimWhitespace",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["true", "false"],
|
||||||
|
required=False,
|
||||||
|
default="true",
|
||||||
|
description="Trim leading/trailing whitespace per segment.",
|
||||||
|
),
|
||||||
|
"includeLineNumbers": WorkflowActionParameter(
|
||||||
|
name="includeLineNumbers",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["true", "false"],
|
||||||
|
required=False,
|
||||||
|
default="false",
|
||||||
|
description="Prefix line numbers when outputMode is lines.",
|
||||||
|
),
|
||||||
|
"includeMetadata": WorkflowActionParameter(
|
||||||
|
name="includeMetadata",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["true", "false"],
|
||||||
|
required=False,
|
||||||
|
default="false",
|
||||||
|
description="Attach filename and offsets to line/chunk entries.",
|
||||||
|
),
|
||||||
|
"csvHeaderRow": WorkflowActionParameter(
|
||||||
|
name="csvHeaderRow",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["true", "false"],
|
||||||
|
required=False,
|
||||||
|
default="true",
|
||||||
|
description="Treat first CSV row as column headers.",
|
||||||
|
),
|
||||||
|
"pdfExtractMode": WorkflowActionParameter(
|
||||||
|
name="pdfExtractMode",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["text", "tables", "images", "all"],
|
||||||
|
required=False,
|
||||||
|
default="all",
|
||||||
|
description="Presentation-layer filter by part type group.",
|
||||||
|
),
|
||||||
|
"markdownPreserveFormatting": WorkflowActionParameter(
|
||||||
|
name="markdownPreserveFormatting",
|
||||||
|
type="str",
|
||||||
|
frontendType=FrontendType.SELECT,
|
||||||
|
frontendOptions=["true", "false"],
|
||||||
|
required=False,
|
||||||
|
default="false",
|
||||||
|
description="Keep markdown markup on text parts when false strips lightly.",
|
||||||
|
),
|
||||||
},
|
},
|
||||||
execute=extractContent.__get__(self, self.__class__)
|
execute=extractContent.__get__(self, self.__class__)
|
||||||
),
|
),
|
||||||
|
|
@ -134,7 +251,7 @@ class MethodContext(MethodBase):
|
||||||
description="Storage scope for keys written by this node",
|
description="Storage scope for keys written by this node",
|
||||||
),
|
),
|
||||||
"assignments": WorkflowActionParameter(
|
"assignments": WorkflowActionParameter(
|
||||||
name="assignments", type="list", required=True,
|
name="assignments", type="List[Any]", required=True,
|
||||||
frontendType=FrontendType.CONTEXT_ASSIGNMENTS,
|
frontendType=FrontendType.CONTEXT_ASSIGNMENTS,
|
||||||
default=[],
|
default=[],
|
||||||
description=(
|
description=(
|
||||||
|
|
@ -183,7 +300,7 @@ class MethodContext(MethodBase):
|
||||||
description="allow = only these keys pass; block = these keys are removed",
|
description="allow = only these keys pass; block = these keys are removed",
|
||||||
),
|
),
|
||||||
"keys": WorkflowActionParameter(
|
"keys": WorkflowActionParameter(
|
||||||
name="keys", type="list", required=True,
|
name="keys", type="List[str]", required=True,
|
||||||
frontendType=FrontendType.JSON,
|
frontendType=FrontendType.JSON,
|
||||||
default=[],
|
default=[],
|
||||||
description="Key paths or glob patterns",
|
description="Key paths or glob patterns",
|
||||||
|
|
@ -214,7 +331,7 @@ class MethodContext(MethodBase):
|
||||||
outputType="Transit",
|
outputType="Transit",
|
||||||
parameters={
|
parameters={
|
||||||
"mappings": WorkflowActionParameter(
|
"mappings": WorkflowActionParameter(
|
||||||
name="mappings", type="list", required=True,
|
name="mappings", type="List[Any]", required=True,
|
||||||
frontendType=FrontendType.MAPPING_TABLE,
|
frontendType=FrontendType.MAPPING_TABLE,
|
||||||
default=[],
|
default=[],
|
||||||
description="List of mapping entries",
|
description="List of mapping entries",
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue