tjidZddlmZddlmZddlmZmZmZm Z ddl Z ddl m Z ddl mZddlmZmZmZmZmZmZmZmZmZmZmZmZmZGd d ZdS) z2Async client for managing assistants in LangGraph.) annotations)Mapping)AnyLiteralcastoverloadN) HttpClient)_quote_path_param) AssistantAssistantSelectFieldAssistantSortByAssistantsSearchResponseAssistantVersionConfigContext GraphSchemaJsonOnConflictBehaviorQueryParamTypes SortOrder Subgraphsc reZdZdZdOdZddddPdZdddddQdZddddRdZ dSddddTdZ dUddddddddddVd+Z ddddddddd,dWd-Z dddd.dXd0Z e dddd1d2dddddd3 dYd@Z e dddd1d2ddddAdddB dZdEZ dddd1d2ddddAdddB d[dHZ ddddddId\dJZ d]dddd^dLZdddd_dNZdS)`AssistantsClienta]Client for managing assistants in LangGraph. This class provides methods to interact with assistants, which are versioned configurations of your graph. ???+ example "Example" ```python client = get_client(url="http://localhost:2024") assistant = await client.assistants.get("assistant_id_123") ``` httpr returnNonec||_dSN)r)selfrs ]/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langgraph_sdk/_async/assistants.py__init__zAssistantsClient.__init__+s  Nheadersparams assistant_idstrr$Mapping[str, str] | Noner%QueryParamTypes | Noner clK|jdt|||d{VS)aGet an assistant by ID. Args: assistant_id: The ID of the assistant to get. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: Assistant: Assistant Object. ???+ example "Example Usage" ```python assistant = await client.assistants.get( assistant_id="my_assistant_id" ) print(assistant) ``` ```shell ---------------------------------------------------- { 'assistant_id': 'my_assistant_id', 'graph_id': 'agent', 'created_at': '2024-06-25T17:10:33.109781+00:00', 'updated_at': '2024-06-25T17:10:33.109781+00:00', 'config': {}, 'metadata': {'created_by': 'system'}, 'version': 1, 'name': 'my_assistant' } ``` /assistants/r#Nrgetr rr&r$r%s r r-zAssistantsClient.get.s^RY]] <,\:: < <#         r"F)xrayr$r%r/ int | booldict[str, list[dict[str, Any]]]cKd|i}|r|||jdt|d||d{VS)aIGet the graph of an assistant by ID. Args: assistant_id: The ID of the assistant to get the graph of. xray: Include graph representation of subgraphs. If an integer value is provided, only subgraphs with a depth less than or equal to the value will be included. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: Graph: The graph information for the assistant in JSON format. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") graph_info = await client.assistants.get_graph( assistant_id="my_assistant_id" ) print(graph_info) ``` ```shell -------------------------------------------------------------------------------------------------------------------------- { 'nodes': [ {'id': '__start__', 'type': 'schema', 'data': '__start__'}, {'id': '__end__', 'type': 'schema', 'data': '__end__'}, {'id': 'agent','type': 'runnable','data': {'id': ['langgraph', 'utils', 'RunnableCallable'],'name': 'agent'}}, ], 'edges': [ {'source': '__start__', 'target': 'agent'}, {'source': 'agent','target': '__end__'} ] } ``` r/r+z/graphr%r$N)updaterr-r )rr&r/r$r% query_paramss r get_graphzAssistantsClient.get_graph]sd~  (    ' ' 'Y]] B,\:: B B B#         r"rcnK|jdt|d||d{VS)aGet the schemas of an assistant by ID. Args: assistant_id: The ID of the assistant to get the schema of. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: GraphSchema: The graph schema for the assistant. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") schema = await client.assistants.get_schemas( assistant_id="my_assistant_id" ) print(schema) ``` ```shell ---------------------------------------------------------------------------------------------------------------------------- { 'graph_id': 'agent', 'state_schema': { 'title': 'LangGraphInput', '$ref': '#/definitions/AgentState', 'definitions': { 'BaseMessage': { 'title': 'BaseMessage', 'description': 'Base abstract Message class. Messages are the inputs and outputs of ChatModels.', 'type': 'object', 'properties': { 'content': { 'title': 'Content', 'anyOf': [ {'type': 'string'}, {'type': 'array','items': {'anyOf': [{'type': 'string'}, {'type': 'object'}]}} ] }, 'additional_kwargs': { 'title': 'Additional Kwargs', 'type': 'object' }, 'response_metadata': { 'title': 'Response Metadata', 'type': 'object' }, 'type': { 'title': 'Type', 'type': 'string' }, 'name': { 'title': 'Name', 'type': 'string' }, 'id': { 'title': 'Id', 'type': 'string' } }, 'required': ['content', 'type'] }, 'AgentState': { 'title': 'AgentState', 'type': 'object', 'properties': { 'messages': { 'title': 'Messages', 'type': 'array', 'items': {'$ref': '#/definitions/BaseMessage'} } }, 'required': ['messages'] } } }, 'context_schema': { 'title': 'Context', 'type': 'object', 'properties': { 'model_name': { 'title': 'Model Name', 'enum': ['anthropic', 'openai'], 'type': 'string' } } } } ``` r+z/schemasr#Nr,r.s r get_schemaszAssistantsClient.get_schemassajY]] D,\:: D D D#         r" namespace str | Nonerecurseboolrc*Kd|i}|ri|t|}|C|jdt|dt|||d{VS|jdt|d||d{VS)aGet the schemas of an assistant by ID. Args: assistant_id: The ID of the assistant to get the schema of. namespace: Optional namespace to filter by. recurse: Whether to recursively get subgraphs. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: Subgraphs: The graph schema for the assistant. r;Nr+z /subgraphs/r3z /subgraphs)dictrr-r )rr&r9r;r$r% get_paramss r get_subgraphszAssistantsClient.get_subgraphss, )  87J7$v,,7J  i0>>iiK\]fKgKgii!'  J0>>JJJ!' r")contextmetadatar& if_existsnamer$ descriptionr%graph_idconfig Config | NonerAContext | NonerBrrCOnConflictBehavior | NonerDrEcKd|i} |r|| d<|r|| d<|r|| d<|r|| d<|r|| d<|r|| d<| r| | d<|jd | || d {VS) aCreate a new assistant. Useful when graph is configurable and you want to create different assistants based on different configurations. Args: graph_id: The ID of the graph the assistant should use. The graph ID is normally set in your langgraph.json configuration. config: Configuration to use for the graph. metadata: Metadata to add to assistant. context: Static context to add to the assistant. !!! version-added "Added in version 0.6.0" 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. 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 params: Optional query parameters to include with the request. Returns: Assistant: The created assistant. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") assistant = await client.assistants.create( graph_id="agent", context={"model_name": "openai"}, metadata={"number":1}, assistant_id="my-assistant-id", if_exists="do_nothing", name="my_name" ) ``` rFrGrArBr&rCrDrEz /assistantsjsonr$r%Nrpost) rrFrGrArBr&rCrDr$rEr%payloads r createzAssistantsClient.create:sf #   ' &GH   )!(GI   +"*GJ   3&2GN #  -#,GK  #"GFO  1%0GM "Y^^ $         r")rFrGrArBrDr$rEr%cKi} |r|| d<||| d<||| d<|r|| d<|r|| d<|r|| d<|jdt|| || d{VS) aUpdate an assistant. Use this to point to a different graph, update the configuration, or change the metadata of an assistant. Args: assistant_id: Assistant to update. graph_id: The ID of the graph the assistant should use. The graph ID is normally set in your langgraph.json configuration. If `None`, assistant will keep pointing to same graph. config: Configuration to use for the graph. context: Static context to add to the assistant. !!! version-added "Added in version 0.6.0" 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 params: Optional query parameters to include with the request. Returns: The updated assistant. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") assistant = await client.assistants.update( assistant_id='e280dad7-8618-443f-87f1-8e41841c180f', graph_id="other-graph", context={"model_name": "anthropic"}, metadata={"number":2} ) ``` rFNrGrArBrDrEr+rL)rpatchr ) rr&rFrGrArBrDr$rEr%rPs r r4zAssistantsClient.updates^#%  +"*GJ    &GH   !(GI   +"*GJ   #"GFO  1%0GM "Y__ <,\:: < < %         r")delete_threadsr$r%rTcKi}|rd|d<|r|||jdt|||pdd{VdS)aDelete an assistant. Args: assistant_id: The assistant ID to delete. delete_threads: If true, delete all threads with `metadata.assistant_id` matching this assistant, along with runs and checkpoints belonging to those threads. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: `None` ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") await client.assistants.delete( assistant_id="my_assistant_id" ) ``` TrTr+Nr#)r4rdeleter )rr&rTr$r%r5s r rVzAssistantsClient.deletes>(*  2-1L) *  (    ' ' 'i <,\:: < <'4           r" r) rBrFrDlimitoffsetsort_by sort_orderselectr$r%rXintrYrZAssistantSortBy | Noner[SortOrder | Noner\!list[AssistantSelectField] | Noneresponse_formatLiteral['object']rc  KdSr rrBrFrDrXrYrZr[r\rar$r%s r searchzAssistantsClient.searchs $'3r"array) rBrFrDrXrYrZr[r\rar$r%Literal['array']list[Assistant]c  KdSrrdres r rfzAssistantsClient.searchs #r"Literal['array', 'object']*AssistantsSearchResponse | list[Assistant]c RK| dvrtd| ||d} |r|| d<|r|| d<|r|| d<|r|| d<|r|| d<|r|| d <d dfd } ttt|jd| | | | dkr| nd d {V}| dkr|dS|S)akSearch for assistants. Args: metadata: Metadata to filter by. Exact match filter for each KV pair. graph_id: The ID of the graph to filter by. The graph ID is normally set in your langgraph.json configuration. name: The name of the assistant to filter by. The filtering logic will match assistants where 'name' is a substring (case insensitive) of the assistant name. limit: The maximum number of results to return. offset: The number of results to skip. sort_by: The field to sort by. sort_order: The order to sort by. select: Specific assistant fields to include in the response. response_format: Controls the response shape. Use `"array"` (default) to return a bare list of assistants, or `"object"` to return a mapping containing assistants plus pagination metadata. Defaults to "array", though this default will be changed to "object" in a future release. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: A list of assistants (when `response_format="array"`) or a mapping with the assistants and the next pagination cursor (when `response_format="object"`). ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") response = await client.assistants.search( metadata = {"name":"my_name"}, graph_id="my_graph_id", limit=5, offset=5, response_format="object" ) next_cursor = response["next"] assistants = response["assistants"] ``` )rgobjectz1response_format must be 'array' or 'object', got rXrYrBrFrDrZr[r\Nresponsehttpx.Responserrc<|jddS)NzX-Pagination-Next)r$r-)rp next_cursors r capture_paginationz3AssistantsClient.search..capture_pagination]s"*../BCCKKKr"z/assistants/searchrn)rMr$r% on_response) assistantsnext)rprqrr) ValueErrorrlistr rrO)rrBrFrDrXrYrZr[r\rar$r%rPrtrvrss @r rfzAssistantsClient.searchsln "5 5 5WOWW # #   +"*GJ   +"*GJ   #"GFO  )!(GI   /$.GL !  ' &GH "&  D D D D D D O)..$2AX2M2M..SW !          h & &",kBB Br")rBrFrDr$r%c|Ki}|r||d<|r||d<|r||d<|jd|||d{VS)aOCount assistants matching filters. Args: metadata: Metadata to filter by. Exact match for each key/value. graph_id: Optional graph id to filter by. name: Optional name to filter by. The filtering logic will match assistants where 'name' is a substring (case insensitive) of the assistant name. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: int: Number of assistants matching the criteria. rBrFrDz/assistants/countrLNrN)rrBrFrDr$r%rPs r countzAssistantsClient.countos,#%  +"*GJ   +"*GJ   #"GFOY^^ gwv$         r"list[AssistantVersion]cK||d}|r||d<|jdt|d|||d{VS)a%List all versions of an assistant. Args: assistant_id: The assistant ID to get versions for. metadata: Metadata to filter versions by. Exact match filter for each KV pair. limit: The maximum number of versions to return. offset: The number of versions to skip. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: A list of assistant versions. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") assistant_versions = await client.assistants.get_versions( assistant_id="my_assistant_id" ) ``` rorBr+z /versionsrLNrrOr )rr&rBrXrYr$r%rPs r get_versionszAssistantsClient.get_versionssD# #   +"*GJ Y^^ E,\:: E E E $         r"versioncxKd|i}|jdt|d|||d{VS)aChange the version of an assistant. Args: assistant_id: The assistant ID to delete. version: The version to change to. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: Assistant Object. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") new_version_assistant = await client.assistants.set_latest( assistant_id="my_assistant_id", version=3 ) ``` rr+z/latestrLNr~)rr&rr$r%rPs r set_latestzAssistantsClient.set_latestsm>$-g"6Y^^ C,\:: C C C $         r")rr rr)r&r'r$r(r%r)rr ) r&r'r/r0r$r(r%r)rr1)r&r'r$r(r%r)rr)NF) r&r'r9r:r;r<r$r(r%r)rrr)rFr:rGrHrArIrBrr&r:rCrJrDr:r$r(rEr:r%r)rr )r&r'rFr:rGrHrArIrBrrDr:r$r(rEr:r%r)rr ) r&r'rTr<r$r(r%r)rr)rBrrFr:rDr:rXr]rYr]rZr^r[r_r\r`rarbr$r(r%r)rr)rBrrFr:rDr:rXr]rYr]rZr^r[r_r\r`rarhr$r(r%r)rri)rBrrFr:rDr:rXr]rYr]rZr^r[r_r\r`rarkr$r(r%r)rrl) rBrrFr:rDr:r$r(r%r)rr])NrWr)r&r'rBrrXr]rYr]r$r(r%r)rr|) r&r'rr]r$r(r%r)rr )__name__ __module__ __qualname____doc__r!r-r6r8r@rQr4rVrrfr{rrrdr"r rrs  -1)- - - - - - - f!,0)- : : : : : : @-1)- y y y y y y |!% $ -1)-$$$$$$R!%E #'#'/3,0"&)-E E E E E E V $ $"&,0"&)-A A A A A A N %,0)- ( ( ( ( ( ( T#*.'+48,0)-'''''X' #*.'+48,3,0)-X&#*.'+486=,0)-]]]]]]D#,0)-      H , -1)-, , , , , , f-1)- & & & & & & & & r"r)r __future__rcollections.abcrtypingrrrrhttpxlanggraph_sdk._async.httpr langgraph_sdk._shared.utilitiesr langgraph_sdk.schemar r r rrrrrrrrrrrrdr"r rsR88""""""######//////////// 000000======"G  G  G  G  G  G  G  G  G  G  r"