AhSrSSKJr SSKJr SSKJr SSKJr SSK J r SSK J r J r SSKJr SS KJr "S S 5rg ) z!Experiment with different models.) annotations)Sequence)Optional)BaseLLM)PromptTemplate)get_color_mapping print_text)Chain)LLMChaincT\rSrSrSrSS Sjjr\SS Sjj5rS SjrSr g) ModelLaboratoryzMA utility to experiment with and compare the performance of different models.NcUHn[U[5(d Sn[U5e[UR5S:waSUR3n[U5e[UR 5S:wdMsSUR 3nM Ub%[U5[U5:wa Sn[U5eXl[[UR 55Vs/sHn[U5PM nn[U5Ul X l gs snf)a~Initialize the ModelLaboratory with chains to experiment with. Args: chains (Sequence[Chain]): A sequence of chains to experiment with. Each chain must have exactly one input and one output variable. names (Optional[List[str]]): Optional list of names corresponding to each chain. If provided, its length must match the number of chains. Raises: ValueError: If any chain is not an instance of `Chain`. ValueError: If a chain does not have exactly one input variable. ValueError: If a chain does not have exactly one output variable. ValueError: If the length of `names` does not match the number of chains. zModelLaboratory should now be initialized with Chains. If you want to initialize with LLMs, use the `from_llms` method instead (`ModelLaboratory.from_llms(...)`)z;Currently only support chains with one input variable, got z!s1v'> ?-k: @sDcUc [S/SS9nUVs/sH n[X2S9PM nnUVs/sHn[U5PM nnU"XES9$s snfs snf)auInitialize the ModelLaboratory with LLMs and an optional prompt. Args: llms (List[BaseLLM]): A list of LLMs to experiment with. prompt (Optional[PromptTemplate]): An optional prompt to use with the LLMs. If provided, the prompt must contain exactly one input variable. Returns: ModelLaboratory: An instance of `ModelLaboratory` initialized with LLMs. _inputz{_input})input_variablestemplate)llmprompt)r)rr r)clsllmsr(r'rrs r from_llmsModelLaboratory.from_llms>s\ >#XJTF>BCds(s2dC%)*TcST*6''D*s AA c"[SUS35 [UR5Hgup#URbURUO [ U5n[ USS9 UR U5n[ XPR[ U5SS9 Mi g)aCompare model outputs on an input text. If a prompt was provided with starting the laboratory, then this text will be fed into the prompt. If no prompt was provided, then the input text is the entire prompt. Args: text: input text to run all models on. zInput:  N)endz )colorr/)print enumeraterrrr runr)rtextrrnameoutputs r compareModelLaboratory.compareTsw &tfB/0!$++.HA$(JJ$:4::a=E D t &YYt_F v%6%6s1v%>F K /)rrr)N)rzSequence[Chain]rzOptional[list[str]])r*z list[BaseLLM]r(zOptional[PromptTemplate]returnr )r4rr:None) __name__ __module__ __qualname____firstlineno____doc__r! classmethodr+r7__static_attributes__r9r r r sFW)V,0(()(  ((*Lr9r N)r@ __future__rcollections.abcrtypingr#langchain_core.language_models.llmsrlangchain_core.prompts.promptrlangchain_core.utils.inputrr langchain.chains.baser langchain.chains.llmr r rCr9r rLs.'"$78D')SLSLr9