Add warm-up graph call

reference.md

@@ -136,6 +136,19 @@ client.batch.create()
 <dl>
 <dd>
 
+**ignore_roles:** `typing.Optional[typing.Sequence[RoleType]]` 
+
+Optional list of message role types to skip during graph ingestion for
+thread_message items in this batch. The messages are still stored and
+retained as context, but no graph extraction is performed for them.
+Has no effect on graph_episode items.
+
+</dd>
+</dl>
+
+<dl>
+<dd>
+
 **metadata:** `typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]` 
 
 </dd>
@@ -2633,6 +2646,76 @@ client.graph.update(
 </dl>
 
 
+</dd>
+</dl>
+</details>
+
+<details><summary><code>client.graph.<a href="src/zep_cloud/graph/client.py">warm</a>(...)</code></summary>
+<dl>
+<dd>
+
+#### 📝 Description
+
+<dl>
+<dd>
+
+<dl>
+<dd>
+
+Hints Zep to warm a graph for low-latency search
+</dd>
+</dl>
+</dd>
+</dl>
+
+#### 🔌 Usage
+
+<dl>
+<dd>
+
+<dl>
+<dd>
+
+```python
+from zep_cloud import Zep
+
+client = Zep(
+    api_key="YOUR_API_KEY",
+)
+client.graph.warm(
+    graph_id="graphId",
+)
+
+```
+</dd>
+</dl>
+</dd>
+</dl>
+
+#### ⚙️ Parameters
+
+<dl>
+<dd>
+
+<dl>
+<dd>
+
+**graph_id:** `str` — The graph_id of the graph to warm.
+
+</dd>
+</dl>
+
+<dl>
+<dd>
+
+**request_options:** `typing.Optional[RequestOptions]` — Request-specific configuration.
+
+</dd>
+</dl>
+</dd>
+</dl>
+
+
 </dd>
 </dl>
 </details>

src/zep_cloud/batch/client.py

@@ -9,6 +9,7 @@
 from ..types.batch_item_list_response import BatchItemListResponse
 from ..types.batch_list_response import BatchListResponse
 from ..types.batch_summary import BatchSummary
+from ..types.role_type import RoleType
 from ..types.success_response import SuccessResponse
 from .raw_client import AsyncRawBatchClient, RawBatchClient
 
@@ -80,6 +81,7 @@ def list(
     def create(
         self,
         *,
+        ignore_roles: typing.Optional[typing.Sequence[RoleType]] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> BatchSummary:
@@ -88,6 +90,12 @@ def create(
 
         Parameters
         ----------
+        ignore_roles : typing.Optional[typing.Sequence[RoleType]]
+            Optional list of message role types to skip during graph ingestion for
+            thread_message items in this batch. The messages are still stored and
+            retained as context, but no graph extraction is performed for them.
+            Has no effect on graph_episode items.
+
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
 
         request_options : typing.Optional[RequestOptions]
@@ -107,7 +115,9 @@ def create(
         )
         client.batch.create()
         """
-        _response = self._raw_client.create(metadata=metadata, request_options=request_options)
+        _response = self._raw_client.create(
+            ignore_roles=ignore_roles, metadata=metadata, request_options=request_options
+        )
         return _response.data
 
     def get(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> BatchSummary:
@@ -375,6 +385,7 @@ async def main() -> None:
     async def create(
         self,
         *,
+        ignore_roles: typing.Optional[typing.Sequence[RoleType]] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> BatchSummary:
@@ -383,6 +394,12 @@ async def create(
 
         Parameters
         ----------
+        ignore_roles : typing.Optional[typing.Sequence[RoleType]]
+            Optional list of message role types to skip during graph ingestion for
+            thread_message items in this batch. The messages are still stored and
+            retained as context, but no graph extraction is performed for them.
+            Has no effect on graph_episode items.
+
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
 
         request_options : typing.Optional[RequestOptions]
@@ -410,7 +427,9 @@ async def main() -> None:
 
         asyncio.run(main())
         """
-        _response = await self._raw_client.create(metadata=metadata, request_options=request_options)
+        _response = await self._raw_client.create(
+            ignore_roles=ignore_roles, metadata=metadata, request_options=request_options
+        )
         return _response.data
 
     async def get(self, batch_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> BatchSummary:

src/zep_cloud/batch/raw_client.py

@@ -21,6 +21,7 @@
 from ..types.batch_item_list_response import BatchItemListResponse
 from ..types.batch_list_response import BatchListResponse
 from ..types.batch_summary import BatchSummary
+from ..types.role_type import RoleType
 from ..types.success_response import SuccessResponse
 
 # this is used as the default value for optional parameters
@@ -126,6 +127,7 @@ def list(
     def create(
         self,
         *,
+        ignore_roles: typing.Optional[typing.Sequence[RoleType]] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[BatchSummary]:
@@ -134,6 +136,12 @@ def create(
 
         Parameters
         ----------
+        ignore_roles : typing.Optional[typing.Sequence[RoleType]]
+            Optional list of message role types to skip during graph ingestion for
+            thread_message items in this batch. The messages are still stored and
+            retained as context, but no graph extraction is performed for them.
+            Has no effect on graph_episode items.
+
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
 
         request_options : typing.Optional[RequestOptions]
@@ -148,6 +156,7 @@ def create(
             "batches",
             method="POST",
             json={
+                "ignore_roles": ignore_roles,
                 "metadata": metadata,
             },
             headers={
@@ -811,6 +820,7 @@ async def list(
     async def create(
         self,
         *,
+        ignore_roles: typing.Optional[typing.Sequence[RoleType]] = OMIT,
         metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[BatchSummary]:
@@ -819,6 +829,12 @@ async def create(
 
         Parameters
         ----------
+        ignore_roles : typing.Optional[typing.Sequence[RoleType]]
+            Optional list of message role types to skip during graph ingestion for
+            thread_message items in this batch. The messages are still stored and
+            retained as context, but no graph extraction is performed for them.
+            Has no effect on graph_episode items.
+
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
 
         request_options : typing.Optional[RequestOptions]
@@ -833,6 +849,7 @@ async def create(
             "batches",
             method="POST",
             json={
+                "ignore_roles": ignore_roles,
                 "metadata": metadata,
             },
             headers={

src/zep_cloud/graph/client.py

@@ -1013,6 +1013,37 @@ def update(
         )
         return _response.data
 
+    def warm(self, graph_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> SuccessResponse:
+        """
+        Hints Zep to warm a graph for low-latency search
+
+        Parameters
+        ----------
+        graph_id : str
+            The graph_id of the graph to warm.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SuccessResponse
+            Warm hint accepted
+
+        Examples
+        --------
+        from zep_cloud import Zep
+
+        client = Zep(
+            api_key="YOUR_API_KEY",
+        )
+        client.graph.warm(
+            graph_id="graphId",
+        )
+        """
+        _response = self._raw_client.warm(graph_id, request_options=request_options)
+        return _response.data
+
 
 class AsyncGraphClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -2120,3 +2151,42 @@ async def main() -> None:
             graph_id, description=description, name=name, request_options=request_options
         )
         return _response.data
+
+    async def warm(self, graph_id: str, *, request_options: typing.Optional[RequestOptions] = None) -> SuccessResponse:
+        """
+        Hints Zep to warm a graph for low-latency search
+
+        Parameters
+        ----------
+        graph_id : str
+            The graph_id of the graph to warm.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SuccessResponse
+            Warm hint accepted
+
+        Examples
+        --------
+        import asyncio
+
+        from zep_cloud import AsyncZep
+
+        client = AsyncZep(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.graph.warm(
+                graph_id="graphId",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.warm(graph_id, request_options=request_options)
+        return _response.data