letta-client 0.1.10__py3-none-any.whl → 0.1.13__py3-none-any.whl

letta_client/base_client.py

@@ -0,0 +1,176 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from .environment import LettaEnvironment
+import httpx
+from .core.client_wrapper import SyncClientWrapper
+from .tools.client import ToolsClient
+from .sources.client import SourcesClient
+from .agents.client import AgentsClient
+from .models.client import ModelsClient
+from .blocks.client import BlocksClient
+from .jobs.client import JobsClient
+from .health.client import HealthClient
+from .providers.client import ProvidersClient
+from .runs.client import RunsClient
+from .tag.client import TagClient
+from .core.client_wrapper import AsyncClientWrapper
+from .tools.client import AsyncToolsClient
+from .sources.client import AsyncSourcesClient
+from .agents.client import AsyncAgentsClient
+from .models.client import AsyncModelsClient
+from .blocks.client import AsyncBlocksClient
+from .jobs.client import AsyncJobsClient
+from .health.client import AsyncHealthClient
+from .providers.client import AsyncProvidersClient
+from .runs.client import AsyncRunsClient
+from .tag.client import AsyncTagClient
+
+
+class LettaBase:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
+
+    Parameters
+    ----------
+    base_url : typing.Optional[str]
+        The base url to use for requests from the client.
+
+    environment : LettaEnvironment
+        The environment to use for requests from the client. from .environment import LettaEnvironment
+
+
+
+        Defaults to LettaEnvironment.LETTA_CLOUD
+
+
+
+    token : typing.Optional[str]
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
+
+    httpx_client : typing.Optional[httpx.Client]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+
+    Examples
+    --------
+    from letta_client import Letta
+
+    client = Letta(
+        token="YOUR_TOKEN",
+    )
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: LettaEnvironment = LettaEnvironment.LETTA_CLOUD,
+        token: typing.Optional[str] = None,
+        timeout: typing.Optional[float] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.Client] = None,
+    ):
+        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
+        self._client_wrapper = SyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            token=token,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else httpx.Client(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
+            if follow_redirects is not None
+            else httpx.Client(timeout=_defaulted_timeout),
+            timeout=_defaulted_timeout,
+        )
+        self.tools = ToolsClient(client_wrapper=self._client_wrapper)
+        self.sources = SourcesClient(client_wrapper=self._client_wrapper)
+        self.agents = AgentsClient(client_wrapper=self._client_wrapper)
+        self.models = ModelsClient(client_wrapper=self._client_wrapper)
+        self.blocks = BlocksClient(client_wrapper=self._client_wrapper)
+        self.jobs = JobsClient(client_wrapper=self._client_wrapper)
+        self.health = HealthClient(client_wrapper=self._client_wrapper)
+        self.providers = ProvidersClient(client_wrapper=self._client_wrapper)
+        self.runs = RunsClient(client_wrapper=self._client_wrapper)
+        self.tag = TagClient(client_wrapper=self._client_wrapper)
+
+
+class AsyncLettaBase:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
+
+    Parameters
+    ----------
+    base_url : typing.Optional[str]
+        The base url to use for requests from the client.
+
+    environment : LettaEnvironment
+        The environment to use for requests from the client. from .environment import LettaEnvironment
+
+
+
+        Defaults to LettaEnvironment.LETTA_CLOUD
+
+
+
+    token : typing.Optional[str]
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
+
+    httpx_client : typing.Optional[httpx.AsyncClient]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+
+    Examples
+    --------
+    from letta_client import AsyncLetta
+
+    client = AsyncLetta(
+        token="YOUR_TOKEN",
+    )
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: LettaEnvironment = LettaEnvironment.LETTA_CLOUD,
+        token: typing.Optional[str] = None,
+        timeout: typing.Optional[float] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.AsyncClient] = None,
+    ):
+        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
+        self._client_wrapper = AsyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            token=token,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else httpx.AsyncClient(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
+            if follow_redirects is not None
+            else httpx.AsyncClient(timeout=_defaulted_timeout),
+            timeout=_defaulted_timeout,
+        )
+        self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)
+        self.sources = AsyncSourcesClient(client_wrapper=self._client_wrapper)
+        self.agents = AsyncAgentsClient(client_wrapper=self._client_wrapper)
+        self.models = AsyncModelsClient(client_wrapper=self._client_wrapper)
+        self.blocks = AsyncBlocksClient(client_wrapper=self._client_wrapper)
+        self.jobs = AsyncJobsClient(client_wrapper=self._client_wrapper)
+        self.health = AsyncHealthClient(client_wrapper=self._client_wrapper)
+        self.providers = AsyncProvidersClient(client_wrapper=self._client_wrapper)
+        self.runs = AsyncRunsClient(client_wrapper=self._client_wrapper)
+        self.tag = AsyncTagClient(client_wrapper=self._client_wrapper)
+
+
+def _get_base_url(*, base_url: typing.Optional[str] = None, environment: LettaEnvironment) -> str:
+    if base_url is not None:
+        return base_url
+    elif environment is not None:
+        return environment.value
+    else:
+        raise Exception("Please pass in either base_url or environment to construct the client")

