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

working on queue, finalization

Browse files
Files changed (1) hide show
  1. Consumer.py +2 -6
Consumer.py CHANGED
@@ -39,7 +39,6 @@ def redis_send(user_id,msg_id,answer):
39
  password=redis_password,
40
  )
41
 
42
- print(f"STORING ANSWER IN REDIS FOR USER_ID {user_id} AND MSG_ID {msg_id}")
43
  success = r.set(f'ANSWER_FOR_USER_ID{user_id}_OF_{msg_id}',json.dumps(answer))
44
  return success
45
 
@@ -53,7 +52,6 @@ def model_call(request,token):
53
  "memory": request['memory']
54
  }
55
 
56
- print(f"MODEL CALL WITH STATE {state} and PROMPT {request['prompt']} and MEMORY {request['memory']} and HT_TOKEN {token} and USER_EMAIL {request['user_email']} and USER_NAME {request['user_name']}")
57
  answer = get_response(request['prompt'], request['memory'],token,state,request['user_email'],request['user_name'])
58
 
59
  # drop unserlizable keys
@@ -66,15 +64,14 @@ def model_call(request,token):
66
  def process_message(recieved_msg):
67
  # decrypt token
68
  token = decrypt_token_from_json(json.loads(recieved_msg['ht_token_encrypted_dumped']))
69
- print(f"TYPE {type(recieved_msg)}, CONTENT {recieved_msg} , TOKEN {token}")
70
  # call the model
71
  model_answer = model_call(recieved_msg,token)
72
  # send answer to redis
73
  user_id = recieved_msg["user_id"]
74
  msg_id = recieved_msg["msg_id"]
75
  redis_send_res = redis_send(user_id,msg_id,model_answer)
76
- print({"STATUS": redis_send_res , "user_id": user_id, "msg_id": msg_id})
77
-
78
 
79
  ##################################################
80
  # CONSUMER METHODS
@@ -93,7 +90,6 @@ def callback(ch, method, properties, body):
93
 
94
  ##### Process Message
95
  process_message(recieved_msg)
96
- print(f"CONSUMER {consumer_id}:::: Processing done")
97
 
98
 
99
  ###### Finalize
 
39
  password=redis_password,
40
  )
41
 
 
42
  success = r.set(f'ANSWER_FOR_USER_ID{user_id}_OF_{msg_id}',json.dumps(answer))
43
  return success
44
 
 
52
  "memory": request['memory']
53
  }
54
 
 
55
  answer = get_response(request['prompt'], request['memory'],token,state,request['user_email'],request['user_name'])
56
 
57
  # drop unserlizable keys
 
64
  def process_message(recieved_msg):
65
  # decrypt token
66
  token = decrypt_token_from_json(json.loads(recieved_msg['ht_token_encrypted_dumped']))
 
67
  # call the model
68
  model_answer = model_call(recieved_msg,token)
69
  # send answer to redis
70
  user_id = recieved_msg["user_id"]
71
  msg_id = recieved_msg["msg_id"]
72
  redis_send_res = redis_send(user_id,msg_id,model_answer)
73
+ print({"STATUS": redis_send_res , "CONSUMER": {consumer_id}}) # add monitoring but still hide user data
74
+
75
 
76
  ##################################################
77
  # CONSUMER METHODS
 
90
 
91
  ##### Process Message
92
  process_message(recieved_msg)
 
93
 
94
 
95
  ###### Finalize