feat(agent): 增强多模态链路与工具调用能力

2026-03-12 00:18:45 +08:00
parent 18db6c50e7
commit 21ba8e4a44
35 changed files with 2057 additions and 829 deletions
@@ -126,21 +126,26 @@ class LiteLLMService:
        temperature: float | None = None,
        max_tokens: int | None = None,
        timeout: float | None = None,
+        response_format: dict[str, Any] | None = None,
        completion_fn: Callable[..., dict[str, Any]] | None = None,
    ) -> LiteLLMResponseWithCost:
        caller = completion_fn or completion
        request_model = model if model.startswith("openai/") else f"openai/{model}"

-        response_any = caller(
-            model=request_model,
-            api_key=self.proxy_api_key,
-            api_base=self.proxy_base_url,
-            messages=messages,
-            temperature=temperature,
-            max_tokens=max_tokens,
-            timeout=timeout,
-            stream=False,
-        )
+        request_kwargs: dict[str, Any] = {
+            "model": request_model,
+            "api_key": self.proxy_api_key,
+            "api_base": self.proxy_base_url,
+            "messages": messages,
+            "temperature": temperature,
+            "max_tokens": max_tokens,
+            "timeout": timeout,
+            "stream": False,
+        }
+        if response_format is not None:
+            request_kwargs["response_format"] = response_format
+
+        response_any = caller(**request_kwargs)
        response = self._normalize_response(response_any)

        usage_raw = response.get("usage")