rowboat/apps/agents/tests/app_client.py
2025-01-14 15:09:14 +05:30

19 lines
No EOL
686 B
Python

from src.utils.common import common_logger, read_json_from_file
logger = common_logger
logger.info("Running swarm_flask_client.py")
import requests
if __name__ == "__main__":
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--sample_request', type=str, required=True, help='Sample request JSON file name under tests/sample_requests/')
args = parser.parse_args()
request = read_json_from_file(f"./tests/sample_requests/{args.sample_request}").get("lastRequest", {})
print("Sending request...")
response = requests.post(
"http://localhost:4040/chat",
json=request
).json()
print("Output: ")
print(response)