sdk: Filter assistants by graph_id (#843)

2026-09-08 18:57:52 +02:00 · 2024-06-26 15:23:08 -07:00
parent 32e8a2dc50
commit c6466da098
1 changed files with 9 additions and 2 deletions
@@ -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,