letta-nightly 0.5.4.dev20241203104336__tar.gz → 0.5.4.dev20241204014655__tar.gz

{letta_nightly-0.5.4.dev20241203104336 → letta_nightly-0.5.4.dev20241204014655}/PKG-INFO

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

{letta_nightly-0.5.4.dev20241203104336 → letta_nightly-0.5.4.dev20241204014655}/letta/client/client.py

@@ -434,6 +434,7 @@ class RESTClient(AbstractClient):
         debug: bool = False,
         default_llm_config: Optional[LLMConfig] = None,
         default_embedding_config: Optional[EmbeddingConfig] = None,
+        headers: Optional[Dict] = None,
     ):
         """
         Initializes a new instance of Client class.
@@ -442,12 +443,16 @@ class RESTClient(AbstractClient):
             auto_save (bool): Whether to automatically save changes.
             user_id (str): The user ID.
             debug (bool): Whether to print debug information.
-            default
+            default_llm_config (Optional[LLMConfig]): The default LLM configuration.
+            default_embedding_config (Optional[EmbeddingConfig]): The default embedding configuration.
+            headers (Optional[Dict]): The additional headers for the REST API.
         """
         super().__init__(debug=debug)
         self.base_url = base_url
         self.api_prefix = api_prefix
         self.headers = {"accept": "application/json", "authorization": f"Bearer {token}"}
+        if headers:
+            self.headers.update(headers)
         self._default_llm_config = default_llm_config
         self._default_embedding_config = default_embedding_config
 

{letta_nightly-0.5.4.dev20241203104336 → letta_nightly-0.5.4.dev20241204014655}/letta/schemas/tool.py

@@ -209,7 +209,7 @@ class ToolRun(LettaBase):
 
 
 class ToolRunFromSource(LettaBase):
+    source_code: str = Field(..., description="The source code of the function.")
     args: str = Field(..., description="The arguments to pass to the tool (as stringified JSON).")
-    name: Optional[str] = Field(..., description="The name of the tool to run.")
-    source_code: str = Field(None, description="The source code of the function.")
+    name: Optional[str] = Field(None, description="The name of the tool to run.")
     source_type: Optional[str] = Field(None, description="The type of the source code.")

{letta_nightly-0.5.4.dev20241203104336 → letta_nightly-0.5.4.dev20241204014655}/letta/server/rest_api/routers/v1/sources.py

@@ -37,7 +37,10 @@ def get_source(
     """
     actor = server.get_user_or_default(user_id=user_id)
 
-    return server.source_manager.get_source_by_id(source_id=source_id, actor=actor)
+    source = server.source_manager.get_source_by_id(source_id=source_id, actor=actor)
+    if not source:
+        raise HTTPException(status_code=404, detail=f"Source with id={source_id} not found.")
+    return source
 
 
 @router.get("/name/{source_name}", response_model=str, operation_id="get_source_id_by_name")
@@ -52,6 +55,8 @@ def get_source_id_by_name(
     actor = server.get_user_or_default(user_id=user_id)
 
     source = server.source_manager.get_source_by_name(source_name=source_name, actor=actor)
+    if not source:
+        raise HTTPException(status_code=404, detail=f"Source with name={source_name} not found.")
     return source.id
 
 
@@ -94,6 +99,8 @@ def update_source(
     Update the name or documentation of an existing data source.
     """
     actor = server.get_user_or_default(user_id=user_id)
+    if not server.source_manager.get_source_by_id(source_id=source_id, actor=actor):
+        raise HTTPException(status_code=404, detail=f"Source with id={source_id} does not exist.")
     return server.source_manager.update_source(source_id=source_id, source_update=source, actor=actor)
 
 

{letta_nightly-0.5.4.dev20241203104336 → letta_nightly-0.5.4.dev20241204014655}/letta/server/rest_api/routers/v1/tools.py

@@ -186,13 +186,27 @@ def run_tool_from_source(
     """
     actor = server.get_user_or_default(user_id=user_id)
 
-    return server.run_tool_from_source(
-        tool_source=request.source_code,
-        tool_source_type=request.source_type,
-        tool_args=request.args,
-        tool_name=request.name,
-        user_id=actor.id,
-    )
+    try:
+        return server.run_tool_from_source(
+            tool_source=request.source_code,
+            tool_source_type=request.source_type,
+            tool_args=request.args,
+            tool_name=request.name,
+            user_id=actor.id,
+        )
+    except LettaToolCreateError as e:
+        # HTTP 400 == Bad Request
+        print(f"Error occurred during tool creation: {e}")
+        # print the full stack trace
+        import traceback
+
+        print(traceback.format_exc())
+        raise HTTPException(status_code=400, detail=str(e))
+
+    except Exception as e:
+        # Catch other unexpected errors and raise an internal server error
+        print(f"Unexpected error occurred: {e}")
+        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")
 
 
 # Specific routes for Composio

{letta_nightly-0.5.4.dev20241203104336 → letta_nightly-0.5.4.dev20241204014655}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "letta-nightly"
-version = "0.5.4.dev20241203104336"
+version = "0.5.4.dev20241204014655"
 packages = [
     {include = "letta"}
 ]