torbjorn/my-pal-mcp-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: simplify Gemini provider initialization using kwargs dict

Browse Source 
As suggested by code review, this reduces code duplication and improves maintainability
This commit is contained in:
dragonfsky
2025-08-22 18:37:11 +08:00
parent 956e8a6927
commit 023940be3e
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
providers/registry.py
View File

@@ -98,13 +98,11 @@ class ModelProviderRegistry:
if not api_key: if not api_key:
return None return None
gemini_base_url = os.getenv("GEMINI_BASE_URL") gemini_base_url = os.getenv("GEMINI_BASE_URL")
provider_kwargs = {"api_key": api_key}
if gemini_base_url: if gemini_base_url:
# Initialize with custom endpoint provider_kwargs["base_url"] = gemini_base_url
provider = provider_class(api_key=api_key, base_url=gemini_base_url)
logging.info(f"Initialized Gemini provider with custom endpoint: {gemini_base_url}") logging.info(f"Initialized Gemini provider with custom endpoint: {gemini_base_url}")
else: provider = provider_class(**provider_kwargs)
# Use default Google endpoint
provider = provider_class(api_key=api_key)
else: else:
if not api_key: if not api_key:
return None return None