From ae9cd3e6346b7dc9d95f01da6fa6029d32ff4770 Mon Sep 17 00:00:00 2001
From: seehi <6580@pm.me>
Date: Thu, 27 Feb 2025 10:54:25 +0800
Subject: [PATCH] support gemini models

---
 metagpt/provider/google_gemini_api.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)