letta_client/client.py

@@ -1,172 +1,55 @@
-# This file was auto-generated by Fern from our API Definition.
-
+from textwrap import dedent
+import inspect
 import typing
-from .environment import LettaEnvironment
-import httpx
-from .core.client_wrapper import SyncClientWrapper
-from .tools.client import ToolsClient
-from .sources.client import SourcesClient
-from .agents.client import AgentsClient
-from .models.client import ModelsClient
-from .blocks.client import BlocksClient
-from .jobs.client import JobsClient
-from .health.client import HealthClient
-from .providers.client import ProvidersClient
-from .runs.client import RunsClient
-from .core.client_wrapper import AsyncClientWrapper
-from .tools.client import AsyncToolsClient
-from .sources.client import AsyncSourcesClient
-from .agents.client import AsyncAgentsClient
-from .models.client import AsyncModelsClient
-from .blocks.client import AsyncBlocksClient
-from .jobs.client import AsyncJobsClient
-from .health.client import AsyncHealthClient
-from .providers.client import AsyncProvidersClient
-from .runs.client import AsyncRunsClient
-
-
-class Letta:
-    """
-    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
-
-    Parameters
-    ----------
-    base_url : typing.Optional[str]
-        The base url to use for requests from the client.
-
-    environment : LettaEnvironment
-        The environment to use for requests from the client. from .environment import LettaEnvironment
-
-
-
-        Defaults to LettaEnvironment.LETTA_CLOUD
-
-
-
-    token : typing.Optional[str]
-    timeout : typing.Optional[float]
-        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
-
-    follow_redirects : typing.Optional[bool]
-        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
 
-    httpx_client : typing.Optional[httpx.Client]
-        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+from .base_client import LettaBase, AsyncLettaBase
+from .core.request_options import RequestOptions
+from .tools.client import ToolsClient as ToolsClientBase
+from .types.letta_schemas_tool_tool import LettaSchemasToolTool
 
-    Examples
-    --------
-    from letta_client import Letta
 
-    client = Letta(
-        token="YOUR_TOKEN",
-    )
-    """
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
 
