truefoundry 0.7.1rc1__tar.gz → 0.7.3rc1__tar.gz

{truefoundry-0.7.1rc1 → truefoundry-0.7.3rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.7.1rc1
+Version: 0.7.3rc1
 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.6.0; 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.1.0
 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.1rc1 → truefoundry-0.7.3rc1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "truefoundry"
-version = "0.7.1rc1"  # This is a placeholder version, it will be replaced when releasing
+version = "0.7.3rc1"  # This is a placeholder version, it will be replaced when releasing
 requires-python = ">=3.8.1,<3.14"
 description = "TrueFoundry CLI"
 authors = [{ name = "TrueFoundry Team", email = "abhishek@truefoundry.com" }]
@@ -16,7 +16,7 @@ dependencies = [
   "importlib-metadata>=4.11.3,<9.0.0",
   "importlib-resources>=5.2.0,<7.0.0",
   "Mako>=1.1.6,<2.0.0",
-  "mcp==1.6.0 ; python_version >= '3.10'",
+  "mcp==1.8.1 ; python_version >= '3.10'",
   "openai>=1.16.2,<2.0.0",
   "packaging>=20.0,<26.0",
   "pydantic>=1.8.2,<3.0.0",
@@ -32,7 +32,7 @@ dependencies = [
   "rich-click>=1.2.1,<2.0.0",
   "rich>=13.7.1,<14.0.0",
   "tqdm>=4.0.0,<5.0.0",
-  "truefoundry-sdk==0.0.16",
+  "truefoundry-sdk==0.1.0",
   "typing-extensions>=4.0",
   "urllib3>=1.26.18,<3",
   "yq>=3.1.0,<4.0.0",

{truefoundry-0.7.1rc1 → truefoundry-0.7.3rc1}/truefoundry/_client.py

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

{truefoundry-0.7.1rc1 → truefoundry-0.7.3rc1}/truefoundry/cli/__main__.py

@@ -100,13 +100,13 @@ def create_truefoundry_cli() -> click.Group:
     cli.add_command(get_terminate_command())
     cli.add_command(get_ml_cli())
     cli.add_command(get_get_command())
+    cli.add_command(get_ask_command())
 
     if not (sys.platform.startswith("win32") or sys.platform.startswith("cygwin")):
         cli.add_command(get_patch_command())
 
     if is_internal_env_set():
         cli.add_command(get_build_command())
-        cli.add_command(get_ask_command())
     return cli
 
 

{truefoundry-0.7.1rc1 → truefoundry-0.7.3rc1}/truefoundry/deploy/_autogen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2025-04-21T06:26:10+00:00
+#   timestamp: 2025-05-03T01:45:46+00:00
 
 from __future__ import annotations
 
@@ -1485,6 +1485,10 @@ 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"
     )
@@ -1552,10 +1556,12 @@ class JobAlert(BaseModel):
         description="List of recipients' email addresses if the notification channel is Email.",
     )
     notification_target: Optional[Union[Email, SlackWebhook, SlackBot]] = None
-    on_start: bool = Field(False, description="Send an alert when the job starts")
-    on_completion: bool = Field(
-        False, description="Send an alert when the job completes"
+    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-0.7.1rc1 → truefoundry-0.7.3rc1}/truefoundry/deploy/lib/clients/ask_client.py

@@ -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-0.7.1rc1/truefoundry/deploy/lib/clients/_mcp_streamable_http.py

@@ -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()