prefect-client 3.3.6.dev2__py3-none-any.whl → 3.3.6.dev3__py3-none-any.whl

prefect/_build_info.py

@@ -1,5 +1,5 @@
 # Generated by versioningit
-__version__ = "3.3.6.dev2"
-__build_date__ = "2025-04-22 08:08:25.558469+00:00"
-__git_commit__ = "562b964535b46f586b9b12808462956ed2f90c5a"
+__version__ = "3.3.6.dev3"
+__build_date__ = "2025-04-23 08:08:36.059139+00:00"
+__git_commit__ = "093237bfa0681dfcb1baeaee9b96da9803b6a879"
 __dirty__ = False

prefect/_versioning.py

@@ -0,0 +1,175 @@
+from __future__ import annotations
+
+import os
+from enum import Enum
+from typing import Any, Callable, Coroutine, Dict, Literal, Optional
+
+from anyio import run_process
+from pydantic import Field, model_validator
+
+from prefect.client.schemas.objects import VersionInfo
+
+
+class SimpleVersionInfo(VersionInfo):
+    type: Literal["prefect:simple"] = "prefect:simple"
+    version: str = Field(default="")
+    branch: Optional[str] = Field(default=None)
+    url: Optional[str] = Field(default=None)
+
+
+class GithubVersionInfo(VersionInfo):
+    type: Literal["vcs:github"] = "vcs:github"
+    version: str
+    branch: str
+    url: str
+    repository: str
+
+    @model_validator(mode="after")
+    def validate_branch(self):
+        if not self.branch:
+            raise ValueError("branch is required when type is 'vcs:github'")
+        return self
+
+
+class GitVersionInfo(VersionInfo):
+    type: Literal["vcs:git"] = "vcs:git"
+    version: str
+    branch: str
+    url: str
+    repository: str
+
+
+async def get_github_version_info(
+    version: Optional[str] = None,
+    branch: Optional[str] = None,
+    repository: Optional[str] = None,
+    url: Optional[str] = None,
+) -> GithubVersionInfo:
+    """Create a GithubVersionInfo object from provided values or environment variables.
+
+    Args:
+        version: The commit SHA, falls back to GITHUB_SHA env var
+        branch: The git branch, falls back to GITHUB_REF env var
+        repository: The repository name, falls back to GITHUB_REPOSITORY env var
+        url: The repository URL, constructed from GITHUB_SERVER_URL/GITHUB_REPOSITORY if not provided
+
+    Returns:
+        A GithubVersionInfo
+
+    Raises:
+        ValueError: If any required fields cannot be determined
+    """
+    version = version or os.getenv("GITHUB_SHA")
+    branch = branch or os.getenv("GITHUB_REF")
+    repository = repository or os.getenv("GITHUB_REPOSITORY")
+    url = url or f"{os.getenv('GITHUB_SERVER_URL')}/{repository}"
+
+    if not version:
+        raise ValueError("version is required - must be provided or set in GITHUB_SHA")
+    if not branch:
+        raise ValueError("branch is required - must be provided or set in GITHUB_REF")
+    if not repository:
+        raise ValueError(
+            "repository is required - must be provided or set in GITHUB_REPOSITORY"
+        )
+
+    return GithubVersionInfo(
+        type="vcs:github",
+        version=version,
+        branch=branch,
+        repository=repository,
+        url=url,
+    )
+
+
+async def get_git_version_info(
+    version: Optional[str] = None,
+    branch: Optional[str] = None,
+    url: Optional[str] = None,
+    repository: Optional[str] = None,
+) -> GitVersionInfo:
+    try:
+        if not version:
+            # Run git command and get stdout
+            result = await run_process(["git", "rev-parse", "HEAD"])
+            # Decode bytes to string and strip whitespace
+            version = result.stdout.decode().strip()
+
+        if not branch:
+            result = await run_process(["git", "rev-parse", "--abbrev-ref", "HEAD"])
+            branch = result.stdout.decode().strip()
+
+        if not repository:
+            result = await run_process(["git", "config", "--get", "remote.origin.url"])
+            remote_url = result.stdout.decode().strip()
+
+            # Extract just the repository name (last part of the path)
+            repo_name = os.path.basename(remote_url.split(":")[1].rstrip(".git"))
+            repository = repo_name
+
+        if not url and repository:
+            # Use the full remote URL as the URL
+            result = await run_process(["git", "config", "--get", "remote.origin.url"])
+            url = result.stdout.decode().strip()
+    except Exception as e:
+        raise ValueError(
+            f"Error getting git version info: {e}. You may not be in a git repository."
+        )
+
+    if not url:
+        raise ValueError("Could not determine git repository URL")
+
+    return GitVersionInfo(
+        type="vcs:git", version=version, branch=branch, url=url, repository=repository
+    )
+
+
+class VersionType(str, Enum):
+    SIMPLE = "prefect:simple"
+    GITHUB = "vcs:github"
+    GIT = "vcs:git"
+    DOCKER = "container:docker"
+
+
+async def get_inferred_version_info(
+    version_type: Optional[str] = None,
+) -> VersionInfo | None:
+    """
+    Attempts to infer version information from the environment.
+
+    Args:
+        version_type: Optional type of version info to get. If provided, only that
+                     type will be attempted.
+
+    Returns:
+        VersionInfo: The inferred version information
+
+    Raises:
+        ValueError: If unable to infer version info from any source
+    """
+    # Map version types to their getter functions
+    type_to_getter: Dict[str, Callable[..., Coroutine[Any, Any, Any]]] = {
+        VersionType.GITHUB: get_github_version_info,
+        VersionType.GIT: get_git_version_info,
+    }
+
+    # Default order of getters to try
+    default_getters = [
+        get_github_version_info,
+        get_git_version_info,
+    ]
+
+    if version_type:
+        if version_type not in type_to_getter:
+            raise ValueError(f"Unknown version type: {version_type}")
+        getters = [type_to_getter[version_type]]
+    else:
+        getters = default_getters
+
+    for getter in getters:
+        try:
+            return await getter()
+        except ValueError:
+            continue
+
+    return None

