WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-29 11:19:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

sdk-py: add versioning for assistants (#1665)

Browse Source 
* add get_versions to python sdk

* fml

* add change_version to sdk

* add assistant name to create

* document assistant name

* change method

* version in payload

* fml

* add js methods and types

* name -> assistant_name

* export onconflictbehavior

* add assistant name to create

* add metadata/limit/offset to sdk

* add assistant_name to update

* assistantversion schema

* export assistnatversion

* add metadata/limit/offset to sdk

* assistantversion schema

* fmt

* fmt

* change version -> set version

* fmt

* assistant_name -> name

* naming
This commit is contained in:
Isaac Francisco
2024-09-20 16:48:19 -07:00
committed by GitHub
parent 41cb54722d
commit 8b95adb1bd
5 changed files with 145 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/sdk-js/src/client.ts
+46
View File
@@ -9,6 +9,7 @@ import {
Thread,
ThreadState,
Cron,
AssistantVersion,
} from "./schema.js";
import { AsyncCaller, AsyncCallerParams } from "./utils/async_caller.js";
import {
@@ -244,6 +245,7 @@ export class AssistantsClient extends BaseClient {
metadata?: Metadata;
assistantId?: string;
ifExists?: OnConflictBehavior;
name?: string;
}): Promise<Assistant> {
return this.fetch<Assistant>("/assistants", {
method: "POST",
@@ -253,6 +255,7 @@ export class AssistantsClient extends BaseClient {
metadata: payload.metadata,
assistant_id: payload.assistantId,
if_exists: payload.ifExists,
name: payload.name,
},
});
}
@@ -269,6 +272,7 @@ export class AssistantsClient extends BaseClient {
graphId?: string;
config?: Config;
metadata?: Metadata;
name?: string;
},
): Promise<Assistant> {
return this.fetch<Assistant>(`/assistants/${assistantId}`, {
@@ -277,6 +281,7 @@ export class AssistantsClient extends BaseClient {
graph_id: payload.graphId,
config: payload.config,
metadata: payload.metadata,
name: payload.name,
},
});
}
@@ -313,6 +318,47 @@ export class AssistantsClient extends BaseClient {
},
});
}
/**
* List all versions of an assistant.
*
* @param assistantId ID of the assistant.
* @returns List of assistant versions.
*/
async getVersions(
assistantId: string,
payload?: {
metadata?: Metadata;
limit?: number;
offset?: number;
},
): Promise<AssistantVersion[]> {
return this.fetch<AssistantVersion[]>(
`/assistants/${assistantId}/versions`,
{
method: "POST",
json: {
metadata: payload?.metadata ?? undefined,
limit: payload?.limit ?? 10,
offset: payload?.offset ?? 0,
},
},
);
}
/**
* Change the version of an assistant.
*
* @param assistantId ID of the assistant.
* @param version The version to change to.
* @returns The updated assistant.
*/
async setLatest(assistantId: string, version: number): Promise<Assistant> {
return this.fetch<Assistant>(`/assistants/${assistantId}/set_latest`, {
method: "POST",
json: { version },
});
}
}
export class ThreadsClient extends BaseClient {
libs/sdk-js/src/index.ts
+3
View File
@@ -2,6 +2,7 @@ export { Client } from "./client.js";
export type {
Assistant,
AssistantVersion,
AssistantGraph,
Config,
DefaultValues,
@@ -12,3 +13,5 @@ export type {
ThreadState,
Cron,
} from "./schema.js";
export type { OnConflictBehavior } from "./types.js";
libs/sdk-js/src/schema.ts
+9 -2
View File
@@ -75,13 +75,20 @@ export interface GraphSchema {
export type Metadata = Optional<Record<string, unknown>>;
export interface Assistant {
export interface AssistantBase {
assistant_id: string;
graph_id: string;
config: Config;
created_at: string;
updated_at: string;
metadata: Metadata;
version: number;
}
export interface AssistantVersion extends AssistantBase {}
export interface Assistant extends AssistantBase {
updated_at: string;
name: string;
}
export type AssistantGraph = Record<string, Array<Record<string, unknown>>>;
libs/sdk-py/langgraph_sdk/client.py
+68 -1
View File
@@ -23,6 +23,7 @@ from httpx._types import QueryParamTypes
import langgraph_sdk
from langgraph_sdk.schema import (
Assistant,
AssistantVersion,
Config,
Cron,
DisconnectMode,
@@ -417,6 +418,7 @@ class AssistantsClient:
metadata: Json = None,
assistant_id: Optional[str] = None,
if_exists: Optional[OnConflictBehavior] = None,
name: Optional[str] = None,
) -> Assistant:
"""Create a new assistant.
@@ -429,6 +431,7 @@ class AssistantsClient:
assistant_id: Assistant ID to use, will default to a random UUID if not provided.
if_exists: How to handle duplicate creation. Defaults to 'raise' under the hood.
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.
Returns:
Assistant: The created assistant.
@@ -440,7 +443,8 @@ class AssistantsClient:
config={"configurable": {"model_name": "openai"}},
metadata={"number":1},
assistant_id="my-assistant-id",
if_exists="do_nothing"
if_exists="do_nothing",
name="my_name"
)
""" # noqa: E501
payload: Dict[str, Any] = {
@@ -454,6 +458,8 @@ class AssistantsClient:
payload["assistant_id"] = assistant_id
if if_exists:
payload["if_exists"] = if_exists
if name:
payload["name"] = name
return await self.http.post("/assistants", json=payload)
async def update(
@@ -463,6 +469,7 @@ class AssistantsClient:
graph_id: Optional[str] = None,
config: Optional[Config] = None,
metadata: Json = None,
name: Optional[str] = None,
) -> Assistant:
"""Update an assistant.
@@ -495,6 +502,8 @@ class AssistantsClient:
payload["config"] = config
if metadata:
payload["metadata"] = metadata
if name:
payload["name"] = name
return await self.http.patch(
f"/assistants/{assistant_id}",
json=payload,
@@ -563,6 +572,64 @@ class AssistantsClient:
json=payload,
)
async def get_versions(
self,
assistant_id: str,
metadata: Json = None,
limit: int = 10,
offset: int = 0,
) -> list[AssistantVersion]:
"""List all versions of an assistant.
Args:
assistant_id: The assistant ID to delete.
Returns:
list[Assistant]: A list of assistants.
Example Usage:
assistant_versions = await client.assistants.get_versions(
assistant_id="my_assistant_id"
)
""" # noqa: E501
payload: Dict[str, Any] = {
"limit": limit,
"offset": offset,
}
if metadata:
payload["metadata"] = metadata
return await self.http.post(
f"/assistants/{assistant_id}/versions", json=payload
)
async def set_latest(self, assistant_id: str, version: int) -> Assistant:
"""Change the version of an assistant.
Args:
assistant_id: The assistant ID to delete.
version: The version to change to.
Returns:
Assistant: Assistant Object.
Example Usage:
new_version_assistant = await client.assistants.set_latest(
assistant_id="my_assistant_id",
version=3
)
""" # noqa: E501
payload: Dict[str, Any] = {"version": version}
return await self.http.post(
f"/assistants/{assistant_id}/set_latest", json=payload
)
class ThreadsClient:
def __init__(self, http: HttpClient) -> None:
libs/sdk-py/langgraph_sdk/schema.py
+19 -4
View File
@@ -57,8 +57,8 @@ class GraphSchema(TypedDict):
Missing if unable to generate JSON schema from graph."""
class Assistant(TypedDict):
"""Assistant model."""
class AssistantBase(TypedDict):
"""Assistant base model."""
assistant_id: str
"""The ID of the assistant."""
@@ -68,10 +68,25 @@ class Assistant(TypedDict):
"""The assistant config."""
created_at: datetime
"""The time the assistant was created."""
updated_at: datetime
"""The last time the assistant was updated."""
metadata: Json
"""The assistant metadata."""
version: int
"""The version of the assistant"""
class AssistantVersion(AssistantBase):
"""Assistant version model."""
pass
class Assistant(AssistantBase):
"""Assistant model."""
updated_at: datetime
"""The last time the assistant was updated."""
name: str
"""The name of the assistant"""
class Thread(TypedDict):