letta-client 0.1.233__py3-none-any.whl → 0.1.234__py3-none-any.whl

letta_client/agents/client.py

@@ -1,55 +1,37 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import datetime as dt
 import typing
-from ..core.client_wrapper import SyncClientWrapper
-from .context.client import ContextClient
-from .tools.client import ToolsClient
-from .sources.client import SourcesClient
-from .folders.client import FoldersClient
-from .files.client import FilesClient
-from .core_memory.client import CoreMemoryClient
-from .blocks.client import BlocksClient
-from .passages.client import PassagesClient
-from .messages.client import MessagesClient
-from .groups.client import GroupsClient
-from .templates.client import TemplatesClient
-from .memory_variables.client import MemoryVariablesClient
+
+from .. import core
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.agent_state import AgentState
-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 ..types.create_block import CreateBlock
-from .types.create_agent_request_tool_rules_item import CreateAgentRequestToolRulesItem
 from ..types.agent_type import AgentType
-from ..types.llm_config import LlmConfig
+from ..types.create_block import CreateBlock
 from ..types.embedding_config import EmbeddingConfig
+from ..types.llm_config import LlmConfig
 from ..types.message_create import MessageCreate
-from .types.create_agent_request_response_format import CreateAgentRequestResponseFormat
-from ..core.serialization import convert_and_respect_annotation_metadata
-from ..core.jsonable_encoder import jsonable_encoder
-from .. import core
-from .types.update_agent_tool_rules_item import UpdateAgentToolRulesItem
-from .types.update_agent_response_format import UpdateAgentResponseFormat
-import datetime as dt
+from .blocks.client import AsyncBlocksClient, BlocksClient
+from .context.client import AsyncContextClient, ContextClient
+from .core_memory.client import AsyncCoreMemoryClient, CoreMemoryClient
+from .files.client import AsyncFilesClient, FilesClient
+from .folders.client import AsyncFoldersClient, FoldersClient
+from .groups.client import AsyncGroupsClient, GroupsClient
+from .memory_variables.client import AsyncMemoryVariablesClient, MemoryVariablesClient
+from .messages.client import AsyncMessagesClient, MessagesClient
+from .passages.client import AsyncPassagesClient, PassagesClient
+from .raw_client import AsyncRawAgentsClient, RawAgentsClient
+from .sources.client import AsyncSourcesClient, SourcesClient
+from .templates.client import AsyncTemplatesClient, TemplatesClient
+from .tools.client import AsyncToolsClient, ToolsClient
 from .types.agents_search_request_search_item import AgentsSearchRequestSearchItem
 from .types.agents_search_request_sort_by import AgentsSearchRequestSortBy
 from .types.agents_search_response import AgentsSearchResponse
-from ..core.client_wrapper import AsyncClientWrapper
-from .context.client import AsyncContextClient
-from .tools.client import AsyncToolsClient
-from .sources.client import AsyncSourcesClient
-from .folders.client import AsyncFoldersClient
-from .files.client import AsyncFilesClient
-from .core_memory.client import AsyncCoreMemoryClient
-from .blocks.client import AsyncBlocksClient
-from .passages.client import AsyncPassagesClient
-from .messages.client import AsyncMessagesClient
-from .groups.client import AsyncGroupsClient
-from .templates.client import AsyncTemplatesClient
-from .memory_variables.client import AsyncMemoryVariablesClient
+from .types.create_agent_request_response_format import CreateAgentRequestResponseFormat
+from .types.create_agent_request_tool_rules_item import CreateAgentRequestToolRulesItem
+from .types.update_agent_response_format import UpdateAgentResponseFormat
+from .types.update_agent_tool_rules_item import UpdateAgentToolRulesItem
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -57,19 +39,41 @@ OMIT = typing.cast(typing.Any, ...)
 
 class AgentsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
