TahaFawzyElshrif commited on
Commit
ceaf913
·
1 Parent(s): 89bc1e6

working on queue, finalization

Browse files
Files changed (2) hide show
  1. Consumer.py +0 -1
  2. app.py +2 -2
Consumer.py CHANGED
@@ -58,7 +58,6 @@ def model_call(request,token):
58
  for k in ["llm","rag_model"]:
59
  answer[k] = ""
60
 
61
- print(f"ANSWER {answer}")
62
  return answer
63
 
64
  def process_message(recieved_msg):
 
58
  for k in ["llm","rag_model"]:
59
  answer[k] = ""
60
 
 
61
  return answer
62
 
63
  def process_message(recieved_msg):
app.py CHANGED
@@ -88,9 +88,9 @@ def call(request: RequestAnswer):
88
  try:
89
  answer = redis_conn.get(f'ANSWER_FOR_USER_ID{request.user_id}_OF_{request.msg_id}')
90
  if answer is None:
91
- return {"status": "pending"}
92
  else:
93
- redis_conn.delete(f'ANSWER_FOR_USER_ID{request.user_id}_OF_{request.msg_id}') # Clean up after fetching
94
  return {"status": "ready", "data": json.loads(answer)}
95
 
96
  except Exception as e:
 
88
  try:
89
  answer = redis_conn.get(f'ANSWER_FOR_USER_ID{request.user_id}_OF_{request.msg_id}')
90
  if answer is None:
91
+ return {"status": "error", "message": "No answer found in Redis."}
92
  else:
93
+ redis_conn.delete(f'ANSWER_FOR_USER_ID{request.user_id}_OF_{request.msg_id}') # Clean up after fetching for memory and better secure as double call is wrong
94
  return {"status": "ready", "data": json.loads(answer)}
95
 
96
  except Exception as e: