letta-nightly 0.12.1.dev20251024104217__py3-none-any.whl → 0.13.0.dev20251024223017__py3-none-any.whl

letta/validators.py

@@ -0,0 +1,105 @@
+import inspect
+import re
+from functools import wraps
+from typing import Annotated
+
+from fastapi import Path
+
+from letta.errors import LettaInvalidArgumentError
+from letta.schemas.enums import PrimitiveType  # PrimitiveType is now in schemas.enums
+
+# Map from PrimitiveType to the actual prefix string (which is just the enum value)
+PRIMITIVE_ID_PREFIXES = {primitive_type: primitive_type.value for primitive_type in PrimitiveType}
+
+
+PRIMITIVE_ID_PATTERNS = {
+    # f-string interpolation gets confused because of the regex's required curly braces {}
+    prefix: re.compile("^" + prefix + "-[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$")
+    for prefix in PRIMITIVE_ID_PREFIXES.values()
+}
+
+
+def _create_path_validator_factory(primitive: str):
+    """
+    Creates a factory function that returns a fresh Path validator.
+
+    This avoids shared state issues when the same validator is used
+    across multiple endpoints with different parameter names.
+    """
+
+    def factory():
+        return Path(
+            description=f"The ID of the {primitive} in the format '{primitive}-<uuid4>'",
+            pattern=PRIMITIVE_ID_PATTERNS[primitive].pattern,
+            examples=[f"{primitive}-123e4567-e89b-42d3-8456-426614174000"],
+            min_length=len(primitive) + 1 + 36,
+            max_length=len(primitive) + 1 + 36,
+        )
+
+    return factory
+
+
+# PATH_VALIDATORS now contains factory functions, not Path objects
+# Usage: folder_id: str = PATH_VALIDATORS[PrimitiveType.FOLDER.value]()
+PATH_VALIDATORS = {primitive_type.value: _create_path_validator_factory(primitive_type.value) for primitive_type in PrimitiveType}
+
+
+# Type aliases for common ID types
+# These can be used directly in route handler signatures for cleaner code
+AgentId = Annotated[str, PATH_VALIDATORS[PrimitiveType.AGENT.value]()]
+ToolId = Annotated[str, PATH_VALIDATORS[PrimitiveType.TOOL.value]()]
+SourceId = Annotated[str, PATH_VALIDATORS[PrimitiveType.SOURCE.value]()]
+BlockId = Annotated[str, PATH_VALIDATORS[PrimitiveType.BLOCK.value]()]
+MessageId = Annotated[str, PATH_VALIDATORS[PrimitiveType.MESSAGE.value]()]
+RunId = Annotated[str, PATH_VALIDATORS[PrimitiveType.RUN.value]()]
+JobId = Annotated[str, PATH_VALIDATORS[PrimitiveType.JOB.value]()]
+GroupId = Annotated[str, PATH_VALIDATORS[PrimitiveType.GROUP.value]()]
+FileId = Annotated[str, PATH_VALIDATORS[PrimitiveType.FILE.value]()]
+FolderId = Annotated[str, PATH_VALIDATORS[PrimitiveType.FOLDER.value]()]
+ArchiveId = Annotated[str, PATH_VALIDATORS[PrimitiveType.ARCHIVE.value]()]
+PassageId = Annotated[str, PATH_VALIDATORS[PrimitiveType.PASSAGE.value]()]
+ProviderId = Annotated[str, PATH_VALIDATORS[PrimitiveType.PROVIDER.value]()]
+SandboxConfigId = Annotated[str, PATH_VALIDATORS[PrimitiveType.SANDBOX_CONFIG.value]()]
+StepId = Annotated[str, PATH_VALIDATORS[PrimitiveType.STEP.value]()]
+IdentityId = Annotated[str, PATH_VALIDATORS[PrimitiveType.IDENTITY.value]()]
+
+
+def raise_on_invalid_id(param_name: str, expected_prefix: PrimitiveType):
+    """
+    Decorator that validates an ID parameter has the expected prefix format.
+    Can be stacked multiple times on the same function to validate different IDs.
+
+    Args:
+        param_name: The name of the function parameter to validate (e.g., "agent_id")
+        expected_prefix: The expected primitive type (e.g., PrimitiveType.AGENT)
+
+    Example:
+        @raise_on_invalid_id(param_name="agent_id", expected_prefix=PrimitiveType.AGENT)
+        @raise_on_invalid_id(param_name="folder_id", expected_prefix=PrimitiveType.FOLDER)
+        def my_function(agent_id: str, folder_id: str):
+            pass
+    """
+
+    def decorator(function):
+        @wraps(function)
+        def wrapper(*args, **kwargs):
+            sig = inspect.signature(function)
+            bound_args = sig.bind(*args, **kwargs)
+            bound_args.apply_defaults()
+
+            if param_name in bound_args.arguments:
+                arg_value = bound_args.arguments[param_name]
+
+                if arg_value is not None:
+                    prefix = PRIMITIVE_ID_PREFIXES[expected_prefix]
+                    if PRIMITIVE_ID_PATTERNS[prefix].match(arg_value) is None:
+                        raise LettaInvalidArgumentError(
+                            message=f"Invalid {expected_prefix.value} ID format: {arg_value}. Expected format: '{prefix}-<uuid4>'",
+                            argument_name=param_name,
+                        )
+
+            return function(*args, **kwargs)
+
+        return wrapper
+
+    return decorator

{letta_nightly-0.12.1.dev20251024104217.dist-info → letta_nightly-0.13.0.dev20251024223017.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: letta-nightly
-Version: 0.12.1.dev20251024104217
+Version: 0.13.0.dev20251024223017
 Summary: Create LLM agents with long-term memory and custom tools
 Author-email: Letta Team <contact@letta.com>
 License: Apache License
@@ -99,7 +99,6 @@ Requires-Dist: pytest-json-report>=1.5.0; extra == 'dev'
 Requires-Dist: pytest-mock>=3.14.0; extra == 'dev'
 Requires-Dist: pytest-order>=1.2.0; extra == 'dev'
 Provides-Extra: experimental
-Requires-Dist: google-cloud-profiler>=4.1.0; extra == 'experimental'
 Requires-Dist: granian[reload,uvloop]>=2.3.2; extra == 'experimental'
 Requires-Dist: uvloop>=0.21.0; extra == 'experimental'
 Provides-Extra: external-tools
@@ -119,6 +118,9 @@ Requires-Dist: pg8000>=1.30.3; extra == 'postgres'
 Requires-Dist: pgvector>=0.2.3; extra == 'postgres'
 Requires-Dist: psycopg2-binary>=2.9.10; extra == 'postgres'
 Requires-Dist: psycopg2>=2.9.10; extra == 'postgres'
+Provides-Extra: profiling
+Requires-Dist: ddtrace>=2.18.2; extra == 'profiling'
+Requires-Dist: google-cloud-profiler>=4.1.0; extra == 'profiling'
 Provides-Extra: redis
 Requires-Dist: redis>=6.2.0; extra == 'redis'
 Provides-Extra: server