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 ]