use-workflow-run-event.ts 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. import {
  2. useWorkflowAgentLog,
  3. useWorkflowFailed,
  4. useWorkflowFinished,
  5. useWorkflowNodeFinished,
  6. useWorkflowNodeIterationFinished,
  7. useWorkflowNodeIterationNext,
  8. useWorkflowNodeIterationStarted,
  9. useWorkflowNodeRetry,
  10. useWorkflowNodeStarted,
  11. useWorkflowStarted,
  12. useWorkflowTextChunk,
  13. useWorkflowTextReplace,
  14. } from '.'
  15. export const useWorkflowRunEvent = () => {
  16. const { handleWorkflowStarted } = useWorkflowStarted()
  17. const { handleWorkflowFinished } = useWorkflowFinished()
  18. const { handleWorkflowFailed } = useWorkflowFailed()
  19. const { handleWorkflowNodeStarted } = useWorkflowNodeStarted()
  20. const { handleWorkflowNodeFinished } = useWorkflowNodeFinished()
  21. const { handleWorkflowNodeIterationStarted } = useWorkflowNodeIterationStarted()
  22. const { handleWorkflowNodeIterationNext } = useWorkflowNodeIterationNext()
  23. const { handleWorkflowNodeIterationFinished } = useWorkflowNodeIterationFinished()
  24. const { handleWorkflowNodeRetry } = useWorkflowNodeRetry()
  25. const { handleWorkflowTextChunk } = useWorkflowTextChunk()
  26. const { handleWorkflowTextReplace } = useWorkflowTextReplace()
  27. const { handleWorkflowAgentLog } = useWorkflowAgentLog()
  28. return {
  29. handleWorkflowStarted,
  30. handleWorkflowFinished,
  31. handleWorkflowFailed,
  32. handleWorkflowNodeStarted,
  33. handleWorkflowNodeFinished,
  34. handleWorkflowNodeIterationStarted,
  35. handleWorkflowNodeIterationNext,
  36. handleWorkflowNodeIterationFinished,
  37. handleWorkflowNodeRetry,
  38. handleWorkflowTextChunk,
  39. handleWorkflowTextReplace,
  40. handleWorkflowAgentLog,
  41. }
  42. }