wehub-resource-sync
33 lines
954 B
Python
33 lines
954 B
Python
from flask import Flask, jsonify, request
|
|
from llama_deploy import ControlPlaneConfig, LlamaDeployClient
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
@app.route("/", methods=["POST"])
|
|
async def run() -> jsonify:
|
|
"""Provides a Flask application to interact with the deployed workflow."""
|
|
|
|
data = request.get_json()
|
|
workflow_name = data.get("workflow", "my_workflow") # Default to 'my_workflow'
|
|
print(workflow_name)
|
|
arguments = data.get("args", {})
|
|
print(arguments)
|
|
|
|
client = LlamaDeployClient(ControlPlaneConfig())
|
|
print("Client success made")
|
|
print(client.control_plane_config)
|
|
session = client.create_session()
|
|
|
|
result = session.run(
|
|
workflow_name, query=arguments["query"], num_steps=arguments["num_steps"]
|
|
)
|
|
print(result)
|
|
|
|
print("Completed workflow")
|
|
return jsonify({"status": 200, "answer": result})
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0", port=8080, debug=False, use_reloader=False)
|