fix: pytest errors

This commit is contained in:
Ida 2026-05-20 17:45:39 +02:00
parent 996cb4a775
commit 53e4573332
2 changed files with 127 additions and 3 deletions

View file

@ -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__)
), ),

View file

@ -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",