From 7debadb3b9e5ae6e511f2b7a42cc3fc1d16cd3ae Mon Sep 17 00:00:00 2001 From: Matthew Zhou Date: Wed, 28 May 2025 10:24:22 -0700 Subject: [PATCH] fix: Change enum to fix composio tests (#2488) --- letta/interfaces/anthropic_streaming_interface.py | 1 - letta/interfaces/openai_streaming_interface.py | 1 - tests/test_tool_schema_parsing.py | 2 +- 3 files changed, 1 insertion(+), 3 deletions(-) diff --git a/letta/interfaces/anthropic_streaming_interface.py b/letta/interfaces/anthropic_streaming_interface.py index 70cb249f..3d9c24da 100644 --- a/letta/interfaces/anthropic_streaming_interface.py +++ b/letta/interfaces/anthropic_streaming_interface.py @@ -38,7 +38,6 @@ from letta.schemas.letta_message_content import ReasoningContent, RedactedReason from letta.schemas.message import Message from letta.schemas.openai.chat_completion_response import FunctionCall, ToolCall from letta.server.rest_api.json_parser import JSONParser, PydanticJSONParser -from letta.tracing import trace_method logger = get_logger(__name__) diff --git a/letta/interfaces/openai_streaming_interface.py b/letta/interfaces/openai_streaming_interface.py index 13c31087..86397ad8 100644 --- a/letta/interfaces/openai_streaming_interface.py +++ b/letta/interfaces/openai_streaming_interface.py @@ -12,7 +12,6 @@ from letta.schemas.message import Message from letta.schemas.openai.chat_completion_response import FunctionCall, ToolCall from letta.server.rest_api.json_parser import OptimisticJSONParser from letta.streaming_utils import JSONInnerThoughtsExtractor -from letta.tracing import trace_method class OpenAIStreamingInterface: diff --git a/tests/test_tool_schema_parsing.py b/tests/test_tool_schema_parsing.py index 8273eb8f..08ad8052 100644 --- a/tests/test_tool_schema_parsing.py +++ b/tests/test_tool_schema_parsing.py @@ -289,7 +289,7 @@ def test_composio_tool_schema_generation(openai_model: str, structured_output: b action_names = [ "GITHUB_STAR_A_REPOSITORY_FOR_THE_AUTHENTICATED_USER", # Simple "CAL_GET_AVAILABLE_SLOTS_INFO", # has an array arg, needs to be converted properly - "SALESFORCE_RETRIEVE_LEAD_DETAILS_BY_ID_WITH_CONDITIONAL_SUPPORT", + "SALESFORCE_RETRIEVE_LEAD_BY_ID", # has an array arg, needs to be converted properly ]