From c6466da0985cfc42c50a62b2ed8d9ad9da8a3451 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@boringbits.io>
Date: Wed, 26 Jun 2024 15:23:08 -0700
Subject: [PATCH] sdk: Filter assistants by graph_id (#843)

---
 libs/sdk-py/langgraph_sdk/client.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/libs/sdk-py/langgraph_sdk/client.py b/libs/sdk-py/langgraph_sdk/client.py
index 97bffdf06..efb4494df 100644
--- a/libs/sdk-py/langgraph_sdk/client.py
+++ b/libs/sdk-py/langgraph_sdk/client.py
@@ -272,15 +272,22 @@ class AssistantsClient:
         await self.http.delete(f"/assistants/{assistant_id}")
 
     async def search(
-        self, *, metadata: Metadata = None, limit: int = 10, offset: int = 0
+        self,
+        *,
+        metadata: Metadata = None,
+        graph_id: Optional[str] = None,
+        limit: int = 10,
+        offset: int = 0,
     ) -> list[Assistant]:
+        """Search for assistants."""
         payload: Dict[str, Any] = {
             "limit": limit,
             "offset": offset,
         }
         if metadata:
             payload["metadata"] = metadata
-        """Search for assistants."""
+        if graph_id:
+            payload["graph_id"] = graph_id
         return await self.http.post(
             "/assistants/search",
             json=payload,