feat(orfix(orchestrator): OrchestratorError to serialize Exception …

…as `str`
PrefectHQ · Nov 2, 2024 · f328387 · f328387
1 parent ae59a63
commit f328387
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/controlflow/events/orchestrator_events.py b/src/controlflow/events/orchestrator_events.py
@@ -1,4 +1,7 @@
-from typing import Literal
+from dataclasses import Field
+from typing import Annotated, Literal
+
+from pydantic.functional_serializers import PlainSerializer
 
 from controlflow.agents.agent import Agent
 from controlflow.events.base import UnpersistedEvent
@@ -21,7 +24,7 @@ class OrchestratorError(UnpersistedEvent):
     event: Literal["orchestrator-error"] = "orchestrator-error"
     persist: bool = False
     orchestrator: Orchestrator
-    error: Exception
+    error: Annotated[Exception, PlainSerializer(lambda x: str(x), return_type=str)]
 
 
 class AgentTurnStart(UnpersistedEvent):