增加extra_params功能

2025-07-31 01:04:14 +08:00
parent 82b5230df1
commit 42a33a406e
4 changed files with 15 additions and 1 deletions
--- a/src/llm_models/model_client/base_client.py
+++ b/src/llm_models/model_client/base_client.py
@@ -83,6 +83,7 @@ class BaseClient:
        | None = None,
        async_response_parser: Callable[[ChatCompletion], tuple[APIResponse, tuple[int, int, int]]] | None = None,
        interrupt_flag: asyncio.Event | None = None,
+        extra_params: dict[str, Any] | None = None,
    ) -> APIResponse:
        """
        获取对话响应
@@ -103,6 +104,7 @@ class BaseClient:
        self,
        model_info: ModelInfo,
        embedding_input: str,
+        extra_params: dict[str, Any] | None = None,
    ) -> APIResponse:
        """
        获取文本嵌入