ai-pipeline-core 0.4.0__py3-none-any.whl → 0.4.2__py3-none-any.whl

ai_pipeline_core/__init__.py

@@ -64,7 +64,7 @@ from .prompt_manager import PromptManager
 from .settings import Settings
 from .testing import disable_run_logger, prefect_test_harness
 
-__version__ = "0.4.0"
+__version__ = "0.4.1"
 
 __all__ = [
     "AIMessageType",

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/deployment/deploy.py

@@ -348,11 +348,12 @@ class Deployer:
 
             self._success(f"Agent '{agent_name}' uploaded ({len(build_info['files'])} files)")
 
-    async def _upload_package(self, tarball: Path):
-        """Upload package tarball to Google Cloud Storage.
+    async def _upload_package(self, tarball: Path, vendor_wheels: list[Path] | None = None):
+        """Upload package tarball and vendor wheels to Google Cloud Storage.
 
         Args:
             tarball: Path to the tarball to upload
+            vendor_wheels: Optional private dependency wheels to upload alongside
         """
         flow_folder = self.config["folder"]
         bucket = self._create_gcs_bucket(flow_folder)
@@ -365,6 +366,10 @@ class Deployer:
 
         self._success(f"Package uploaded to {flow_folder}/{tarball.name}")
 
+        for wheel in vendor_wheels or []:
+            await bucket.write_path(wheel.name, wheel.read_bytes())
+            self._success(f"Vendor wheel uploaded: {wheel.name}")
+
     async def _deploy_via_api(self, agent_builds: dict[str, dict[str, Any]] | None = None):
         """Create or update Prefect deployment using RunnerDeployment pattern.
 
@@ -420,7 +425,8 @@ class Deployer:
                     "stream_output": True,
                     "directory": "{{ pull_code.directory }}",
                     # Use uv for fast installation (worker has it installed)
-                    "script": f"uv pip install --system ./{self.config['tarball']}",
+                    # --find-links . resolves private dependencies from co-uploaded wheels
+                    "script": f"uv pip install --system --find-links . ./{self.config['tarball']}",
                 }
             },
         ]
@@ -488,8 +494,16 @@ class Deployer:
         # Phase 2: Build agent bundles (if configured)
         agent_builds = self._build_agents()
 
-        # Phase 3: Upload flow package
-        await self._upload_package(tarball)
+        # Phase 3: Upload flow package (include private dependency wheels from agent builds)
+        vendor_wheels: list[Path] = []
+        if agent_builds:
+            seen: set[str] = set()
+            for build_info in agent_builds.values():
+                for filename, filepath in build_info["files"].items():
+                    if filename.endswith(".whl") and filename not in seen and "cli_agents" in filename:
+                        vendor_wheels.append(filepath)
+                        seen.add(filename)
+        await self._upload_package(tarball, vendor_wheels)
 
         # Phase 4: Upload agent bundles
         await self._upload_agents(agent_builds)

ai_pipeline_core/pipeline/options.py

