2025-01-13 18:20:38 +05:30
|
|
|
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__":
|
2025-01-14 15:09:14 +05:30
|
|
|
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", {})
|
2025-01-13 18:20:38 +05:30
|
|
|
print("Sending request...")
|
|
|
|
|
response = requests.post(
|
2025-01-14 15:09:14 +05:30
|
|
|
"http://localhost:4040/chat",
|
2025-01-13 18:20:38 +05:30
|
|
|
json=request
|
|
|
|
|
).json()
|
|
|
|
|
print("Output: ")
|
2025-01-14 15:09:14 +05:30
|
|
|
print(response)
|