Update app.py

app.py

@@ -17,7 +17,8 @@ from dotenv import load_dotenv
 
 import chainlit as cl
 from chainlit import user_session
-
+from chainlit.context import init_ws_context
+from chainlit.session import WebsocketSession # --'Session is disconnected' rt7E1rI_uhficQm8AAAA 
 from langchain import hub
 from langchain_openai import OpenAI
 from langchain.chains import LLMChain
@@ -31,22 +32,22 @@ from langchain_core.messages import BaseMessage
 from pydantic import BaseModel, Field
 from typing import List
 
-class InMemoryHistory(BaseChatMessageHistory, BaseModel):
-    messages: List[BaseMessage] = Field(default_factory=list)
+#class InMemoryHistory(BaseChatMessageHistory, BaseModel):
+    #messages: List[BaseMessage] = Field(default_factory=list)
 
-    def add_messages(self, messages: List[BaseMessage]) -> None:
-        self.messages.extend(messages)
+    #def add_messages(self, messages: List[BaseMessage]) -> None:
+        #self.messages.extend(messages)
 
-    def clear(self) -> None:
-        self.messages = []
+    #def clear(self) -> None:
+        #self.messages = []
 
 # --storing chat histories
-chat_histories = {}
+#chat_histories = {}
 
-def get_session_history(session_id: str) -> BaseChatMessageHistory:
-    if session_id not in chat_histories:
-        chat_histories[session_id] = InMemoryHistory()
-    return chat_histories[session_id]
+#def get_session_history(session_id: str) -> BaseChatMessageHistory:
+    #if session_id not in chat_histories:
+        #chat_histories[session_id] = InMemoryHistory()
+    #return chat_histories[session_id]
 
 # ---------------------------------------------------for backend looks, example file:----------------------------------
 #with open('/home/user/.local/lib/python3.10/site-packages/socketio/async_server.py', 'r') as file:
@@ -54,6 +55,7 @@ def get_session_history(session_id: str) -> BaseChatMessageHistory:
     #print("see line 640:", content)
 # ------------------------------------------------------the end--------------------------------------------------------
 
+
 load_dotenv()
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 auth_token = os.environ.get("DAYSOFF_API_TOKEN")
@@ -95,7 +97,7 @@ async def set_starters():
             ),
         cl.Starter(
             label="Metric Space Self-Identity Framework",
-            message="Explain the Metric Space Self-Identity Framework like I'm five years old.",
+            message="Explain the Metric Space Self-Identity Framework like I'm six years old.",
             icon="/public/learn.svg",
             ),
         cl.Starter(
@@ -112,6 +114,15 @@ async def set_starters():
     
 @cl.on_chat_start # <————————————————— the Chainlit context
 def setup_multiple_chains():
+
+    # --config Chainlit logging
+    cl.config.run.log_level = "INFO"
+    print("Session id:", cl.user_session.get("id"))
+
+    #session_id: str
+    #ws_session = WebsocketSession.get_by_id(session_id=session_id)
+    #init_ws_context(ws_session)
+    
     llm = OpenAI(
         model="gpt-3.5-turbo-instruct",
         temperature=0.7,