a bgf@sddlmZddlZddlmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZer|ddlmZddlmZeeZGd d d e ZGd d d e ZdS) ) annotationsN) TYPE_CHECKINGDictIterableIteratorListOptionalUnioncast)BaseChatLoader) ChatSession)loadClientRunc@sleZdZdZddddddZedd d d d Zedd d d dZeddd ddZddddZ dS)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. NzIterable[Union[str, Run]]Optional['Client'])runsclientcCs"ddlm}||_|p||_dS)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.clientrrr)selfrrrrx/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/chat_loaders/langsmith.py__init__s zLangSmithRunChatLoader.__init__z'Run'r )llm_runreturncCs$t|}t|}|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)rZ chat_sessionrrrr_load_single_chat_session)s   z0LangSmithRunChatLoader._load_single_chat_sessioncCsv|jdkrtd|jd|jvr4td|j|jsBtdt|jd}t|jddd}t||gd 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 runZ generationsrmessage)r#)run_type ValueErrorinputsoutputsr r )rr#Z message_chunkrrrr7s  z1LangSmithRunChatLoader._get_messages_from_llm_runzOptional[List[Dict]]cCs2|jdkrtd|j|jp"ididS)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"Zinvocation_paramsr)r%r&extraget)rrrrrIs z2LangSmithRunChatLoader._get_functions_from_llm_runIterator[ChatSession]rc csddlm}|jD]}z6t|dr(|}n |j|}|t||}|VWqty}z2t d|dt |WYd}~qWYd}~qd}~00qdS)a  Lazy load the chat sessions from the iterable of run IDs. This method fetches the runs and converts them to chat sessions on-the-fly, yielding one session at a time. :return: Iterator of chat sessions containing messages. rridzCould not load run z: N) langsmith.schemasrrhasattrrZread_runr r r&loggerwarningrepr)rrZrun_objrunsessionerrr lazy_loadUs     z LangSmithRunChatLoader.lazy_load)N) __name__ __module__ __qualname____doc__r staticmethodr rrr6rrrrrs   rc@s4eZdZdZdddddddZd d d d ZdS) LangSmithDatasetChatLoaderz Load chat sessions from a LangSmith dataset with the "chat" data type. Attributes: dataset_name (str): The name of the LangSmith dataset. client (Client): Instance of LangSmith client for fetching data. N)rstrr) dataset_namerc CsVzddlm}Wn.ty>}ztd|WYd}~n d}~00||_|pN||_dS)a Initialize a new LangSmithChatDatasetLoader instance. :param dataset_name: The name of the LangSmith dataset. :param client: An instance of LangSmith client; if not provided, a new client instance will be created. rrzkThe LangSmith client is required to load LangSmith datasets. Please install it with `pip install langsmith`N)rr ImportErrorr>r)rr>rrr5rrrrvsz#LangSmithDatasetChatLoader.__init__r+r,c#sTddlm|jj|jd}|D].}tfdd|dgD|ddVq d S) 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)r>csg|]}|qSr)Zconvert_dict_to_message).0mZ oai_adapterrr sz8LangSmithDatasetChatLoader.lazy_load..r#r)r#rN)Zlangchain_community.adaptersr@rZread_dataset_openai_finetuningr>r r*)rdataZ data_pointrrCrr6s   z$LangSmithDatasetChatLoader.lazy_load)r7r8r9r:rr6rrrrr<msr<) __future__rloggingtypingrrrrrrr r Zlangchain_core.chat_loadersr Zlangchain_core.chat_sessionsr Zlangchain_core.load.loadr rrr.r getLoggerr7r0rr<rrrrs (      \