NgfddlmZddlZddlmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZer ddlmZddlmZejeZGd d e ZGd d e ZdS) ) annotationsN) TYPE_CHECKINGDictIterableIteratorListOptionalUnioncast)BaseChatLoader) ChatSession)loadClientRuncneZdZdZ dddZedd Zedd ZeddZddZ 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']cBddlm}||_|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)selfrrrs f/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/chat_loaders/langsmith.py__init__zLangSmithRunChatLoader.__init__s3 ,+++++ ( llm_run'Run'returnr c|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)r chat_sessionr#s r_load_single_chat_sessionz0LangSmithRunChatLoader._load_single_chat_session)sB.HHQQ *FFwOO  2(1L %rcZ|jdkrtd|jd|jvrtd|j|jstdt |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 )rr+ message_chunks rr$z1LangSmithRunChatLoader._get_messages_from_llm_run7s  u $ $Qw?OQQRR R W^ + +R'.RRSS S ;9:: :'' 3W_--m????rOptional[List[Dict]]c|jdkrtd|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*invocation_paramsr#)r.r/extraget)rs rr%z2LangSmithRunChatLoader._get_functions_from_llm_runIsX  u $ $Qw?OQQRR R #(()z8LangSmithDatasetChatLoader.lazy_load..s7 77::rr+r#)r+r#N)langchain_community.adaptersrTrread_dataset_openai_finetuningrOr r7)rdata data_pointrYs @rrEz$LangSmithDatasetChatLoader.lazy_loads GFFFFF{99*:    J'^^J;;%..55       r)rOrPrrrF)rGrHrIrJrrErLrrrNrNmsWKO))))))&rrN) __future__rloggingtypingrrrrrr r r langchain_core.chat_loadersr langchain_core.chat_sessionsr langchain_core.load.loadrrrr;r getLoggerrGr>rrNrLrrrfs7""""""WWWWWWWWWWWWWWWWWWWW666666444444))))))&''''''%%%%%%  8 $ $YYYYY^YYYx2222222222r