Skip to content

Commit 023940b

Browse files
committed
refactor: simplify Gemini provider initialization using kwargs dict
As suggested by code review, this reduces code duplication and improves maintainability
1 parent 956e8a6 commit 023940b

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

providers/registry.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -98,13 +98,11 @@ def get_provider(cls, provider_type: ProviderType, force_new: bool = False) -> O
9898
if not api_key:
9999
return None
100100
gemini_base_url = os.getenv("GEMINI_BASE_URL")
101+
provider_kwargs = {"api_key": api_key}
101102
if gemini_base_url:
102-
# Initialize with custom endpoint
103-
provider = provider_class(api_key=api_key, base_url=gemini_base_url)
103+
provider_kwargs["base_url"] = gemini_base_url
104104
logging.info(f"Initialized Gemini provider with custom endpoint: {gemini_base_url}")
105-
else:
106-
# Use default Google endpoint
107-
provider = provider_class(api_key=api_key)
105+
provider = provider_class(**provider_kwargs)
108106
else:
109107
if not api_key:
110108
return None

0 commit comments

Comments
 (0)