camparchimedes commited on
Commit
0c0882f
·
verified ·
1 Parent(s): 16bafb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -187,8 +187,6 @@ async def handle_message(message: cl.Message):
187
 
188
 
189
  """
190
-
191
-
192
  @cl.on_message
193
  async def handle_message(message: cl.Message):
194
  user_message = message.content
@@ -197,14 +195,15 @@ async def handle_message(message: cl.Message):
197
 
198
  booking_pattern = r'\b[A-Z]{6}\d{6}\b'
199
  endpoint_url = "https://aivisions.no/data/daysoff/api/v1/booking/"
200
- auth_token = auth_token
 
201
 
202
  async def process_message():
203
  if re.search(booking_pattern, user_message):
204
  bestillingskode = re.search(booking_pattern, user_message).group(0)
205
  question = f"Retrieve information for booking ID {bestillingskode}"
206
 
207
- response = await api_chain.acall(
208
  {
209
  "url": endpoint_url,
210
  "method": "POST",
@@ -220,7 +219,7 @@ async def handle_message(message: cl.Message):
220
  callbacks=[cl.AsyncLangchainCallbackHandler()],
221
  )
222
  else:
223
- response = await llm_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
224
 
225
  response_key = "output" if "output" in response else "text"
226
  await cl.Message(response.get(response_key, "")).send()
 
187
 
188
 
189
  """
 
 
190
  @cl.on_message
191
  async def handle_message(message: cl.Message):
192
  user_message = message.content
 
195
 
196
  booking_pattern = r'\b[A-Z]{6}\d{6}\b'
197
  endpoint_url = "https://aivisions.no/data/daysoff/api/v1/booking/"
198
+ #auth_token = f"Bearer {auth_token}
199
+
200
 
201
  async def process_message():
202
  if re.search(booking_pattern, user_message):
203
  bestillingskode = re.search(booking_pattern, user_message).group(0)
204
  question = f"Retrieve information for booking ID {bestillingskode}"
205
 
206
+ response = await api_chain.ainvoke(
207
  {
208
  "url": endpoint_url,
209
  "method": "POST",
 
219
  callbacks=[cl.AsyncLangchainCallbackHandler()],
220
  )
221
  else:
222
+ response = await llm_chain.ainvoke(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
223
 
224
  response_key = "output" if "output" in response else "text"
225
  await cl.Message(response.get(response_key, "")).send()