Update api_data.py

api_data.py

@@ -1,10 +1,6 @@
 # ================================
-# {𝗗𝗮𝘆𝘀𝗼𝗳𝗳 ᵉˣᵖᵃⁿˢᵉ ˢʸˢᵗᵉᵐ} | 𝘴𝘶𝘱𝘱𝘰𝘳𝘵
+# api-data.py
 # ================================
-# V2 | try: chainlit==1.3.0 for better inteerface
-# ———————————————————————————————>
-
-# ———————————————————————————————>
 
 import json
 import asyncio
@@ -14,19 +10,9 @@ import requests
 import chainlit as cl
 from dotenv import load_dotenv
 
-
-from chromadb.config import Settings
-from chromadb import Client
-
-# Initialize ChromaDB client
-chroma_client = Client(Settings(chroma_db_impl="duckdb+parquet", persist_directory="./session_data"))
-
-
-
 # --------------------------------=== API URL ===-----------------------------------
 API_URL = "https://firmahytte.daysoff.no/api/bookings"
 
-
 # --------------------------------=== environment ===-------------------------------
 load_dotenv()
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
@@ -64,12 +50,12 @@ async def async_post_request(url, headers, data):
 async def booking_agent_system(message: cl.Message, msg: cl.Message) -> bool:
 
     user_message = message.content  # -- doc: extract message text
-    booking_pattern = r'\b[A-Z]{6}\d{6}\b' # -- raw string (r'') 
+    booking_pattern = r'\b[A-Z]{6}\d{6}\b' # -- raw string (r'')
     match = re.search(booking_pattern, user_message)
 
     if match:
         preliminary = (
-            f"Takk for bookingnummer 👍🏻 "  # ——> f"Takk for bookingnummer \\b[A-Z]{{6}}\\d{{6}} ", OBS! → \b, {}, and \d must be double-escaped inside an f-string.
+            f"Takk for bookingnummer 👍🏻 "
             "Jeg skal sjekke dette for deg nå. Vennligst et øyeblikk bare..\n"
         )
         for token in preliminary.split():
@@ -88,10 +74,11 @@ async def booking_agent_system(message: cl.Message, msg: cl.Message) -> bool:
         }
 
         try:
+
             response = await async_post_request(API_URL, headers, payload)
             response.raise_for_status()
             response_data = response.json()
-            booking_data = response_data.get('data', {}) 
+            booking_data = response_data.get('data', {})
 
             print("API Response Data:", json.dumps(booking_data, indent=2))
 
@@ -100,18 +87,7 @@ async def booking_agent_system(message: cl.Message, msg: cl.Message) -> bool:
             if "error" in booking_data:
                 raise APIResponseError(booking_data["error"])
 
-
-            # Store booking data in ChromaDB
-            collection = chroma_client.create_collection(name=f"session_{message.user_id}")
-            collection.add(
-                documents=[json.dumps(booking_data)],
-                metadatas=[{"booking_number": booking_number}],
-                ids=[booking_number]
-            )
-
-
-
-            table = (  
+            table = (
                 "\n\n## 𝘽𝙤𝙤𝙠𝙞𝙣𝙜 𝙞𝙣𝙛𝙤𝙧𝙢𝙖𝙨𝙟𝙤𝙣\n\n"
                 f"| 𝐵𝑜𝑜𝑘𝑖𝑛𝑔𝑛𝑢𝑚𝑚𝑒𝑟 | {booking_data.get('order_number', 'N/A')} |\n"
                 f"|:-----|:------------------|\n"
@@ -141,14 +117,14 @@ async def booking_agent_system(message: cl.Message, msg: cl.Message) -> bool:
             await msg.stream_token(table)
             await asyncio.sleep(0.07)
             await msg.send()
-            
+
             header = (
                 "\n\n# 𝗩𝗶𝗸𝘁𝗶𝗴 𝗜𝗻𝗳𝗼𝗿𝗺𝗮𝘀𝗷𝗼𝗻 𝗼𝗺 𝗢𝗽𝗽𝗵𝗼𝗹𝗱𝗲𝘁\n\n"
             )
             await msg.stream_token(header)
             await asyncio.sleep(0.07)
             await msg.send()
-            
+
             # 𝘿𝙚𝙨𝙘𝙧𝙞𝙥𝙩𝙞𝙤𝙣
             markdown = (
                 "\n\n# 🇩 🇪 🇸 🇨 🇷 🇮 🇵 🇹 🇮 🇴 🇳\n\n"
@@ -168,7 +144,7 @@ async def booking_agent_system(message: cl.Message, msg: cl.Message) -> bool:
             await asyncio.sleep(0.07)
             await msg.send()
 
-            
+
         except (APIConnectionError, APIResponseError, BookingNotFoundError) as e:
             error_msg = (
                 f"🙄 {str(e)} "
@@ -177,7 +153,7 @@ async def booking_agent_system(message: cl.Message, msg: cl.Message) -> bool:
                 await msg.stream_token(token + " ")
                 #await asyncio.sleep(0.01)
             await msg.send()
-            return True  # --——> returns True, bcoz handled booking query even if with error
+            return True
 
         except requests.exceptions.RequestException as e:
             unexpected = (