Upload 157 files

src/apis/controllers/chat_controller.py

@@ -52,19 +52,19 @@ async def save_history(user_id, human_message, ai_message, intent):
             {"session_id": user_id}, {"intent": intent}
         )
         logger.info("History updated")
-    # history_cache = await get_key_redis(f"chat_history_{user_id}")
-    # if history_cache is not None:
-    #     history_cache = eval(history_cache)
-    #     history_cache["message"] = (
-    #         history_cache["message"] + messages_add_to_history_dict
-    #     )
-    #     history_cache["intent"] = intent
-    #     await set_key_redis(
-    #         f"chat_history_{user_id}",
-    #         str(history_cache),
-    #     )
-    #     return {"message": "History updated"}
-    # await set_key_redis(f"chat_history_{user_id}", str(messages_add_to_history_cache))
+    history_cache = await get_key_redis(f"chat_history_{user_id}")
+    if history_cache is not None:
+        history_cache = eval(history_cache)
+        history_cache["message"] = (
+            history_cache["message"] + messages_add_to_history_dict
+        )
+        history_cache["intent"] = intent
+        await set_key_redis(
+            f"chat_history_{user_id}",
+            str(history_cache),
+        )
+        return {"message": "History updated"}
+    await set_key_redis(f"chat_history_{user_id}", str(messages_add_to_history_cache))
     return {"message": "History created"}
 
 
@@ -211,10 +211,10 @@ async def get_intent_function(session_id):
     return record["intent"]
 
 
-async def get_history_function(session_id):
-    history = chat_messages_history(session_id, 50)
+async def get_history_function(user_id, background_tasks: BackgroundTasks):
+    history = chat_messages_history(user_id, 50)
     try:
-        # history_messages = await get_key_redis(f"chat_history_{session_id}")
+        history_messages = await get_key_redis(f"chat_history_{user_id}")
         history_messages = None
         if not history_messages:
             logger.info("History not found in redis")
@@ -222,9 +222,11 @@ async def get_history_function(session_id):
             history_messages = [
                 i.model_dump(include=["type", "content"]) for i in history_messages
             ]
-            intent = await get_intent_function(session_id)
+            intent = await get_intent_function(user_id)
             logger.info(F"INTENT {intent}")
-            return {"message": history_messages, "intent": intent}
+            result = {"message": history_messages, "intent": intent}
+            background_tasks.add_task(set_key_redis, f"chat_history_{user_id}", str(result))
+            return result
         history_messages = eval(history_messages)
         return history_messages
     except Exception as e:

src/apis/models/post_models.py

@@ -18,7 +18,6 @@ class Comment(BaseDocument):
         }
     }
 
-
 class Reaction(BaseDocument):
     user_id: str = Field("", description="User's id")
     post_id: str = Field("", description="Post's id")

src/apis/routes/chat_route.py

@@ -47,10 +47,10 @@ async def get_chat_history(user: user_dependency, background_tasks: BackgroundTa
     if user is None:
         return JSONResponse(content={"message": "User not found"}, status_code=404)
     user_id = user["id"]
-    result = await get_history_function(user_id)
+    result = await get_history_function(user_id, background_tasks)
     if result is None:
         return JSONResponse(content={"message": [], "intent": None}, status_code=500)
-    # background_tasks.add_task(set_key_redis, f"chat_history_{user_id}", str(result))
+    background_tasks.add_task(set_key_redis, f"chat_history_{user_id}", str(result))
     return JSONResponse(content=result, status_code=200)
 
 

src/apis/routes/post_router.py

@@ -67,12 +67,12 @@ async def get_post(post_id: str):
 
 @router.get("/list", status_code=status.HTTP_200_OK)
 async def list_all_posts(background_tasks: BackgroundTasks):
-    # result = await get_key_redis("all_posts")
+    result = await get_key_redis("all_posts")
     result = None
     result = eval(result) if result else None
     if not result:
         result = await list_all_posts_controller()
-        # background_tasks.add_task(set_key_redis, "all_posts", str(result),20)
+        background_tasks.add_task(set_key_redis, "all_posts", str(result),20)
     if result["status"] == "error":
         return JSONResponse(content=result, status_code=404)
     else: