Edmundoogaz commited on
Commit
0d78f52
·
1 Parent(s): 43acfaf

add rag model

Browse files
Files changed (1) hide show
  1. app.py +20 -1
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 = "response"
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)}