Update app.py

app.py

@@ -116,38 +116,35 @@ async def handle_message(message: cl.Message):
     api_chain = cl.user_session.get("api_chain") 
 
     booking_pattern = r'\b[A-Z]{6}\d{6}\b'
-    base_url = "https://aivisions.no/data/daysoff/api/v1/booking/"
-    headers = {
-        "Authorization": f"Bearer {auth_token}",
-        "Content-Type": "application/json"
-    }
-
-    #from langchain.requests import Requests
-
-    #requests = Requests(
-        #headers=headers,
-        #verify=True  # @SSL verification
-    #)
-
+    endpoint_url = "https://aivisions.no/data/daysoff/api/v1/booking/"
+    
     if re.search(booking_pattern, user_message):
-        bestillingskode = re.search(booking_pattern, user_message).group(0)
-        question = f"Retrieve information for booking ID {bestillingskode}" 
-        # ~ainvoke
-        response = await api_chain.acall(                           
-            {
-            "question": question,
-            "input": {
-                "url": base_url,
-                "method": "POST",
-                "headers": headers,
-                "body": {"booking_id": bestillingskode}
-            }
-        },
+        booking_id = re.search(booking_pattern, user_message).group(0)
+        headers = {"Authorization": f"Bearer {auth_token}", "Content-Type": "application/json"}
+      
+        # --------------------------------ALT 3--------------------------------
+        response = await api_chain.acall({
+            "booking_id": booking_id,
+            "headers": headers #{"Authorization": f"Bearer {auth_token}"},
+        }),
         callbacks=[cl.AsyncLangchainCallbackHandler()])
+        # ----------------------------------------------------------------------
+
         
     else:
         response = await llm_chain.acall(user_message, callbacks=[cl.AsyncLangchainCallbackHandler()])
 
     response_key = "output" if "output" in response else "text"
     await cl.Message(response.get(response_key, "")).send()
-    return message.content 
+    return message.content 
+
+
+
+    #from langchain.requests import Requests
+
+    #requests = Requests(
+        #headers=headers,
+        #verify=True  # @SSL verification
+    #)
+    
+    #response = requests.get(f"{endpoint_url}?booking_id={bestillingskode}", headers=headers)