Merge pull request #1725 from seehi/fix-gemini-model

Support gemini models
This commit is contained in:
better629 2025-03-10 01:46:34 +08:00 committed by GitHub
commit de368d818f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -50,7 +50,7 @@ class GeminiLLM(BaseLLM):
self.__init_gemini(config)
self.config = config
self.model = "gemini-pro" # so far only one model
self.model = config.model
self.pricing_plan = self.config.pricing_plan or self.model
self.llm = GeminiGenerativeModel(model_name=self.model)