mirror of
https://github.com/Alishahryar1/free-claude-code.git
synced 2026-07-03 14:05:26 +02:00
Move nim_settings from shared base class to NvidiaNimProvider (#78)
This commit is contained in:
@@ -20,10 +20,9 @@ class NvidiaNimProvider(OpenAICompatibleProvider):
|
||||
provider_name="NIM",
|
||||
base_url=config.base_url or NVIDIA_NIM_BASE_URL,
|
||||
api_key=config.api_key,
|
||||
nim_settings=nim_settings,
|
||||
)
|
||||
self._nim_settings = nim_settings
|
||||
|
||||
def _build_request_body(self, request: Any) -> dict:
|
||||
"""Internal helper for tests and shared building."""
|
||||
assert self._nim_settings is not None
|
||||
return build_request_body(request, self._nim_settings)
|
||||
|
||||
@@ -34,13 +34,11 @@ class OpenAICompatibleProvider(BaseProvider):
|
||||
provider_name: str,
|
||||
base_url: str,
|
||||
api_key: str,
|
||||
nim_settings: Any = None,
|
||||
):
|
||||
super().__init__(config)
|
||||
self._provider_name = provider_name
|
||||
self._api_key = api_key
|
||||
self._base_url = base_url.rstrip("/")
|
||||
self._nim_settings = nim_settings
|
||||
self._global_rate_limiter = GlobalRateLimiter.get_instance(
|
||||
rate_limit=config.rate_limit,
|
||||
rate_window=config.rate_window,
|
||||
|
||||
Reference in New Issue
Block a user