vvzvlad/allora
1
0
Fork 0
Code Pull Requests Packages Releases Activity

Major upgrade

Browse Source
This commit is contained in:
clementupshot 2024-06-17 10:37:20 +02:00
parent 911555f358
commit 819b5b1e9d
No known key found for this signature in database
2 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
app.py
View File

@ -34,17 +34,13 @@ def generate_inference(token):
"""Generate inference for given token.""" """Generate inference for given token."""
if not token or token != "ETH": if not token or token != "ETH":
error_msg = "Token is required" if not token else "Token not supported" error_msg = "Token is required" if not token else "Token not supported"
return Response( return Response(json.dumps({"error": error_msg}), status=400, mimetype='application/json')
json.dumps({"error": error_msg}), status=400, mimetype="application/json"
)
try: try:
inference = get_eth_inference() inference = get_eth_inference()
return jsonify({"value": str(inference)}) return Response(str(inference), status=200)
except Exception as e: except Exception as e:
return Response( return Response(json.dumps({"error": str(e)}), status=500, mimetype='application/json')
json.dumps({"error": str(e)}), status=500, mimetype="application/json"
)
@app.route("/update") @app.route("/update")

22
main.py
View File

@ -9,19 +9,23 @@ INFERENCE_ADDRESS = os.environ["INFERENCE_API_ADDRESS"]
def process(token_name): def process(token_name):
response = requests.get(f"{INFERENCE_ADDRESS}/inference/{token_name}") response = requests.get(f"{INFERENCE_ADDRESS}/inference/{token_name}")
content = response.text content = response.text
print(content) return content
if __name__ == "__main__": if __name__ == "__main__":
# Your code logic with the parsed argument goes here # Your code logic with the parsed argument goes here
try: try:
if len(sys.argv) >= 3: if len(sys.argv) < 5:
# Not using to discriminate by topicId for simplicity. value = json.dumps({"error": f"Not enough arguments provided: {len(sys.argv)}, expected 4 arguments: topic_id, blockHeight, blockHeightEval, default_arg"})
# topic_id = sys.argv[1]
token_name = sys.argv[2]
else: else:
token_name = "ETH" topic_id = sys.argv[1]
process(token_name=token_name) blockHeight = sys.argv[2]
blockHeightEval = sys.argv[3]
default_arg = sys.argv[4]
response_inference = process(token_name=default_arg)
response_dict = {"infererValue": response_inference}
value = json.dumps(response_dict)
except Exception as e: except Exception as e:
response = json.dumps({"error": {str(e)}}) value = json.dumps({"error": {str(e)}})
print(response) print(value)