chore: fix ci issues

This commit is contained in:
Novice
2025-12-16 15:14:42 +08:00
parent ff57848268
commit 0c4c268003
6 changed files with 56 additions and 51 deletions

View File

@ -1,6 +1,5 @@
import threading
from collections.abc import Mapping, Sequence
from typing import Any
from collections.abc import Sequence
from sqlalchemy import Engine
from sqlalchemy.orm import sessionmaker
@ -109,9 +108,6 @@ class WorkflowRunService:
run_id=run_id,
)
if workflow_run:
workflow_run.outputs_as_generation = self._are_all_generation_outputs(workflow_run.outputs_dict)
return workflow_run
def get_workflow_runs_count(
@ -165,32 +161,3 @@ class WorkflowRunService:
app_id=app_model.id,
workflow_run_id=run_id,
)
@staticmethod
def _are_all_generation_outputs(outputs: Mapping[str, Any]) -> bool:
if not outputs:
return False
allowed_sequence_types = {"reasoning", "content", "tool_call"}
for value in outputs.values():
if not isinstance(value, Mapping):
return False
content = value.get("content")
reasoning_content = value.get("reasoning_content")
tool_calls = value.get("tool_calls")
sequence = value.get("sequence")
if not isinstance(content, str):
return False
if not isinstance(reasoning_content, list) or any(not isinstance(item, str) for item in reasoning_content):
return False
if not isinstance(tool_calls, list) or any(not isinstance(item, Mapping) for item in tool_calls):
return False
if not isinstance(sequence, list) or any(
not isinstance(item, Mapping) or item.get("type") not in allowed_sequence_types for item in sequence
):
return False
return True