Update app.py

app.py

@@ -36,8 +36,8 @@ auth_token = os.getenv("DAYSOFF_API_TOKEN")
 class EnhancedRequestsPostTool(RequestsPostTool):
     def __init__(self, requests_wrapper, llm, api_docs, api_url_prompt, api_response_prompt):
         super().__init__(requests_wrapper=requests_wrapper, allow_dangerous_requests=True)
-        self.url_chain = LLMChain(llm=llm, prompt=api_url_prompt)
-        self.response_chain = LLMChain(llm=llm, prompt=api_response_prompt)
+        self.url_chain = LLMChain(llm=llm, prompt=api_url_prompt) 
+        self.response_chain = LLMChain(llm=llm, prompt=api_response_prompt) 
         self.api_docs = api_docs
 
     async def ainvoke(self, input_data, callbacks=None):
@@ -58,7 +58,7 @@ class EnhancedRequestsPostTool(RequestsPostTool):
             "api_response": api_response
         })
         
-        return formatted_response # when to call?
+        return formatted_response 
     
     
 daysoff_assistant_template = """
@@ -148,8 +148,6 @@ def setup_multiple_chains():
     cl.user_session.set("llm_chain", llm_chain)
     cl.user_session.set("post_tool", post_tool)
     
-    
-    
 
 @cl.on_message
 async def handle_message(message: cl.Message):
@@ -157,49 +155,12 @@ async def handle_message(message: cl.Message):
     llm_chain = cl.user_session.get("llm_chain")
     post_tool = cl.user_session.get("post_tool")
 
-    # Match booking pattern
     booking_pattern = r'\b[A-Z]{6}\d{6}\b'
     endpoint_url = "https://aivisions.no/data/daysoff/api/v1/booking/"
 
     match = re.search(booking_pattern, user_message)
     if match:
-        bestillingskode = match.group(0)
-        post_data = {
-            "url": endpoint_url,
-            "body": {
-                "booking_id": bestillingskode
-            }
-        }
-        try:
-            response = await post_tool.acall(
-                json.dumps(post_data),
-                callbacks=[cl.AsyncLangchainCallbackHandler()]
-            )
-            response_key = "output" if "output" in response else "text"
-            await cl.Message(content=response.get(response_key, "")).send()
-        except Exception as e:
-            await cl.Message(content=f"Error: {str(e)}").send()
-    else:
-        response = await llm_chain.ainvoke(
-            {"chat_history": "", "question": user_message},
-            callbacks=[cl.AsyncLangchainCallbackHandler()]
-        )
-        await cl.Message(content=response).send()
-
-"""
-async def handle_message(message: cl.Message):
-    user_message = message.content
-    llm_chain = cl.user_session.get("llm_chain")
-    cl.user_session.set("post_tool", post_tool)
-
-    # --------------------------------AD VAL.II.7--------------------------------
-    booking_pattern = r'\b[A-Z]{6}\d{6}\b'
-    endpoint_url = "https://aivisions.no/data/daysoff/api/v1/booking/"
-
-    match = re.search(booking_pattern, user_message)
-    if match:
-        bestillingskode = match.group(0)   
-        
+        bestillingskode = match.group()
         post_data = {
             "url": endpoint_url,
             "body": {
@@ -211,19 +172,16 @@ async def handle_message(message: cl.Message):
             json.dumps(post_data),
             callbacks=[cl.AsyncLangchainCallbackHandler()]
         )
-    # --------------------------------AD VAL.@End--------------------------------
-    
+               
     else:
-        response = await llm_chain.ainvoke(
-            user_message, 
-            callbacks=[cl.AsyncLangchainCallbackHandler()]
+        response = await llm_chain.ainvoke(      
+            user_message,                                # {"chat_history": "", "question": user_message}
         )
+        await cl.Message(content=response).send()
 
     response_key = "output" if "output" in response else "text"
 
-    
     await cl.Message(response.get(response_key, "")).send()
     return message.content
 
 
-"""