prefect/client/orchestration/_deployments/client.py

@@ -153,13 +153,14 @@ class DeploymentClient(BaseClient):
             if getattr(deployment_create, field) is None:
                 exclude.add(field)
 
-        json = deployment_create.model_dump(mode="json", exclude=exclude)
+        payload = deployment_create.model_dump(mode="json", exclude=exclude)
+        if deployment_create.version_info:
+            payload["version_info"] = deployment_create.version_info.model_dump(
+                mode="json"
+            )
+
+        response = self.request("POST", "/deployments/", json=payload)
 
-        response = self.request(
-            "POST",
-            "/deployments/",
-            json=json,
-        )
         deployment_id = response.json().get("id")
         if not deployment_id:
             raise RequestError(f"Malformed response: {response}")
@@ -179,15 +180,28 @@ class DeploymentClient(BaseClient):
         deployment_id: UUID,
         deployment: "DeploymentUpdate",
     ) -> None:
+        exclude_if_none = [
+            "version_info",
+        ]
+
+        exclude = {"name", "flow_name", "triggers"}
+        for field in exclude_if_none:
+            if getattr(deployment, field) is None:
+                exclude.add(field)
+
+        payload = deployment.model_dump(
+            mode="json",
+            exclude_unset=True,
+            exclude=exclude,
+        )
+        if deployment.version_info:
+            payload["version_info"] = deployment.version_info.model_dump(mode="json")
+
         self.request(
             "PATCH",
             "/deployments/{id}",
             path_params={"id": deployment_id},
-            json=deployment.model_dump(
-                mode="json",
-                exclude_unset=True,
-                exclude={"name", "flow_name", "triggers"},
-            ),
+            json=payload,
         )
 
     def _create_deployment_from_schema(self, schema: "DeploymentCreate") -> UUID:
@@ -733,13 +747,13 @@ class DeploymentAsyncClient(BaseAsyncClient):
             if getattr(deployment_create, field) is None:
                 exclude.add(field)
 
-        json = deployment_create.model_dump(mode="json", exclude=exclude)
+        payload = deployment_create.model_dump(mode="json", exclude=exclude)
+        if deployment_create.version_info:
+            payload["version_info"] = deployment_create.version_info.model_dump(
+                mode="json"
+            )
 
