a dg+@sDdZddlmZddlZddlmZmZmZmZm Z m Z ddl Z ddl m Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZmZmZdd l m!Z"dd l#m!Z$dd l%m!Z&dd l'm!Z(GdddeZ)Gddde)Z*Gddde)Z+Gddde)Z,Gddde)Z-Gddde)Z.dS)z CPAL Chain and its subchains ) annotationsN)AnyClassVarDictListOptionalType)BaseLanguageModel)Chain)LLMChain)PydanticOutputParser)CallbackManagerForChainRun)PromptTemplate)Constant) CausalModelInterventionModelNarrativeModel QueryModel StoryModel)templatec@seZdZUded<ejjZded<ejjZ ded<dZ ded<dZ d ed <e d d d dZ edd ddZedd ddZe ddddddZdddddddZdS)_BaseStoryElementChainr chainstr input_key output_keyNz,ClassVar[Optional[Type[pydantic.BaseModel]]]pydantic_modelzClassVar[Optional[str]]rr )returncCs&|jdurtd|jt|jdS)z(Parse LLM output into a pydantic object.Nz#pydantic_model not implemented for )Zpydantic_object)rNotImplementedError__name__r )clsr n/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_experimental/cpal/base.pyparser1s   z_BaseStoryElementChain.parserz List[str]cCs|jgS)z7Return the input keys. :meta private: )r)selfr r r! input_keys:sz!_BaseStoryElementChain.input_keyscCs |jg}|S)z8Return the output keys. :meta private: )r)r#Z _output_keysr r r! output_keysBsz"_BaseStoryElementChain.output_keysr rllmkwargsrc Ks@|fdt|ttjjg|d|jd|iddi|S)NrrZformat_instructions)input_variablesrZpartial_variablesr'prompt) r rrnarrative_inputvaluegetrr"Zget_format_instructionsrr'r(r r r!from_univariate_promptKs   z-_BaseStoryElementChain.from_univariate_promptDict[str, Any]$Optional[CallbackManagerForChainRun])inputs run_managerrcCs6|j||j}|j|}tjj|tj jdiS)N) rrunr __class__r"parser chain_datar- chain_answer)r#r3r4 completionZ pydantic_datar r r!_call_s z_BaseStoryElementChain._call)N)r __module__ __qualname____annotations__rr,r-rr9rrr classmethodr"propertyr$r%r0r;r r r r!r(s   rc@s*eZdZUdZeZded<eZded<dS)NarrativeChainzhDecompose the narrative into its story elements. - causal model - query - intervention "ClassVar[Type[pydantic.BaseModel]]r ClassVar[str]rN) rr<r=__doc__rrr>narrative_templaterr r r r!rAls  rAc@s*eZdZUdZeZded<eZded<dS) CausalChainz:Translate the causal narrative into a stack of operations.rBrrCrN) rr<r=rDrrr>causal_templaterr r r r!rFxs  rFc@s*eZdZUdZeZded<eZded<dS)InterventionChainz5Set the hypothetical conditions for the causal model.rBrrCrN) rr<r=rDrrr>intervention_templaterr r r r!rHs  rHc@s*eZdZUdZeZded<eZded<dS) QueryChainarQuery the outcome table using SQL. *Security note*: This class implements an AI technique that generates SQL code. If those SQL commands are executed, it's critical to ensure they use credentials that are narrowly-scoped to only include the permissions this chain needs. Failure to do so may result in data corruption or loss, since this chain may attempt commands like `DROP TABLE` or `INSERT` if appropriately prompted. The best way to guard against such negative outcomes is to (as appropriate) limit the permissions granted to the credentials used with this chain. rBrrCrN) rr<r=rDrrr>query_templaterr r r r!rJs  rJc@seZdZUdZded<dZded<dZded<dZd ed <dZd ed <e j dd Z ded<e ddddddZ ddddddddZdddddZdS) CPALChainaCausal program-aided language (CPAL) chain implementation. *Security note*: The building blocks of this class include the implementation of an AI technique that generates SQL code. If those SQL commands are executed, it's critical to ensure they use credentials that are narrowly-scoped to only include the permissions this chain needs. Failure to do so may result in data corruption or loss, since this chain may attempt commands like `DROP TABLE` or `INSERT` if appropriately prompted. The best way to guard against such negative outcomes is to (as appropriate) limit the permissions granted to the credentials used with this chain. r r'NzOptional[NarrativeChain]narrative_chainzOptional[CausalChain] causal_chainzOptional[InterventionChain]intervention_chainzOptional[QueryChain] query_chain)defaultr_storyrr&c KsP|f|t|tddgdddtj|dtj|dtj|dtj|dd|S)ainstantiation depends on component chains *Security note*: The building blocks of this class include the implementation of an AI technique that generates SQL code. If those SQL commands are executed, it's critical to ensure they use credentials that are narrowly-scoped to only include the permissions this chain needs. Failure to do so may result in data corruption or loss, since this chain may attempt commands like `DROP TABLE` or `INSERT` if appropriately prompted. The best way to guard against such negative outcomes is to (as appropriate) limit the permissions granted to the credentials used with this chain. question query_resultzFSummarize this answer '{query_result}' to this question '{question}'? )r)rr*r')r'rrMrNrOrP)r rrAr0rFrHrJr/r r r!r0s     z CPALChain.from_univariate_promptr1r2)r3r4r(rc Ks|jdurtj|jd|_|jdur4tj|jd|_|jdurNtj|jd|_|jdurht j|jd|_||t j j t j j }t||jt j j ||jt j j ||jt j j d}||_dddddd}|pt}|j|d|jdd |jd dd ddd d }|j|d|jdd |jd |jjjr`td|jd|jnt|jjjdd} t j j ||j | i|} | S)NrU)Zcausal_operationsZ interventionqueryr)titledrcSs |d|S)N r rWrXr r r!pretty_print_strsz)CPALChain._call..pretty_print_strzstory outcome datagreenz )colorendverbosedictcSs|dtj|ddS)NrY)indent)jsondumpsrZr r r!pretty_print_dictsz*CPALChain._call..pretty_print_dictz query databluez9unanswerable, query and outcome are incoherent outcome: z query: r)!rMrAr0r'rNrFrOrHrPrJrr,r-r8rZ story_plotZstory_hypotheticalZstory_outcome_questionrRr Zget_noop_managerZon_textZ_outcome_tableZ to_stringr_rVr`Z _result_tableempty ValueErrorfloatvaluesr) r#r3r4r(Z narrativeZstoryr[Z _run_managerrerToutputr r r!r;sn            zCPALChain._callNone)r(rcKs|jjjfi|dS)z CPAL chain can draw its resulting DAG. Usage in a jupyter notebook: >>> from IPython.display import SVG >>> cpal_chain.draw(path="graph.svg") >>> SVG('graph.svg') N)rRZ_networkx_wrapperZ draw_graphviz)r#r(r r r!draw%s zCPALChain.draw)N)rr<r=rDr>rMrNrOrPpydanticZ PrivateAttrrRr?r0r;rnr r r r!rLs      &WrL)/rD __future__rrctypingrrrrrrroZlangchain.base_languager Zlangchain.chains.baser Zlangchain.chains.llmr Zlangchain.output_parsersr Z langchain_core.callbacks.managerr Zlangchain_core.prompts.promptrZ%langchain_experimental.cpal.constantsrZ"langchain_experimental.cpal.modelsrrrrrZ7langchain_experimental.cpal.templates.univariate.causalrrGZ=langchain_experimental.cpal.templates.univariate.interventionrIZ:langchain_experimental.cpal.templates.univariate.narrativerEZ6langchain_experimental.cpal.templates.univariate.queryrKrrArFrHrJrLr r r r!s,             D