PyPI - truefoundry - Versions diffs - 0.7.2__py3-none-any.whl → 0.9.0rc1__py3-none-any.whl - Mend

truefoundry 0.7.2py3-none-any.whl → 0.9.0rc1py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of truefoundry might be problematic. Click here for more details.

Files changed (13) hide show

truefoundry/_client.py CHANGED Viewed

@@ -33,4 +33,4 @@ class _LazyTrueFoundry(truefoundry_sdk.TrueFoundry):
         )
-client = _LazyTrueFoundry().v1
+client = _LazyTrueFoundry()

truefoundry/cli/util.py CHANGED Viewed

@@ -51,6 +51,9 @@ def handle_exception(exception):
             title="Command Failed",
             border_style="red",
         )
+    elif isinstance(exception, KeyError):
+        # KeyError messages in Python just suck - they tell you the key that was missing, but not the context
+        console.print_exception(show_locals=False, max_frames=1)
     else:
         print_dict_as_table_panel(
             {"Error": str(exception)},

truefoundry/common/entities.py CHANGED Viewed

@@ -109,6 +109,9 @@ class PythonSDKConfig(BaseModel):
     use_sfy_server_auth_apis: Optional[bool] = Field(
         alias="useSFYServerAuthAPIs", default=False
     )
+    python_build_default_image_tag: str = Field(
+        alias="pythonBuildDefaultImageTag", default="3.11"
+    )
 class DeviceCode(BaseModel):

truefoundry/deploy/_autogen/models.py CHANGED Viewed

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2025-05-03T01:45:46+00:00
+#   timestamp: 2025-05-15T05:28:11+00:00
 from __future__ import annotations
@@ -360,13 +360,6 @@ class Image(BaseModel):
     )
-class JobEvent(str, Enum):
-    START = "START"
-    SUCCEEDED = "SUCCEEDED"
-    FAILED = "FAILED"
-    TERMINATED = "TERMINATED"
 class Claim(BaseModel):
     key: str
     values: List[str]