-        response = await self.request(
-            "POST",
-            "/deployments/",
-            json=json,
-        )
+        response = await self.request("POST", "/deployments/", json=payload)
         deployment_id = response.json().get("id")
         if not deployment_id:
             raise RequestError(f"Malformed response: {response}")
@@ -761,15 +775,28 @@ class DeploymentAsyncClient(BaseAsyncClient):
         deployment_id: UUID,
         deployment: "DeploymentUpdate",
     ) -> None:
+        exclude_if_none = [
+            "version_info",
+        ]
+
+        exclude = {"name", "flow_name", "triggers"}
+        for field in exclude_if_none:
+            if getattr(deployment, field) is None:
+                exclude.add(field)
+
+        payload = deployment.model_dump(
+            mode="json",
+            exclude_unset=True,
+            exclude=exclude,
+        )
+        if deployment.version_info:
+            payload["version_info"] = deployment.version_info.model_dump(mode="json")
+
         await self.request(
             "PATCH",
             "/deployments/{id}",
             path_params={"id": deployment_id},
-            json=deployment.model_dump(
-                mode="json",
-                exclude_unset=True,
-                exclude={"name", "flow_name", "triggers"},
-            ),
+            json=payload,
         )
 
     async def _create_deployment_from_schema(self, schema: "DeploymentCreate") -> UUID:

prefect/deployments/runner.py

@@ -29,6 +29,8 @@ Example:
 
 """
 
+from __future__ import annotations
+
 import importlib
 import tempfile
 from datetime import datetime, timedelta
@@ -63,6 +65,7 @@ from prefect.client.schemas.filters import WorkerFilter, WorkerFilterStatus
 from prefect.client.schemas.objects import (
     ConcurrencyLimitConfig,
     ConcurrencyOptions,
+    VersionInfo,
 )
 from prefect.client.schemas.schedules import (
     SCHEDULE_TYPES,
@@ -281,7 +284,10 @@ class RunnerDeployment(BaseModel):
         return reconcile_schedules_runner(values)
 
     async def _create(
-        self, work_pool_name: Optional[str] = None, image: Optional[str] = None
+        self,
+        work_pool_name: Optional[str] = None,
+        image: Optional[str] = None,
+        version_info: VersionInfo | None = None,
     ) -> UUID:
         work_pool_name = work_pool_name or self.work_pool_name
 
@@ -312,6 +318,7 @@ class RunnerDeployment(BaseModel):
                 work_queue_name=self.work_queue_name,
                 work_pool_name=work_pool_name,
                 version=self.version,
+                version_info=version_info,
                 paused=self.paused,
                 schedules=self.schedules,
                 concurrency_limit=self.concurrency_limit,
@@ -367,7 +374,12 @@ class RunnerDeployment(BaseModel):
 
             return deployment_id
 
-    async def _update(self, deployment_id: UUID, client: PrefectClient):
+    async def _update(
+        self,
+        deployment_id: UUID,
+        client: PrefectClient,
+        version_info: VersionInfo | None,
+    ):
         parameter_openapi_schema = self._parameter_openapi_schema.model_dump(
             exclude_unset=True
         )
@@ -388,6 +400,7 @@ class RunnerDeployment(BaseModel):
             deployment_id,
             deployment=DeploymentUpdate(
                 **update_payload,
+                version_info=version_info,
                 parameter_openapi_schema=parameter_openapi_schema,
             ),
         )
@@ -428,7 +441,10 @@ class RunnerDeployment(BaseModel):
 
     @sync_compatible
     async def apply(
-        self, work_pool_name: Optional[str] = None, image: Optional[str] = None
+        self,
+        work_pool_name: Optional[str] = None,
+        image: Optional[str] = None,
+        version_info: VersionInfo | None = None,
     ) -> UUID:
         """
         Registers this deployment with the API and returns the deployment's ID.
@@ -439,7 +455,7 @@ class RunnerDeployment(BaseModel):
             image: The registry, name, and tag of the Docker image to
                 use for this deployment. Only used when the deployment is
                 deployed to a work pool.
