Spaces:
Running
on
Zero
Running
on
Zero
"""This is just a demo to test api.py.""" | |
from time import sleep | |
import requests | |
def post_data(url, extra_arguments): | |
""" | |
Make an HTTP POST request with extra_arguments as data. | |
Parameters | |
---------- | |
url : str | |
The URL to which the POST request should be sent. | |
extra_arguments : dict | |
A dictionary of data that needs to be sent in the POST request. | |
Returns | |
------- | |
response | |
The response from the server. | |
""" | |
response = requests.post(url, json=extra_arguments) | |
return response | |
if __name__ == "__main__": | |
URL_BASE = "http://127.0.0.1:8000" | |
arguments = { | |
"input": "We are writing snake in python. MVC components split \ | |
in separate files. Keyboard control.", # our prompt | |
"additional_input": {"improve_option": False}, | |
} | |
# create a task | |
response = post_data(f"{URL_BASE}/agent/tasks", arguments) | |
print(response.json()) | |
task_id = response.json()["task_id"] | |
sleep(1) # this is not needed | |
# execute the step for our task | |
response = post_data(f"{URL_BASE}/agent/tasks/{task_id}/steps", {}) | |
print(response.json()) | |