support gemini models

2026-08-31 01:43:02 +08:00 · 2025-02-27 10:54:25 +08:00
parent 07d4be2df3
commit ae9cd3e634
1 changed files with 1 additions and 1 deletions
@@ -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)