sdk-py: Add values to get/list threads

This commit is contained in:
Nuno Campos
2024-08-30 14:23:10 -07:00
parent 3943637202
commit 770132dd79
2 changed files with 15 additions and 12 deletions
+9 -6
View File
@@ -27,7 +27,7 @@ from langgraph_sdk.schema import (
Cron,
DisconnectMode,
GraphSchema,
Metadata,
Json,
MultitaskStrategy,
OnCompletionBehavior,
OnConflictBehavior,
@@ -414,7 +414,7 @@ class AssistantsClient:
graph_id: Optional[str],
config: Optional[Config] = None,
*,
metadata: Metadata = None,
metadata: Json = None,
assistant_id: Optional[str] = None,
if_exists: Optional[OnConflictBehavior] = None,
) -> Assistant:
@@ -462,7 +462,7 @@ class AssistantsClient:
*,
graph_id: Optional[str] = None,
config: Optional[Config] = None,
metadata: Metadata = None,
metadata: Json = None,
) -> Assistant:
"""Update an assistant.
@@ -524,7 +524,7 @@ class AssistantsClient:
async def search(
self,
*,
metadata: Metadata = None,
metadata: Json = None,
graph_id: Optional[str] = None,
limit: int = 10,
offset: int = 0,
@@ -600,7 +600,7 @@ class ThreadsClient:
async def create(
self,
*,
metadata: Metadata = None,
metadata: Json = None,
thread_id: Optional[str] = None,
if_exists: Optional[OnConflictBehavior] = None,
) -> Thread:
@@ -675,7 +675,8 @@ class ThreadsClient:
async def search(
self,
*,
metadata: Metadata = None,
metadata: Json = None,
values: Json = None,
status: Optional[ThreadStatus] = None,
limit: int = 10,
offset: int = 0,
@@ -708,6 +709,8 @@ class ThreadsClient:
}
if metadata:
payload["metadata"] = metadata
if values:
payload["values"] = values
if status:
payload["status"] = status
return await self.http.post(
+6 -6
View File
@@ -1,7 +1,7 @@
from datetime import datetime
from typing import Any, Literal, Optional, Sequence, TypedDict, Union
Metadata = Optional[dict[str, Any]]
Json = Optional[dict[str, Any]]
RunStatus = Literal["pending", "running", "error", "success", "timeout", "interrupted"]
@@ -70,7 +70,7 @@ class Assistant(TypedDict):
"""The time the assistant was created."""
updated_at: datetime
"""The last time the assistant was updated."""
metadata: Metadata
metadata: Json
"""The assistant metadata."""
@@ -81,11 +81,11 @@ class Thread(TypedDict):
"""The time the thread was created."""
updated_at: datetime
"""The last time the thread was updated."""
metadata: Metadata
metadata: Json
"""The thread metadata."""
status: ThreadStatus
"""The status of the thread, one of 'idle', 'busy', 'interrupted'."""
values: dict
values: Json
"""The current state of the thread."""
@@ -97,7 +97,7 @@ class ThreadState(TypedDict):
received."""
checkpoint_id: str
"""The ID of the checkpoint."""
metadata: Metadata
metadata: Json
"""Metadata for this state"""
created_at: Optional[str]
"""Timestamp of state creation"""
@@ -118,7 +118,7 @@ class Run(TypedDict):
"""The last time the run was updated."""
status: RunStatus
"""The status of the run. One of 'pending', 'running', "error", 'success', "timeout", "interrupted"."""
metadata: Metadata
metadata: Json
"""The run metadata."""
multitask_strategy: MultitaskStrategy
"""Strategy to handle concurrent runs on the same thread."""