-    def __init__(
-        self,
-        *,
-        base_url: typing.Optional[str] = None,
-        environment: LettaEnvironment = LettaEnvironment.LETTA_CLOUD,
-        token: typing.Optional[str] = None,
-        timeout: typing.Optional[float] = None,
-        follow_redirects: typing.Optional[bool] = True,
-        httpx_client: typing.Optional[httpx.Client] = None,
-    ):
-        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
-        self._client_wrapper = SyncClientWrapper(
-            base_url=_get_base_url(base_url=base_url, environment=environment),
-            token=token,
-            httpx_client=httpx_client
-            if httpx_client is not None
-            else httpx.Client(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
-            if follow_redirects is not None
-            else httpx.Client(timeout=_defaulted_timeout),
-            timeout=_defaulted_timeout,
-        )
-        self.tools = ToolsClient(client_wrapper=self._client_wrapper)
-        self.sources = SourcesClient(client_wrapper=self._client_wrapper)
-        self.agents = AgentsClient(client_wrapper=self._client_wrapper)
-        self.models = ModelsClient(client_wrapper=self._client_wrapper)
-        self.blocks = BlocksClient(client_wrapper=self._client_wrapper)
-        self.jobs = JobsClient(client_wrapper=self._client_wrapper)
-        self.health = HealthClient(client_wrapper=self._client_wrapper)
-        self.providers = ProvidersClient(client_wrapper=self._client_wrapper)
-        self.runs = RunsClient(client_wrapper=self._client_wrapper)
-
-
-class AsyncLetta:
-    """
-    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
-
-    Parameters
-    ----------
-    base_url : typing.Optional[str]
-        The base url to use for requests from the client.
 
-    environment : LettaEnvironment
-        The environment to use for requests from the client. from .environment import LettaEnvironment
+class Letta(LettaBase):
 
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.tools = ExtendedToolsClient(client_wrapper=self._client_wrapper)
 
 
-        Defaults to LettaEnvironment.LETTA_CLOUD
+class AsyncLetta(AsyncLettaBase):
 
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.tools = ExtendedToolsClient(client_wrapper=self._client_wrapper)
 
 
-    token : typing.Optional[str]
-    timeout : typing.Optional[float]
-        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+class ToolsClient(ToolsClientBase):
 
-    follow_redirects : typing.Optional[bool]
-        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
-
-    httpx_client : typing.Optional[httpx.AsyncClient]
-        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
-
-    Examples
-    --------
-    from letta_client import AsyncLetta
-
-    client = AsyncLetta(
-        token="YOUR_TOKEN",
-    )
-    """
-
-    def __init__(
+    def create_from_function(
         self,
         *,
-        base_url: typing.Optional[str] = None,
-        environment: LettaEnvironment = LettaEnvironment.LETTA_CLOUD,
-        token: typing.Optional[str] = None,
-        timeout: typing.Optional[float] = None,
-        follow_redirects: typing.Optional[bool] = True,
-        httpx_client: typing.Optional[httpx.AsyncClient] = None,
-    ):
-        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
-        self._client_wrapper = AsyncClientWrapper(
-            base_url=_get_base_url(base_url=base_url, environment=environment),
-            token=token,
-            httpx_client=httpx_client
-            if httpx_client is not None
-            else httpx.AsyncClient(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
-            if follow_redirects is not None
-            else httpx.AsyncClient(timeout=_defaulted_timeout),
-            timeout=_defaulted_timeout,
+        function: typing.Callable,
+        name: typing.Optional[str] = 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,
+    ) -> LettaSchemasToolTool:
+        source_code = dedent(inspect.getsource(function))
+        return self.create(
+            source_code=source_code,
+            name=name,
+            description=description,
+            tags=tags,
+            source_type=source_type,
+            json_schema=json_schema,
+            return_char_limit=return_char_limit,
+            request_options=request_options,
         )
-        self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)
-        self.sources = AsyncSourcesClient(client_wrapper=self._client_wrapper)
-        self.agents = AsyncAgentsClient(client_wrapper=self._client_wrapper)
-        self.models = AsyncModelsClient(client_wrapper=self._client_wrapper)
-        self.blocks = AsyncBlocksClient(client_wrapper=self._client_wrapper)
-        self.jobs = AsyncJobsClient(client_wrapper=self._client_wrapper)
-        self.health = AsyncHealthClient(client_wrapper=self._client_wrapper)
-        self.providers = AsyncProvidersClient(client_wrapper=self._client_wrapper)
-        self.runs = AsyncRunsClient(client_wrapper=self._client_wrapper)
-
-
-def _get_base_url(*, base_url: typing.Optional[str] = None, environment: LettaEnvironment) -> str:
-    if base_url is not None:
-        return base_url
-    elif environment is not None:
-        return environment.value
-    else:
-        raise Exception("Please pass in either base_url or environment to construct the client")

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.10",
+            "X-Fern-SDK-Version": "0.1.13",
         }
         if self.token is not None:
             headers["Authorization"] = f"Bearer {self.token}"

letta_client/tag/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

letta_client/tag/client.py

@@ -0,0 +1,169 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.client_wrapper import SyncClientWrapper
+import typing
+from ..core.request_options import RequestOptions
+from ..core.unchecked_base_model import construct_type
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.http_validation_error import HttpValidationError
+from json.decoder import JSONDecodeError
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper
+
+
+class TagClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def list_tags(
+        self,
+        *,
+        cursor: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        query_text: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[str]:
+        """
+        Get a list of all tags in the database
+
+        Parameters
+        ----------
+        cursor : typing.Optional[str]
+
+        limit : typing.Optional[int]
+
+        query_text : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[str]
+            Successful Response
+
+        Examples
+        --------
+        from letta_client import Letta
+
+        client = Letta(
+            token="YOUR_TOKEN",
+        )
+        client.tag.list_tags()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/tags/",
+            method="GET",
+            params={
+                "cursor": cursor,
+                "limit": limit,
+                "query_text": query_text,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[str],
+                    construct_type(
+                        type_=typing.List[str],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    typing.cast(
+                        HttpValidationError,
+                        construct_type(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncTagClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def list_tags(
+        self,
+        *,
+        cursor: typing.Optional[str] = None,
+        limit: typing.Optional[int] = None,
+        query_text: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[str]:
+        """
+        Get a list of all tags in the database
+
+        Parameters
+        ----------
+        cursor : typing.Optional[str]
+
+        limit : typing.Optional[int]
+
+        query_text : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[str]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from letta_client import AsyncLetta
+
+        client = AsyncLetta(
+            token="YOUR_TOKEN",
+        )
+
+
+        async def main() -> None:
+            await client.tag.list_tags()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/tags/",
+            method="GET",
+            params={
+                "cursor": cursor,
+                "limit": limit,
+                "query_text": query_text,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[str],
+                    construct_type(
+                        type_=typing.List[str],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    typing.cast(
+                        HttpValidationError,
+                        construct_type(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)