sse_client.py 708 B

12345678910111213141516171819202122
  1. import requests
  2. import sseclient
  3. def stream_agent_response(prompt, session_id):
  4. url = f"http://127.0.0.1:8000/stream?prompt={prompt}&session_id={session_id}"
  5. # response = requests.get(url, stream=True)
  6. client = sseclient.SSEClient(url)
  7. print("AI输出:", end="", flush=True)
  8. for event in client:
  9. print(event.data, end="", flush=True)
  10. if "?" in event.data:
  11. break
  12. if __name__ == "__main__":
  13. # 示例:多轮对话
  14. session_id = "user123"
  15. while True:
  16. user_input = input("你: ")
  17. if user_input.strip().lower() in ["exit", "quit"]:
  18. break
  19. stream_agent_response(user_input, session_id)
  20. print() # 换行