o Zhf@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@sReZdZdZ ddddZedd dZedddZedddZdddZ 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. NrunsIterable[Union[str, Run]]clientOptional['Client']cCs"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)selfrrrra/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/chat_loaders/langsmith.py__init__s zLangSmithRunChatLoader.__init__llm_run'Run'returnr cCs$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|jdkr td|jd|jvrtd|j|js!tdt|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 ValueErrorZinputsZoutputsr r )rr%Z message_chunkrrrr 7s  z1LangSmithRunChatLoader._get_messages_from_llm_runOptional[List[Dict]]cCs2|jdkr td|j|jpididS)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)rrrrr!Is z2LangSmithRunChatLoader._get_functions_from_llm_runIterator[ChatSession]c csddlm}|jD]<}zt|dr|}n|j|}|t||}|VWq tyF}zt d|dt |WYd}~q d}~wwdS)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)rrrr)rrrr )rrrr)rr,) __name__ __module__ __qualname____doc__r staticmethodr"r r!r6rrrrrs     rc@s*eZdZdZddddd Zdd 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)r dataset_namestrrrc CsJzddlm}Wnty}ztd|d}~ww||_|p!||_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__rr,c#sVddlm|jj|jd}|D]}tfdd|dgD|ddVqd 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.adaptersrArZread_dataset_openai_finetuningr>r r+)rdataZ data_pointrrDrr6s    z$LangSmithDatasetChatLoader.lazy_load)r>r?rrr7)r8r9r:r;rr6rrrrr=msr=) __future__rloggingtypingrrrrrrr r Zlangchain_core.chat_loadersr Zlangchain_core.chat_sessionsr Zlangchain_core.load.loadr rrr.r getLoggerr8r0rr=rrrrs (      \