Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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
|