Update app.py

app.py

@@ -28,8 +28,6 @@ from langchain.memory.buffer import ConversationBufferMemory
 load_dotenv()
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 auth_token = os.environ.get("CHAINLIT_AUTH_SECRET")
-if not auth_token.startswith("Bearer "):
-    auth_token = f"Bearer {auth_token}"
 
 # API endpoint
 API_URL = "https://aivisions.no/data/daysoff/api/v1/booking/"
@@ -120,6 +118,9 @@ async def handle_message(message: cl.Message):
         except requests.exceptions.RequestException as e:
             await cl.Message(content=f"Request failed: {str(e)}").send()
     else:
-       
-        response = await llm_chain.ainvoke(user_message)
-        await cl.Message(content=response).send()
+        response = await llm_chain.ainvoke(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