@@ -623,9 +616,11 @@ class PythonBuild(BaseModel):
     """
     type: Literal["tfy-python-buildpack"] = Field(..., description="")
-    python_version: constr(regex=r"^\d+(\.\d+){1,2}([\-\.a-z0-9]+)?$") = Field(
-        "3.11",
-        description="Python version to run your application. Should be one of the tags listed on [Official Python Docker Page](https://hub.docker.com/_/python)",
+    python_version: Optional[constr(regex=r"^\d+(\.\d+){1,2}([\-\.a-z0-9]+)?$")] = (
+        Field(
+            None,
+            description="Python version to run your application. Should be one of the tags listed on [Official Python Docker Page](https://hub.docker.com/_/python)",
+        )
     )
     build_context_path: str = Field(
         "./", description="Build path relative to project root path."
@@ -963,9 +958,11 @@ class TaskPythonBuild(BaseModel):
         None,
         description="FQN of the container registry. If you can't find your registry here,\nadd it through the [Integrations](/integrations?tab=docker-registry) page",
     )
-    python_version: constr(regex=r"^\d+(\.\d+){1,2}([\-\.a-z0-9]+)?$") = Field(
-        "3.11",
-        description="Python version to run your application. Should be one of the tags listed on [Official Python Docker Page](https://hub.docker.com/_/python)",
+    python_version: Optional[constr(regex=r"^\d+(\.\d+){1,2}([\-\.a-z0-9]+)?$")] = (
+        Field(
+            None,
+            description="Python version to run your application. Should be one of the tags listed on [Official Python Docker Page](https://hub.docker.com/_/python)",
+        )
     )
     requirements_path: Optional[str] = Field(
         None,
@@ -1060,13 +1057,6 @@ class WorkbenchImage(BaseModel):
     )
-class WorkflowEvent(str, Enum):
-    SUCCEEDED = "SUCCEEDED"
-    FAILED = "FAILED"
-    ABORTED = "ABORTED"
-    TIMED_OUT = "TIMED_OUT"
 class ArtifactsDownload(BaseModel):
     """
     Download and cache models in a volume to enhance loading speeds and reduce costs by avoiding repeated downloads. [Docs](https://docs.truefoundry.com/docs/download-and-cache-models)
@@ -1485,10 +1475,6 @@ class WorkflowAlert(BaseModel):
     """
     notification_target: Optional[Union[Email, SlackWebhook, SlackBot]] = None
-    events: List[WorkflowEvent] = Field(
-        ...,
-        description="Specify the events to send alerts for, it should be one of the following: SUCCEEDED, FAILED, ABORTED, TIMED_OUT",
-    )
     on_completion: bool = Field(
         False, description="Send an alert when the job completes"
     )
@@ -1556,10 +1542,6 @@ class JobAlert(BaseModel):
         description="List of recipients' email addresses if the notification channel is Email.",
     )
     notification_target: Optional[Union[Email, SlackWebhook, SlackBot]] = None
-    events: List[JobEvent] = Field(
-        ...,
-        description="Specify the events to send alerts for, it should be one of the following: START, SUCCEEDED, FAILED, TERMINATED",
-    )
     on_start: bool = Field(False, description="Send an alert when the job starts")
     on_completion: bool = False
     on_failure: bool = Field(True, description="Send an alert when the job fails")

truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py CHANGED Viewed

@@ -36,6 +36,10 @@ def build(
     build_configuration: PythonBuild,
     extra_opts: Optional[List[str]] = None,
 ):
+    if not build_configuration.python_version:
+        raise ValueError(
+            "`python_version` is required for `tfy-python-buildpack` builder"
+        )
     mount_python_package_manager_conf_secret = (
         has_python_package_manager_conf_secret(extra_opts) if extra_opts else False
     )

truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py CHANGED Viewed

@@ -174,6 +174,10 @@ def generate_dockerfile_content(
     requirements_destination_path = (
         "/tmp/requirements.txt" if requirements_path else None
     )
+    if not build_configuration.python_version:
+        raise ValueError(
+            "`python_version` is required for `tfy-python-buildpack` builder"
+        )
     if package_manager == PythonPackageManager.PIP.value:
         python_packages_install_command = generate_pip_install_command(
             requirements_path=requirements_destination_path,

truefoundry/deploy/lib/clients/ask_client.py CHANGED Viewed

@@ -1,10 +1,7 @@
 try:
     from mcp import ClientSession
+    from mcp.client.streamable_http import streamablehttp_client
     from mcp.types import TextContent
-    from truefoundry.deploy.lib.clients._mcp_streamable_http import (
-        streamablehttp_client,
-    )
 except ImportError:
     import sys
@@ -75,7 +72,7 @@ class AskClient:
             (
                 read_stream,
                 write_stream,
-                self._terminate_cb,
+                _,
             ) = await self._streams_context.__aenter__()
             self._session_context = ClientSession(
                 read_stream=read_stream, write_stream=write_stream
@@ -99,7 +96,6 @@ class AskClient:
     async def cleanup(self):
         """Properly close all async contexts opened during session initialization."""
-        await self._terminate_cb()
         for context in [
             getattr(self, "_session_context", None),
             getattr(self, "_streams_context", None),

truefoundry/deploy/v2/lib/deploy.py CHANGED Viewed

@@ -25,6 +25,19 @@ from truefoundry.pydantic_v1 import BaseModel
 Component = TypeVar("Component", bound=BaseModel)
+def _set_python_version_if_missing(build_spec: autogen_models.PythonBuild) -> None:
+    if build_spec.python_version is None:
+        client = ServiceFoundryServiceClient()
+        server_default_python_version = (
+            client.python_sdk_config.python_build_default_image_tag
+        )
+        logger.warning(
+            f"No python version was provided in the spec, "
+            f"using the default python version ({server_default_python_version}) from the server."
+        )
+        build_spec.python_version = server_default_python_version
 def _handle_if_local_source(component: Component, workspace_fqn: str) -> Component:
     if (
         hasattr(component, "image")
@@ -33,7 +46,7 @@ def _handle_if_local_source(component: Component, workspace_fqn: str) -> Compone
     ):
         new_component = component.copy(deep=True)
-        if component.image.build_source.local_build:
+        if new_component.image.build_source.local_build:
             if not env_has_docker():
                 logger.warning(
                     "Did not find Docker locally installed on this system, image will be built remotely. "
@@ -64,25 +77,33 @@ def _handle_if_local_source(component: Component, workspace_fqn: str) -> Compone
             local_build = False
         if local_build:
+            if isinstance(new_component.image.build_spec, autogen_models.PythonBuild):
+                _set_python_version_if_missing(new_component.image.build_spec)
             # We are to build the image locally, push and update `image` in spec
-            logger.info("Building image for %s '%s'", component.type, component.name)
+            logger.info(
+                "Building image for %s '%s'", new_component.type, new_component.name
+            )
             new_component.image = local_source_to_image(
-                build=component.image,
-                docker_registry_fqn=component.image.docker_registry,
+                build=new_component.image,
+                docker_registry_fqn=new_component.image.docker_registry,
                 workspace_fqn=workspace_fqn,
-                component_name=component.name,
+                component_name=new_component.name,
             )
         else:
             # We'll build image on TrueFoundry servers, upload the source and update image.build_source
-            logger.info("Uploading code for %s '%s'", component.type, component.name)
+            logger.info(
+                "Uploading code for %s '%s'", new_component.type, new_component.name
+            )
             client = ServiceFoundryServiceClient()
             new_component.image.build_source = local_source_to_remote_source(
-                local_source=component.image.build_source,
+                local_source=new_component.image.build_source,
                 workspace_fqn=workspace_fqn,
-                component_name=component.name,
+                component_name=new_component.name,
                 upload_code_package=client.upload_code_package,
             )
-            logger.debug("Uploaded code for %s '%s'", component.type, component.name)
+            logger.debug(
+                "Uploaded code for %s '%s'", new_component.type, new_component.name
+            )
         return new_component
     return component

{truefoundry-0.7.2.dist-info → truefoundry-0.9.0rc1.dist-info}/METADATA RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.7.2
+Version: 0.9.0rc1
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1
@@ -14,7 +14,7 @@ Requires-Dist: gitpython<4.0.0,>=3.1.43
 Requires-Dist: importlib-metadata<9.0.0,>=4.11.3
 Requires-Dist: importlib-resources<7.0.0,>=5.2.0
 Requires-Dist: mako<2.0.0,>=1.1.6
-Requires-Dist: mcp==1.7.1; python_version >= '3.10'
+Requires-Dist: mcp==1.8.1; python_version >= '3.10'
 Requires-Dist: numpy<3.0.0,>=1.23.0
 Requires-Dist: openai<2.0.0,>=1.16.2
 Requires-Dist: packaging<26.0,>=20.0
@@ -31,7 +31,7 @@ Requires-Dist: requirements-parser<0.12.0,>=0.11.0
 Requires-Dist: rich-click<2.0.0,>=1.2.1
 Requires-Dist: rich<14.0.0,>=13.7.1
 Requires-Dist: tqdm<5.0.0,>=4.0.0
-Requires-Dist: truefoundry-sdk==0.0.16
+Requires-Dist: truefoundry-sdk<0.2.0,>=0.1.1
 Requires-Dist: typing-extensions>=4.0
 Requires-Dist: urllib3<3,>=1.26.18
 Requires-Dist: yq<4.0.0,>=3.1.0

{truefoundry-0.7.2.dist-info → truefoundry-0.9.0rc1.dist-info}/RECORD RENAMED Viewed

@@ -1,5 +1,5 @@
 truefoundry/__init__.py,sha256=VVpO-Awh1v93VOURe7hank8QpeSPc0dCykwr14GOFsw,967
-truefoundry/_client.py,sha256=VQEfRvPE7nuqq--q28cpmnIYPG3RH52RSifIFOzzvTg,1420
+truefoundry/_client.py,sha256=Y3qHi_Lg4Sx6GNvsjAHIoAfFr8PJnqgCrXmpNAI3ECg,1417
 truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
 truefoundry/pydantic_v1.py,sha256=jSuhGtz0Mbk1qYu8jJ1AcnIDK4oxUsdhALc4spqstmM,345
 truefoundry/version.py,sha256=bqiT4Q-VWrTC6P4qfK43mez-Ppf-smWfrl6DcwV7mrw,137
@@ -33,13 +33,13 @@ truefoundry/cli/config.py,sha256=f7z0_gmYZiNImB7Bxz0AnOlrxY2X4lFnX4jYW1I7NHQ,139
 truefoundry/cli/console.py,sha256=9-dMy4YPisCJQziRKTg8Qa0UJnOGl1soiUnJjsnLDvE,242
 truefoundry/cli/const.py,sha256=dVHPo1uAiDSSMXwXoT2mR5kNQjExT98QNVRz98Hz_Ts,510
 truefoundry/cli/display_util.py,sha256=s0_eWUUAK1dbmqW5h_qAG93roH81dh-g1nLjuQVFm6k,5130
-truefoundry/cli/util.py,sha256=BjDUG4WU8iF540YdrBtBQI2WqfNAYyk9iuMiKq7b6jk,3705
+truefoundry/cli/util.py,sha256=7DmKXY5OPslPu2LO6vrUUfDtoHeo12sJTDUA0GOi8IM,3922
 truefoundry/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/common/auth_service_client.py,sha256=N3YxKlx63r6cPZqbgb2lqBOPI69ShB7D7RCIq4FSCjc,7949
 truefoundry/common/constants.py,sha256=eWcElAYIVb0jnHUAcsHvgnkdKf2E1nCg_Ybbi8ibxF0,4365
 truefoundry/common/credential_file_manager.py,sha256=1yEk1Zm2xS4G0VDFwKSZ4w0VUrcPWQ1nJnoBaz9xyKA,4251
 truefoundry/common/credential_provider.py,sha256=_OhJ2XFlDaVsrUO-FyywxctcGGqDdC2pgcvwEKqQD0Q,4071
-truefoundry/common/entities.py,sha256=ko33kesGy3vI9NJ5Ganq8HpnaURTOHictr6h75764no,3893
+truefoundry/common/entities.py,sha256=b4R6ss06-ygDS3C4Tqa_GOq5LFKDYbt7x4Mghnfz6yo,4007
 truefoundry/common/exceptions.py,sha256=jkU0N7hV_P-EhXeud4I5vuB9glXXZSWPf8LcH04mSbw,459
 truefoundry/common/request_utils.py,sha256=e9qrAQ1MutU7JALDKcucmNd0KQEVBqgW3yx0w1zeHIU,5700
 truefoundry/common/servicefoundry_client.py,sha256=2fYhdVPSvLXz5C5tosOq86JD8WM3IRUIy1VO9deDxZI,3340
@@ -50,7 +50,7 @@ truefoundry/common/utils.py,sha256=j3QP0uOsaGD_VmDDR68JTwoYE1okkAq6OqpVkzVf48Q,6
 truefoundry/common/warnings.py,sha256=rs6BHwk7imQYedo07iwh3TWEOywAR3Lqhj0AY4khByg,504
 truefoundry/deploy/__init__.py,sha256=6D22iiCgd5xlzBaG34q9Cx4rGgwf5qIAKQrOCgaCXYY,2746
 truefoundry/deploy/python_deploy_codegen.py,sha256=AainOFR20XvhNeztJkLPWGZ40lAT_nwc-ZmG77Kum4o,6525
-truefoundry/deploy/_autogen/models.py,sha256=p6sZVLmrFbzujNW6PzjGbPA1OiNhq5TgX5tH0evJvRE,72038
+truefoundry/deploy/_autogen/models.py,sha256=gGH63evQTTnU0fEjtNgCsD0aqIxhdp316GL3Xb65NJk,71461
 truefoundry/deploy/builder/__init__.py,sha256=nGQiR3r16iumRy7xbVQ6q-k0EApmijspsfVpXDE-9po,4953
 truefoundry/deploy/builder/constants.py,sha256=amUkHoHvVKzGv0v_knfiioRuKiJM0V0xW0diERgWiI0,508
 truefoundry/deploy/builder/docker_service.py,sha256=sm7GWeIqyrKaZpxskdLejZlsxcZnM3BTDJr6orvPN4E,3948
@@ -59,8 +59,8 @@ truefoundry/deploy/builder/builders/__init__.py,sha256=tlFLXqyDaKLd4iZbo4Hcu_8gO
 truefoundry/deploy/builder/builders/dockerfile.py,sha256=XMbMlPUTMPCyaHl7jJQY1ODtlRkpI61PcvgG6Ck5jNc,1522
 truefoundry/deploy/builder/builders/tfy_notebook_buildpack/__init__.py,sha256=RGWGqY8xOF7vycUPJd10N7ZzahWv24lO0anrOPtLuDU,1796
 truefoundry/deploy/builder/builders/tfy_notebook_buildpack/dockerfile_template.py,sha256=rQgdvKmAT9HArVW4TAG5yd2QTKRs3S5LJ9RQbc_EkHE,2518
-truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py,sha256=wnPwIIArn1H_g4CeTNl5W7EaqRq5FbWDnJpd6E4RSDY,1864
-truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=2KnYTL6yhy0KdiTjSkRfEj_gyCYtmDa2ZsnB06tOpT8,8973
+truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py,sha256=_fjqHKn80qKi68SAMMALge7_A6e1sTsQWichw8uoGIw,2025
+truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=f4l3fH21E2b8W3-JotMKc0AdPcCxV7LRPxxYJa7z_UQ,9134
 truefoundry/deploy/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/cli/commands/__init__.py,sha256=f7sXiQK9UuxDJmvBa-QCFNyumUpGGMhZbCxdwJzWXwQ,1116
 truefoundry/deploy/cli/commands/apply_command.py,sha256=Y2e_C8HVpo8CssVod-3JRz-89qStC5JRaNzJ7O2mRlY,2039
@@ -95,8 +95,7 @@ truefoundry/deploy/lib/session.py,sha256=fLdgR6ZDp8-hFl5NTON4ngnWLsMzGxvKtfpDOOw
 truefoundry/deploy/lib/util.py,sha256=J7r8San2wKo48A7-BlH2-OKTlBO67zlPjLEhMsL8os0,1059
 truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
 truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/_mcp_streamable_http.py,sha256=oO_pdgVy0BkksLDZo-arcnaJ4gSaTJmMMZF9u-uF7Mk,10851
-truefoundry/deploy/lib/clients/ask_client.py,sha256=PINeyjZF9IMkphU6ln6JVKlFJrHPhkkE5wolFBrp_f8,13589
+truefoundry/deploy/lib/clients/ask_client.py,sha256=77106708EC16wsi2M1n1_5HgOVboEZoq9_obKsf24M0,13494
 truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=fmRlPYCimk1ZLbMgdzfJVCbcKRCVnFYL5T3j2uJA0Tc,27037
 truefoundry/deploy/lib/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/dao/application.py,sha256=oMszpueXPUfTUuN_XdKwoRjQyqAgWHhZ-10cbprCVdM,9226
@@ -108,7 +107,7 @@ truefoundry/deploy/lib/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 truefoundry/deploy/lib/model/entity.py,sha256=Bp9sLB-M5INCpw5lPmFdygHWS1zvnLicnSiSCi2iqhQ,8591
 truefoundry/deploy/v2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/v2/lib/__init__.py,sha256=WEiVMZXOVljzEE3tpGJil14liIn_PCDoACJ6b3tZ6sI,188
-truefoundry/deploy/v2/lib/deploy.py,sha256=EOqVQTCMrvs_Kwp8Ft0FZYlRErPuFq4eDBDR1qEZ7_E,11767
+truefoundry/deploy/v2/lib/deploy.py,sha256=kAnh6RO4ci7AVjlIoN1Sr5FmcOU7nbkwNvbrS802spY,12625
 truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=G5BzMIbap8pgDX1eY-TITruUxQdkKhYtBmRwLL6lDeY,14342
 truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=xbHFD3pURflvCm8EODPvjfvRrv67mlSrjPUknY8SMB8,4060
 truefoundry/deploy/v2/lib/models.py,sha256=ogc1UYs1Z2nBdGSKCrde9sk8d0GxFKMkem99uqO5CmM,1148
@@ -376,7 +375,7 @@ truefoundry/workflow/remote_filesystem/__init__.py,sha256=LQ95ViEjJ7Ts4JcCGOxMPs
 truefoundry/workflow/remote_filesystem/logger.py,sha256=em2l7D6sw7xTLDP0kQSLpgfRRCLpN14Qw85TN7ujQcE,1022
 truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=xcT0wQmQlgzcj0nP3tJopyFSVWT1uv3nhiTIuwfXYeg,12342
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=nSGPZu0Gyd_jz0KsEE-7w_BmnTD8CVF1S8cUJoxaCbc,13305
-truefoundry-0.7.2.dist-info/METADATA,sha256=wjXiGA-c9eUYLeH9Qcy2QfYLBmi9FOa1j-R5AABgtgg,2459
-truefoundry-0.7.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truefoundry-0.7.2.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
-truefoundry-0.7.2.dist-info/RECORD,,
+truefoundry-0.9.0rc1.dist-info/METADATA,sha256=Jke--bMLUmA_EhDKqqVyPODCMgbsIHC0W7yH2nRsjqk,2468
+truefoundry-0.9.0rc1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.9.0rc1.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.9.0rc1.dist-info/RECORD,,

truefoundry/deploy/lib/clients/_mcp_streamable_http.py DELETED Viewed

@@ -1,264 +0,0 @@
-"""
-StreamableHTTP Client Transport Module
-# From https://github.com/modelcontextprotocol/python-sdk/pull/573
-This module implements the StreamableHTTP transport for MCP clients,
-providing support for HTTP POST requests with optional SSE streaming responses
-and session management.
-"""
-import logging
-from contextlib import asynccontextmanager
-from datetime import timedelta
-from typing import Any
-import anyio
-import httpx
-from httpx_sse import EventSource, aconnect_sse
-from mcp.types import (
-    ErrorData,
-    JSONRPCError,
-    JSONRPCMessage,
-    JSONRPCNotification,
-    JSONRPCRequest,
-)
-logger = logging.getLogger(__name__)
-# Header names
-MCP_SESSION_ID_HEADER = "mcp-session-id"
-LAST_EVENT_ID_HEADER = "last-event-id"
-# Content types
-CONTENT_TYPE_JSON = "application/json"
-CONTENT_TYPE_SSE = "text/event-stream"
-@asynccontextmanager
-async def streamablehttp_client(
-    url: str,
-    headers: dict[str, Any] | None = None,
-    timeout: timedelta = timedelta(seconds=30),
-    sse_read_timeout: timedelta = timedelta(seconds=60 * 5),
-):
-    """
-    Client transport for StreamableHTTP.
-    `sse_read_timeout` determines how long (in seconds) the client will wait for a new
-    event before disconnecting. All other HTTP operations are controlled by `timeout`.
-    Yields:
-        Tuple of (read_stream, write_stream, terminate_callback)
-    """
-    read_stream_writer, read_stream = anyio.create_memory_object_stream[
-        JSONRPCMessage | Exception
-    ](0)
-    write_stream, write_stream_reader = anyio.create_memory_object_stream[
-        JSONRPCMessage
-    ](0)
-    async def get_stream():
-        """
-        Optional GET stream for server-initiated messages
-        """
-        nonlocal session_id
-        try:
-            # Only attempt GET if we have a session ID
-            if not session_id:
-                return
-            get_headers = request_headers.copy()
-            get_headers[MCP_SESSION_ID_HEADER] = session_id
-            async with aconnect_sse(
-                client,
-                "GET",
-                url,
-                headers=get_headers,
-                timeout=httpx.Timeout(timeout.seconds, read=sse_read_timeout.seconds),
-            ) as event_source:
-                event_source.response.raise_for_status()
-                logger.debug("GET SSE connection established")
-                async for sse in event_source.aiter_sse():
-                    if sse.event == "message":
-                        try:
-                            message = JSONRPCMessage.model_validate_json(sse.data)
-                            logger.debug(f"GET message: {message}")
-                            await read_stream_writer.send(message)
-                        except Exception as exc:
-                            logger.error(f"Error parsing GET message: {exc}")
-                            await read_stream_writer.send(exc)
-                    else:
-                        logger.warning(f"Unknown SSE event from GET: {sse.event}")
-        except Exception as exc:
-            # GET stream is optional, so don't propagate errors
-            logger.debug(f"GET stream error (non-fatal): {exc}")
-    async def post_writer(client: httpx.AsyncClient):
-        nonlocal session_id
-        try:
-            async with write_stream_reader:
-                async for message in write_stream_reader:
-                    # Add session ID to headers if we have one
-                    post_headers = request_headers.copy()
-                    if session_id:
-                        post_headers[MCP_SESSION_ID_HEADER] = session_id
-                    logger.debug(f"Sending client message: {message}")
-                    # Handle initial initialization request
-                    is_initialization = (
-                        isinstance(message.root, JSONRPCRequest)
-                        and message.root.method == "initialize"
-                    )
-                    if (
-                        isinstance(message.root, JSONRPCNotification)
-                        and message.root.method == "notifications/initialized"
-                    ):
-                        tg.start_soon(get_stream)
-                    async with client.stream(
-                        "POST",
-                        url,
-                        json=message.model_dump(
-                            by_alias=True, mode="json", exclude_none=True
-                        ),
-                        headers=post_headers,
-                    ) as response:
-                        if response.status_code == 202:
-                            logger.debug("Received 202 Accepted")
-                            continue
-                        # Check for 404 (session expired/invalid)
-                        if response.status_code == 404:
-                            if isinstance(message.root, JSONRPCRequest):
-                                jsonrpc_error = JSONRPCError(
-                                    jsonrpc="2.0",
-                                    id=message.root.id,
-                                    error=ErrorData(
-                                        code=32600,
-                                        message="Session terminated",
-                                    ),
-                                )
-                                await read_stream_writer.send(
-                                    JSONRPCMessage(jsonrpc_error)
-                                )
-                                continue
-                        if not response.is_success:
-                            _response_content = await response.aread()
-                            logger.error(
-                                f"Response: {response.status_code} {_response_content}"
-                            )
-                            response.raise_for_status()
-                        # Extract session ID from response headers
-                        if is_initialization:
-                            new_session_id = response.headers.get(MCP_SESSION_ID_HEADER)
-                            if new_session_id:
-                                session_id = new_session_id
-                                logger.info(f"Received session ID: {session_id}")
-                        # Handle different response types
-                        content_type = response.headers.get("content-type", "").lower()
-                        if content_type.startswith(CONTENT_TYPE_JSON):
-                            try:
-                                content = await response.aread()
-                                json_message = JSONRPCMessage.model_validate_json(
-                                    content
-                                )
-                                await read_stream_writer.send(json_message)
-                            except Exception as exc:
-                                logger.error(f"Error parsing JSON response: {exc}")
-                                await read_stream_writer.send(exc)
-                        elif content_type.startswith(CONTENT_TYPE_SSE):
-                            # Parse SSE events from the response
-                            try:
-                                event_source = EventSource(response)
-                                async for sse in event_source.aiter_sse():
-                                    if sse.event == "message":
-                                        try:
-                                            await read_stream_writer.send(
-                                                JSONRPCMessage.model_validate_json(
-                                                    sse.data
-                                                )
-                                            )
-                                        except Exception as exc:
-                                            logger.exception("Error parsing message")
-                                            await read_stream_writer.send(exc)
-                                    else:
-                                        logger.warning(f"Unknown event: {sse.event}")
-                            except Exception as e:
-                                logger.exception("Error reading SSE stream:")
-                                await read_stream_writer.send(e)
-                        else:
-                            # For 202 Accepted with no body
-                            if response.status_code == 202:
-                                logger.debug("Received 202 Accepted")
-                                continue
-                            error_msg = f"Unexpected content type: {content_type}"
-                            logger.error(error_msg)
-                            await read_stream_writer.send(ValueError(error_msg))
-        except Exception as exc:
-            logger.error(f"Error in post_writer: {exc}")
-        finally:
-            await read_stream_writer.aclose()
-            await write_stream.aclose()
-    async def terminate_session():
-        """
-        Terminate the session by sending a DELETE request.
-        """
-        nonlocal session_id
-        if not session_id:
-            return  # No session to terminate
-        try:
-            delete_headers = request_headers.copy()
-            delete_headers[MCP_SESSION_ID_HEADER] = session_id
-            response = await client.delete(
-                url,
-                headers=delete_headers,
-            )
-            if response.status_code == 405:
-                # Server doesn't allow client-initiated termination
-                logger.debug("Server does not allow session termination")
-            elif response.status_code != 200:
-                logger.warning(f"Session termination failed: {response.status_code}")
-        except Exception as exc:
-            logger.warning(f"Session termination failed: {exc}")
-    async with anyio.create_task_group() as tg:
-        try:
-            logger.debug(f"Connecting to StreamableHTTP endpoint: {url}")
-            # Set up headers with required Accept header
-            request_headers = {
-                "Accept": f"{CONTENT_TYPE_JSON}, {CONTENT_TYPE_SSE}",
-                "Content-Type": CONTENT_TYPE_JSON,
-                **(headers or {}),
-            }
-            # Track session ID if provided by server
-            session_id: str | None = None
-            async with httpx.AsyncClient(
-                headers=request_headers,
-                timeout=httpx.Timeout(timeout.seconds, read=sse_read_timeout.seconds),
-                follow_redirects=True,
-            ) as client:
-                tg.start_soon(post_writer, client)
-                try:
-                    yield read_stream, write_stream, terminate_session
-                finally:
-                    tg.cancel_scope.cancel()
-        finally:
-            await read_stream_writer.aclose()
-            await write_stream.aclose()

{truefoundry-0.7.2.dist-info → truefoundry-0.9.0rc1.dist-info}/WHEEL RENAMED Viewed

File without changes

{truefoundry-0.7.2.dist-info → truefoundry-0.9.0rc1.dist-info}/entry_points.txt RENAMED Viewed

File without changes

truefoundry 0.7.2__py3-none-any.whl → 0.9.0rc1__py3-none-any.whl

Potentially problematic release.

truefoundry 0.7.2py3-none-any.whl → 0.9.0rc1py3-none-any.whl