feat: 实现 AgentScope ReAct Runner 两阶段执行并重构事件处理

2026-03-16 09:01:01 +08:00
parent 072c09d99d
commit dcceb48d84
51 changed files with 5015 additions and 5663 deletions
@@ -118,6 +118,31 @@ class LiteLLMService:
            + normalized_completion_tokens * selected_tier.output_cost_per_token
        )

+    def build_usage_metadata(
+        self,
+        *,
+        model: str,
+        usage_summary: dict[str, int] | None,
+    ) -> dict[str, Any]:
+        summary = usage_summary or {}
+        input_tokens = max(int(summary.get("input_tokens", 0) or 0), 0)
+        output_tokens = max(int(summary.get("output_tokens", 0) or 0), 0)
+        latency_ms = max(int(summary.get("latency_ms", 0) or 0), 0)
+        cached_prompt_tokens = max(int(summary.get("cached_prompt_tokens", 0) or 0), 0)
+        cost = self.calculate_cost(
+            model=model,
+            prompt_tokens=input_tokens,
+            completion_tokens=output_tokens,
+            cached_prompt_tokens=cached_prompt_tokens,
+        )
+        return {
+            "model": model,
+            "inputTokens": input_tokens,
+            "outputTokens": output_tokens,
+            "cost": cost,
+            "latencyMs": latency_ms,
+        }
+
    def run_completion_with_cost(
        self,
        *,