From a83bb998f7480a6c5b9045fd349232c6add51323 Mon Sep 17 00:00:00 2001
From: Eric <eric@king.rocks>
Date: Fri, 4 Oct 2024 00:08:09 -0400
Subject: [PATCH] Remove reference to Gemini vision

---
 camel/model_backend.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/camel/model_backend.py b/camel/model_backend.py
index 897a26c..9d5f1de 100644
--- a/camel/model_backend.py
+++ b/camel/model_backend.py
@@ -175,8 +175,7 @@ class GeminiModel(ModelBackend):
         num_prompt_tokens = model.count_tokens(string).total_tokens
         num_max_token_map = {
             "gemini-1.5-pro": 1000000,
-            "gemini-1.5-flash": 1000000,
-            "gemini-pro-vision": 12288,
+            "gemini-1.5-flash": 1000000
         }
         num_max_token = num_max_token_map[self.model_type.value]
         num_max_completion_tokens = num_max_token - num_prompt_tokens