@@ -10,7 +10,7 @@ class FlowOptions(BaseSettings):
     for environment variable overrides. Immutable after creation.
     """
 
-    model_config = SettingsConfigDict(frozen=True, extra="ignore")
+    model_config = SettingsConfigDict(frozen=True, extra="allow")
 
 
 __all__ = ["FlowOptions"]

{ai_pipeline_core-0.4.0.dist-info → ai_pipeline_core-0.4.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-pipeline-core
-Version: 0.4.0
+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.0.dist-info → ai_pipeline_core-0.4.2.dist-info}/RECORD

@@ -1,13 +1,13 @@
-ai_pipeline_core/__init__.py,sha256=_GM0O3dDQuCvXQD44dyXKvzLOZk2htwwFS0mXpvxJQU,3270
+ai_pipeline_core/__init__.py,sha256=REZQInD3-LSShbonlXFpbe-IfjEtr90kkEx3JFxgfMA,3270
 ai_pipeline_core/exceptions.py,sha256=csAl7vq6xjSFBF8-UM9WZODCbhsOdOG5zH6IbA8iteM,1280
 ai_pipeline_core/prompt_manager.py,sha256=3wFkL5rrjtUT1cLInkgyhS8hKnO4MeD1cdXAEuLhgoE,9459
 ai_pipeline_core/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_pipeline_core/settings.py,sha256=BUz8JEFfJQrdE4rNOhQWwxnTrfekLjWkoy-3wDZQ7PY,5142
 ai_pipeline_core/testing.py,sha256=jIRrLxNvTwdamucfJoHET2qMeRhhMZV9uEJXO5vAfis,279
 ai_pipeline_core/deployment/__init__.py,sha256=wTkVK6gcEQvqBajFMTAuodRONpN25yHbR1jtcumf0WQ,900
-ai_pipeline_core/deployment/base.py,sha256=ros0VzvkNCwbPgG9D49ceKSjTq857iRHzvW-uhiaNUE,34750
+ai_pipeline_core/deployment/base.py,sha256=ZdXrO2tt2H5bTRkxV0EA1_v6mvuCxfYMUlmi4bpsq8c,34780
 ai_pipeline_core/deployment/contract.py,sha256=a1qbHhneTGB27oSOUy79CUIhOIzOoq37M63XoIMzA4Y,1952
-ai_pipeline_core/deployment/deploy.py,sha256=89W0w22cRkK_yMrn9iuH5L4dvnlMu31eojhJIKHtK2E,21991
+ai_pipeline_core/deployment/deploy.py,sha256=TCF4fH5f-K1ADODZHEyf-x7PJzDbv4qtWxlpoCe_mTs,22909
 ai_pipeline_core/deployment/helpers.py,sha256=yVtGFUs4AFXkpLkiQ_ale0nXXt5btfWSb5PAbikQHNs,3312
 ai_pipeline_core/deployment/progress.py,sha256=5tVD9nW0N-b8Z2BxazcWCWHFpLu6pJ-eqPmRyj68X6Y,3591
 ai_pipeline_core/deployment/remote.py,sha256=tOexisKEeeBoHLGYZWqcjr2H-nqqYc6kvoDL72AW78w,4661
@@ -69,8 +69,8 @@ ai_pipeline_core/observability/_tracking/_service.py,sha256=diK-0qJg4HU-BHgpN1NT
 ai_pipeline_core/observability/_tracking/_writer.py,sha256=xZjwYyIxDzzzPxqkKjYAYOyNP4uvKXZ-r_u-APSV_x8,9246
 ai_pipeline_core/pipeline/__init__.py,sha256=uMv1jwSyq8Ym8Hbn5097twBJLdwN1iMeqnVM4EWyrhA,282
 ai_pipeline_core/pipeline/decorators.py,sha256=CDJAeOjGLt5Ewc0Jc9zEuwLZwKyutOv89LSRS9dcXmI,37456
-ai_pipeline_core/pipeline/options.py,sha256=Y--5-DxzxR1Ul4GltGPP3JUIk8bw1GlUdZ3IDX8UIHQ,439
-ai_pipeline_core-0.4.0.dist-info/METADATA,sha256=hH5B9XsY_NT4bCB1W-LvBEG5LYR7SyaXdt4Z75PWvEE,29947
-ai_pipeline_core-0.4.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-ai_pipeline_core-0.4.0.dist-info/licenses/LICENSE,sha256=kKj8mfbdWwkyG3U6n7ztB3bAZlEwShTkAsvaY657i3I,1074
-ai_pipeline_core-0.4.0.dist-info/RECORD,,
+ai_pipeline_core/pipeline/options.py,sha256=KF4FcT085-IwX8r649v0a9ua5xnApM0qG2wJHWbq39A,438
+ai_pipeline_core-0.4.2.dist-info/METADATA,sha256=9oibNB9_K3sQ2K3rwPeV77tZ-t4kt0Kxrnhlhp6ySjY,29947
+ai_pipeline_core-0.4.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+ai_pipeline_core-0.4.2.dist-info/licenses/LICENSE,sha256=kKj8mfbdWwkyG3U6n7ztB3bAZlEwShTkAsvaY657i3I,1074
+ai_pipeline_core-0.4.2.dist-info/RECORD,,