letta-nightly 0.7.30.dev20250603104343__py3-none-any.whl → 0.8.0.dev20250604104349__py3-none-any.whl

letta/templates/sandbox_code_file.py.j2

@@ -0,0 +1,47 @@
+{{ 'from __future__ import annotations' if future_import else '' }}
+from typing import *
+import pickle
+import sys
+import base64
+import struct
+import hashlib
+
+{# Additional imports to support agent state #}
+{% if inject_agent_state %}
+import letta
+from letta import *
+{% endif %}
+
+{# Add schema code if available #}
+{{ schema_imports or ''}}
+
+{# Load agent state #}
+agent_state = {{ 'pickle.loads(' ~ agent_state_pickle ~ ')' if agent_state_pickle else 'None' }}
+
+{{ tool_args }}
+
+{# The tool's source code #}
+{{ tool_source_code }}
+
+{# Invoke the function and store the result in a global variable #}
+{{ local_sandbox_result_var_name }} = {
+    "results": {{ invoke_function_call }},
+    "agent_state": agent_state
+}
+
+{{ local_sandbox_result_var_name }}_pkl = pickle.dumps({{ local_sandbox_result_var_name }})
+
+{% if wrap_print_with_markers %}
+{# Combine everything to flush and write at once. #}
+data_checksum = hashlib.md5({{ local_sandbox_result_var_name }}_pkl).hexdigest().encode('ascii')
+{{ local_sandbox_result_var_name }}_msg = (
+  {{ start_marker }} +
+  struct.pack('>I', len({{ local_sandbox_result_var_name }}_pkl)) +
+  data_checksum +
+  {{ local_sandbox_result_var_name }}_pkl
+)
+sys.stdout.buffer.write({{ local_sandbox_result_var_name }}_msg)
+sys.stdout.buffer.flush()
+{% else %}
+base64.b64encode({{ local_sandbox_result_var_name }}_pkl).decode('utf-8')
+{% endif %}

letta/templates/template_helper.py

@@ -0,0 +1,16 @@
+import os
+
+from jinja2 import Environment, FileSystemLoader, StrictUndefined
+
+TEMPLATE_DIR = os.path.dirname(__file__)
+jinja_env = Environment(
+    loader=FileSystemLoader(TEMPLATE_DIR),
+    undefined=StrictUndefined,
+    trim_blocks=True,
+    lstrip_blocks=True,
+)
+
+
+def render_template(template_name: str, **kwargs):
+    template = jinja_env.get_template(template_name)
+    return template.render(**kwargs)

letta/tracing.py

@@ -16,6 +16,8 @@ from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
 from opentelemetry.trace import Status, StatusCode
 
+from letta import __version__ as letta_version
+
 tracer = trace.get_tracer(__name__)
 _is_tracing_initialized = False
 _excluded_v1_endpoints_regex: List[str] = [
@@ -24,6 +26,7 @@ _excluded_v1_endpoints_regex: List[str] = [
     # "^GET /v1/agents/(?P<agent_id>[^/]+)/archival-memory$",
     # "^GET /v1/agents/(?P<agent_id>[^/]+)/sources$",
     # r"^POST /v1/voice-beta/.*/chat/completions$",
+    "^GET /v1/health$",
 ]
 
 
@@ -80,6 +83,31 @@ async def update_trace_attributes(request: Request):
     if user_id:
         span.set_attribute("user.id", user_id)
 
+    # Add organization_id if available
+    organization_id = request.headers.get("x-organization-id")
+    if organization_id:
+        span.set_attribute("organization.id", organization_id)
+
+    # Add project_id if available
+    project_id = request.headers.get("x-project-id")
+    if project_id:
+        span.set_attribute("project.id", project_id)
+
+    # Add agent_id if available
+    agent_id = request.headers.get("x-agent-id")
+    if agent_id:
+        span.set_attribute("agent.id", agent_id)
+
+    # Add template_id if available
+    template_id = request.headers.get("x-template-id")
+    if template_id:
+        span.set_attribute("template.id", template_id)
+
+    # Add base_template_id if available
+    base_template_id = request.headers.get("x-base-template-id")
+    if base_template_id:
+        span.set_attribute("base_template.id", base_template_id)
+
     # Add request body if available
     try:
         body = await request.json()
@@ -124,7 +152,8 @@ def setup_tracing(
         resource=Resource.create(
             {
                 "service.name": service_name,
-                "device.id": uuid.getnode(),  # MAC address as unique device identifier
+                "device.id": uuid.getnode(),  # MAC address as unique device identifier,
+                "letta.version": letta_version,
             }
         )
     )

letta/types/__init__.py

@@ -0,0 +1,7 @@
+from typing import Any, TypeAlias
+
+from pydantic import JsonValue
+
+JsonDict: TypeAlias = dict[str, JsonValue]
+
+__all__ = ["JsonDict", "JsonValue"]

letta/utils.py

@@ -1018,7 +1018,7 @@ def sanitize_filename(filename: str) -> str:
         base = base[:max_base_length]
 
     # Append a unique UUID suffix for uniqueness
-    unique_suffix = uuid.uuid4().hex
+    unique_suffix = uuid.uuid4().hex[:4]
     sanitized_filename = f"{base}_{unique_suffix}{ext}"
 
     # Return the sanitized filename
@@ -1034,6 +1034,20 @@ def get_friendly_error_msg(function_name: str, exception_name: str, exception_me
     return error_msg
 
 
+def parse_stderr_error_msg(stderr_txt: str, last_n_lines: int = 3) -> tuple[str, str]:
+    """
+    Parses out from the last `last_n_line` of `stderr_txt` the Exception type and message.
+    """
+    index = -(last_n_lines + 1)
+    pattern = r"(\w+(?:Error|Exception)): (.+?)$"
+    for line in stderr_txt.split("\n")[:index:-1]:
+        if "Error" in line or "Exception" in line:
+            match = re.search(pattern, line)
+            if match:
+                return match.group(1), match.group(2)
+    return "", ""
+
+
 def run_async_task(coro: Coroutine[Any, Any, Any]) -> Any:
     """
     Safely runs an asynchronous coroutine in a synchronous context.
@@ -1074,3 +1088,13 @@ def log_telemetry(logger: Logger, event: str, **kwargs):
 
 def make_key(*args, **kwargs):
     return str((args, tuple(sorted(kwargs.items()))))
+
+
+def safe_create_task(coro, logger: Logger, label: str = "background task"):
+    async def wrapper():
+        try:
+            await coro
+        except Exception as e:
+            logger.exception(f"{label} failed with {type(e).__name__}: {e}")
+
+    return asyncio.create_task(wrapper())

{letta_nightly-0.7.30.dev20250603104343.dist-info → letta_nightly-0.8.0.dev20250604104349.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: letta-nightly
-Version: 0.7.30.dev20250603104343
+Version: 0.8.0.dev20250604104349
 Summary: Create LLM agents with long-term memory and custom tools
 License: Apache License
 Author: Letta Team
@@ -16,6 +16,7 @@ Provides-Extra: bedrock
 Provides-Extra: cloud-tool-sandbox
 Provides-Extra: desktop
 Provides-Extra: dev
+Provides-Extra: experimental
 Provides-Extra: external-tools
 Provides-Extra: google
 Provides-Extra: postgres
@@ -27,7 +28,8 @@ Requires-Dist: aiosqlite (>=0.21.0,<0.22.0)
 Requires-Dist: alembic (>=1.13.3,<2.0.0)
 Requires-Dist: anthropic (>=0.49.0,<0.50.0)
 Requires-Dist: apscheduler (>=3.11.0,<4.0.0)
-Requires-Dist: asyncpg (>=0.30.0,<0.31.0)
+Requires-Dist: async-lru (>=2.0.5,<3.0.0)
+Requires-Dist: asyncpg (>=0.30.0,<0.31.0) ; extra == "postgres"
 Requires-Dist: autoflake (>=2.3.0,<3.0.0) ; extra == "dev" or extra == "all"
 Requires-Dist: black[jupyter] (>=24.2.0,<25.0.0) ; extra == "dev" or extra == "all"
 Requires-Dist: boto3 (>=1.36.24,<2.0.0) ; extra == "bedrock"
@@ -43,6 +45,7 @@ Requires-Dist: faker (>=36.1.0,<37.0.0)
 Requires-Dist: fastapi (>=0.115.6,<0.116.0) ; extra == "server" or extra == "desktop" or extra == "all"
 Requires-Dist: firecrawl-py (>=1.15.0,<2.0.0)
 Requires-Dist: google-genai (>=1.15.0,<2.0.0) ; extra == "google"
+Requires-Dist: granian[reload,uvloop] (>=2.3.2,<3.0.0) ; extra == "experimental" or extra == "all"
 Requires-Dist: grpcio (>=1.68.1,<2.0.0)
 Requires-Dist: grpcio-tools (>=1.68.1,<2.0.0)
 Requires-Dist: html2text (>=2020.1.16,<2021.0.0)
@@ -59,6 +62,7 @@ Requires-Dist: locust (>=2.31.5,<3.0.0) ; extra == "dev" or extra == "desktop" o
 Requires-Dist: marshmallow-sqlalchemy (>=1.4.1,<2.0.0)
 Requires-Dist: matplotlib (>=3.10.1,<4.0.0)
 Requires-Dist: mcp (>=1.3.0,<2.0.0)
+Requires-Dist: mistralai (>=1.8.1,<2.0.0)
 Requires-Dist: nltk (>=3.8.1,<4.0.0)
 Requires-Dist: numpy (>=1.26.2,<2.0.0)
 Requires-Dist: openai (>=1.60.0,<2.0.0)
@@ -97,6 +101,7 @@ Requires-Dist: tavily-python (>=0.7.2,<0.8.0)
 Requires-Dist: tqdm (>=4.66.1,<5.0.0)
 Requires-Dist: typer[all] (>=0.9.0,<0.10.0)
 Requires-Dist: uvicorn (>=0.24.0.post1,<0.25.0) ; extra == "server" or extra == "desktop" or extra == "all"
+Requires-Dist: uvloop (>=0.21.0,<0.22.0) ; extra == "experimental" or extra == "all"
 Requires-Dist: wikipedia (>=1.4.0,<2.0.0) ; extra == "external-tools" or extra == "tests" or extra == "desktop" or extra == "all"
 Description-Content-Type: text/markdown