support gemini models

This commit is contained in:
seehi 2025-02-27 10:54:25 +08:00
parent 07d4be2df3
commit ae9cd3e634

View file

@ -51,7 +51,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)