Skip to content

Commit c469195

Browse files
committed
ran pre commit
1 parent 59392dd commit c469195

6 files changed

+17
-10
lines changed

spanish/rag_csv.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,8 @@
5757
# Ahora podemos usar los resultados para generar una respuesta
5858
SYSTEM_MESSAGE = """
5959
Eres un asistente útil que responde preguntas sobre automóviles basándote en un conjunto de datos de autos híbridos.
60-
Debes utilizar el conjunto de datos para responder las preguntas, no debes proporcionar ninguna información que no esté en las fuentes proporcionadas.
60+
Debes utilizar el conjunto de datos para responder las preguntas, no debes
61+
proporcionar ninguna información que no esté en las fuentes proporcionadas.
6162
"""
6263

6364
response = client.chat.completions.create(

spanish/rag_documents_flow.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
else:
3333
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
3434
MODEL_NAME = os.environ["OPENAI_MODEL"]
35-
35+
3636
# Indexar los datos del JSON - cada objeto tiene id, texto y embedding
3737
with open("rag_ingested_chunks.json") as file:
3838
documents = json.load(file)
@@ -67,4 +67,4 @@
6767
)
6868

6969
print(f"\nRespuesta de {MODEL_NAME} en {API_HOST}: \n")
70-
print(response.choices[0].message.content)
70+
print(response.choices[0].message.content)

spanish/rag_documents_hybrid.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@
4141
documents_by_id = {doc["id"]: doc for doc in documents}
4242
index = lunr(ref="id", fields=["text"], documents=documents)
4343

44+
4445
def full_text_search(query, limit):
4546
"""
4647
Realizar una búsqueda de texto completo en los documentos indexados.
@@ -139,4 +140,4 @@ def hybrid_search(query, limit):
139140
)
140141

141142
print(f"\nRespuesta de {MODEL_NAME} en {API_HOST}: \n")
142-
print(response.choices[0].message.content)
143+
print(response.choices[0].message.content)

spanish/rag_ingested_chunks.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -97145,4 +97145,4 @@
9714597145
0.014291062951087952
9714697146
]
9714797147
}
97148-
]
97148+
]

spanish/rag_multiturn.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -53,9 +53,10 @@ def search(query):
5353
return matches_table
5454

5555

56-
SYSTEM_MESSAGE = """
56+
SYSTEM_MESSAGE = """
5757
Eres un asistente útil que responde preguntas sobre automóviles basándote en un conjunto de datos de autos híbridos.
58-
Debes utilizar el conjunto de datos para responder las preguntas, no debes proporcionar ninguna información que no esté en las fuentes proporcionadas.
58+
Debes utilizar el conjunto de datos para responder las preguntas, no debes proporcionar ninguna información que no
59+
esté en las fuentes proporcionadas.
5960
"""
6061
messages = [{"role": "system", "content": SYSTEM_MESSAGE}]
6162

spanish/rag_queryrewrite.py

+7-3
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
else:
3333
client = openai.OpenAI(api_key=os.environ["OPENAI_KEY"])
3434
MODEL_NAME = os.environ["OPENAI_MODEL"]
35-
35+
3636
# Indexar los datos del CSV
3737
with open("hybridos.csv") as file:
3838
reader = csv.reader(file)
@@ -62,7 +62,8 @@ def search(query):
6262

6363
SYSTEM_MESSAGE = """
6464
Eres un asistente útil que responde preguntas sobre automóviles basándose en un conjunto de datos de coches híbridos.
65-
Debes utilizar el conjunto de datos para responder las preguntas, no debes proporcionar información que no esté en las fuentes proporcionadas.
65+
Debes utilizar el conjunto de datos para responder las preguntas, no debes proporcionar información que no
66+
esté en las fuentes proporcionadas.
6667
"""
6768
messages = [{"role": "system", "content": SYSTEM_MESSAGE}]
6869

@@ -75,7 +76,10 @@ def search(query):
7576
temperature=0.3,
7677
messages=[
7778
{"role": "system", "content": QUERY_REWRITE_SYSTEM_MESSAGE},
78-
{"role": "user", "content": f"Nueva pregunta del usuario:{question}\n\nHistorial de conversación:{messages}"},
79+
{
80+
"role": "user",
81+
"content": f"Nueva pregunta del usuario:{question}\n\nHistorial de conversación:{messages}",
82+
},
7983
],
8084
)
8185
search_query = response.choices[0].message.content

0 commit comments

Comments
 (0)