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

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

@@ -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 → ai_pipeline_core-0.4.2}/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-0.4.0 → 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.0 → ai_pipeline_core-0.4.2}/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-0.4.0 → ai_pipeline_core-0.4.2}/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 → ai_pipeline_core-0.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ai-pipeline-core"
-version = "0.4.0"
+version = "0.4.2"
 description = "Core utilities for AI-powered processing pipelines using prefect"
 readme = "README.md"
 license = {text = "MIT"}
@@ -228,7 +228,7 @@ reportUnusedCoroutine = "error"
 reportImplicitStringConcatenation = false
 
 [tool.bumpversion]
-current_version = "0.4.0"
+current_version = "0.4.1"
 commit = true
 tag = true
 tag_name = "v{new_version}"