fix [功能请求]: GPT-4 Turbo模型的Token长度限制似乎有问题？ #953

modules/models/OpenAI.py

@@ -25,8 +25,6 @@ class OpenAIClient(BaseLLMModel):
         top_p=1.0,
         user_name=""
     ) -> None:
-        if model_name in MODEL_METADATA:
-            model_name = MODEL_METADATA[model_name]["model_name"]
         super().__init__(
             model_name=model_name,
             temperature=temperature,

modules/models/OpenAIVision.py

@@ -32,7 +32,7 @@ class OpenAIVisionClient(BaseLLMModel):
         user_name=""
     ) -> None:
         super().__init__(
-            model_name=MODEL_METADATA[model_name]["model_name"],
+            model_name=model_name,
             temperature=temperature,
             top_p=top_p,
             system_prompt=system_prompt,

modules/models/base_model.py

@@ -213,7 +213,10 @@ class BaseLLMModel:
     ) -> None:
         self.history = []
         self.all_token_counts = []
-        self.model_name = model_name
+        if model_name in MODEL_METADATA:
+            self.model_name = MODEL_METADATA[model_name]["model_name"]
+        else:
+            self.model_name = model_name
         self.model_type = ModelType.get_type(model_name)
         try:
             self.token_upper_limit = MODEL_METADATA[model_name]["token_limit"]