From 290e1b4dc65a7feb6b0b12c9ab60b3b30d1f263b Mon Sep 17 00:00:00 2001 From: Guillaume Aquilina Date: Tue, 18 Mar 2025 22:34:15 -0400 Subject: [PATCH] feat: add feedback token --- pyproject.toml | 2 +- workflowai/core/client/_models.py | 3 +++ workflowai/core/domain/run.py | 2 ++ 3 files changed, 6 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index c6892fc..f41cf23 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "workflowai" -version = "0.6.0" +version = "0.6.1" description = "Python SDK for WorkflowAI" authors = ["Guillaume Aquilina "] readme = "README.md" diff --git a/workflowai/core/client/_models.py b/workflowai/core/client/_models.py index ac888ba..75e61cf 100644 --- a/workflowai/core/client/_models.py +++ b/workflowai/core/client/_models.py @@ -129,6 +129,8 @@ class RunResponse(BaseModel): tool_calls: Optional[list[ToolCall]] = None tool_call_requests: Optional[list[ToolCallRequest]] = None + feedback_token: Optional[str] = None + def to_domain( self, task_id: str, @@ -151,6 +153,7 @@ def to_domain( cost_usd=self.cost_usd, tool_calls=safe_map_list(self.tool_calls, tool_call_to_domain), tool_call_requests=safe_map_list(self.tool_call_requests, tool_call_request_to_domain), + feedback_token=self.feedback_token, ) diff --git a/workflowai/core/domain/run.py b/workflowai/core/domain/run.py index 759fe4a..80db3dc 100644 --- a/workflowai/core/domain/run.py +++ b/workflowai/core/domain/run.py @@ -46,6 +46,8 @@ class Run(BaseModel, Generic[AgentOutput]): description="An error that occurred during the run. Only provided if the run failed.", ) + feedback_token: Optional[str] = None + _agent: Optional["_AgentBase[AgentOutput]"] = None def __eq__(self, other: object) -> bool: