debug

Consumer.py

@@ -42,23 +42,40 @@ def redis_send(user_id,msg_id,answer):
     success = r.set(f'ANSWER_FOR_USER_ID{user_id}_OF_{msg_id}',json.dumps(answer))
     return success
 
-def model_call(request,token):
-    # fill with last state 
-    try:
-        state = json.loads(request['last_state']) 
-    except Exception:
-        state: ProblemState = {
-            "question": request['prompt'],
-            "memory": request['memory']
-        }
-
-    answer = get_response(request['prompt'], request['memory'],token,state,request['user_email'],request['user_name'])
-    
-    # drop unserlizable keys
-    for k in ["llm","rag_model"]:
-        answer[k] = ""
-
-    print(answer)
+
+def model_call(request, token): 
+    # تأكد إن request dict
+    if isinstance(request, str):
+        request = json.loads(request)
+
+    # fill with last state  
+    try: 
+        state = json.loads(request.get('last_state', "")) if request.get('last_state') else {}
+    except Exception: 
+        state = {}
+
+    # fallback لو مفيش state
+    if not state:
+        state = { 
+            "question": request.get('prompt', ""), 
+            "memory": request.get('memory', []) 
+        } 
+
+    answer = get_response(
+        request.get('prompt', ""), 
+        request.get('memory', []),
+        token,
+        state,
+        request.get('user_email', ""),
+        request.get('user_name', "")
+    ) 
+     
+    # drop unserializable keys 
+    for k in ["llm", "rag_model"]: 
+        if k in answer:
+            answer[k] = "" 
+
+    print(answer) 
     return answer
 
 def process_message(recieved_msg):
@@ -120,4 +137,18 @@ def start_consumer():
 if __name__ == "__main__":
     print(f"Starting New Consumer {consumer_id}...")
     #start_consumer()
-    model_call("""prompt='اتعلم كام لغة برمجة ؟' ht_token_encrypted_dumped='{"iv":"4fc9f781a608d1973a6c891a","data":"219240045595c56e063583038736e0287f09f2dc2f60c123e2b31270d31c425f2b1e36c249","tag":"296601cb1b77ded17eba99dba2153d71"}' user_email='egyptyahoo2@gmail.com' user_name='yahoo egypt' user_id='gEU1QJ3qaLUXL5nGnTWZgLyDfIj1' msg_id='688514' memory=['اتعلم كام لغة برمجة ؟'] last_state=''""",os.environ["test_hf_token"])
+
+    model_call({
+        "prompt": "اتعلم كام لغة برمجة ؟",
+        "ht_token_encrypted_dumped": {
+            "iv": "4fc9f781a608d1973a6c891a",
+            "data": "219240045595c56e063583038736e0287f09f2dc2f60c123e2b31270d31c425f2b1e36c249",
+            "tag": "296601cb1b77ded17eba99dba2153d71"
+        },
+        "user_email": "egyptyahoo2@gmail.com",
+        "user_name": "yahoo egypt",
+        "user_id": "gEU1QJ3qaLUXL5nGnTWZgLyDfIj1",
+        "msg_id": "688514",
+        "memory": ["اتعلم كام لغة برمجة ؟"],
+        "last_state": ""
+    }, os.environ["test_hf_token"])