Spaces:
Sleeping
Sleeping
TahaFawzyElshrif commited on
Commit ·
ceaf913
1
Parent(s): 89bc1e6
working on queue, finalization
Browse files- Consumer.py +0 -1
- 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": "
|
| 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:
|