7|hfddlmZddlZddlmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZer ddlmZddlmZej,eZGd d e ZGd d e Zy) ) annotationsN) TYPE_CHECKINGDictIterableIteratorListOptionalUnioncast)BaseChatLoader) ChatSession)loadClientRunc`eZdZdZ d d dZed dZed dZed dZd dZ y) LangSmithRunChatLoaderz Load chat sessions from a list of LangSmith "llm" runs. Attributes: runs (Iterable[Union[str, Run]]): The list of LLM run IDs or run objects. client (Client): Instance of LangSmith client for fetching data. Nc>ddlm}||_|xs||_y)z Initialize a new LangSmithRunChatLoader instance. :param runs: List of LLM run IDs or run objects. :param client: An instance of LangSmith client, if not provided, a new client instance will be created. rrN)langsmith.clientrrunsclient)selfrrrs i/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/chat_loaders/langsmith.py__init__zLangSmithRunChatLoader.__init__s , ( chtj|}tj|}|r||d<|S)z Convert an individual LangSmith LLM run to a ChatSession. :param llm_run: The LLM run object. :return: A chat session representing the run's data. functions)r_get_messages_from_llm_run_get_functions_from_llm_run)llm_run chat_sessionrs r_load_single_chat_sessionz0LangSmithRunChatLoader._load_single_chat_session)s7.HHQ *FFwO (1L %rcV|jdk7rtd|jd|jvrtd|j|js tdt |jd}t |jddd}t ||gz S) z Extract messages from a LangSmith LLM run. :param llm_run: The LLM run object. :return: ChatSession with the extracted messages. llmExpected run of type llm. Got: messagesz"Run has no 'messages' inputs. Got zCannot convert pending run generationsrmessage)r')run_type ValueErrorinputsoutputsrr )r!r' message_chunks rrz1LangSmithRunChatLoader._get_messages_from_llm_run7s   u $>w?O?O>PQR R W^^ +A'..AQRS S9: :' 3W__-m??rc|jdk7rtd|j|jxsijdijdS)z Extract functions from a LangSmith LLM run if they exist. :param llm_run: The LLM run object. :return: Functions from the run or None. r%r&invocation_paramsr)r*r+extraget)r!s rr z2LangSmithRunChatLoader._get_functions_from_llm_runIsT   u $>w?O?O>PQR R #(()s rrz#LangSmithDatasetChatLoader.__init__vsL  /)(  A  s : 5:c #Kddlm}|jj|j}|D]M}t |j dgDcgc]}|j|c}|j dOycc}ww)a: Lazy load the chat sessions from the specified LangSmith dataset. This method fetches the chat data from the dataset and converts each data point to chat sessions on-the-fly, yielding one session at a time. :return: Iterator of chat sessions containing messages. r)openai)rNr'r)r'rN)langchain_community.adaptersrPrread_dataset_openai_finetuningrNr r2convert_dict_to_message)r oai_adapterdata data_pointms rr?z$LangSmithDatasetChatLoader.lazy_loads G{{99**:  J(^^J; 77:%..5   sA B B %"B)rNstrrr@rC)rDrErFrGrr?rIrrrKrKmsKO)&rrK) __future__rloggingtypingrrrrrr r r langchain_core.chat_loadersr langchain_core.chat_sessionsr langchain_core.load.loadrrrr5r getLoggerrDr8rrKrIrrr`sS"WWW64)'%   8 $Y^Yx22r