camparchimedes commited on
Commit
8318d4c
ยท
verified ยท
1 Parent(s): 2535a3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -8,6 +8,8 @@ import time
8
  import json
9
  import torch
10
 
 
 
11
  from api_docs_mck import api_docs_str
12
  #from daysoff import daysoff_str ## make daysoff.py, put json info in dict.
13
  #from personvernpolicy import personvernpolicy_str ## make personvernpolicy.py, put json info in dict.
@@ -21,6 +23,7 @@ from langchain_core.prompts import PromptTemplate
21
  from langchain_huggingface import HuggingFaceEndpoint
22
  from langchain.memory.buffer import ConversationBufferMemory
23
 
 
24
 
25
  HUGGINGFACEHUB_API_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
26
  BOOKING_ID = re.compile(r'\b[A-Z]{6}\d{6}\b')
@@ -121,7 +124,7 @@ def setup_multiple_chains():
121
 
122
  @cl.on_message
123
  async def handle_message(message: cl.Message):
124
- user_message = message.content #.lower()
125
  llm_chain = cl.user_session.get("llm_chain")
126
  api_chain = cl.user_session.get("api_chain")
127
 
@@ -135,6 +138,7 @@ async def handle_message(message: cl.Message):
135
  #if booked:
136
 
137
  if re.search(r'\b[A-Z]{6}\d{6}\b', user_message): # ex., "Hei, har du booking info for EQJLCQ362149?"
 
138
  response = await api_chain.acall(user_message,
139
  callbacks=[cl.AsyncLangchainCallbackHandler()])
140
 
@@ -142,6 +146,8 @@ async def handle_message(message: cl.Message):
142
  response = await llm_chain.acall(user_message,
143
  callbacks=[cl.AsyncLangchainCallbackHandler()])
144
 
 
 
145
  response_key = "output" if "output" in response else "text"
146
  await cl.Message(response.get(response_key, "")).send()
147
  return message.content
 
8
  import json
9
  import torch
10
 
11
+ import logging
12
+
13
  from api_docs_mck import api_docs_str
14
  #from daysoff import daysoff_str ## make daysoff.py, put json info in dict.
15
  #from personvernpolicy import personvernpolicy_str ## make personvernpolicy.py, put json info in dict.
 
23
  from langchain_huggingface import HuggingFaceEndpoint
24
  from langchain.memory.buffer import ConversationBufferMemory
25
 
26
+ logging.basicConfig(level=logging.DEBUG)
27
 
28
  HUGGINGFACEHUB_API_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
29
  BOOKING_ID = re.compile(r'\b[A-Z]{6}\d{6}\b')
 
124
 
125
  @cl.on_message
126
  async def handle_message(message: cl.Message):
127
+ user_message = message.content.lower()
128
  llm_chain = cl.user_session.get("llm_chain")
129
  api_chain = cl.user_session.get("api_chain")
130
 
 
138
  #if booked:
139
 
140
  if re.search(r'\b[A-Z]{6}\d{6}\b', user_message): # ex., "Hei, har du booking info for EQJLCQ362149?"
141
+ logging.debug(f"User message: {user_message}")
142
  response = await api_chain.acall(user_message,
143
  callbacks=[cl.AsyncLangchainCallbackHandler()])
144
 
 
146
  response = await llm_chain.acall(user_message,
147
  callbacks=[cl.AsyncLangchainCallbackHandler()])
148
 
149
+ logging.debug(f"API response: {response}")
150
+
151
  response_key = "output" if "output" in response else "text"
152
  await cl.Message(response.get(response_key, "")).send()
153
  return message.content