Waflon commited on
Commit
3532415
1 Parent(s): 7200106

Update modelo.py

Browse files
Files changed (1) hide show
  1. modelo.py +7 -5
modelo.py CHANGED
@@ -8,10 +8,12 @@ from langchain_openai import ChatOpenAI
8
  from langchain_community.vectorstores import FAISS
9
  from langchain_community.document_loaders import HuggingFaceDatasetLoader
10
  from langchain_community.embeddings import HuggingFaceEmbeddings
 
 
 
11
 
12
-
13
- def get_chain(openai_key):
14
- os.environ["OPENAI_API_KEY"] = openai_key # agregada en la config de hugginface
15
  #Embeddings que transforman a vectores densos multidimensionales las preguntas del SII
16
  embeddings = HuggingFaceEmbeddings(
17
  model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2", # Ruta a modelo Pre entrenado
@@ -37,8 +39,8 @@ def get_chain(openai_key):
37
 
38
  prompt_template = """Usa los siguientes fragmentos de contextos para responder una pregunta al final. Por favor sigue las siguientes reglas:
39
  1. Si la pregunta requiere vinculos, por favor retornar solamente las vinculos de los vinculos sin respuesta
40
- 2. Si no sabes la respuesta, no inventes una respuesta. Solamente di **No pude encontrar la respuesta definitiva, pero tal vez quieras ver los siguientes vinculos** y agregalos a la lista de vinculos.
41
- 3. Si encuentras la respuesta, escribe una respuesta concisa y agrega la lista de vinculos que sean usadas **directamente** para derivar la respuesta.
42
 
43
  {contexto}
44
 
 
8
  from langchain_community.vectorstores import FAISS
9
  from langchain_community.document_loaders import HuggingFaceDatasetLoader
10
  from langchain_community.embeddings import HuggingFaceEmbeddings
11
+ from dotenv import load_dotenv
12
+ load_dotenv()
13
+ #os.environ["OPENAI_API_KEY"] = st.secrets("OPENAI_API_KEY")
14
 
15
+ def get_chain():
16
+ # agregada en la config de hugginface
 
17
  #Embeddings que transforman a vectores densos multidimensionales las preguntas del SII
18
  embeddings = HuggingFaceEmbeddings(
19
  model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2", # Ruta a modelo Pre entrenado
 
39
 
40
  prompt_template = """Usa los siguientes fragmentos de contextos para responder una pregunta al final. Por favor sigue las siguientes reglas:
41
  1. Si la pregunta requiere vinculos, por favor retornar solamente las vinculos de los vinculos sin respuesta
42
+ 2. Si no sabes la respuesta, no inventes una respuesta. Solamente di **No pude encontrar la respuesta definitiva, pero, tal vez quieras ver los siguientes vínculos** y agregalos a la lista de vínculos.
43
+ 3. Si encuentras la respuesta, escribe una respuesta concisa y agrega la lista de víinculos relevantes para derivar la respuesta.
44
 
45
  {contexto}
46