Update main.py

main.py

@@ -426,6 +426,7 @@ app = FastAPI(title="Sattva AI API")
 
 class ChatRequest(BaseModel):
     user_id: str
+    username: str
     message: str
     mode: str
 
@@ -461,7 +462,8 @@ async def cron_test():
 @app.post("/chat", response_model=ChatResponse)
 async def chat_endpoint(request: ChatRequest):
     try:
-        username = request.user_id
+        username = request.username
+        userid = request.user_id
         user_input = request.message
         mode = request.mode
         config = RailsConfig.from_path("./config")
@@ -510,9 +512,9 @@ async def chat_endpoint(request: ChatRequest):
     #         # pass
      
         ncon = 3
-        chat_history_str = extract_chats(collection, username, ncon) 
+        chat_history_str = extract_chats(collection, userid, ncon) 
         
-        prev_goalandfocus = extract_goalfocus(collection, username)
+        prev_goalandfocus = extract_goalfocus(collection, userid)
 
         context_goal = f'''
        Ohk, so you are an expert in navigating paths through human conversations. So, let's say if someone is telling you about how they are feeling, what they did
@@ -602,7 +604,8 @@ async def chat_endpoint(request: ChatRequest):
 
         if collection is not None:
             chat_document = {
-                "user_id": username,
+                "user_id": userid,
+                "username": username,
                 "timestamp": datetime.now(),
                 "conversation": {
                     "human": user_input,