Standardize LLM clients and expose configurable base URLs.

2025-10-09 02:23:08 +08:00
parent 960608cee9
commit 49d4893780
22 changed files with 189 additions and 62 deletions
@@ -72,7 +72,8 @@ class DeepSearchAgent:
        if self.config.default_llm_provider == "deepseek":
            return DeepSeekLLM(
                api_key=self.config.deepseek_api_key,
-                model_name=self.config.deepseek_model
+                model_name=self.config.deepseek_model,
+                base_url=self.config.deepseek_base_url
            )
        elif self.config.default_llm_provider == "openai":
            return OpenAILLM(
@@ -82,7 +83,8 @@ class DeepSearchAgent:
        elif self.config.default_llm_provider == "kimi":
            return KimiLLM(
                api_key=self.config.kimi_api_key,
-                model_name=self.config.kimi_model
+                model_name=self.config.kimi_model,
+                base_url=self.config.kimi_base_url
            )
        else:
            raise ValueError(f"不支持的LLM提供商: {self.config.default_llm_provider}")