Spaces:
Running
Running
File size: 690 Bytes
d32424b 12d3e1a d32424b 12d3e1a d32424b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from setup.easy_imports import RunnablePassthrough, create_retrieval_chain
class Chain:
def create_prompt_model_chain(self, prompt, model):
return prompt | model
def create_prompt_model_retriever_chain(self, prompt, model, retriever):
chain = prompt | model
return create_retrieval_chain(retriever, chain)
def invoke_retrieval_chain(self, chain, busca_no_vetor):
resposta = chain.invoke({"input": busca_no_vetor})
class Resposta:
def __init__(self, resposta):
self.final_answer = resposta["answer"].content
self.complete_obj = resposta
return Resposta(resposta)
chain = Chain()
|