Spaces:
Sleeping
Sleeping
update consumer to clearly output what needed
Browse files- 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 |
-
|
| 79 |
GRAPH,
|
| 80 |
request.get('prompt', request.get('msg', "")),
|
| 81 |
token,
|
|
@@ -85,7 +85,11 @@ def model_call(request, token):
|
|
| 85 |
|
| 86 |
)
|
| 87 |
|
| 88 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|