camparchimedes commited on
Commit
1fbce43
·
verified ·
1 Parent(s): a5625e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -6
app.py CHANGED
@@ -72,7 +72,13 @@ daysoff_assistant_prompt = PromptTemplate(
72
 
73
  async def async_post_request(url, headers, data):
74
  return await asyncio.to_thread(requests.post, url, headers=headers, json=data)
75
-
 
 
 
 
 
 
76
  @cl.set_starters
77
  async def set_starters():
78
  return [
@@ -140,8 +146,7 @@ def setup_multiple_chains():
140
  async def handle_message(message: cl.Message):
141
  user_message = message.content
142
  llm_chain = cl.user_session.get("llm_chain")
143
- session_id = cl.user_session.get("id")
144
-
145
  booking_pattern = r'\b[A-Z]{6}\d{6}\b'
146
  match = re.search(booking_pattern, user_message)
147
 
@@ -154,7 +159,7 @@ async def handle_message(message: cl.Message):
154
  payload = {"booking_id": bestillingskode}
155
 
156
  try:
157
- response = await async_post_request(API_URL, headers, payload)
158
  response.raise_for_status()
159
  booking_data = response.json()
160
 
@@ -175,7 +180,7 @@ async def handle_message(message: cl.Message):
175
  )
176
 
177
  combined_message = f"### Her er informasjon for {bestillingskode}:\n\n{table}"
178
- await cl.Message(content=combined_message).send()
179
 
180
  except Exception as e:
181
  await cl.Message(content=f"Error processing booking data: {str(e)}").send()
@@ -194,7 +199,7 @@ async def handle_message(message: cl.Message):
194
  "chat_history": ""
195
  }, callbacks=[cl.AsyncLangchainCallbackHandler()])
196
 
197
- await cl.Message(content=response["text"]).send()
198
 
199
  except Exception as e:
200
  await cl.Message(content=f"Error: {str(e)}").send()
 
72
 
73
  async def async_post_request(url, headers, data):
74
  return await asyncio.to_thread(requests.post, url, headers=headers, json=data)
75
+
76
+
77
+ async def keepalive():
78
+ while True:
79
+ await asyncio.sleep(10)
80
+ await cl.Message(content="").send()
81
+
82
  @cl.set_starters
83
  async def set_starters():
84
  return [
 
146
  async def handle_message(message: cl.Message):
147
  user_message = message.content
148
  llm_chain = cl.user_session.get("llm_chain")
149
+
 
150
  booking_pattern = r'\b[A-Z]{6}\d{6}\b'
151
  match = re.search(booking_pattern, user_message)
152
 
 
159
  payload = {"booking_id": bestillingskode}
160
 
161
  try:
162
+ response = await async_post_request(API_URL, headers, payload)
163
  response.raise_for_status()
164
  booking_data = response.json()
165
 
 
180
  )
181
 
182
  combined_message = f"### Her er informasjon for {bestillingskode}:\n\n{table}"
183
+ await cl.make_async.Message(content=combined_message).send()
184
 
185
  except Exception as e:
186
  await cl.Message(content=f"Error processing booking data: {str(e)}").send()
 
199
  "chat_history": ""
200
  }, callbacks=[cl.AsyncLangchainCallbackHandler()])
201
 
202
+ await cl.make_async.Message(content=response["text"]).send()
203
 
204
  except Exception as e:
205
  await cl.Message(content=f"Error: {str(e)}").send()