Update app.py

app.py

@@ -32,30 +32,30 @@ daysoff_assistant_booking_prompt= PromptTemplate(
 
 api_url_template = """
 Given the following API Documentation for Daysoff's official
-booking information API: {api_docs_mck}
+booking information API: {daysoff_api_docs}
 Your task is to construct the most efficient API URL to answer
 the user's question, ensuring the
 call is optimized to include only the necessary information.
 Question: {question}
 """
-api_url_prompt = PromptTemplate(input_variables=['api_docs_mck', 'question'],
+api_url_prompt = PromptTemplate(input_variables=['daysoff_api_docs', 'question'],
                                 template=api_url_template)
 
 api_response_template = """"
-With the API Documentation for Daysoff's official API: {api_docs_mck}
+With the API Documentation for Daysoff's official API: {daysoff_api_docs}
 and the specific user question: {question} in mind,
-and given this API URL: {api_url} for querying, here is the
-response from Daysoff's API: {api_response}.
+and given this API URL: {base_url} for querying, here is the
+response from Daysoff's API: {response}.
 Please provide user with their booking information,
 focusing on delivering the answer with clarity and conciseness,
 as if a human customer service agent is providing this information.
 Adapt to user's language. By default, you speak Norwegian.
 Booking information:
 """
-api_response_prompt = PromptTemplate(input_variables=['api_docs_mck',
+api_response_prompt = PromptTemplate(input_variables=['daysoff_api_docs',
                                                       'question',
-                                                      'api_url',
-                                                      'api_response'],
+                                                      'base_url',
+                                                      'response'],
                                      template=api_response_template)
 
 @cl.on_chat_start
@@ -73,9 +73,9 @@ def setup_multiple_chains():
 
     cl.user_session.set("llm_chain", llm_chain)
 
-    api_chain = APIChain(
+    api_chain = APIChain.from_llm_and_api_docs(
         llm=llm,
-        api_docs_mck=daysoff_api_docs,
+        api_docs=daysoff_api_docs,
         api_url_prompt=api_url_prompt,
         api_response_prompt=api_response_prompt,
         verbose=True,
@@ -113,7 +113,11 @@ async def handle_message(message: cl.Message):
         re.search(keyword, user_message, re.IGNORECASE) 
         for keyword in BOOKING_KEYWORDS + [BOOKING_ID]
     )
+"""
+ match = re.search(r'\b[A-Z]{6}\d{6}\b', user_message) 
+    return match.group(0) if match else None
 
+"""
     if is_booking_query:
         response = await api_chain.acall(user_message,
                                          callbacks=[cl.AsyncLangchainCallbackHandler()])