vellum-ai 1.2.5__py3-none-any.whl → 1.3.1__py3-none-any.whl

vellum/__init__.py

@@ -643,11 +643,15 @@ from .client.types import (
     WorkflowPushResponse,
     WorkflowReleaseTagRead,
     WorkflowReleaseTagWorkflowDeploymentHistoryItem,
+    WorkflowRequestAudioInputRequest,
     WorkflowRequestChatHistoryInputRequest,
+    WorkflowRequestDocumentInputRequest,
+    WorkflowRequestImageInputRequest,
     WorkflowRequestInputRequest,
     WorkflowRequestJsonInputRequest,
     WorkflowRequestNumberInputRequest,
     WorkflowRequestStringInputRequest,
+    WorkflowRequestVideoInputRequest,
     WorkflowResultEvent,
     WorkflowResultEventOutputData,
     WorkflowResultEventOutputDataArray,
@@ -1359,11 +1363,15 @@ __all__ = [
     "WorkflowPushResponse",
     "WorkflowReleaseTagRead",
     "WorkflowReleaseTagWorkflowDeploymentHistoryItem",
+    "WorkflowRequestAudioInputRequest",
     "WorkflowRequestChatHistoryInputRequest",
+    "WorkflowRequestDocumentInputRequest",
+    "WorkflowRequestImageInputRequest",
     "WorkflowRequestInputRequest",
     "WorkflowRequestJsonInputRequest",
     "WorkflowRequestNumberInputRequest",
     "WorkflowRequestStringInputRequest",
+    "WorkflowRequestVideoInputRequest",
     "WorkflowResultEvent",
     "WorkflowResultEventOutputData",
     "WorkflowResultEventOutputDataArray",

vellum/client/README.md

@@ -3,7 +3,7 @@
 [![fern shield](https://img.shields.io/badge/%F0%9F%8C%BF-Built%20with%20Fern-brightgreen)](https://buildwithfern.com?utm_source=github&utm_medium=github&utm_campaign=readme&utm_source=https%3A%2F%2Fgithub.com%2Fvellum-ai%2Fvellum-python-sdks)
 [![pypi](https://img.shields.io/pypi/v/vellum-ai)](https://pypi.python.org/pypi/vellum-ai)
 
-The Vellum Python library provides convenient access to the Vellum API from Python.
+The Vellum Python library provides convenient access to the Vellum APIs from Python.
 
 ## API Docs
 You can find Vellum's complete API docs at [docs.vellum.ai](https://docs.vellum.ai/api-reference/introduction/getting-started).

vellum/client/core/client_wrapper.py

@@ -27,10 +27,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "vellum-ai/1.2.5",
+            "User-Agent": "vellum-ai/1.3.1",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "1.2.5",
+            "X-Fern-SDK-Version": "1.3.1",
             **(self.get_custom_headers() or {}),
         }
         if self._api_version is not None:

vellum/client/reference.md

@@ -6222,7 +6222,6 @@ client = Vellum(
 )
 client.workflow_sandboxes.deploy_workflow(
     id="id",
-    workflow_id="workflow_id",
 )
 
 ```
@@ -6247,14 +6246,6 @@ client.workflow_sandboxes.deploy_workflow(
 <dl>
 <dd>
 
-**workflow_id:** `str` — An ID identifying the Workflow you'd like to deploy.
-    
-</dd>
-</dl>
-
-<dl>
-<dd>
-
 **workflow_deployment_id:** `typing.Optional[str]` — The Vellum-generated ID of the Workflow Deployment you'd like to update. Cannot specify both this and workflow_deployment_name. Leave null to create a new Workflow Deployment.
     
 </dd>

vellum/client/resources/workflow_sandboxes/client.py

@@ -35,7 +35,6 @@ class WorkflowSandboxesClient:
     def deploy_workflow(
         self,
         id: str,
-        workflow_id: str,
         *,
         workflow_deployment_id: typing.Optional[str] = OMIT,
         workflow_deployment_name: typing.Optional[str] = OMIT,
@@ -50,9 +49,6 @@ class WorkflowSandboxesClient:
         id : str
             A UUID string identifying this workflow sandbox.
 
-        workflow_id : str
-            An ID identifying the Workflow you'd like to deploy.
-
         workflow_deployment_id : typing.Optional[str]
             The Vellum-generated ID of the Workflow Deployment you'd like to update. Cannot specify both this and workflow_deployment_name. Leave null to create a new Workflow Deployment.
 
@@ -86,12 +82,10 @@ class WorkflowSandboxesClient:
         )
         client.workflow_sandboxes.deploy_workflow(
             id="id",
-            workflow_id="workflow_id",
         )
         """
         _response = self._raw_client.deploy_workflow(
             id,
-            workflow_id,
             workflow_deployment_id=workflow_deployment_id,
             workflow_deployment_name=workflow_deployment_name,
             label=label,
@@ -168,7 +162,6 @@ class AsyncWorkflowSandboxesClient:
     async def deploy_workflow(
         self,
         id: str,
-        workflow_id: str,
         *,
         workflow_deployment_id: typing.Optional[str] = OMIT,
         workflow_deployment_name: typing.Optional[str] = OMIT,
@@ -183,9 +176,6 @@ class AsyncWorkflowSandboxesClient:
         id : str
             A UUID string identifying this workflow sandbox.
 
-        workflow_id : str
-            An ID identifying the Workflow you'd like to deploy.
-
         workflow_deployment_id : typing.Optional[str]
             The Vellum-generated ID of the Workflow Deployment you'd like to update. Cannot specify both this and workflow_deployment_name. Leave null to create a new Workflow Deployment.
 
@@ -224,7 +214,6 @@ class AsyncWorkflowSandboxesClient:
         async def main() -> None:
             await client.workflow_sandboxes.deploy_workflow(
                 id="id",
-                workflow_id="workflow_id",
             )
 
 
@@ -232,7 +221,6 @@ class AsyncWorkflowSandboxesClient:
         """
         _response = await self._raw_client.deploy_workflow(
             id,
-            workflow_id,
             workflow_deployment_id=workflow_deployment_id,
             workflow_deployment_name=workflow_deployment_name,
             label=label,

vellum/client/resources/workflow_sandboxes/raw_client.py

@@ -24,7 +24,6 @@ class RawWorkflowSandboxesClient:
     def deploy_workflow(
         self,
         id: str,
-        workflow_id: str,
         *,
         workflow_deployment_id: typing.Optional[str] = OMIT,
         workflow_deployment_name: typing.Optional[str] = OMIT,
@@ -39,9 +38,6 @@ class RawWorkflowSandboxesClient:
         id : str
             A UUID string identifying this workflow sandbox.
 
-        workflow_id : str
-            An ID identifying the Workflow you'd like to deploy.
-
         workflow_deployment_id : typing.Optional[str]
             The Vellum-generated ID of the Workflow Deployment you'd like to update. Cannot specify both this and workflow_deployment_name. Leave null to create a new Workflow Deployment.
 
@@ -66,7 +62,7 @@ class RawWorkflowSandboxesClient:
 
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"v1/workflow-sandboxes/{jsonable_encoder(id)}/workflows/{jsonable_encoder(workflow_id)}/deploy",
+            f"v1/workflow-sandboxes/{jsonable_encoder(id)}/deploy",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
             json={
@@ -165,7 +161,6 @@ class AsyncRawWorkflowSandboxesClient:
     async def deploy_workflow(
         self,
         id: str,
-        workflow_id: str,
         *,
         workflow_deployment_id: typing.Optional[str] = OMIT,
         workflow_deployment_name: typing.Optional[str] = OMIT,
@@ -180,9 +175,6 @@ class AsyncRawWorkflowSandboxesClient:
         id : str
             A UUID string identifying this workflow sandbox.
 
-        workflow_id : str
-            An ID identifying the Workflow you'd like to deploy.
-
         workflow_deployment_id : typing.Optional[str]
             The Vellum-generated ID of the Workflow Deployment you'd like to update. Cannot specify both this and workflow_deployment_name. Leave null to create a new Workflow Deployment.
 
@@ -207,7 +199,7 @@ class AsyncRawWorkflowSandboxesClient:
 
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"v1/workflow-sandboxes/{jsonable_encoder(id)}/workflows/{jsonable_encoder(workflow_id)}/deploy",
+            f"v1/workflow-sandboxes/{jsonable_encoder(id)}/deploy",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
             json={

vellum/client/types/__init__.py

@@ -667,11 +667,15 @@ from .workflow_push_exec_config import WorkflowPushExecConfig
 from .workflow_push_response import WorkflowPushResponse
 from .workflow_release_tag_read import WorkflowReleaseTagRead
 from .workflow_release_tag_workflow_deployment_history_item import WorkflowReleaseTagWorkflowDeploymentHistoryItem
+from .workflow_request_audio_input_request import WorkflowRequestAudioInputRequest
 from .workflow_request_chat_history_input_request import WorkflowRequestChatHistoryInputRequest
+from .workflow_request_document_input_request import WorkflowRequestDocumentInputRequest
+from .workflow_request_image_input_request import WorkflowRequestImageInputRequest
 from .workflow_request_input_request import WorkflowRequestInputRequest
 from .workflow_request_json_input_request import WorkflowRequestJsonInputRequest
 from .workflow_request_number_input_request import WorkflowRequestNumberInputRequest
 from .workflow_request_string_input_request import WorkflowRequestStringInputRequest
+from .workflow_request_video_input_request import WorkflowRequestVideoInputRequest
 from .workflow_result_event import WorkflowResultEvent
 from .workflow_result_event_output_data import WorkflowResultEventOutputData
 from .workflow_result_event_output_data_array import WorkflowResultEventOutputDataArray
@@ -1326,11 +1330,15 @@ __all__ = [
     "WorkflowPushResponse",
     "WorkflowReleaseTagRead",
     "WorkflowReleaseTagWorkflowDeploymentHistoryItem",
+    "WorkflowRequestAudioInputRequest",
     "WorkflowRequestChatHistoryInputRequest",
+    "WorkflowRequestDocumentInputRequest",
+    "WorkflowRequestImageInputRequest",
     "WorkflowRequestInputRequest",
     "WorkflowRequestJsonInputRequest",
     "WorkflowRequestNumberInputRequest",
     "WorkflowRequestStringInputRequest",
+    "WorkflowRequestVideoInputRequest",
     "WorkflowResultEvent",
     "WorkflowResultEventOutputData",
     "WorkflowResultEventOutputDataArray",

vellum/client/types/deployment_read.py

@@ -13,6 +13,10 @@ from .vellum_variable import VellumVariable
 
 
 class DeploymentRead(UniversalBaseModel):
+    """
+    A Prompt Deployment's full details.
+    """
+
     id: str
     created: dt.datetime
     label: str = pydantic.Field()
@@ -35,11 +39,7 @@ class DeploymentRead(UniversalBaseModel):
 
     environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
     """
-    The environment this deployment is used in
-    
-    * `DEVELOPMENT` - Development
-    * `STAGING` - Staging
-    * `PRODUCTION` - Production
+    Deprecated. The value returned will always be 'PRODUCTION'.
     """
 
     last_deployed_on: dt.datetime

vellum/client/types/slim_deployment_read.py

@@ -13,6 +13,10 @@ from .vellum_variable import VellumVariable
 
 
 class SlimDeploymentRead(UniversalBaseModel):
+    """
+    A subset of a Prompt Deployment's full details.
+    """
+
     id: str
     created: dt.datetime
     label: str = pydantic.Field()
@@ -35,11 +39,7 @@ class SlimDeploymentRead(UniversalBaseModel):
 
     environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
     """
-    The environment this deployment is used in
-    
-    * `DEVELOPMENT` - Development
-    * `STAGING` - Staging
-    * `PRODUCTION` - Production
+    Deprecated. The value returned will always be 'PRODUCTION'.
     """
 
     last_deployed_on: dt.datetime

vellum/client/types/slim_workflow_deployment.py

@@ -13,6 +13,10 @@ from .vellum_variable import VellumVariable
 
 
 class SlimWorkflowDeployment(UniversalBaseModel):
+    """
+    A subset of a Workflow Deployment's full details.
+    """
+
     id: str
     name: str = pydantic.Field()
     """
@@ -34,11 +38,7 @@ class SlimWorkflowDeployment(UniversalBaseModel):
 
     environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
     """
-    The environment this workflow deployment is used in
-    
-    * `DEVELOPMENT` - Development
-    * `STAGING` - Staging
-    * `PRODUCTION` - Production
+    Deprecated. The value returned will always be 'PRODUCTION'.
     """
 
     created: dt.datetime

vellum/client/types/workflow_deployment_read.py

@@ -13,6 +13,10 @@ from .vellum_variable import VellumVariable
 
 
 class WorkflowDeploymentRead(UniversalBaseModel):
+    """
+    A Workflow Deployment's full details.
+    """
+
     id: str
     name: str = pydantic.Field()
     """
@@ -34,11 +38,7 @@ class WorkflowDeploymentRead(UniversalBaseModel):
 
     environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
     """
-    The environment this workflow deployment is used in
-    
-    * `DEVELOPMENT` - Development
-    * `STAGING` - Staging
-    * `PRODUCTION` - Production
+    Deprecated. The value returned will always be 'PRODUCTION'.
     """
 
     created: dt.datetime

vellum/client/types/workflow_request_audio_input_request.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_audio_request import VellumAudioRequest
+
+
+class WorkflowRequestAudioInputRequest(UniversalBaseModel):
+    """
+    The input for an audio variable in a Workflow.
+    """
+
+    name: str = pydantic.Field()
+    """
+    The variable's name, as defined in the Workflow.
+    """
+
+    type: typing.Literal["AUDIO"] = "AUDIO"
+    value: VellumAudioRequest
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/workflow_request_document_input_request.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_document_request import VellumDocumentRequest
+
+
+class WorkflowRequestDocumentInputRequest(UniversalBaseModel):
+    """
+    The input for a document variable in a Workflow.
+    """
+
+    name: str = pydantic.Field()
+    """
+    The variable's name, as defined in the Workflow.
+    """
+
+    type: typing.Literal["DOCUMENT"] = "DOCUMENT"
+    value: VellumDocumentRequest
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/workflow_request_image_input_request.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_image_request import VellumImageRequest
+
+
+class WorkflowRequestImageInputRequest(UniversalBaseModel):
+    """
+    The input for an image variable in a Workflow.
+    """
+
+    name: str = pydantic.Field()
+    """
+    The variable's name, as defined in the Workflow.
+    """
+
+    type: typing.Literal["IMAGE"] = "IMAGE"
+    value: VellumImageRequest
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/workflow_request_input_request.py

@@ -2,14 +2,22 @@
 
 import typing
 
+from .workflow_request_audio_input_request import WorkflowRequestAudioInputRequest
 from .workflow_request_chat_history_input_request import WorkflowRequestChatHistoryInputRequest
+from .workflow_request_document_input_request import WorkflowRequestDocumentInputRequest
+from .workflow_request_image_input_request import WorkflowRequestImageInputRequest
 from .workflow_request_json_input_request import WorkflowRequestJsonInputRequest
 from .workflow_request_number_input_request import WorkflowRequestNumberInputRequest
 from .workflow_request_string_input_request import WorkflowRequestStringInputRequest
+from .workflow_request_video_input_request import WorkflowRequestVideoInputRequest
 
 WorkflowRequestInputRequest = typing.Union[
     WorkflowRequestStringInputRequest,
     WorkflowRequestJsonInputRequest,
     WorkflowRequestChatHistoryInputRequest,
     WorkflowRequestNumberInputRequest,
+    WorkflowRequestAudioInputRequest,
+    WorkflowRequestVideoInputRequest,
+    WorkflowRequestImageInputRequest,
+    WorkflowRequestDocumentInputRequest,
 ]

vellum/client/types/workflow_request_video_input_request.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_video_request import VellumVideoRequest
+
+
+class WorkflowRequestVideoInputRequest(UniversalBaseModel):
+    """
+    The input for a video variable in a Workflow.
+    """
+
+    name: str = pydantic.Field()
+    """
+    The variable's name, as defined in the Workflow.
+    """
+
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: VellumVideoRequest
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/workflow_request_audio_input_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.workflow_request_audio_input_request import *

vellum/types/workflow_request_document_input_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.workflow_request_document_input_request import *

vellum/types/workflow_request_image_input_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.workflow_request_image_input_request import *

vellum/types/workflow_request_video_input_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.workflow_request_video_input_request import *

vellum/workflows/events/types.py

@@ -7,7 +7,6 @@ from pydantic import Field, GetCoreSchemaHandler, Tag, ValidationInfo
 from pydantic_core import CoreSchema, core_schema
 
 from vellum.client.core.pydantic_utilities import UniversalBaseModel
-from vellum.client.types.span_link import SpanLink
 from vellum.workflows.state.encoder import DefaultStateEncoder
 from vellum.workflows.types.definition import VellumCodeResourceDefinition
 from vellum.workflows.types.utils import datetime_now
@@ -86,6 +85,12 @@ class ExternalParentContext(BaseParentContext):
     type: Literal["EXTERNAL"] = "EXTERNAL"
 
 
+class SpanLink(UniversalBaseModel):
+    trace_id: str
+    type: Literal["TRIGGERED_BY", "PREVIOUS_SPAN", "ROOT_SPAN"]
+    span_context: "ParentContext"
+
+
 def _cast_parent_context_discriminator(v: Any) -> Any:
     if v in PARENT_CONTEXT_TYPES:
         return v

vellum/workflows/events/workflow.py

@@ -1,3 +1,4 @@
+import logging
 from uuid import UUID
 from typing import TYPE_CHECKING, Any, Dict, Generic, Iterable, Literal, Optional, Type, Union
 from typing_extensions import TypeGuard
@@ -25,6 +26,8 @@ from .types import BaseEvent, default_serializer
 if TYPE_CHECKING:
     from vellum.workflows.workflows.base import BaseWorkflow
 
+logger = logging.getLogger(__name__)
+
 
 class _BaseWorkflowExecutionBody(UniversalBaseModel):
     workflow_definition: Type["BaseWorkflow"]
@@ -107,8 +110,12 @@ class WorkflowExecutionInitiatedEvent(_BaseWorkflowEvent, Generic[InputsType, St
     ) -> WorkflowExecutionInitiatedBody[InputsType, StateType]:
         context = info.context if info and hasattr(info, "context") else {}
         if context and "event_enricher" in context and callable(context["event_enricher"]):
-            event = context["event_enricher"](self)
-            return event.body
+            try:
+                event = context["event_enricher"](self)
+                return event.body
+            except Exception as e:
+                logger.exception(f"Error in event_enricher: {e}")
+                return body
         else:
             return body
 

vellum/workflows/integrations/tests/test_mcp_service.py

@@ -1,10 +1,11 @@
+import pytest
 import asyncio
 import json
 from unittest import mock
 
 from vellum.workflows.constants import AuthorizationType
 from vellum.workflows.integrations.mcp_service import MCPHttpClient, MCPService
-from vellum.workflows.types.definition import MCPServer
+from vellum.workflows.types.definition import MCPServer, MCPToolDefinition
 
 
 def test_mcp_http_client_sse_response():
@@ -118,3 +119,107 @@ def test_mcp_service_api_key_auth():
 
     # THEN the custom API key header should be set correctly
     assert headers == {"X-API-Key": "api-key-123"}
+
+
+@pytest.mark.asyncio
+async def test_mcp_http_client_empty_response():
+    """Test that empty responses are handled gracefully"""
+    # GIVEN a mock response that returns empty content
+    mock_response = mock.Mock()
+    mock_response.headers = {"content-type": "application/json"}
+    mock_response.text = ""
+
+    # AND a mock httpx client that returns this response
+    with mock.patch("vellum.workflows.integrations.mcp_service.httpx.AsyncClient") as mock_client_class:
+        mock_client = mock.AsyncMock()
+        mock_client.post.return_value = mock_response
+        mock_client_class.return_value = mock_client
+
+        # WHEN we call initialize with an empty response
+        # THEN it should raise an exception about empty response
+        async with MCPHttpClient("https://test.server.com", {}) as client:
+            with pytest.raises(Exception, match="Empty response received from server"):
+                await client.initialize()
+
+
+@pytest.mark.asyncio
+async def test_mcp_http_client_invalid_sse_json():
+    """Test that invalid JSON in SSE data is handled"""
+    # GIVEN an SSE response with invalid JSON
+    invalid_sse = """event: message
+data: {invalid json}
+
+"""
+
+    mock_response = mock.Mock()
+    mock_response.headers = {"content-type": "text/event-stream"}
+    mock_response.text = invalid_sse
+
+    with mock.patch("vellum.workflows.integrations.mcp_service.httpx.AsyncClient") as mock_client_class:
+        mock_client = mock.AsyncMock()
+        mock_client.post.return_value = mock_response
+        mock_client_class.return_value = mock_client
+
+        # WHEN we call initialize with invalid SSE data
+        # THEN it should raise an exception about no valid JSON
+        async with MCPHttpClient("https://test.server.com", {}) as client:
+            with pytest.raises(Exception, match="No valid JSON data found in SSE response"):
+                await client.initialize()
+
+
+def test_mcp_service_hydrate_tool_definitions():
+    """Test tool definition hydration with SSE responses"""
+    # GIVEN an MCP server configuration
+    sample_mcp_server = MCPServer(
+        name="test-server",
+        url="https://test.mcp.server.com/mcp",
+        authorization_type=AuthorizationType.BEARER_TOKEN,
+        bearer_token_value="test-token-123",
+    )
+
+    # AND a mock MCP service that returns tools via SSE
+    with mock.patch("vellum.workflows.integrations.mcp_service.asyncio.run") as mock_run:
+        mock_run.return_value = [
+            {
+                "name": "resolve-library-id",
+                "description": "Resolves library names to IDs",
+                "inputSchema": {
+                    "type": "object",
+                    "properties": {"libraryName": {"type": "string"}},
+                    "required": ["libraryName"],
+                },
+            }
+        ]
+
+        # WHEN we hydrate tool definitions
+        service = MCPService()
+        tool_definitions = service.hydrate_tool_definitions(sample_mcp_server)
+
+        # THEN we should get properly formatted MCPToolDefinition objects
+        assert len(tool_definitions) == 1
+        assert isinstance(tool_definitions[0], MCPToolDefinition)
+        assert tool_definitions[0].name == "resolve-library-id"
+        assert tool_definitions[0].description == "Resolves library names to IDs"
+        assert tool_definitions[0].server == sample_mcp_server
+        assert tool_definitions[0].parameters == {
+            "type": "object",
+            "properties": {"libraryName": {"type": "string"}},
+            "required": ["libraryName"],
+        }
+
+
+def test_mcp_service_list_tools_handles_errors():
+    """Test that SSE parsing errors are handled gracefully"""
+    # GIVEN an MCP server configuration
+    sample_mcp_server = MCPServer(name="test-server", url="https://test.mcp.server.com/mcp")
+
+    # AND a mock that raises an exception during SSE parsing
+    with mock.patch("vellum.workflows.integrations.mcp_service.asyncio.run") as mock_run:
+        mock_run.side_effect = Exception("SSE parsing failed")
+
+        # WHEN we try to list tools
+        service = MCPService()
+        tools = service.list_tools(sample_mcp_server)
+
+        # THEN we should get an empty list instead of crashing
+        assert tools == []