letta-nightly 0.8.11.dev20250708104317__py3-none-any.whl → 0.8.12.dev20250709104346__py3-none-any.whl

letta/__init__.py

@@ -5,8 +5,7 @@ try:
     __version__ = version("letta")
 except PackageNotFoundError:
     # Fallback for development installations
-    __version__ = "0.8.11"
-
+    __version__ = "0.8.12"
 
 if os.environ.get("LETTA_VERSION"):
     __version__ = os.environ["LETTA_VERSION"]

letta/jobs/llm_batch_job_polling.py

@@ -6,6 +6,7 @@ from letta.agents.letta_agent_batch import LettaAgentBatch
 from letta.jobs.helpers import map_anthropic_batch_job_status_to_job_status, map_anthropic_individual_batch_item_status_to_job_status
 from letta.jobs.types import BatchPollingResult, ItemUpdateInfo
 from letta.log import get_logger
+from letta.otel.tracing import trace_method
 from letta.schemas.enums import JobStatus, ProviderType
 from letta.schemas.letta_response import LettaBatchResponse
 from letta.schemas.llm_batch_job import LLMBatchJob
@@ -37,6 +38,7 @@ class BatchPollingMetrics:
         logger.info(f"[Poll BatchJob] Updated {self.updated_items_count} items for newly completed batch(es).")
 
 
+@trace_method
 async def fetch_batch_status(server: SyncServer, batch_job: LLMBatchJob) -> BatchPollingResult:
     """
     Fetch the current status of a single batch job from the provider.
@@ -60,6 +62,7 @@ async def fetch_batch_status(server: SyncServer, batch_job: LLMBatchJob) -> Batc
         return BatchPollingResult(batch_job.id, JobStatus.running, None)
 
 
+@trace_method
 async def fetch_batch_items(server: SyncServer, batch_id: str, batch_resp_id: str) -> List[ItemUpdateInfo]:
     """
     Fetch individual item results for a completed batch.
@@ -86,6 +89,7 @@ async def fetch_batch_items(server: SyncServer, batch_id: str, batch_resp_id: st
     return updates
 
 
+@trace_method
 async def poll_batch_updates(server: SyncServer, batch_jobs: List[LLMBatchJob], metrics: BatchPollingMetrics) -> List[BatchPollingResult]:
     """
     Poll for updates to multiple batch jobs concurrently.
@@ -113,6 +117,7 @@ async def poll_batch_updates(server: SyncServer, batch_jobs: List[LLMBatchJob],
     return results
 
 
+@trace_method
 async def process_completed_batches(
     server: SyncServer, batch_results: List[BatchPollingResult], metrics: BatchPollingMetrics
 ) -> List[ItemUpdateInfo]:
@@ -161,6 +166,7 @@ async def process_completed_batches(
     return item_updates
 
 
+@trace_method
 async def poll_running_llm_batches(server: "SyncServer") -> List[LettaBatchResponse]:
     """
     Cron job to poll all running LLM batch jobs and update their polling responses in bulk.

letta/jobs/scheduler.py

@@ -45,6 +45,7 @@ async def _try_acquire_lock_and_start_scheduler(server: SyncServer) -> bool:
                     text("SELECT pg_try_advisory_lock(CAST(:lock_key AS bigint))"), {"lock_key": ADVISORY_LOCK_KEY}
                 )
                 acquired_lock = result.scalar()
+                await lock_session.commit()
 
         if not acquired_lock:
             if lock_session:
@@ -90,7 +91,7 @@ async def _try_acquire_lock_and_start_scheduler(server: SyncServer) -> bool:
             logger.warning("Attempting to release lock due to error during startup.")
             try:
                 _advisory_lock_session = lock_session
-                await _release_advisory_lock()
+                await _release_advisory_lock(lock_session)
             except Exception as unlock_err:
                 logger.error(f"Failed to release lock during error handling: {unlock_err}", exc_info=True)
             finally:
@@ -140,11 +141,11 @@ async def _background_lock_retry_loop(server: SyncServer):
             await asyncio.sleep(settings.poll_lock_retry_interval_seconds)
 
 
-async def _release_advisory_lock():
+async def _release_advisory_lock(lock_session=None):
     """Releases the advisory lock using the stored session."""
     global _advisory_lock_session
 
-    lock_session = _advisory_lock_session
+    lock_session = _advisory_lock_session or lock_session
     _advisory_lock_session = None
 
     if lock_session is not None:
