taha454 commited on
Commit
9e4211b
·
verified ·
1 Parent(s): acad1c7

update consumer to clearly output what needed

Browse files
Files changed (1) hide show
  1. Consumer.py +7 -3
Consumer.py CHANGED
@@ -73,9 +73,9 @@ def model_call(request, token):
73
  if isinstance(request, str):
74
  request = json.loads(request)
75
 
76
-
77
 
78
- answer = get_response(
79
  GRAPH,
80
  request.get('prompt', request.get('msg', "")),
81
  token,
@@ -85,7 +85,11 @@ def model_call(request, token):
85
 
86
  )
87
 
88
- return answer
 
 
 
 
89
 
90
  def process_message(recieved_msg):
91
  # decrypt token
 
73
  if isinstance(request, str):
74
  request = json.loads(request)
75
 
76
+
77
 
78
+ response = get_response(
79
  GRAPH,
80
  request.get('prompt', request.get('msg', "")),
81
  token,
 
85
 
86
  )
87
 
88
+ if "question" in response.keys():
89
+ return response["answer"]
90
+ elif "report" in response.keys():
91
+ return response["report"]
92
+ return response["answer"]
93
 
94
  def process_message(recieved_msg):
95
  # decrypt token