-        self.context = ContextClient(client_wrapper=self._client_wrapper)
-        self.tools = ToolsClient(client_wrapper=self._client_wrapper)
-        self.sources = SourcesClient(client_wrapper=self._client_wrapper)
-        self.folders = FoldersClient(client_wrapper=self._client_wrapper)
-        self.files = FilesClient(client_wrapper=self._client_wrapper)
-        self.core_memory = CoreMemoryClient(client_wrapper=self._client_wrapper)
-        self.blocks = BlocksClient(client_wrapper=self._client_wrapper)
-        self.passages = PassagesClient(client_wrapper=self._client_wrapper)
-        self.messages = MessagesClient(client_wrapper=self._client_wrapper)
-        self.groups = GroupsClient(client_wrapper=self._client_wrapper)
-        self.templates = TemplatesClient(client_wrapper=self._client_wrapper)
-        self.memory_variables = MemoryVariablesClient(client_wrapper=self._client_wrapper)
+        self._raw_client = RawAgentsClient(client_wrapper=client_wrapper)
+        self.context = ContextClient(client_wrapper=client_wrapper)
+
+        self.tools = ToolsClient(client_wrapper=client_wrapper)
+
+        self.sources = SourcesClient(client_wrapper=client_wrapper)
+
+        self.folders = FoldersClient(client_wrapper=client_wrapper)
+
+        self.files = FilesClient(client_wrapper=client_wrapper)
+
+        self.core_memory = CoreMemoryClient(client_wrapper=client_wrapper)
+
+        self.blocks = BlocksClient(client_wrapper=client_wrapper)
+
+        self.passages = PassagesClient(client_wrapper=client_wrapper)
+
+        self.messages = MessagesClient(client_wrapper=client_wrapper)
+
+        self.groups = GroupsClient(client_wrapper=client_wrapper)
+
+        self.templates = TemplatesClient(client_wrapper=client_wrapper)
+
+        self.memory_variables = MemoryVariablesClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawAgentsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawAgentsClient
+        """
+        return self._raw_client
 
     def list(
         self,
@@ -162,51 +166,25 @@ class AgentsClient:
         )
         client.agents.list()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/agents/",
-            method="GET",
-            params={
-                "name": name,
-                "tags": tags,
-                "match_all_tags": match_all_tags,
-                "before": before,
-                "after": after,
-                "limit": limit,
-                "query_text": query_text,
-                "project_id": project_id,
-                "template_id": template_id,
-                "base_template_id": base_template_id,
-                "identity_id": identity_id,
-                "identifier_keys": identifier_keys,
-                "include_relationships": include_relationships,
-                "ascending": ascending,
-                "sort_by": sort_by,
-            },
+        _response = self._raw_client.list(
+            name=name,
+            tags=tags,
+            match_all_tags=match_all_tags,
+            before=before,
+            after=after,
+            limit=limit,
+            query_text=query_text,
+            project_id=project_id,
+            template_id=template_id,
+            base_template_id=base_template_id,
+            identity_id=identity_id,
+            identifier_keys=identifier_keys,
+            include_relationships=include_relationships,
+            ascending=ascending,
+            sort_by=sort_by,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[AgentState],
-                    construct_type(
-                        type_=typing.List[AgentState],  # 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)
+        return _response.data
 
     def create(
         self,
@@ -400,93 +378,51 @@ class AgentsClient:
         )
         client.agents.create()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/agents/",
-            method="POST",
-            json={
-                "name": name,
-                "memory_blocks": convert_and_respect_annotation_metadata(
-                    object_=memory_blocks, annotation=typing.Sequence[CreateBlock], direction="write"
-                ),
-                "tools": tools,
-                "tool_ids": tool_ids,
-                "source_ids": source_ids,
-                "block_ids": block_ids,
-                "tool_rules": convert_and_respect_annotation_metadata(
-                    object_=tool_rules, annotation=typing.Sequence[CreateAgentRequestToolRulesItem], direction="write"
-                ),
-                "tags": tags,
-                "system": system,
-                "agent_type": agent_type,
-                "llm_config": convert_and_respect_annotation_metadata(
-                    object_=llm_config, annotation=LlmConfig, direction="write"
-                ),
-                "embedding_config": convert_and_respect_annotation_metadata(
-                    object_=embedding_config, annotation=EmbeddingConfig, direction="write"
-                ),
-                "initial_message_sequence": convert_and_respect_annotation_metadata(
-                    object_=initial_message_sequence, annotation=typing.Sequence[MessageCreate], direction="write"
-                ),
-                "include_base_tools": include_base_tools,
-                "include_multi_agent_tools": include_multi_agent_tools,
-                "include_base_tool_rules": include_base_tool_rules,
-                "include_default_source": include_default_source,
-                "description": description,
-                "metadata": metadata,
-                "model": model,
-                "embedding": embedding,
-                "context_window_limit": context_window_limit,
-                "embedding_chunk_size": embedding_chunk_size,
-                "max_tokens": max_tokens,
-                "max_reasoning_tokens": max_reasoning_tokens,
-                "enable_reasoner": enable_reasoner,
-                "from_template": from_template,
-                "template": template,
-                "project": project,
-                "tool_exec_environment_variables": tool_exec_environment_variables,
-                "memory_variables": memory_variables,
-                "project_id": project_id,
-                "template_id": template_id,
-                "base_template_id": base_template_id,
-                "identity_ids": identity_ids,
-                "message_buffer_autoclear": message_buffer_autoclear,
-                "enable_sleeptime": enable_sleeptime,
-                "response_format": convert_and_respect_annotation_metadata(
-                    object_=response_format, annotation=CreateAgentRequestResponseFormat, direction="write"
-                ),
-                "timezone": timezone,
-                "max_files_open": max_files_open,
-                "per_file_view_window_char_limit": per_file_view_window_char_limit,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = self._raw_client.create(
+            name=name,
+            memory_blocks=memory_blocks,
+            tools=tools,
+            tool_ids=tool_ids,
+            source_ids=source_ids,
+            block_ids=block_ids,
+            tool_rules=tool_rules,
+            tags=tags,
+            system=system,
+            agent_type=agent_type,
+            llm_config=llm_config,
+            embedding_config=embedding_config,
+            initial_message_sequence=initial_message_sequence,
+            include_base_tools=include_base_tools,
+            include_multi_agent_tools=include_multi_agent_tools,
+            include_base_tool_rules=include_base_tool_rules,
+            include_default_source=include_default_source,
+            description=description,
+            metadata=metadata,
+            model=model,
+            embedding=embedding,
+            context_window_limit=context_window_limit,
+            embedding_chunk_size=embedding_chunk_size,
+            max_tokens=max_tokens,
+            max_reasoning_tokens=max_reasoning_tokens,
+            enable_reasoner=enable_reasoner,
+            from_template=from_template,
+            template=template,
+            project=project,
+            tool_exec_environment_variables=tool_exec_environment_variables,
+            memory_variables=memory_variables,
+            project_id=project_id,
+            template_id=template_id,
+            base_template_id=base_template_id,
+            identity_ids=identity_ids,
+            message_buffer_autoclear=message_buffer_autoclear,
+            enable_sleeptime=enable_sleeptime,
+            response_format=response_format,
+            timezone=timezone,
+            max_files_open=max_files_open,
+            per_file_view_window_char_limit=per_file_view_window_char_limit,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentState,
-                    construct_type(
-                        type_=AgentState,  # 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)
+        return _response.data
 
     def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
         """
