Spaces:
Sleeping
Sleeping
File size: 1,213 Bytes
3fdcc70 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import os
import requests
__ALL__ = ["tog", "task_decomposer"]
HOST = PORT = os.environ.get("TOG_SERVICE_HOST", "localhost")
PORT = os.environ.get("TOG_SERVICE_PORT", 10052)
def setup(host="localhost", port=10052):
global HOST, PORT
HOST = host
PORT = port
def tog(request, subtasks, **kwargs):
host = kwargs.get("host", HOST)
port = kwargs.get("port", PORT)
stream = kwargs.get("stream", False)
url = f"http://{host}:{port}/tog"
data = {"request": request, "subtasks": subtasks, "stream": stream}
response = requests.post(url, data=data, stream=stream)
# if not stream:
# response = response.content.decode("utf-8")
# print(f"response.json(): {response.json()}")
return response.json()
def task_decomposer(request, **kwargs):
host = kwargs.get("host", HOST)
port = kwargs.get("port", PORT)
stream = kwargs.get("stream", False)
url = f"http://{host}:{port}/task_decomposer"
data = {"request": request, "stream": stream}
response = requests.post(url, data=data, stream=stream)
# if not stream:
# response = response.content.decode("utf-8")
# return response.content.decode("utf-8")
return response.json()
|