Update app.py

app.py

@@ -1,26 +1,8 @@
 
 # ===========================================
-# ver01.01-5.workload-----app.py
+# "the-very-latest"-----app.py ✍🏽🎬
 # ===========================================
 
-def print_scripts(file_path):
-    try:
-        with open(file_path, 'r') as file:
-            print(f"\n--- Contents of {file_path} ---\n")
-            print(file.read())
-            print(f"--- End of {file_path} ---\n")
-    except Exception as e:
-        print(f"Could not read {file_path}: {e}")
-
-
-if __name__ == "__main__":
-    
-    file_1 = "/home/user/.local/lib/python3.10/site-packages/chainlit/socket.py"
-    file_2 = "/home/user/.local/lib/python3.10/site-packages/socketio/async_server.py"
-
-    print_scripts(file_1)
-    print_scripts(file_2)
-
 import asyncio
 import concurrent.futures
 import os
@@ -40,12 +22,11 @@ from langchain.memory.buffer import ConversationBufferMemory
 from langchain.memory import ConversationTokenBufferMemory
 from langchain.memory import ConversationSummaryMemory
 
-#from api_docs_mck import api_docs_str 
-from api_docs import api_docs_str, auth_token
+from api_docs import api_docs_str
 
 load_dotenv()
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
-auth = os.environ.get("CHAINLIT_AUTH_SECRET")
+auth_token = os.getenv("DAYSOFF_API_TOKEN")
 
 daysoff_assistant_template = """
 #You are a customer support assistant (’kundeservice AI assistent’) for Daysoff. 
@@ -92,16 +73,6 @@ api_response_prompt = PromptTemplate(
     template=api_response_template
 )
 
-#async def on_chat_start():
-    #app_user = cl.user_session.get("user")
-    #user_env = cl.user_session.get("env")
-    #await cl.Message(f"Hello {app_user.identifier}").send()
-
-#@cl.on_chat_start
-#async def on_chat_start():
-    #app_user = cl.user_session.get("user")
-    #await cl.Message(f"Hello {app_user}").send()
-    
 @cl.on_chat_start
 def setup_multiple_chains():
     llm = OpenAI(
@@ -127,86 +98,21 @@ def setup_multiple_chains():
     
     cl.user_session.set("llm_chain", llm_chain)
 
-    
     api_chain = APIChain.from_llm_and_api_docs(
         llm=llm,
         api_docs=api_docs_str,
         api_url_prompt=api_url_prompt,
         api_response_prompt=api_response_prompt,
+        headers={
+            "Authorization": f"Bearer {auth_token}",
+            "Content-Type": "application/json"
+        },
         verbose=True,
-        limit_to_domains=None #["http://0.0.0.0:7860/"]
+        limit_to_domains=["aivisions.no"]
     )
 
     cl.user_session.set("api_chain", api_chain)
 
-
-"""
-@cl.on_message
-async def handle_message(message: cl.Message):
-    user_message = message.content #.lower()
-    llm_chain = cl.user_session.get("llm_chain")
-    api_chain = cl.user_session.get("api_chain")
-  
-    booking_pattern = r'\b[A-Z]{6}\d{6}\b' 
-    endpoint_url = "https://aivisions.no/data/daysoff/api/v1/booking/"
-
-    
-    #auth_token = f"Bearer {auth_token}
-    auth_token = "abc123def456ghi789jkl012mno345pqr678stu901vwx234yz567"
-
-
-    # --GET method
-    if re.search(booking_pattern, user_message):  
-        bestillingskode = re.search(booking_pattern, user_message).group(0)  
-        question = f"Retrieve information for booking ID {endpoint_url}?search={bestillingskode}&auth_token={auth_token}"
-" 
-        response = await api_chain.acall(
-            {
-                "bestillingskode": bestillingskode,
-                "question": question,
-                "auth_token": auth_token
-              
-            },
-            callbacks=[cl.AsyncLangchainCallbackHandler()])
-    
-"""
-
-
-"""
-    # --POST method, booking_id@body
-    if re.search(booking_pattern, user_message):  
-        bestillingskode = re.search(booking_pattern, user_message).group(0)  
-        question = f"Retrieve information for booking ID {bestillingskode}"
-
-        response = await api_chain.acall(
-            {
-                "url": endpoint_url,  
-                "method": "POST",     
-                "headers": {
-                    "Authorization": f"Bearer {auth_token}",
-                    "Content-Type": "application/json"
-                },
-                "body": {             
-                    "booking_id": bestillingskode
-                },
-                "question": question
-            },
-            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
-
-    #endpoint_url = "https://670dccd0073307b4ee447f2f.mockapi.io/daysoff/api/V1/booking"
-    
-    
-    
-"""
 @cl.on_message
 async def handle_message(message: cl.Message):
     user_message = message.content 
@@ -214,36 +120,38 @@ 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'
-    endpoint_url = "https://aivisions.no/data/daysoff/api/v1/booking/"
-    auth_token = "abc123def456ghi789jkl012mno345pqr678stu901vwx234yz567"
+    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
+    #)
 
     if re.search(booking_pattern, user_message):
         bestillingskode = re.search(booking_pattern, user_message).group(0)
-        question = f"Retrieve information for booking ID {endpoint_url}?auth_token={auth_token}&booking_id={bestillingskode}"
-
-        response = await api_chain.acall(                           # ~ainvoke
-            {
-                "bestillingskode": bestillingskode,
-                "question": question
-            },
-            callbacks=[cl.AsyncLangchainCallbackHandler()])
+        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}
+            }
+        },
+        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
-
-
-    # --concurrent execution@ThreadPoolExecutor, resolve the ’Could not reach the server' error?
-    #loop = asyncio.get_running_loop()
-    #with concurrent.futures.ThreadPoolExecutor() as pool:
-        #await loop.run_in_executor(pool, lambda: asyncio.run(process_message()))
-
-
-
-
-    
+    return message.content