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"]