Spaces:
Sleeping
Sleeping
Commit
·
0d78f52
1
Parent(s):
43acfaf
add rag model
Browse files
app.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import chainlit as cl
|
|
|
2 |
|
3 |
import os
|
4 |
from dotenv import load_dotenv
|
@@ -6,6 +7,24 @@ load_dotenv()
|
|
6 |
|
7 |
@cl.on_message
|
8 |
async def main(message: cl.Message):
|
|
|
9 |
await cl.Message(
|
10 |
-
content =
|
11 |
).send()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import chainlit as cl
|
2 |
+
import requests
|
3 |
|
4 |
import os
|
5 |
from dotenv import load_dotenv
|
|
|
7 |
|
8 |
@cl.on_message
|
9 |
async def main(message: cl.Message):
|
10 |
+
response = call_backend(message.content)
|
11 |
await cl.Message(
|
12 |
+
content = response
|
13 |
).send()
|
14 |
+
|
15 |
+
def call_backend(user_message):
|
16 |
+
try:
|
17 |
+
url = os.environ.get("RAG_MODEL")
|
18 |
+
headers = {"Content-Type": "application/json"}
|
19 |
+
data = {"query": user_message}
|
20 |
+
response = requests.post(url, json=data, headers=headers, timeout=(5, 50))
|
21 |
+
response.raise_for_status()
|
22 |
+
backend_response = response.json()
|
23 |
+
print(backend_response)
|
24 |
+
return backend_response['respuesta']
|
25 |
+
except requests.exceptions.RequestException as e:
|
26 |
+
print(f"Error al llamar al backend: {e}")
|
27 |
+
return {"error": str(e)}
|
28 |
+
except requests.exceptions.Timeout as e:
|
29 |
+
print(f"Error de timeout: {e}")
|
30 |
+
return {"error": str(e)}
|