-
+            version_info: Version information for the deployment.
         Returns:
             The ID of the created deployment.
         """
@@ -448,13 +464,13 @@ class RunnerDeployment(BaseModel):
             try:
                 deployment = await client.read_deployment_by_name(self.full_name)
             except ObjectNotFound:
-                return await self._create(work_pool_name, image)
+                return await self._create(work_pool_name, image, version_info)
             else:
                 if image:
                     self.job_variables["image"] = image
                 if work_pool_name:
                     self.work_pool_name = work_pool_name
-                return await self._update(deployment.id, client)
+                return await self._update(deployment.id, client, version_info)
 
     async def _create_slas(self, deployment_id: UUID, client: PrefectClient):
         if not isinstance(self._sla, list):

prefect/workers/base.py

@@ -720,6 +720,15 @@ class BaseWorker(abc.ABC, Generic[C, V, R]):
         if self._runs_task_group is None:
             raise RuntimeError("Worker not properly initialized")
 
+        from prefect.results import get_result_store
+
+        current_result_store = get_result_store()
+        if current_result_store.result_storage is None and flow.result_storage is None:
+            self._logger.warning(
+                f"Flow {flow.name!r} has no result storage configured. Please configure "
+                "result storage for the flow if you want to retrieve the result for the flow run."
+            )
+
         flow_run = await self._runs_task_group.start(
             partial(
                 self._submit_adhoc_run,

{prefect_client-3.3.6.dev2.dist-info → prefect_client-3.3.6.dev3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.3.6.dev2
+Version: 3.3.6.dev3
 Summary: Workflow orchestration and management.
 Project-URL: Changelog, https://github.com/PrefectHQ/prefect/releases
 Project-URL: Documentation, https://docs.prefect.io

{prefect_client-3.3.6.dev2.dist-info → prefect_client-3.3.6.dev3.dist-info}/RECORD

@@ -1,8 +1,9 @@
 prefect/.prefectignore,sha256=awSprvKT0vI8a64mEOLrMxhxqcO-b0ERQeYpA2rNKVQ,390
 prefect/__init__.py,sha256=iCdcC5ZmeewikCdnPEP6YBAjPNV5dvfxpYCTpw30Hkw,3685
 prefect/__main__.py,sha256=WFjw3kaYJY6pOTA7WDOgqjsz8zUEUZHCcj3P5wyVa-g,66
-prefect/_build_info.py,sha256=EDorkrJUYtVWxr5Aoxi9ooVoj5KtlRaQZ-bkFwmKNr4,185
+prefect/_build_info.py,sha256=F2YAuWqa_cJqxuGkhaAz5fIexdkNGcMj0zohhfmWD4s,185
 prefect/_result_records.py,sha256=S6QmsODkehGVSzbMm6ig022PYbI6gNKz671p_8kBYx4,7789
+prefect/_versioning.py,sha256=4zp4Dl9dJWsoItj4AAhRxYtP3CMdo-7nG0dyv3Xz4nU,5361
 prefect/_waiters.py,sha256=Ia2ITaXdHzevtyWIgJoOg95lrEXQqNEOquHvw3T33UQ,9026
 prefect/agent.py,sha256=dPvG1jDGD5HSH7aM2utwtk6RaJ9qg13XjkA0lAIgQmY,287
 prefect/artifacts.py,sha256=dMBUOAWnUamzjb5HSqwB5-GR2Qb-Gxee26XG5NDCUuw,22720
@@ -98,7 +99,7 @@ prefect/client/orchestration/_blocks_types/client.py,sha256=alA4xD-yp3mycAbzMyRu
 prefect/client/orchestration/_concurrency_limits/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/client/orchestration/_concurrency_limits/client.py,sha256=r_oyY7hQbgyG1rntwe7WWcsraQHBKhk6MOPFUAHWiVc,23678
 prefect/client/orchestration/_deployments/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prefect/client/orchestration/_deployments/client.py,sha256=I2-RhU15lbEI1_JX_-WJErfp3lkvTCJHja-p4y8hzLA,42923
+prefect/client/orchestration/_deployments/client.py,sha256=a_sUmznQC3lBLxXfu4hIIbszbUc28sWXtCioQoNgZPk,43876
 prefect/client/orchestration/_flow_runs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/client/orchestration/_flow_runs/client.py,sha256=fjh5J-LG8tsny7BGYEvynbuGuHDAudYHpx-PamL0GYQ,32220
 prefect/client/orchestration/_flows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -136,7 +137,7 @@ prefect/deployments/__init__.py,sha256=_wb7NxDKhq11z9MjYsPckmT3o6MRhGLRgCV9TmvYt
 prefect/deployments/base.py,sha256=YY7g8MN6qzjNEjEA8wQXPxCrd47WnACIUeSRtI4nrEk,11849
 prefect/deployments/deployments.py,sha256=K3Rgnpjxo_T8I8LMwlq24OKqZiZBTE8-YnPg-YGUStM,171
 prefect/deployments/flow_runs.py,sha256=NYe-Bphsy6ENLqSSfywQuX5cRZt-uVgzqGmOsf3Sqw4,7643
-prefect/deployments/runner.py,sha256=QNKdKsLEzx3TE7T_3dJcK7bGvTCvNLY2WCpgDxiQDno,54368
+prefect/deployments/runner.py,sha256=_VqbkXvPVvdyFVkRsr5emi26cJmu5-2uhtVUoE0EkNA,54805
 prefect/deployments/schedules.py,sha256=2eL1-w8qXtwKVkgfUK7cuamwpKK3X6tN1QYTDa_gWxU,2190
 prefect/deployments/steps/__init__.py,sha256=Dlz9VqMRyG1Gal8dj8vfGpPr0LyQhZdvcciozkK8WoY,206
 prefect/deployments/steps/core.py,sha256=ulSgBFSx1lhBt1fP-UxebrernkumBDlympR6IPffV1g,6900
@@ -311,13 +312,13 @@ prefect/utilities/schema_tools/__init__.py,sha256=At3rMHd2g_Em2P3_dFQlFgqR_EpBwr
 prefect/utilities/schema_tools/hydration.py,sha256=NkRhWkNfxxFmVGhNDfmxdK_xeKaEhs3a42q83Sg9cT4,9436
 prefect/utilities/schema_tools/validation.py,sha256=Wix26IVR-ZJ32-6MX2pHhrwm3reB-Q4iB6_phn85OKE,10743
 prefect/workers/__init__.py,sha256=EaM1F0RZ-XIJaGeTKLsXDnfOPHzVWk5bk0_c4BVS44M,64
-prefect/workers/base.py,sha256=l7ghNeSJh6DXiDdsnLmQor2-b82YaZJIz5TpFSn2dFY,59171
+prefect/workers/base.py,sha256=TpGj8laGPQoCS_OiE__oSaNWOgN1REUzwedy-G0E8yQ,59601
 prefect/workers/block.py,sha256=dPvG1jDGD5HSH7aM2utwtk6RaJ9qg13XjkA0lAIgQmY,287
 prefect/workers/cloud.py,sha256=dPvG1jDGD5HSH7aM2utwtk6RaJ9qg13XjkA0lAIgQmY,287
 prefect/workers/process.py,sha256=Yi5D0U5AQ51wHT86GdwtImXSefe0gJf3LGq4r4z9zwM,11090
 prefect/workers/server.py,sha256=2pmVeJZiVbEK02SO6BEZaBIvHMsn6G8LzjW8BXyiTtk,1952
 prefect/workers/utilities.py,sha256=VfPfAlGtTuDj0-Kb8WlMgAuOfgXCdrGAnKMapPSBrwc,2483
-prefect_client-3.3.6.dev2.dist-info/METADATA,sha256=3OCxGaU3XixlWqbKkkhN9eW5vxz22tq5NWzfp19Ctgo,7471
-prefect_client-3.3.6.dev2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-prefect_client-3.3.6.dev2.dist-info/licenses/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
-prefect_client-3.3.6.dev2.dist-info/RECORD,,
+prefect_client-3.3.6.dev3.dist-info/METADATA,sha256=BTogX-QjhTZkFVfTFe00RyeaKFsJdyDChxWSZxjgfCQ,7471
+prefect_client-3.3.6.dev3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+prefect_client-3.3.6.dev3.dist-info/licenses/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
+prefect_client-3.3.6.dev3.dist-info/RECORD,,