This commit is contained in:
khazic 2024-03-26 16:00:40 +08:00
parent 43a19d9b66
commit 5b90286843
5 changed files with 30 additions and 30 deletions

View File

@ -11,7 +11,7 @@ if __name__ == '__main__':
provider_manager = ModelManager( provider_manager = ModelManager(
provider_name_to_provider_records_dict={ provider_name_to_provider_records_dict={
'openai': { 'openai': {
'openai_api_key': "sk- ", 'openai_api_key': "sk-4M9LYF",
} }
}, },
provider_name_to_provider_model_records_dict={} provider_name_to_provider_model_records_dict={}

View File

@ -7,18 +7,18 @@ from typing import Optional
from pydantic import BaseModel from pydantic import BaseModel
from model_providers.core.entities.model_entities import ModelStatus, ModelWithProviderEntity, SimpleModelProviderEntity from chatchat_model_providers.core.entities.model_entities import ModelStatus, ModelWithProviderEntity, SimpleModelProviderEntity
from model_providers.core.entities.provider_entities import CustomConfiguration from chatchat_model_providers.core.entities.provider_entities import CustomConfiguration
from model_providers.core.model_runtime.entities.model_entities import FetchFrom, ModelType from chatchat_model_providers.core.model_runtime.entities.model_entities import FetchFrom, ModelType
from model_providers.core.model_runtime.entities.provider_entities import ( from chatchat_model_providers.core.model_runtime.entities.provider_entities import (
ConfigurateMethod, ConfigurateMethod,
CredentialFormSchema, CredentialFormSchema,
FormType, FormType,
ProviderEntity, ProviderEntity,
) )
from model_providers.core.model_runtime.model_providers import model_provider_factory from chatchat_model_providers.core.model_runtime.model_providers import model_provider_factory
from model_providers.core.model_runtime.model_providers.__base.ai_model import AIModel from chatchat_model_providers.core.model_runtime.model_providers.__base.ai_model import AIModel
from model_providers.core.model_runtime.model_providers.__base.model_provider import ModelProvider from chatchat_model_providers.core.model_runtime.model_providers.__base.model_provider import ModelProvider
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)

View File

@ -3,7 +3,7 @@ from typing import Optional
from pydantic import BaseModel from pydantic import BaseModel
from model_providers.core.model_runtime.entities.model_entities import ModelType from chatchat_model_providers.core.model_runtime.entities.model_entities import ModelType
class RestrictModel(BaseModel): class RestrictModel(BaseModel):

View File

@ -1,21 +1,21 @@
from collections.abc import Generator from collections.abc import Generator
from typing import IO, Optional, Union, cast from typing import IO, Optional, Union, cast
from model_providers.core.entities.provider_configuration import ProviderModelBundle from chatchat_model_providers.core.entities.provider_configuration import ProviderModelBundle
from model_providers.core.errors.error import ProviderTokenNotInitError from chatchat_model_providers.errors.error import ProviderTokenNotInitError
from model_providers.core.model_runtime.callbacks.base_callback import Callback from chatchat_model_providers.core.model_runtime.callbacks.base_callback import Callback
from model_providers.core.model_runtime.entities.llm_entities import LLMResult from chatchat_model_providers.core.model_runtime.entities.llm_entities import LLMResult
from model_providers.core.model_runtime.entities.message_entities import PromptMessage, PromptMessageTool from chatchat_model_providers.core.model_runtime.entities.message_entities import PromptMessage, PromptMessageTool
from model_providers.core.model_runtime.entities.model_entities import ModelType from chatchat_model_providers.core.model_runtime.entities.model_entities import ModelType
from model_providers.core.model_runtime.entities.rerank_entities import RerankResult from chatchat_model_providers.core.model_runtime.entities.rerank_entities import RerankResult
from model_providers.core.model_runtime.entities.text_embedding_entities import TextEmbeddingResult from chatchat_model_providers.core.model_runtime.entities.text_embedding_entities import TextEmbeddingResult
from model_providers.core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel from chatchat_model_providers.core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
from model_providers.core.model_runtime.model_providers.__base.moderation_model import ModerationModel from chatchat_model_providers.core.model_runtime.model_providers.__base.moderation_model import ModerationModel
from model_providers.core.model_runtime.model_providers.__base.rerank_model import RerankModel from chatchat_model_providers.core.model_runtime.model_providers.__base.rerank_model import RerankModel
from model_providers.core.model_runtime.model_providers.__base.speech2text_model import Speech2TextModel from chatchat_model_providers.core.model_runtime.model_providers.__base.speech2text_model import Speech2TextModel
from model_providers.core.model_runtime.model_providers.__base.text_embedding_model import TextEmbeddingModel from chatchat_model_providers.core.model_runtime.model_providers.__base.text_embedding_model import TextEmbeddingModel
from model_providers.core.model_runtime.model_providers.__base.tts_model import TTSModel from chatchat_model_providers.core.model_runtime.model_providers.__base.tts_model import TTSModel
from model_providers.core.provider_manager import ProviderManager from chatchat_model_providers.core.provider_manager import ProviderManager
def _fetch_credentials_from_bundle(provider_model_bundle: ProviderModelBundle, model: str) -> dict: def _fetch_credentials_from_bundle(provider_model_bundle: ProviderModelBundle, model: str) -> dict:

View File

@ -5,21 +5,21 @@ from typing import Optional
from sqlalchemy.exc import IntegrityError from sqlalchemy.exc import IntegrityError
from model_providers.core.entities.model_entities import DefaultModelEntity, DefaultModelProviderEntity from chatchat_model_providers.core.entities.model_entities import DefaultModelEntity, DefaultModelProviderEntity
from model_providers.core.entities.provider_configuration import ProviderConfiguration, ProviderConfigurations, \ from chatchat_model_providers.core.entities.provider_configuration import ProviderConfiguration, ProviderConfigurations, \
ProviderModelBundle ProviderModelBundle
from model_providers.core.entities.provider_entities import ( from chatchat_model_providers.core.entities.provider_entities import (
CustomConfiguration, CustomConfiguration,
CustomModelConfiguration, CustomModelConfiguration,
CustomProviderConfiguration, CustomProviderConfiguration,
) )
from model_providers.core.model_runtime.entities.model_entities import ModelType from chatchat_model_providers.core.model_runtime.entities.model_entities import ModelType
from model_providers.core.model_runtime.entities.provider_entities import ( from chatchat_model_providers.core.model_runtime.entities.provider_entities import (
CredentialFormSchema, CredentialFormSchema,
FormType, FormType,
ProviderEntity, ProviderEntity,
) )
from model_providers.core.model_runtime.model_providers import model_provider_factory from chatchat_model_providers.core.model_runtime.model_providers import model_provider_factory
class ProviderManager: class ProviderManager: