Se corrigen algunos errores en los modelos de ollama

This commit is contained in:
mongar
2024-05-04 21:34:43 -05:00
parent 24685ebb99
commit 6d3cafc579
9 changed files with 38 additions and 26 deletions

View File

@@ -68,7 +68,7 @@ class LangChainTools:
"""
# db_name = _file_name.replace(".pdf", "").replace(" ", "_").lower()
st.success(_file_name)
# st.success(_file_name)
if "db_name" not in st.session_state.keys():
st.session_state.db_name = (
@@ -265,7 +265,7 @@ class LangChainTools:
# Pregunta: {question}
# Respuesta en español:"""
conversation.combine_docs_chain.llm_chain.prompt.template = template
conversation.question_generator.prompt.template = "Dado el siguiente diálogo y una pregunta de seguimiento, reformula la pregunta de seguimiento para que sea una pregunta independiente, en su idioma original.\n\nHistorial del chat:\n{chat_history}\nPregunta de seguimiento: {question}\nPregunta independiente:"
# conversation.combine_docs_chain.llm_chain.prompt.template = template
# conversation.question_generator.prompt.template = "Dado el siguiente diálogo y una pregunta de seguimiento, reformula la pregunta de seguimiento para que sea una pregunta independiente, en su idioma original.\n\nHistorial del chat:\n{chat_history}\nPregunta de seguimiento: {question}\nPregunta independiente:"
return conversation

View File

@@ -39,7 +39,7 @@ class PdfLangChain:
loader = PyPDFLoader(_self.file_path)
_self.docs = loader.load()
st.success(f"Se carga el pdf : {_self.file_path}")
# st.success(f"Se carga el pdf : {_self.file_path}")
return _self.docs
def split_docs(self, data: list) -> list:
@@ -60,6 +60,6 @@ class PdfLangChain:
)
self.docs_split = splitter.split_documents(data)
st.success(f"{self.file_path[3][:200]}")
# st.success(f"{self.file_path[3][:200]}")
return self.docs_split