From 32974f8e2fd50ab2daaaa7c04ff8ea0ec1febb68 Mon Sep 17 00:00:00 2001
From: glide-the <2533736852@qq.com>
Date: Thu, 28 Mar 2024 09:11:08 +0800
Subject: [PATCH] =?UTF-8?q?=E5=B0=81=E8=A3=85ModelManager?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 model-providers/model_providers/__init__.py     | 17 +++++++++++++++++
 .../core/bootstrap/openai_protocol.py           |  2 +-
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/model-providers/model_providers/__init__.py b/model-providers/model_providers/__init__.py
index e69de29b..486e453c 100644
--- a/model-providers/model_providers/__init__.py
+++ b/model-providers/model_providers/__init__.py
@@ -0,0 +1,17 @@
+from chatchat.configs import MODEL_PLATFORMS
+from model_providers.core.model_manager import ModelManager
+
+def _to_custom_provide_configuration():
+    provider_name_to_provider_records_dict = {}
+    provider_name_to_provider_model_records_dict = {}
+    return provider_name_to_provider_records_dict, provider_name_to_provider_model_records_dict
+
+# 基于配置管理器创建的模型实例
+provider_manager = ModelManager(
+    provider_name_to_provider_records_dict={
+        'openai': {
+            'openai_api_key': "sk-4M9LYF",
+        }
+    },
+    provider_name_to_provider_model_records_dict={}
+)
\ No newline at end of file
diff --git a/model-providers/model_providers/core/bootstrap/openai_protocol.py b/model-providers/model_providers/core/bootstrap/openai_protocol.py
index 4d1b2c52..690475fd 100644
--- a/model-providers/model_providers/core/bootstrap/openai_protocol.py
+++ b/model-providers/model_providers/core/bootstrap/openai_protocol.py
@@ -1,7 +1,7 @@
 import time
 from enum import Enum
 from typing import Any, Dict, List, Optional, Union
-from pydantic.v1 import BaseModel, Field, root_validator
+from pydantic import BaseModel, Field, root_validator
 from typing_extensions import Literal