test_embedding.py 681 B

1234567891011121314151617181920212223
  1. import requests
  2. import time
  3. import json
  4. session = requests.Session()
  5. def post_request():
  6. for _ in range(100):
  7. start_time = time.time()
  8. _json = {"test":1,"sentence":"testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest"}
  9. resp = session.post("http://localhost:17130/embedding", json=_json)
  10. if resp.status_code==200:
  11. content = resp.content.decode("utf-8")
  12. _d = json.loads(content)
  13. if _d.get("success"):
  14. print(resp.status_code,_d.get("success"),time.time()-start_time)
  15. if __name__ == '__main__':
  16. post_request()