Update app.py
Browse files
app.py
CHANGED
@@ -154,26 +154,25 @@ async def handle_message(message: cl.Message):
|
|
154 |
|
155 |
|
156 |
if match:
|
157 |
-
|
158 |
-
|
159 |
-
|
160 |
-
|
161 |
-
|
162 |
-
|
163 |
-
"
|
164 |
-
"booking_id": bestillingskode
|
165 |
-
}
|
166 |
}
|
|
|
167 |
|
168 |
-
|
169 |
-
|
170 |
-
|
171 |
-
|
172 |
|
173 |
if response:
|
174 |
try:
|
175 |
response_data = json.loads(response)
|
176 |
-
|
177 |
booking_info = {
|
178 |
"booking_id": response_data.get("booking_id", "N/A"),
|
179 |
"full_name": response_data.get("full_name", "N/A"),
|
@@ -187,14 +186,14 @@ async def handle_message(message: cl.Message):
|
|
187 |
}
|
188 |
|
189 |
await cl.Message(content=booking_info).send()
|
190 |
-
|
191 |
except json.JSONDecodeError:
|
192 |
-
|
193 |
-
|
194 |
-
|
195 |
else:
|
196 |
response = await llm_chain.ainvoke(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
|
197 |
|
|
|
198 |
response_key = "output" if "output" in response else "text"
|
199 |
await cl.Message(response.get(response_key, "")).send()
|
200 |
return message.content
|
|
|
154 |
|
155 |
|
156 |
if match:
|
157 |
+
bestillingskode = match.group()
|
158 |
+
endpoint_url = "https://aivisions.no/data/daysoff/api/v1/booking/"
|
159 |
+
|
160 |
+
post_data = {
|
161 |
+
"url": endpoint_url,
|
162 |
+
"body": {
|
163 |
+
"booking_id": bestillingskode
|
|
|
|
|
164 |
}
|
165 |
+
}
|
166 |
|
167 |
+
response = await post_tool.ainvoke(
|
168 |
+
post_data,
|
169 |
+
[cl.AsyncLangchainCallbackHandler()]
|
170 |
+
)
|
171 |
|
172 |
if response:
|
173 |
try:
|
174 |
response_data = json.loads(response)
|
175 |
+
|
176 |
booking_info = {
|
177 |
"booking_id": response_data.get("booking_id", "N/A"),
|
178 |
"full_name": response_data.get("full_name", "N/A"),
|
|
|
186 |
}
|
187 |
|
188 |
await cl.Message(content=booking_info).send()
|
189 |
+
|
190 |
except json.JSONDecodeError:
|
191 |
+
await cl.Message(content="Error: Invalid response format.").send()
|
192 |
+
|
|
|
193 |
else:
|
194 |
response = await llm_chain.ainvoke(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
|
195 |
|
196 |
+
|
197 |
response_key = "output" if "output" in response else "text"
|
198 |
await cl.Message(response.get(response_key, "")).send()
|
199 |
return message.content
|