§
    è˜ºhC  ã                  ó¶   — d Z ddlmZ ddlmZmZmZmZ ddlm	Z	 ddl
mZ ddlmZ ddlmZ ddlmZ dd	lmZ dd
lmZmZmZmZ ddlmZ  G d„ de¦  «        ZdS )z;
Question answering over an RDF or OWL graph using SPARQL.
é    )Úannotations)ÚAnyÚDictÚListÚOptional)ÚRdfGraph)ÚBaseLanguageModel)ÚBasePromptTemplate)ÚField)ÚCallbackManagerForChainRun)ÚChain)ÚSPARQL_GENERATION_SELECT_PROMPTÚSPARQL_GENERATION_UPDATE_PROMPTÚSPARQL_INTENT_PROMPTÚSPARQL_QA_PROMPT©ÚLLMChainc                  óÚ   — e Zd ZU dZ ed¬¦  «        Zded<   ded<   ded<   ded	<   ded
<   dZded<   dZded<   e	d&d„¦   «         Z
e	d&d„¦   «         Zeeeeedœd'd„¦   «         Z	 d(d)d%„ZdS )*ÚGraphSparqlQAChainaë  Question-answering against an RDF or OWL graph by generating SPARQL statements.

    *Security note*: Make sure that the database connection uses credentials
        that are narrowly-scoped to only include necessary permissions.
        Failure to do so may result in data corruption or loss, since the calling
        code may attempt commands that would result in deletion, mutation
        of data if appropriately prompted or reading sensitive data if such
        data is present in the database.
        The best way to guard against such negative outcomes is to (as appropriate)
        limit the permissions granted to the credentials used with this tool.

        See https://python.langchain.com/docs/security for more information.
    T)Úexcluder   Úgraphr   Úsparql_generation_select_chainÚsparql_generation_update_chainÚsparql_intent_chainÚqa_chainÚqueryÚstrÚ	input_keyÚresultÚ
output_keyÚreturnú	List[str]c                ó   — | j         gS ©N)r   )Úselfs    úh/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain/chains/graph_qa/sparql.pyÚ
input_keyszGraphSparqlQAChain.input_keys/   s   € à”ÐÐó    c                ó   — | j         g}|S r$   )r    )r%   Ú_output_keyss     r&   Úoutput_keyszGraphSparqlQAChain.output_keys3   s   € àœÐ(ˆØÐr(   )Ú	qa_promptÚsparql_select_promptÚsparql_update_promptÚsparql_intent_promptÚllmr	   r,   r
   r-   r.   r/   Úkwargsr   c               ó¤   — t          ||¬¦  «        }t          ||¬¦  «        }t          ||¬¦  «        }	t          ||¬¦  «        }
 | d|||	|
dœ|¤ŽS )zInitialize from LLM.)r0   Úprompt)r   r   r   r   © r   )Úclsr0   r,   r-   r.   r/   r1   r   r   r   r   s              r&   Úfrom_llmzGraphSparqlQAChain.from_llm8   s†   € õ  ¨IÐ6Ñ6Ô6ˆÝ)1°cÐBVÐ)WÑ)WÔ)WÐ&Ý)1°cÐBVÐ)WÑ)WÔ)WÐ&Ý&¨3Ð7KÐLÑLÔLÐàˆsð 
ØØ+IØ+IØ 3ð	
ð 
ð
 ð
ð 
ð 	
r(   NÚinputsúDict[str, Any]Úrun_managerú$Optional[CallbackManagerForChainRun]úDict[str, str]c                ó  — |pt          j        ¦   «         }|                     ¦   «         }|| j                 }| j                             d|i|¬¦  «        }|                     ¦   «         }d|v rd|vr
| j        }d}n!d|v rd|vr
| j        }d}nt          d¦  «        ‚| 
                    dd| j        ¬¦  «         | 
                    |d	d| j        ¬
¦  «         |                     || j        j        dœ|¬¦  «        }	| 
                    dd| j        ¬¦  «         | 
                    |	d	d| j        ¬
¦  «         |dk    r�| j                             |	¦  «        }
| 
                    dd| j        ¬¦  «         | 
                    t          |
