From 6e54f74fa916b0bf5fed20e0a0f15e883267f87f Mon Sep 17 00:00:00 2001
From: Eugene Yurtsev <eyurtsev@gmail.com>
Date: Wed, 2 Apr 2025 17:26:14 -0400
Subject: [PATCH] sdk-py: Add option to set description via client sdk (#4147)

Co-authored-by: William FH <13333726+hinthornw@users.noreply.github.com>
---
 libs/sdk-py/langgraph_sdk/client.py | 20 ++++++++++++++++++++
 libs/sdk-py/langgraph_sdk/schema.py |  2 ++
 2 files changed, 22 insertions(+)

diff --git a/libs/sdk-py/langgraph_sdk/client.py b/libs/sdk-py/langgraph_sdk/client.py
index ae1deea2a..95d756bd3 100644
--- a/libs/sdk-py/langgraph_sdk/client.py
+++ b/libs/sdk-py/langgraph_sdk/client.py
@@ -631,6 +631,7 @@ class AssistantsClient:
         if_exists: Optional[OnConflictBehavior] = None,
         name: Optional[str] = None,
         headers: Optional[dict[str, str]] = None,
+        description: Optional[str] = None,
     ) -> Assistant:
         """Create a new assistant.
 
@@ -645,6 +646,8 @@ class AssistantsClient:
                 Must be either 'raise' (raise error if duplicate), or 'do_nothing' (return existing assistant).
             name: The name of the assistant. Defaults to 'Untitled' under the hood.
             headers: Optional custom headers to include with the request.
+            description: Optional description of the assistant.
+                The description field is available for langgraph-api server version>=0.0.45
 
         Returns:
             Assistant: The created assistant.
@@ -673,6 +676,8 @@ class AssistantsClient:
             payload["if_exists"] = if_exists
         if name:
             payload["name"] = name
+        if description:
+            payload["description"] = description
         return await self.http.post("/assistants", json=payload, headers=headers)
 
     async def update(
@@ -684,6 +689,7 @@ class AssistantsClient:
         metadata: Json = None,
         name: Optional[str] = None,
         headers: Optional[dict[str, str]] = None,
+        description: Optional[str] = None,
     ) -> Assistant:
         """Update an assistant.
 
@@ -697,6 +703,8 @@ class AssistantsClient:
             metadata: Metadata to merge with existing assistant metadata.
             name: The new name for the assistant.
             headers: Optional custom headers to include with the request.
+            description: Optional description of the assistant.
+                The description field is available for langgraph-api server version>=0.0.45
 
         Returns:
             Assistant: The updated assistant.
@@ -720,6 +728,8 @@ class AssistantsClient:
             payload["metadata"] = metadata
         if name:
             payload["name"] = name
+        if description:
+            payload["description"] = description
         return await self.http.patch(
             f"/assistants/{assistant_id}",
             json=payload,
@@ -3055,6 +3065,7 @@ class SyncAssistantsClient:
         if_exists: Optional[OnConflictBehavior] = None,
         name: Optional[str] = None,
         headers: Optional[dict[str, str]] = None,
+        description: Optional[str] = None,
     ) -> Assistant:
         """Create a new assistant.
 
@@ -3069,6 +3080,8 @@ class SyncAssistantsClient:
                 Must be either 'raise' (raise error if duplicate), or 'do_nothing' (return existing assistant).
             name: The name of the assistant. Defaults to 'Untitled' under the hood.
             headers: Optional custom headers to include with the request.
+            description: Optional description of the assistant.
+                The description field is available for langgraph-api server version>=0.0.45
 
         Returns:
             Assistant: The created assistant.
@@ -3097,6 +3110,8 @@ class SyncAssistantsClient:
             payload["if_exists"] = if_exists
         if name:
             payload["name"] = name
+        if description:
+            payload["description"] = description
         return self.http.post("/assistants", json=payload, headers=headers)
 
     def update(
@@ -3108,6 +3123,7 @@ class SyncAssistantsClient:
         metadata: Json = None,
         name: Optional[str] = None,
         headers: Optional[dict[str, str]] = None,
+        description: Optional[str] = None,
     ) -> Assistant:
         """Update an assistant.
 
@@ -3121,6 +3137,8 @@ class SyncAssistantsClient:
             metadata: Metadata to merge with existing assistant metadata.
             name: The new name for the assistant.
             headers: Optional custom headers to include with the request.
+            description: Optional description of the assistant.
+                The description field is available for langgraph-api server version>=0.0.45
 
         Returns:
             Assistant: The updated assistant.
@@ -3144,6 +3162,8 @@ class SyncAssistantsClient:
             payload["metadata"] = metadata
         if name:
             payload["name"] = name
+        if description:
+            payload["description"] = description
         return self.http.patch(
             f"/assistants/{assistant_id}",
             json=payload,
diff --git a/libs/sdk-py/langgraph_sdk/schema.py b/libs/sdk-py/langgraph_sdk/schema.py
index f6b8bb164..70c4eb610 100644
--- a/libs/sdk-py/langgraph_sdk/schema.py
+++ b/libs/sdk-py/langgraph_sdk/schema.py
@@ -171,6 +171,8 @@ class AssistantBase(TypedDict):
     """The version of the assistant"""
     name: str
     """The name of the assistant"""
+    description: Optional[str]
+    """The description of the assistant"""
 
 
 class AssistantVersion(AssistantBase):