ai-pipeline-core 0.4.1__tar.gz → 0.4.2__tar.gz

{ai_pipeline_core-0.4.1 → ai_pipeline_core-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-pipeline-core
-Version: 0.4.1
+Version: 0.4.2
 Summary: Core utilities for AI-powered processing pipelines using prefect
 Project-URL: Homepage, https://github.com/bbarwik/ai-pipeline-core
 Project-URL: Repository, https://github.com/bbarwik/ai-pipeline-core

{ai_pipeline_core-0.4.1 → ai_pipeline_core-0.4.2}/ai_pipeline_core/deployment/base.py

@@ -132,7 +132,7 @@ class FlowCallable(Protocol):
     output_document_types: list[type[Document]]
     estimated_minutes: int
 
-    def __call__(self, project_name: str, documents: list[Document], flow_options: FlowOptions) -> Any:  # type: ignore[type-arg]
+    def __call__(self, project_name: str, documents: list[Document], flow_options: FlowOptions | dict[str, Any]) -> Any:  # type: ignore[type-arg]
         """Execute the flow with standard pipeline signature."""
         ...
 
@@ -519,7 +519,7 @@ class PipelineDeployment(Generic[TOptions, TResult]):
                     current_docs = input_docs
 
                 try:
-                    await active_flow(project_name, current_docs, options)
+                    await active_flow(project_name, current_docs, options.model_dump())
                 except Exception as e:
                     # Upload partial results on failure
                     if context.output_documents_urls and store:

{ai_pipeline_core-0.4.1 → ai_pipeline_core-0.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ai-pipeline-core"
-version = "0.4.1"
+version = "0.4.2"
 description = "Core utilities for AI-powered processing pipelines using prefect"
 readme = "README.md"
 license = {text = "MIT"}