¦  «        d	d| j        ¬
¦  «         |                      ||
dœ|¬¦  «        }|| j        j                 }n2|dk    r| j                             |	¦  «         d}nt          d¦  «        ‚| j        |iS )zt
        Generate SPARQL query, use it to retrieve a response from the gdb and answer
        the question.
        r3   )Ú	callbacksÚSELECTÚUPDATEzuI am sorry, but this prompt seems to fit none of the currently supported SPARQL query types, i.e., SELECT and UPDATE.zIdentified intent:ú
)ÚendÚverboseÚgreen)ÚcolorrA   rB   )r3   ÚschemazGenerated SPARQL:zFull Context:)r3   Úcontextz-Successfully inserted triples into the graph.zUnsupported SPARQL query type.)r   Úget_noop_managerÚ	get_childr   r   ÚrunÚstripr   r   Ú
ValueErrorÚon_textrB   r   Ú
get_schemar   r   r   r    Úupdate)r%   r7   r9   Ú_run_managerr=   r3   Ú_intentÚintentÚsparql_generation_chainÚgenerated_sparqlrF   r   Úress                r&   Ú_callzGraphSparqlQAChain._callQ   sf  € ð #ÐSÕ&@Ô&QÑ&SÔ&SˆØ ×*Ò*Ñ,Ô,ˆ	Ø˜œÔ'ˆàÔ*×.Ò.°¸&Ð/AÈYÐ.ÑWÔWˆØ—’‘”ˆà�vÐÐ (°&Ð"8Ð"8Ø&*Ô&IÐ#ØˆFˆFØ˜ÐÐ H°FÐ$:Ð$:Ø&*Ô&IÐ#ØˆFˆFåðIñô ð ð
 	×ÒÐ1°tÀTÄ\ÐÑRÔRÐRØ×Ò˜V¨7¸ÀdÄlÐÑSÔSÐSà2×6Ò6Ø¨¬Ô)>Ð?Ð?È9ð 7ñ 
ô 
Ðð 	×ÒÐ0°dÀDÄLÐÑQÔQÐQØ×ÒØ G°¸t¼|ð 	ñ 	
ô 	
ð 	
ð �XÒÐØ”j×&Ò&Ð'7Ñ8Ô8ˆGà× Ò  °dÀDÄLÐ ÑQÔQÐQØ× Ò Ý�G‘” G°¸t¼|ð !ñ ô ð ð —]’]Ø!¨gÐ6Ð6Ø#ð #ñ ô ˆFð ˜œÔ1Ô2ˆCˆCØ�xÒÐØŒJ×ÒÐ.Ñ/Ô/Ð/ØAˆCˆCåÐ=Ñ>Ô>Ð>Ø” Ð%Ð%r(   )r!   r"   )r0   r	   r,   r
   r-   r
   r.   r
   r/   r
   r1   r   r!   r   r$   )r7   r8   r9   r:   r!   r;   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   Ú__annotations__r   r    Úpropertyr'   r+   Úclassmethodr   r   r   r   r6   rU   r4   r(   r&   r   r      s'  € € € € € € ðð ð �e DÐ)Ñ)Ô)€EÐ)Ð)Ð)Ñ)Ø,Ð,Ð,Ñ,Ø,Ð,Ð,Ñ,Ø!Ð!Ð!Ñ!ØÐÐÑØ€IÐÐÐÑØ€JÐÐÐÑàð ð  ð  ñ „Xð ð ðð ð ñ „Xðð ð
 )9Ø3RØ3RØ3Gð
ð 
ð 
ð 
ð 
ñ „[ð
ð6 =Að9&ð 9&ð 9&ð 9&ð 9&ð 9&ð 9&r(   r   N)rY   Ú
__future__r   Útypingr   r   r   r   Ú$langchain_community.graphs.rdf_graphr   Úlangchain_core.language_modelsr	   Úlangchain_core.prompts.baser
   Úlangchain_core.pydantic_v1r   Úlangchain.callbacks.managerr   Úlangchain.chains.baser   Ú!langchain.chains.graph_qa.promptsr   r   r   r   Úlangchain.chains.llmr   r   r4   r(   r&   ú<module>rg      sA  ððð ð #Ð "Ð "Ð "Ð "Ð "à ,Ð ,Ð ,Ð ,Ð ,Ð ,Ð ,Ð ,Ð ,Ð ,Ð ,Ð ,à 9Ð 9Ð 9Ð 9Ð 9Ð 9Ø <Ð <Ð <Ð <Ð <Ð <Ø :Ð :Ð :Ð :Ð :Ð :Ø ,Ð ,Ð ,Ð ,Ð ,Ð ,à BÐ BÐ BÐ BÐ BÐ BØ 'Ð 'Ð 'Ð 'Ð 'Ð 'ðð ð ð ð ð ð ð ð ð ð ð ð *Ð )Ð )Ð )Ð )Ð )ðr&ð r&ð r&ð r&ð r&˜ñ r&ô r&ð r&ð r&ð r&r(   