Update README.md, NOTICE.md and app_client.py

This commit is contained in:
akhisud3195 2025-01-14 15:09:14 +05:30
parent cb8d845c19
commit 244811ac57
3 changed files with 100 additions and 22 deletions

View file

@ -1,27 +1,19 @@
from src.utils.common import common_logger, read_json_from_file
logger = common_logger
logger.info("Running swarm_flask_client.py")
import requests
from pprint import pprint
if __name__ == "__main__":
request = read_json_from_file("./tests/sample_requests/example4.json").get("lastRequest", {})
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...")
logger.info("Sending request...")
response = requests.post(
"http://localhost:4040/chat",
"http://localhost:4040/chat",
json=request
).json()
print("Output: ")
logger.info(f"Output: ")
# for k, v in response.items():
# print(f"{k}: {v}")
# print('*'*200)
# print('*'*200)
# logger.info(f"{k}: {v}")
# logger.info('*'*200)
# logger.info('='*200)
pprint(response, indent=2)
logger.info(response)
print(response)