taha454 commited on
Commit
8a1d4f0
·
verified ·
1 Parent(s): 942afe5

debug json loading code

Browse files
Files changed (3) hide show
  1. Consumer.py +7 -1
  2. Queue_Producer.py +3 -1
  3. app.py +1 -1
Consumer.py CHANGED
@@ -97,7 +97,13 @@ def model_call(request, token):
97
 
98
  def process_message(recieved_msg):
99
  # decrypt token
100
- token = decrypt_token_from_json(json.loads(recieved_msg['ht_token_encrypted_dumped']))
 
 
 
 
 
 
101
  # call the model
102
  model_answer = model_call(recieved_msg,token)
103
  # send answer to redis
 
97
 
98
  def process_message(recieved_msg):
99
  # decrypt token
100
+ token_data = recieved_msg["ht_token_encrypted_dumped"]
101
+
102
+ if isinstance(token_data, str):
103
+ token_data = json.loads(token_data)
104
+
105
+ token = decrypt_token_from_json(token_data)
106
+
107
  # call the model
108
  model_answer = model_call(recieved_msg,token)
109
  # send answer to redis
Queue_Producer.py CHANGED
@@ -1,6 +1,8 @@
1
  # Producer
2
  import pika
3
  import os
 
 
4
  QUEUE_NAME = os.environ["QUEUE_NAME"]
5
 
6
 
@@ -10,7 +12,7 @@ def send_message(RABBITMQ_CONNECTION,data: dict):
10
 
11
  channel.queue_declare(queue=QUEUE_NAME, durable=True)
12
 
13
- message = str(data)
14
 
15
  channel.basic_publish(
16
  exchange='',
 
1
  # Producer
2
  import pika
3
  import os
4
+ import json
5
+
6
  QUEUE_NAME = os.environ["QUEUE_NAME"]
7
 
8
 
 
12
 
13
  channel.queue_declare(queue=QUEUE_NAME, durable=True)
14
 
15
+ message = json.dumps(data) #str(data)
16
 
17
  channel.basic_publish(
18
  exchange='',
app.py CHANGED
@@ -142,7 +142,7 @@ def get_health():
142
  def call(request: RequestModel):
143
  global RABBITMQ_CONNECTION
144
 
145
- data = json.dumps(request.model_dump())
146
 
147
  redis_send(
148
  REDIS_CONN,
 
142
  def call(request: RequestModel):
143
  global RABBITMQ_CONNECTION
144
 
145
+ data =request.model_dump()
146
 
147
  redis_send(
148
  REDIS_CONN,