Merge pull request #1586 from rahultayal22/rah_google_vertex_issue

Fixed params issue in Google Vertex ai
This commit is contained in:
Mark Backman
2025-04-16 14:56:46 -04:00
committed by GitHub
2 changed files with 6 additions and 1 deletions

View File

@@ -65,7 +65,9 @@ class GoogleVertexLLMService(OpenAILLMService):
base_url = self._get_base_url(params)
self._api_key = self._get_api_token(credentials, credentials_path)
super().__init__(api_key=self._api_key, base_url=base_url, model=model, **kwargs)
super().__init__(
api_key=self._api_key, base_url=base_url, model=model, params=params, **kwargs
)
@staticmethod
def _get_base_url(params: InputParams) -> str: