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