@@ -512,34 +448,8 @@ class AgentsClient:
         )
         client.agents.count()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/agents/count",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    int,
-                    construct_type(
-                        type_=int,  # 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)
+        _response = self._raw_client.count(request_options=request_options)
+        return _response.data
 
     def export_file(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> str:
         """
@@ -569,34 +479,8 @@ class AgentsClient:
             agent_id="agent_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/export",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    str,
-                    construct_type(
-                        type_=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)
+        _response = self._raw_client.export_file(agent_id, request_options=request_options)
+        return _response.data
 
     def import_file(
         self,
@@ -646,45 +530,15 @@ class AgentsClient:
         )
         client.agents.import_file()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/agents/import",
-            method="POST",
-            params={
-                "append_copy_suffix": append_copy_suffix,
-                "override_existing_tools": override_existing_tools,
-                "project_id": project_id,
-                "strip_messages": strip_messages,
-            },
-            data={},
-            files={
-                "file": file,
-            },
+        _response = self._raw_client.import_file(
+            file=file,
+            append_copy_suffix=append_copy_suffix,
+            override_existing_tools=override_existing_tools,
+            project_id=project_id,
+            strip_messages=strip_messages,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentState,
-                    construct_type(
-                        type_=AgentState,  # 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)
+        return _response.data
 
     def retrieve(
         self,
@@ -723,37 +577,10 @@ class AgentsClient:
             agent_id="agent_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}",
-            method="GET",
-            params={
-                "include_relationships": include_relationships,
-            },
-            request_options=request_options,
+        _response = self._raw_client.retrieve(
+            agent_id, include_relationships=include_relationships, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentState,
-                    construct_type(
-                        type_=AgentState,  # 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)
+        return _response.data
 
     def delete(
         self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -785,34 +612,8 @@ class AgentsClient:
             agent_id="agent_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}",
-            method="DELETE",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    construct_type(
-                        type_=typing.Optional[typing.Any],  # 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)
+        _response = self._raw_client.delete(agent_id, request_options=request_options)
+        return _response.data
 
     def modify(
         self,
@@ -955,75 +756,38 @@ class AgentsClient:
             agent_id="agent_id",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}",
-            method="PATCH",
-            json={
-                "name": name,
-                "tool_ids": tool_ids,
-                "source_ids": source_ids,
-                "block_ids": block_ids,
-                "tags": tags,
-                "system": system,
-                "tool_rules": convert_and_respect_annotation_metadata(
-                    object_=tool_rules, annotation=typing.Sequence[UpdateAgentToolRulesItem], direction="write"
-                ),
-                "llm_config": convert_and_respect_annotation_metadata(
-                    object_=llm_config, annotation=LlmConfig, direction="write"
-                ),
-                "embedding_config": convert_and_respect_annotation_metadata(
-                    object_=embedding_config, annotation=EmbeddingConfig, direction="write"
-                ),
-                "message_ids": message_ids,
-                "description": description,
-                "metadata": metadata,
-                "tool_exec_environment_variables": tool_exec_environment_variables,
-                "project_id": project_id,
-                "template_id": template_id,
-                "base_template_id": base_template_id,
-                "identity_ids": identity_ids,
-                "message_buffer_autoclear": message_buffer_autoclear,
-                "model": model,
-                "embedding": embedding,
-                "enable_sleeptime": enable_sleeptime,
-                "response_format": convert_and_respect_annotation_metadata(
-                    object_=response_format, annotation=UpdateAgentResponseFormat, direction="write"
-                ),
-                "last_run_completion": last_run_completion,
-                "last_run_duration_ms": last_run_duration_ms,
-                "timezone": timezone,
-                "max_files_open": max_files_open,
-                "per_file_view_window_char_limit": per_file_view_window_char_limit,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = self._raw_client.modify(
+            agent_id,
+            name=name,
+            tool_ids=tool_ids,
+            source_ids=source_ids,
+            block_ids=block_ids,
+            tags=tags,
+            system=system,
+            tool_rules=tool_rules,
+            llm_config=llm_config,
+            embedding_config=embedding_config,
+            message_ids=message_ids,
+            description=description,
+            metadata=metadata,
+            tool_exec_environment_variables=tool_exec_environment_variables,
+            project_id=project_id,
+            template_id=template_id,
+            base_template_id=base_template_id,
+            identity_ids=identity_ids,
+            message_buffer_autoclear=message_buffer_autoclear,
+            model=model,
+            embedding=embedding,
+            enable_sleeptime=enable_sleeptime,
+            response_format=response_format,
+            last_run_completion=last_run_completion,
+            last_run_duration_ms=last_run_duration_ms,
+            timezone=timezone,
+            max_files_open=max_files_open,
+            per_file_view_window_char_limit=per_file_view_window_char_limit,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentState,
-                    construct_type(
-                        type_=AgentState,  # 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)
+        return _response.data
 
     def summarize_agent_conversation(
         self, agent_id: str, *, max_message_length: int, request_options: typing.Optional[RequestOptions] = None
@@ -1061,31 +825,10 @@ class AgentsClient:
             max_message_length=1,
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/summarize",
-            method="POST",
-            params={
-                "max_message_length": max_message_length,
-            },
-            request_options=request_options,
+        _response = self._raw_client.summarize_agent_conversation(
+            agent_id, max_message_length=max_message_length, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            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)
+        return _response.data
 
     def search(
         self,
@@ -1138,56 +881,56 @@ class AgentsClient:
         )
         client.agents.search()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/agents/search",
-            method="POST",
-            json={
-                "search": convert_and_respect_annotation_metadata(
-                    object_=search, annotation=typing.Sequence[AgentsSearchRequestSearchItem], direction="write"
-                ),
-                "project_id": project_id,
-                "combinator": combinator,
-                "limit": limit,
-                "after": after,
-                "sortBy": sort_by,
-                "ascending": ascending,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = self._raw_client.search(
+            search=search,
+            project_id=project_id,
+            combinator=combinator,
+            limit=limit,
+            after=after,
+            sort_by=sort_by,
+            ascending=ascending,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentsSearchResponse,
-                    construct_type(
-                        type_=AgentsSearchResponse,  # 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)
+        return _response.data
 
 
 class AsyncAgentsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-        self.context = AsyncContextClient(client_wrapper=self._client_wrapper)
-        self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)
-        self.sources = AsyncSourcesClient(client_wrapper=self._client_wrapper)
-        self.folders = AsyncFoldersClient(client_wrapper=self._client_wrapper)
-        self.files = AsyncFilesClient(client_wrapper=self._client_wrapper)
-        self.core_memory = AsyncCoreMemoryClient(client_wrapper=self._client_wrapper)
-        self.blocks = AsyncBlocksClient(client_wrapper=self._client_wrapper)
-        self.passages = AsyncPassagesClient(client_wrapper=self._client_wrapper)
-        self.messages = AsyncMessagesClient(client_wrapper=self._client_wrapper)
-        self.groups = AsyncGroupsClient(client_wrapper=self._client_wrapper)
-        self.templates = AsyncTemplatesClient(client_wrapper=self._client_wrapper)
-        self.memory_variables = AsyncMemoryVariablesClient(client_wrapper=self._client_wrapper)
+        self._raw_client = AsyncRawAgentsClient(client_wrapper=client_wrapper)
+        self.context = AsyncContextClient(client_wrapper=client_wrapper)
+
+        self.tools = AsyncToolsClient(client_wrapper=client_wrapper)
+
+        self.sources = AsyncSourcesClient(client_wrapper=client_wrapper)
+
+        self.folders = AsyncFoldersClient(client_wrapper=client_wrapper)
+
+        self.files = AsyncFilesClient(client_wrapper=client_wrapper)
+
+        self.core_memory = AsyncCoreMemoryClient(client_wrapper=client_wrapper)
+
+        self.blocks = AsyncBlocksClient(client_wrapper=client_wrapper)
+
+        self.passages = AsyncPassagesClient(client_wrapper=client_wrapper)
+
+        self.messages = AsyncMessagesClient(client_wrapper=client_wrapper)
+
+        self.groups = AsyncGroupsClient(client_wrapper=client_wrapper)
+
+        self.templates = AsyncTemplatesClient(client_wrapper=client_wrapper)
+
+        self.memory_variables = AsyncMemoryVariablesClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawAgentsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawAgentsClient
+        """
+        return self._raw_client
 
     async def list(
         self,
@@ -1288,51 +1031,25 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/agents/",
-            method="GET",
-            params={
-                "name": name,
-                "tags": tags,
-                "match_all_tags": match_all_tags,
-                "before": before,
-                "after": after,
-                "limit": limit,
-                "query_text": query_text,
-                "project_id": project_id,
-                "template_id": template_id,
-                "base_template_id": base_template_id,
-                "identity_id": identity_id,
-                "identifier_keys": identifier_keys,
-                "include_relationships": include_relationships,
-                "ascending": ascending,
-                "sort_by": sort_by,
-            },
+        _response = await self._raw_client.list(
+            name=name,
+            tags=tags,
+            match_all_tags=match_all_tags,
+            before=before,
+            after=after,
+            limit=limit,
+            query_text=query_text,
+            project_id=project_id,
+            template_id=template_id,
+            base_template_id=base_template_id,
+            identity_id=identity_id,
+            identifier_keys=identifier_keys,
+            include_relationships=include_relationships,
+            ascending=ascending,
+            sort_by=sort_by,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[AgentState],
-                    construct_type(
-                        type_=typing.List[AgentState],  # 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)
+        return _response.data
 
     async def create(
         self,
@@ -1534,93 +1251,51 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/agents/",
-            method="POST",
-            json={
-                "name": name,
-                "memory_blocks": convert_and_respect_annotation_metadata(
-                    object_=memory_blocks, annotation=typing.Sequence[CreateBlock], direction="write"
-                ),
-                "tools": tools,
-                "tool_ids": tool_ids,
-                "source_ids": source_ids,
-                "block_ids": block_ids,
-                "tool_rules": convert_and_respect_annotation_metadata(
-                    object_=tool_rules, annotation=typing.Sequence[CreateAgentRequestToolRulesItem], direction="write"
-                ),
-                "tags": tags,
-                "system": system,
-                "agent_type": agent_type,
-                "llm_config": convert_and_respect_annotation_metadata(
-                    object_=llm_config, annotation=LlmConfig, direction="write"
-                ),
-                "embedding_config": convert_and_respect_annotation_metadata(
-                    object_=embedding_config, annotation=EmbeddingConfig, direction="write"
-                ),
-                "initial_message_sequence": convert_and_respect_annotation_metadata(
-                    object_=initial_message_sequence, annotation=typing.Sequence[MessageCreate], direction="write"
-                ),
-                "include_base_tools": include_base_tools,
-                "include_multi_agent_tools": include_multi_agent_tools,
-                "include_base_tool_rules": include_base_tool_rules,
-                "include_default_source": include_default_source,
-                "description": description,
-                "metadata": metadata,
-                "model": model,
-                "embedding": embedding,
-                "context_window_limit": context_window_limit,
-                "embedding_chunk_size": embedding_chunk_size,
-                "max_tokens": max_tokens,
-                "max_reasoning_tokens": max_reasoning_tokens,
-                "enable_reasoner": enable_reasoner,
-                "from_template": from_template,
-                "template": template,
-                "project": project,
-                "tool_exec_environment_variables": tool_exec_environment_variables,
-                "memory_variables": memory_variables,
-                "project_id": project_id,
-                "template_id": template_id,
-                "base_template_id": base_template_id,
-                "identity_ids": identity_ids,
-                "message_buffer_autoclear": message_buffer_autoclear,
-                "enable_sleeptime": enable_sleeptime,
-                "response_format": convert_and_respect_annotation_metadata(
-                    object_=response_format, annotation=CreateAgentRequestResponseFormat, direction="write"
-                ),
-                "timezone": timezone,
-                "max_files_open": max_files_open,
-                "per_file_view_window_char_limit": per_file_view_window_char_limit,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = await self._raw_client.create(
+            name=name,
+            memory_blocks=memory_blocks,
+            tools=tools,
+            tool_ids=tool_ids,
+            source_ids=source_ids,
+            block_ids=block_ids,
+            tool_rules=tool_rules,
+            tags=tags,
+            system=system,
+            agent_type=agent_type,
+            llm_config=llm_config,
+            embedding_config=embedding_config,
+            initial_message_sequence=initial_message_sequence,
+            include_base_tools=include_base_tools,
+            include_multi_agent_tools=include_multi_agent_tools,
+            include_base_tool_rules=include_base_tool_rules,
+            include_default_source=include_default_source,
+            description=description,
+            metadata=metadata,
+            model=model,
+            embedding=embedding,
+            context_window_limit=context_window_limit,
+            embedding_chunk_size=embedding_chunk_size,
+            max_tokens=max_tokens,
+            max_reasoning_tokens=max_reasoning_tokens,
+            enable_reasoner=enable_reasoner,
+            from_template=from_template,
+            template=template,
+            project=project,
+            tool_exec_environment_variables=tool_exec_environment_variables,
+            memory_variables=memory_variables,
+            project_id=project_id,
+            template_id=template_id,
+            base_template_id=base_template_id,
+            identity_ids=identity_ids,
+            message_buffer_autoclear=message_buffer_autoclear,
+            enable_sleeptime=enable_sleeptime,
+            response_format=response_format,
+            timezone=timezone,
+            max_files_open=max_files_open,
+            per_file_view_window_char_limit=per_file_view_window_char_limit,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentState,
-                    construct_type(
-                        type_=AgentState,  # 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)
+        return _response.data
 
     async def count(self, *, request_options: typing.Optional[RequestOptions] = None) -> int:
         """
@@ -1654,34 +1329,8 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/agents/count",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    int,
-                    construct_type(
-                        type_=int,  # 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)
+        _response = await self._raw_client.count(request_options=request_options)
+        return _response.data
 
     async def export_file(self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> str:
         """
@@ -1719,34 +1368,8 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/export",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    str,
-                    construct_type(
-                        type_=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)
+        _response = await self._raw_client.export_file(agent_id, request_options=request_options)
+        return _response.data
 
     async def import_file(
         self,
@@ -1804,45 +1427,15 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/agents/import",
-            method="POST",
-            params={
-                "append_copy_suffix": append_copy_suffix,
-                "override_existing_tools": override_existing_tools,
-                "project_id": project_id,
-                "strip_messages": strip_messages,
-            },
-            data={},
-            files={
-                "file": file,
-            },
+        _response = await self._raw_client.import_file(
+            file=file,
+            append_copy_suffix=append_copy_suffix,
+            override_existing_tools=override_existing_tools,
+            project_id=project_id,
+            strip_messages=strip_messages,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentState,
-                    construct_type(
-                        type_=AgentState,  # 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)
+        return _response.data
 
     async def retrieve(
         self,
@@ -1889,37 +1482,10 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}",
-            method="GET",
-            params={
-                "include_relationships": include_relationships,
-            },
-            request_options=request_options,
+        _response = await self._raw_client.retrieve(
+            agent_id, include_relationships=include_relationships, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentState,
-                    construct_type(
-                        type_=AgentState,  # 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)
+        return _response.data
 
     async def delete(
         self, agent_id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -1959,34 +1525,8 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}",
-            method="DELETE",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    construct_type(
-                        type_=typing.Optional[typing.Any],  # 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)
+        _response = await self._raw_client.delete(agent_id, request_options=request_options)
+        return _response.data
 
     async def modify(
         self,
@@ -2137,75 +1677,38 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}",
-            method="PATCH",
-            json={
-                "name": name,
-                "tool_ids": tool_ids,
-                "source_ids": source_ids,
-                "block_ids": block_ids,
-                "tags": tags,
-                "system": system,
-                "tool_rules": convert_and_respect_annotation_metadata(
-                    object_=tool_rules, annotation=typing.Sequence[UpdateAgentToolRulesItem], direction="write"
-                ),
-                "llm_config": convert_and_respect_annotation_metadata(
-                    object_=llm_config, annotation=LlmConfig, direction="write"
-                ),
-                "embedding_config": convert_and_respect_annotation_metadata(
-                    object_=embedding_config, annotation=EmbeddingConfig, direction="write"
-                ),
-                "message_ids": message_ids,
-                "description": description,
-                "metadata": metadata,
-                "tool_exec_environment_variables": tool_exec_environment_variables,
-                "project_id": project_id,
-                "template_id": template_id,
-                "base_template_id": base_template_id,
-                "identity_ids": identity_ids,
-                "message_buffer_autoclear": message_buffer_autoclear,
-                "model": model,
-                "embedding": embedding,
-                "enable_sleeptime": enable_sleeptime,
-                "response_format": convert_and_respect_annotation_metadata(
-                    object_=response_format, annotation=UpdateAgentResponseFormat, direction="write"
-                ),
-                "last_run_completion": last_run_completion,
-                "last_run_duration_ms": last_run_duration_ms,
-                "timezone": timezone,
-                "max_files_open": max_files_open,
-                "per_file_view_window_char_limit": per_file_view_window_char_limit,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = await self._raw_client.modify(
+            agent_id,
+            name=name,
+            tool_ids=tool_ids,
+            source_ids=source_ids,
+            block_ids=block_ids,
+            tags=tags,
+            system=system,
+            tool_rules=tool_rules,
+            llm_config=llm_config,
+            embedding_config=embedding_config,
+            message_ids=message_ids,
+            description=description,
+            metadata=metadata,
+            tool_exec_environment_variables=tool_exec_environment_variables,
+            project_id=project_id,
+            template_id=template_id,
+            base_template_id=base_template_id,
+            identity_ids=identity_ids,
+            message_buffer_autoclear=message_buffer_autoclear,
+            model=model,
+            embedding=embedding,
+            enable_sleeptime=enable_sleeptime,
+            response_format=response_format,
+            last_run_completion=last_run_completion,
+            last_run_duration_ms=last_run_duration_ms,
+            timezone=timezone,
+            max_files_open=max_files_open,
+            per_file_view_window_char_limit=per_file_view_window_char_limit,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentState,
-                    construct_type(
-                        type_=AgentState,  # 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)
+        return _response.data
 
     async def summarize_agent_conversation(
         self, agent_id: str, *, max_message_length: int, request_options: typing.Optional[RequestOptions] = None
@@ -2251,31 +1754,10 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/agents/{jsonable_encoder(agent_id)}/summarize",
-            method="POST",
-            params={
-                "max_message_length": max_message_length,
-            },
-            request_options=request_options,
+        _response = await self._raw_client.summarize_agent_conversation(
+            agent_id, max_message_length=max_message_length, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            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)
+        return _response.data
 
     async def search(
         self,
@@ -2336,36 +1818,14 @@ class AsyncAgentsClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/agents/search",
-            method="POST",
-            json={
-                "search": convert_and_respect_annotation_metadata(
-                    object_=search, annotation=typing.Sequence[AgentsSearchRequestSearchItem], direction="write"
-                ),
-                "project_id": project_id,
-                "combinator": combinator,
-                "limit": limit,
-                "after": after,
-                "sortBy": sort_by,
-                "ascending": ascending,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = await self._raw_client.search(
+            search=search,
+            project_id=project_id,
+            combinator=combinator,
+            limit=limit,
+            after=after,
+            sort_by=sort_by,
+            ascending=ascending,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    AgentsSearchResponse,
-                    construct_type(
-                        type_=AgentsSearchResponse,  # 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)
+        return _response.data