letta-client 0.1.77__py3-none-any.whl → 0.1.78__py3-none-any.whl

letta_client/client.py

@@ -7,6 +7,7 @@ from abc import abstractmethod
 from .base_client import AsyncLettaBase, LettaBase
 from .core.request_options import RequestOptions
 from .tools.client import ToolsClient as ToolsClientBase
+from .tools.client import AsyncToolsClient as AsyncToolsClientBase
 from .types.tool import Tool
 
 # this is used as the default value for optional parameters
@@ -24,7 +25,7 @@ class AsyncLetta(AsyncLettaBase):
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
-        self.tools = ToolsClient(client_wrapper=self._client_wrapper)
+        self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)
 
 
 class BaseTool(Tool):
@@ -380,3 +381,258 @@ class ToolsClient(ToolsClientBase):
             return_char_limit=tool.return_char_limit or OMIT,
             request_options=request_options,
         )
+    
+
+class AsyncToolsClient(AsyncToolsClientBase):
+
+    async def create_from_function(
+        self,
+        *,
+        func: typing.Callable,
+        args_schema: typing.Optional[typing.Type[BaseModel]] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        source_type: typing.Optional[str] = OMIT,
+        json_schema: typing.Optional[
+            typing.Dict[str, typing.Optional[typing.Any]]
+        ] = OMIT,
+        return_char_limit: typing.Optional[int] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Tool:
+        """
+        Create a new tool from a callable
+
+        Parameters
+        ----------
+        func : typing.Callable
+            The callable to create the tool from.
+        
+        args_schema : typing.Optional[typing.Type[BaseModel]]
+            The arguments schema of the function, as a Pydantic model.
+
+        description : typing.Optional[str]
+            The description of the tool.
+
+        tags : typing.Optional[typing.Sequence[str]]
+            Metadata tags.
+
+        source_type : typing.Optional[str]
+            The source type of the function.
+
+        json_schema : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            The JSON schema of the function (auto-generated from source_code if not provided)
+
+        return_char_limit : typing.Optional[int]
+            The maximum number of characters in the response.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Tool
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        
+        def add_two_numbers(a: int, b: int) -> int:
+            return a + b
+        
+        await client.tools.create_from_function(
+            func=add_two_numbers,
+        )
+
+        class InventoryEntryData(BaseModel):
+            data: InventoryEntry
+            quantity_change: int
+
+        def manage_inventory(data: InventoryEntry, quantity_change: int) -> bool:
+            pass
+        
+        await client.tools.create_from_function(
+            func=manage_inventory,
+            args_schema=InventoryEntryData,
+        )
+        """
+        source_code = dedent(inspect.getsource(func))
+        args_json_schema = args_schema.model_json_schema() if args_schema and args_schema != OMIT else None
+        return await self.create(
+            source_code=source_code,
+            args_json_schema=args_json_schema,
+            description=description,
+            tags=tags,
+            source_type=source_type,
+            json_schema=json_schema,
+            return_char_limit=return_char_limit,
+            request_options=request_options,
+        )
+
+
+    async def upsert_from_function(
+        self,
+        *,
+        func: typing.Callable,
+        args_schema: typing.Optional[typing.Type[BaseModel]] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        source_type: typing.Optional[str] = OMIT,
+        json_schema: typing.Optional[
+            typing.Dict[str, typing.Optional[typing.Any]]
+        ] = OMIT,
+        return_char_limit: typing.Optional[int] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Tool:
+        """
+        Create or update a tool from a callable
+
+        Parameters
+        ----------
+        func : typing.Callable
+            The callable to create or update the tool from.
+        
+        args_schema : typing.Optional[typing.Type[BaseModel]]
+            The arguments schema of the function, as a Pydantic model.
+
+        description : typing.Optional[str]
+            The description of the tool.
+
+        tags : typing.Optional[typing.Sequence[str]]
+            Metadata tags.
+
+        source_type : typing.Optional[str]
+            The source type of the function.
+
+        json_schema : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            The JSON schema of the function (auto-generated from source_code if not provided)
+
+        return_char_limit : typing.Optional[int]
+            The maximum number of characters in the response.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Tool
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+
+        def add_two_numbers(a: int, b: int) -> int:
+            return a + b
+        
+        await client.tools.upsert_from_function(
+            func=add_two_numbers,
+        )
+
+        class InventoryEntryData(BaseModel):
+            data: InventoryEntry
+            quantity_change: int
+
+        def manage_inventory(data: InventoryEntry, quantity_change: int) -> bool:
+            pass
+        
+        await client.tools.upsert_from_function(
+            func=manage_inventory,
+            args_schema=InventoryEntryData,
+        )
+        """
+        source_code = dedent(inspect.getsource(func))
+        args_json_schema = args_schema.model_json_schema() if args_schema and args_schema != OMIT else None
+        return await self.upsert(
+            source_code=source_code,
+            args_json_schema=args_json_schema,
+            description=description,
+            tags=tags,
+            source_type=source_type,
+            json_schema=json_schema,
+            return_char_limit=return_char_limit,
+            request_options=request_options,
+        )
+    
+    async def add(
+        self,
+        *,
+        tool: BaseTool,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Tool:
+        """
+        Add a tool to Letta from a custom Tool class
+
+        Parameters
+        ----------
+        tool : BaseTool
+            The tool object to be added.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Tool
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+
+        class InventoryItem(BaseModel):
+            sku: str  # Unique product identifier
+            name: str  # Product name
+            price: float  # Current price
+            category: str  # Product category (e.g., "Electronics", "Clothing")
+
+        class InventoryEntry(BaseModel):
+            timestamp: int  # Unix timestamp of the transaction
+            item: InventoryItem  # The product being updated
+            transaction_id: str  # Unique identifier for this inventory update
+
+        class InventoryEntryData(BaseModel):
+            data: InventoryEntry
+            quantity_change: int  # Change in quantity (positive for additions, negative for removals)
+
+        class ManageInventoryTool(BaseTool):
+            name: str = "manage_inventory"
+            args_schema: Type[BaseModel] = InventoryEntryData
+            description: str = "Update inventory catalogue with a new data entry"
+            tags: List[str] = ["inventory", "shop"]
+
+            def run(self, data: InventoryEntry, quantity_change: int) -> bool:
+                '''
+                Implementation of the manage_inventory tool
+                '''
+                print(f"Updated inventory for {data.item.name} with a quantity change of {quantity_change}")
+                return True
+                
+        await client.tools.add(
+            tool=ManageInventoryTool()
+        )
+        """
+        source_code = tool.get_source_code()
+        args_json_schema = tool.args_schema.model_json_schema() if tool.args_schema else None
+        return await self.upsert(
+            source_code=source_code,
+            args_json_schema=args_json_schema or OMIT,
+            description=tool.description or OMIT,
+            tags=tool.tags or OMIT,
+            source_type=tool.source_type or OMIT,
+            json_schema=tool.json_schema or OMIT,
+            return_char_limit=tool.return_char_limit or OMIT,
+            request_options=request_options,
+        )

letta_client/core/client_wrapper.py

@@ -16,7 +16,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "letta-client",
-            "X-Fern-SDK-Version": "0.1.77",
+            "X-Fern-SDK-Version": "0.1.78",
         }
         if self.token is not None:
             headers["Authorization"] = f"Bearer {self.token}"

{letta_client-0.1.77.dist-info → letta_client-0.1.78.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: letta-client
-Version: 0.1.77
+Version: 0.1.78
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Intended Audience :: Developers

{letta_client-0.1.77.dist-info → letta_client-0.1.78.dist-info}/RECORD

@@ -231,10 +231,10 @@ letta_client/agents/types/update_agent_tool_rules_item.py,sha256=O0W83nOK8oMx_6N
 letta_client/base_client.py,sha256=bF1nEcPGrhNqelaWz3DgROYdroGsBFCjpAHHri0uq7I,8629
 letta_client/blocks/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 letta_client/blocks/client.py,sha256=LE9dsHaBxFLC3G035f0VpNDG7XKWRK8y9OXpeFCMvUw,30082
-letta_client/client.py,sha256=xdSrD4IkWokZHujowd1r7zESBoVgKGNvo6RqgZ3f0Fg,12808
+letta_client/client.py,sha256=k2mZqqEWciVmEQHgipjCK4kQILk74hpSqzcdNwdql9A,21212
 letta_client/core/__init__.py,sha256=OKbX2aCZXgHCDUsCouqv-OiX32xA6eFFCKIUH9M5Vzk,1591
 letta_client/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-letta_client/core/client_wrapper.py,sha256=C7fBDdx_orCk4CP53xXubQTaQL1EQeWcRsOh94IY__M,1997
+letta_client/core/client_wrapper.py,sha256=JO6fN9Yvpg2o0vMf-HNqjk5yv0wFGsDDnc-w-Rm8MhQ,1997
 letta_client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 letta_client/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 letta_client/core/http_client.py,sha256=Z77OIxIbL4OAB2IDqjRq_sYa5yNYAWfmdhdCSSvh6Y4,19552
@@ -689,6 +689,6 @@ letta_client/voice/__init__.py,sha256=ZrZEuXIukVGhsfM-i0dIFfqjeSOBMPeEgDva7Vvnip
 letta_client/voice/client.py,sha256=2KKJiteGk5HQM79ne1jOPl_ZyUTfZM_gXNdZZ_ndPU8,6485
 letta_client/voice/types/__init__.py,sha256=hBLJcrom99DkDxxsVRU2ni8kPx6SsCy8gtAJvNOz26w,199
 letta_client/voice/types/create_voice_chat_completions_request.py,sha256=K4__83rXRCshfdobyAmH-5fUDJQ_PeSQetTUeC4Abk0,381
-letta_client-0.1.77.dist-info/METADATA,sha256=PvdfXj-r_Bus9_HKv1zHMjP3h_f_EZ23RRAiq3pPRb0,5041
-letta_client-0.1.77.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-letta_client-0.1.77.dist-info/RECORD,,
+letta_client-0.1.78.dist-info/METADATA,sha256=VxRfLmu98Hu1fY_4Gju0oKLYXZUvVp88drgfAQaV78E,5041
+letta_client-0.1.78.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+letta_client-0.1.78.dist-info/RECORD,,