opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-09-08 02:43:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(api): resolve new agent beta task typing

Browse Source
This commit is contained in:
hjlarry 2026-08-25 09:41:38 +08:00
parent a13d201184
commit 0f37c8b1e0
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/tasks/new_agent_beta_task.py
View File

@ -118,15 +118,14 @@ def register_new_agent_beta_workflow_publish_after_commit(
reject_on_worker_lost=True,
)
def ensure_new_agent_beta_participation_task(self, source_id: str, source_type: str = "revision") -> None:
if source_type == "revision":
ensure = BillingService.ensure_new_agent_beta_revision
elif source_type == "workflow":
ensure = BillingService.ensure_new_agent_beta_workflow
else:
if source_type not in {"revision", "workflow"}:
raise ValueError(f"Unsupported New Agent Beta source type: {source_type}")
try:
ensure(source_id)
if source_type == "revision":
BillingService.ensure_new_agent_beta_revision(source_id)
else:
BillingService.ensure_new_agent_beta_workflow(source_id)
except Exception as exc:
if self.request.retries >= _MAX_RETRIES:
logger.exception(