c ihdZddlmZddlmZddlmZmZmZm Z ddl Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZmZmZmZGdd ZdS) z2Async client for managing assistants in LangGraph.) annotations)Mapping)AnyLiteralcastoverloadN) HttpClient) 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 C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langgraph_sdk/_async/assistants.py__init__zAssistantsClient.__init__*s  Nheadersparams assistant_idstrr#Mapping[str, str] | Noner$QueryParamTypes | Noner cRK|jd|||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"Nrgetrr%r#r$s rr,zAssistantsClient.get-sSRY]] )< ) )76#         r!F)xrayr#r$r. int | booldict[str, list[dict[str, Any]]]cKd|i}|r|||jd|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,)rr%r.r#r$ query_paramss r get_graphzAssistantsClient.get_graphZs{d~  (    ' ' 'Y]] /< / / / g#         r!rcTK|jd|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_schemassVjY]] 1< 1 1 176#         r! namespace str | NonerecurseboolrcKd|i}|ri|t|}|)|jd|d|||d{VS|jd|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/r2z /subgraphs)dictrr,)rr%r8r:r#r$ get_paramss r get_subgraphszAssistantsClient.get_subgraphs s, )  87J7$v,,7J  C|CC CC!'  7|777!' r!)contextmetadatar% if_existsnamer# descriptionr$graph_idconfig Config | Noner@Context | NonerArrBOnConflictBehavior | NonerCrDcKd|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" ) ``` rErFr@rAr%rBrCrDz /assistantsjsonr#r$Nrpost) rrErFr@rAr%rBrCr#rDr$payloads rcreatezAssistantsClient.create3sf #   ' &GH   )!(GI   +"*GJ   3&2GN #  -#,GK  #"GFO  1%0GM "Y^^ $         r!)rErFr@rArCr#rDr$cKi} |r|| d<||| d<||| d<|r|| d<|r|| d<|r|| d<|jd|| || 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} ) ``` rENrFr@rArCrDr*rK)rpatch) rr%rErFr@rArCr#rDr$rOs rr3zAssistantsClient.updatezs^#%  +"*GJ    &GH   !(GI   +"*GJ   #"GFO  1%0GM "Y__ )< ) ) %         r!)delete_threadsr#r$rScKi}|rd|d<|r|||jd|||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" ) ``` TrSr*Nr")r3rdelete)rr%rSr#r$r4s rrUzAssistantsClient.deletes>(*  2-1L) *  (    ' ' 'i )< ) )'4           r! r) rArErClimitoffsetsort_by sort_orderselectr#r$rWintrXrYAssistantSortBy | NonerZSortOrder | Noner[!list[AssistantSelectField] | Noneresponse_formatLiteral['object']r c  KdSr rrArErCrWrXrYrZr[r`r#r$s rsearchzAssistantsClient.searchs $'3r!array) rArErCrWrXrYrZr[r`r#r$Literal['array']list[Assistant]c  KdSrrcrds rrezAssistantsClient.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"] ``` )rfobjectz1response_format must be 'array' or 'object', got rWrXrArErCrYrZr[Nresponsehttpx.Responserrc<|jddS)NzX-Pagination-Next)r#r,)ro next_cursors rcapture_paginationz3AssistantsClient.search..capture_paginationVs"*../BCCKKKr!z/assistants/searchrm)rLr#r$ on_response) assistantsnext)rorprr) ValueErrorrlistr rrN)rrArErCrWrXrYrZr[r`r#r$rOrsrurrs @rrezAssistantsClient.search sln "5 5 5WOWW # #   +"*GJ   +"*GJ   #"GFO  )!(GI   /$.GL !  ' &GH "&  D D D D D D O)..$2AX2M2M..SW !          h & &",kBB Br!)rArErCr#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. rArErCz/assistants/countrKNrM)rrArErCr#r$rOs rcountzAssistantsClient.counths,#%  +"*GJ   +"*GJ   #"GFOY^^ gwv$         r!list[AssistantVersion]cnK||d}|r||d<|jd|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" ) ``` rnrAr*z /versionsrKNrM)rr%rArWrXr#r$rOs r get_versionszAssistantsClient.get_versionssD# #   +"*GJ Y^^ 2< 2 2 2 $         r!versionc^Kd|i}|jd|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 ) ``` r~r*z/latestrKNrM)rr%r~r#r$rOs r set_latestzAssistantsClient.set_latestsd>$-g"6Y^^ 0< 0 0 0 $         r!)rr rr)r%r&r#r'r$r(rr ) r%r&r.r/r#r'r$r(rr0)r%r&r#r'r$r(rr)NF) r%r&r8r9r:r;r#r'r$r(rrr)rEr9rFrGr@rHrArr%r9rBrIrCr9r#r'rDr9r$r(rr )r%r&rEr9rFrGr@rHrArrCr9r#r'rDr9r$r(rr ) r%r&rSr;r#r'r$r(rr)rArrEr9rCr9rWr\rXr\rYr]rZr^r[r_r`rar#r'r$r(rr )rArrEr9rCr9rWr\rXr\rYr]rZr^r[r_r`rgr#r'r$r(rrh)rArrEr9rCr9rWr\rXr\rYr]rZr^r[r_r`rjr#r'r$r(rrk) rArrEr9rCr9r#r'r$r(rr\)NrVr)r%r&rArrWr\rXr\r#r'r$r(rr{) r%r&r~r\r#r'r$r(rr )__name__ __module__ __qualname____doc__r r,r5r7r?rPr3rUrrerzr}rrcr!rrrs  -1)- + + + + + + b!,0)- 8 8 8 8 8 8 |-1)- w w w w w w x!% $ -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.schemar r r r rrrrrrrrrrrcr!rrs@88""""""######//////////// 000000"A  A  A  A  A  A  A  A  A  A  r!