diff --git a/metagpt/provider/google_gemini_api.py b/metagpt/provider/google_gemini_api.py index 5c1b92503..de534216d 100644 --- a/metagpt/provider/google_gemini_api.py +++ b/metagpt/provider/google_gemini_api.py @@ -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)