coze_workflow.py 668 B

123456789101112131415161718192021222324
  1. # -*- coding: utf-8 -*-
  2. """
  3. @author: yq
  4. @time: 2025/1/14
  5. @desc:
  6. """
  7. from cozepy import Coze, Stream, WorkflowEvent, WorkflowEventType, COZE_CN_BASE_URL
  8. from cozepy.auth import TokenAuth
  9. coze = Coze(auth=TokenAuth("pat_NoyDg48ZoXd4swDSXeDXLrHaQR73G9Ygr8qjxxvRfvyITwIwL7dQUjqfgcL8hL6j"), base_url=COZE_CN_BASE_URL)
  10. stream: Stream[WorkflowEvent] = coze.workflows.runs.stream(
  11. workflow_id='7483418763824693287', parameters={"query": "北京小米"})
  12. for event in stream:
  13. if event.event == WorkflowEventType.MESSAGE:
  14. print(event.message)
  15. elif event.event == WorkflowEventType.ERROR:
  16. print(event.error)
  17. if __name__ == "__main__":
  18. pass