camparchimedes commited on
Commit
2f0ac37
·
verified ·
1 Parent(s): a41ad93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -6
app.py CHANGED
@@ -76,8 +76,6 @@ api_response_prompt = PromptTemplate(
76
  template=api_response_template
77
  )
78
 
79
-
80
-
81
  @cl.on_chat_start
82
  def setup_multiple_chains():
83
  llm = OpenAI(
@@ -116,7 +114,7 @@ def setup_multiple_chains():
116
 
117
  requests_wrapper = TextRequestsWrapper(
118
  headers={
119
- "Authorization": auth_token,
120
  "Content-Type": "application/json"
121
  }
122
  )
@@ -146,16 +144,33 @@ async def handle_message(message: cl.Message):
146
  bestillingskode = match.group()
147
  post_data = {
148
  "url": endpoint_url,
149
- "body": {
150
- "booking_id": bestillingskode
151
- }
152
  }
 
153
 
154
  response = await post_tool.ainvoke(
155
  json.dumps(post_data),
156
  config={"callbacks": [cl.AsyncLangchainCallbackHandler()]}
157
  )
158
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
159
 
160
  else:
161
  response = await llm_chain.ainvoke(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
 
76
  template=api_response_template
77
  )
78
 
 
 
79
  @cl.on_chat_start
80
  def setup_multiple_chains():
81
  llm = OpenAI(
 
114
 
115
  requests_wrapper = TextRequestsWrapper(
116
  headers={
117
+ "Authorization": f"Bearer <auth_token>",
118
  "Content-Type": "application/json"
119
  }
120
  )
 
144
  bestillingskode = match.group()
145
  post_data = {
146
  "url": endpoint_url,
147
+ "data": {
148
+ "booking_id": bestillingskode
 
149
  }
150
+ }
151
 
152
  response = await post_tool.ainvoke(
153
  json.dumps(post_data),
154
  config={"callbacks": [cl.AsyncLangchainCallbackHandler()]}
155
  )
156
 
157
+ if response:
158
+ booking_data = json.loads(response)
159
+ formatted_response = f"""
160
+ | Field | Value |
161
+ |-------|--------|
162
+ | Booking ID | {booking_data['booking_id']} |
163
+ | Name | {booking_data['full_name']} |
164
+ | Amount | {booking_data['amount']} kr |
165
+ | Check-in | {booking_data['checkin']} |
166
+ | Check-out | {booking_data['checkout']} |
167
+ | Address | {booking_data['address']} |
168
+ | User ID | {booking_data['user_id']} |
169
+ | Info | {booking_data['infotext']} |
170
+ | Included | {booking_data['included']} |
171
+ """
172
+ await cl.Message(formatted_response).send()
173
+
174
 
175
  else:
176
  response = await llm_chain.ainvoke(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])