From c31521c7ad9e0dca3a4a4f18b6249927ea0fe10a Mon Sep 17 00:00:00 2001 From: cthomas Date: Tue, 7 Oct 2025 11:52:54 -0700 Subject: [PATCH] feat: add conditional importing for lettuce client (#5212) --- letta/server/rest_api/routers/v1/agents.py | 2 +- letta/server/rest_api/routers/v1/runs.py | 2 +- letta/services/lettuce/__init__.py | 6 ++++++ 3 files changed, 8 insertions(+), 2 deletions(-) create mode 100644 letta/services/lettuce/__init__.py diff --git a/letta/server/rest_api/routers/v1/agents.py b/letta/server/rest_api/routers/v1/agents.py index 8fe3c769..174306db 100644 --- a/letta/server/rest_api/routers/v1/agents.py +++ b/letta/server/rest_api/routers/v1/agents.py @@ -56,7 +56,7 @@ from letta.serialize_schemas.pydantic_agent_schema import AgentSchema from letta.server.rest_api.dependencies import HeaderParams, get_headers, get_letta_server from letta.server.rest_api.redis_stream_manager import create_background_stream_processor, redis_sse_stream_generator from letta.server.server import SyncServer -from letta.services.lettuce.lettuce_client import LettuceClient +from letta.services.lettuce import LettuceClient from letta.services.run_manager import RunManager from letta.settings import settings from letta.utils import safe_create_shielded_task, safe_create_task, truncate_file_visible_content diff --git a/letta/server/rest_api/routers/v1/runs.py b/letta/server/rest_api/routers/v1/runs.py index 067afbf3..913a2a89 100644 --- a/letta/server/rest_api/routers/v1/runs.py +++ b/letta/server/rest_api/routers/v1/runs.py @@ -22,7 +22,7 @@ from letta.server.rest_api.streaming_response import ( cancellation_aware_stream_wrapper, ) from letta.server.server import SyncServer -from letta.services.lettuce.lettuce_client import LettuceClient +from letta.services.lettuce import LettuceClient from letta.services.run_manager import RunManager from letta.settings import settings diff --git a/letta/services/lettuce/__init__.py b/letta/services/lettuce/__init__.py new file mode 100644 index 00000000..f8e75a73 --- /dev/null +++ b/letta/services/lettuce/__init__.py @@ -0,0 +1,6 @@ +try: + from .lettuce_client import LettuceClient +except ImportError: + from .lettuce_client_base import LettuceClient + +__all__ = ["LettuceClient"]