o yft@sddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddlm Z dd lmZdd lmZdd lmZmZdd lmZdd lZddlmZddlmZdd lZddlmZddlmZdd lZddlm Z dd l!Z!e eZ"eZ#ddgZ$e"j%edgddgdgde!j&dZ'GdddeZ(e")dde(fddZ*e")dde(fd d!Z+e")d"de(fd#d!Z+d S)$)FastAPI)CORSMiddleware)OpenAIEmbeddings)LLMChain) ChatOpenAI)ConversationBufferMemory)PromptTemplate)create_qa_with_sources_chain)StuffDocumentsChain)ConversationalRetrievalChain)r HTTPException)MilvusN) BaseModel) Embeddings) Procesador)vectores) load_dotenvzhttp://192.168.100.18:3000zhttp://otro-dominio.com*T) allow_originsallow_credentials allow_methods allow_headers OPEN_AI_KEYc@seZdZUeed<dS)MessageRequestmessageN)__name__ __module__ __qualname__str__annotations__r r @/home/marcatoma/Documents/GitHub/back-agrota-serv-ia/app/main.pyr*s rz /chat-agrotarequestc stddtd}tdtd}t|j}d}t|}t||d}tdd d }t |}td d d gd}t |d|d} t ||j ddid|| d} t dt | t dt d|t d| t |t dt |t dt | | d|ji} t| } t | | dS)N?gpt-4o temperaturemodelopenai_api_keyztext-embedding-3-large)r'r(uToma la última pregunta del historial de chat y únela a la pregunta de entrada de seguimiento y crea una pregunta independiente. Historial de Chat: {chat_history} Entrada de seguimiento: {question} Pregunta independiente:llmprompt chat_historyT) memory_keyreturn_messagesuu 1.Actua como un asistente virtual de la empresa Agrota que se dedica a la venta de productos agricola. 2.No puedes buscar en internet 3.No digas que eres creado por OpenAI 4.Solamente responde basandote en el contexto entregado 5.Recuerda que eres un asistente de la empresa Agrota dedicada a la venta de materiales agrícolas 6.Responde solamente en el idioma Español 7.Si la información del contexto entregado no es suficiente responde con: No encuentro esa información en mi base de conocimiento. Contexto Entregado: {page_content} Source: {document} page_contentdocument)templateinput_variablescontext) llm_chaindocument_variable_namedocument_promptk ) search_kwargs)question_generator retrievermemorycombine_docs_chainz******* retrieval_qa ********z***************rz pg vector questionanswer)rrrrrr from_templaterrr r r as_retrieverprintrunjsonloads) r"r* embeddingsvectordbcondense_question_promptcondense_question_chainr<qa_chain doc_promptfinal_qa_chain retrieval_qaresponse responseDictr r r! read_root-s\       rPz/consulta_endpointc szsT                H%