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
@@ -60,7 +60,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(
@@ -70,7 +71,8 @@ class DeepSearchAgent:
        elif self.config.default_llm_provider == "gemini":
            return GeminiLLM(
                api_key=self.config.gemini_api_key,
-                model_name=self.config.gemini_model
+                model_name=self.config.gemini_model,
+                base_url=self.config.gemini_base_url
            )
        else:
            raise ValueError(f"不支持的LLM提供商: {self.config.default_llm_provider}")