From 023940be3e38a7eedbc8bf8404a4a5afc50f8398 Mon Sep 17 00:00:00 2001 From: dragonfsky Date: Fri, 22 Aug 2025 18:37:11 +0800 Subject: [PATCH] refactor: simplify Gemini provider initialization using kwargs dict As suggested by code review, this reduces code duplication and improves maintainability --- providers/registry.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/providers/registry.py b/providers/registry.py index b4d4cb1..7a1b94e 100644 --- a/providers/registry.py +++ b/providers/registry.py @@ -98,13 +98,11 @@ class ModelProviderRegistry: if not api_key: return None gemini_base_url = os.getenv("GEMINI_BASE_URL") + provider_kwargs = {"api_key": api_key} if gemini_base_url: - # Initialize with custom endpoint - provider = provider_class(api_key=api_key, base_url=gemini_base_url) + provider_kwargs["base_url"] = gemini_base_url logging.info(f"Initialized Gemini provider with custom endpoint: {gemini_base_url}") - else: - # Use default Google endpoint - provider = provider_class(api_key=api_key) + provider = provider_class(**provider_kwargs) else: if not api_key: return None