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