@@ -152,6 +153,7 @@ async def _release_advisory_lock():
         try:
             await lock_session.execute(text("SELECT pg_advisory_unlock(CAST(:lock_key AS bigint))"), {"lock_key": ADVISORY_LOCK_KEY})
             logger.info(f"Executed pg_advisory_unlock for lock {ADVISORY_LOCK_KEY}")
+            await lock_session.commit()
         except Exception as e:
             logger.error(f"Error executing pg_advisory_unlock: {e}", exc_info=True)
         finally:

letta/server/rest_api/routers/v1/blocks.py

@@ -101,6 +101,14 @@ async def retrieve_block(
 @router.get("/{block_id}/agents", response_model=List[AgentState], operation_id="list_agents_for_block")
 async def list_agents_for_block(
     block_id: str,
+    include_relationships: list[str] | None = Query(
+        None,
+        description=(
+            "Specify which relational fields (e.g., 'tools', 'sources', 'memory') to include in the response. "
+            "If not provided, all relationships are loaded by default. "
+            "Using this can optimize performance by reducing unnecessary joins."
+        ),
+    ),
     server: SyncServer = Depends(get_letta_server),
     actor_id: Optional[str] = Header(None, alias="user_id"),
 ):
@@ -110,7 +118,9 @@ async def list_agents_for_block(
     """
     actor = await server.user_manager.get_actor_or_default_async(actor_id=actor_id)
     try:
-        agents = await server.block_manager.get_agents_for_block_async(block_id=block_id, actor=actor)
+        agents = await server.block_manager.get_agents_for_block_async(
+            block_id=block_id, include_relationships=include_relationships, actor=actor
+        )
         return agents
     except NoResultFound:
         raise HTTPException(status_code=404, detail=f"Block with id={block_id} not found")

letta/services/block_manager.py

@@ -286,14 +286,20 @@ class BlockManager:
 
     @trace_method
     @enforce_types
-    async def get_agents_for_block_async(self, block_id: str, actor: PydanticUser) -> List[PydanticAgentState]:
+    async def get_agents_for_block_async(
+        self,
+        block_id: str,
+        actor: PydanticUser,
+        include_relationships: Optional[List[str]] = None,
+    ) -> List[PydanticAgentState]:
         """
         Retrieve all agents associated with a given block.
         """
         async with db_registry.async_session() as session:
             block = await BlockModel.read_async(db_session=session, identifier=block_id, actor=actor)
             agents_orm = block.agents
-            return await asyncio.gather(*[agent.to_pydantic_async() for agent in agents_orm])
+            agents = await asyncio.gather(*[agent.to_pydantic_async(include_relationships=include_relationships) for agent in agents_orm])
+            return agents
 
     @trace_method
     @enforce_types

letta/services/tool_sandbox/local_sandbox.py

@@ -158,6 +158,7 @@ class AsyncToolSandboxLocal(AsyncToolSandboxBase):
             if not settings.debug:
                 os.remove(temp_file_path)
 
+    @trace_method
     async def _prepare_venv(self, local_configs, venv_path: str, env: Dict[str, str]):
         """
         Prepare virtual environment asynchronously (in a background thread).
@@ -174,11 +175,12 @@ class AsyncToolSandboxLocal(AsyncToolSandboxBase):
             )
             log_event(name="finish create_venv_for_local_sandbox")
 
-        log_event(name="start install_pip_requirements_for_sandbox", attributes={"local_configs": local_configs.model_dump_json()})
-        await asyncio.to_thread(
-            install_pip_requirements_for_sandbox, local_configs, upgrade=True, user_install_if_no_venv=False, env=env, tool=self.tool
-        )
-        log_event(name="finish install_pip_requirements_for_sandbox", attributes={"local_configs": local_configs.model_dump_json()})
+        if local_configs.pip_requirements or (self.tool and self.tool.pip_requirements):
+            log_event(name="start install_pip_requirements_for_sandbox", attributes={"local_configs": local_configs.model_dump_json()})
+            await asyncio.to_thread(
+                install_pip_requirements_for_sandbox, local_configs, upgrade=True, user_install_if_no_venv=False, env=env, tool=self.tool
+            )
+            log_event(name="finish install_pip_requirements_for_sandbox", attributes={"local_configs": local_configs.model_dump_json()})
 
     @trace_method
     async def _execute_tool_subprocess(

{letta_nightly-0.8.11.dev20250708104317.dist-info → letta_nightly-0.8.12.dev20250709104346.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: letta-nightly
-Version: 0.8.11.dev20250708104317
+Version: 0.8.12.dev20250709104346
 Summary: Create LLM agents with long-term memory and custom tools
 License: Apache License
 Author: Letta Team

{letta_nightly-0.8.11.dev20250708104317.dist-info → letta_nightly-0.8.12.dev20250709104346.dist-info}/RECORD

@@ -1,4 +1,4 @@
-letta/__init__.py,sha256=MY4Eqs94gdtwy8X6KBPrk0zKk7LUqwyQWvQU-ssrpes,1223
+letta/__init__.py,sha256=O18yG1Lw_7vBqjWzHjZ1QG1xU3TTjRvuxL3iG4G-GxM,1222
 letta/agent.py,sha256=esW2W5hBzO7aPr7ghEDb_fLnUxgYqBYDq_VWtQDrB0c,89153
 letta/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/agents/base_agent.py,sha256=Z1jgCTMFRTLnaLRcfdo8TmsP8tuCYqNcOM8ov9kviMA,6869
@@ -74,8 +74,8 @@ letta/interfaces/openai_streaming_interface.py,sha256=FtVFHCw2KTwqqNAlCsggpINXX9
 letta/interfaces/utils.py,sha256=c6jvO0dBYHh8DQnlN-B0qeNC64d3CSunhfqlFA4pJTY,278
 letta/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/jobs/helpers.py,sha256=kO4aj954xsQ1RAmkjY6LQQ7JEIGuhaxB1e9pzrYKHAY,914
-letta/jobs/llm_batch_job_polling.py,sha256=r_6D5RcqEJQgrdh-rnN7vdLD0GAQl-GGmIfCnV0naHQ,10299
-letta/jobs/scheduler.py,sha256=bnwvgT_72ULlKmSFG-26T6tfmiEzxSrm9ARl1MuOEvA,8818
+letta/jobs/llm_batch_job_polling.py,sha256=HUCTa1lTOiLAB_8m95RUfeNJa4lxlF8paGdCV1NqOeA,10413
+letta/jobs/scheduler.py,sha256=_KkOp0uL8C7WpZEwCvHYtvtEjU6RBFXzcS8dnv7fhUM,8947
 letta/jobs/types.py,sha256=K8GKEnqEgAT6Kq4F2hUrBC4ZAFM9OkfOjVMStzxKuXQ,742
 letta/llm_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/llm_api/anthropic.py,sha256=tbMy4483TySrEmbXD3juM6TpPRrV9_M3Fgp59sDBcqE,47935
@@ -299,7 +299,7 @@ letta/server/rest_api/routers/openai/chat_completions/__init__.py,sha256=47DEQpj
 letta/server/rest_api/routers/openai/chat_completions/chat_completions.py,sha256=QBWab1fn2LXVDMtc6li3gOzmrNzDiUw5WUJsMeeMZII,5076
 letta/server/rest_api/routers/v1/__init__.py,sha256=JfSSttkEWu0W18NVVDxl8AGnd8Qhj0BXJNxntOB7070,1768
 letta/server/rest_api/routers/v1/agents.py,sha256=i0zS-tEZlMf2gfVy-NOq-4mgicJHQBpqRJTLsRUU6L4,52709
-letta/server/rest_api/routers/v1/blocks.py,sha256=bxezefb92YWkM9mTuamhqq9AQPPRwhKeCejFKJqwD9s,4817
+letta/server/rest_api/routers/v1/blocks.py,sha256=MArBBnC7k9bc-Z1xMf46aH4ij6qhKAQAOoK9KjiXatU,5257
 letta/server/rest_api/routers/v1/embeddings.py,sha256=PRaQlrmEXPiIdWsTbadrFsv3Afyv5oEFUdhgHA8FTi8,989
 letta/server/rest_api/routers/v1/groups.py,sha256=kR_oAuwPd9q-DaeK4Q6Xqu1XlXTXkwEvf2hH7tOiVuw,10978
 letta/server/rest_api/routers/v1/health.py,sha256=MoOjkydhGcJXTiuJrKIB0etVXiRMdTa51S8RQ8-50DQ,399
@@ -336,7 +336,7 @@ letta/server/ws_api/protocol.py,sha256=5mDgpfNZn_kNwHnpt5Dsuw8gdNH298sgxTGed3etz
 letta/server/ws_api/server.py,sha256=cBSzf-V4zT1bL_0i54OTI3cMXhTIIxqjSRF8pYjk7fg,5835
 letta/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/services/agent_manager.py,sha256=7xdWIchSUmlPCnNRsXjzd-9EKyk05t9WpGaerMwJLkg,121953
-letta/services/block_manager.py,sha256=YwDGdy6f6MNXVXVOxIMOOP6IEWT8h-k5uQlveof0pyE,22744
+letta/services/block_manager.py,sha256=7EliXd0-LpSRwD2LbyjFpH5uiBdrtdZ6YLgb2_wKs3o,22905
 letta/services/context_window_calculator/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/services/context_window_calculator/context_window_calculator.py,sha256=H0-Ello1DHV28MnzMseWrg--jarDc6YwCcgwPlWjtZk,6527
 letta/services/context_window_calculator/token_counter.py,sha256=Ai9-aPkNvhhMTj9zlvdiQAdVqroTzIyAn0TrHpHNQZY,2954
@@ -397,7 +397,7 @@ letta/services/tool_manager.py,sha256=6VI3mjXa6-vrPA74Aar_-ti9Rn3EEBq5TIiayyzoVA
 letta/services/tool_sandbox/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 letta/services/tool_sandbox/base.py,sha256=Vt4CnxuY5otUD6Kv8PpJNrAtl9eI8tjfcwkOdtUFwKg,7917
 letta/services/tool_sandbox/e2b_sandbox.py,sha256=TrWWav56H1AsnaKgNZuq0RI-FeWHOZvOubtUywPH72s,11125
-letta/services/tool_sandbox/local_sandbox.py,sha256=RQ3iSZqP1nndSdU8pN8GoNDbQr8PRuRIK2-BinIYtK4,11810
+letta/services/tool_sandbox/local_sandbox.py,sha256=mDMoBUl0asu03p7gp2BlzBCna68esMqIRwWF_dzteCA,11937
 letta/services/user_manager.py,sha256=Neik-mxXgf9jc9jBiiBIlK38UukJonUy9NRS2soFR98,10405
 letta/settings.py,sha256=lWaLL1t06s9pp4VK5ojQvTiI7D85VcJPjaVKak5LFxs,11304
 letta/streaming_interface.py,sha256=c-T7zoMTXGXFwDWJJXrv7UypeMPXwPOmNHeuuh0b9zk,16398
@@ -410,8 +410,8 @@ letta/templates/summary_request_text.j2,sha256=ZttQwXonW2lk4pJLYzLK0pmo4EO4EtUUI
 letta/templates/template_helper.py,sha256=uHWO1PukgMoIIvgqQdPyHq3o3CQ6mcjUjTGvx9VLGkk,409
 letta/types/__init__.py,sha256=hokKjCVFGEfR7SLMrtZsRsBfsC7yTIbgKPLdGg4K1eY,147
 letta/utils.py,sha256=4segcFYPNsPrzMpiouYoV6Qzj4TIHuqtCyzVwAMildM,36172
-letta_nightly-0.8.11.dev20250708104317.dist-info/LICENSE,sha256=mExtuZ_GYJgDEI38GWdiEYZizZS4KkVt2SF1g_GPNhI,10759
-letta_nightly-0.8.11.dev20250708104317.dist-info/METADATA,sha256=oeNwjzCwH4x7dMt85jJZ80dzNHIpOIC4rMrHz0bNpdQ,22892
-letta_nightly-0.8.11.dev20250708104317.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-letta_nightly-0.8.11.dev20250708104317.dist-info/entry_points.txt,sha256=2zdiyGNEZGV5oYBuS-y2nAAgjDgcC9yM_mHJBFSRt5U,40
-letta_nightly-0.8.11.dev20250708104317.dist-info/RECORD,,
+letta_nightly-0.8.12.dev20250709104346.dist-info/LICENSE,sha256=mExtuZ_GYJgDEI38GWdiEYZizZS4KkVt2SF1g_GPNhI,10759
+letta_nightly-0.8.12.dev20250709104346.dist-info/METADATA,sha256=NBLNREmyI2WWhwKP0V7qqVZ7ucr040lLC5g1HGmhTiY,22892
+letta_nightly-0.8.12.dev20250709104346.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+letta_nightly-0.8.12.dev20250709104346.dist-info/entry_points.txt,sha256=2zdiyGNEZGV5oYBuS-y2nAAgjDgcC9yM_mHJBFSRt5U,40
+letta_nightly-0.8.12.dev20250709104346.dist-info/RECORD,,