diff --git a/model-providers/model_providers/core/model_runtime/model_providers/cohere/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/cohere/llm/llm.py index fde1a2a1..a2031792 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/cohere/llm/llm.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/cohere/llm/llm.py @@ -1,5 +1,5 @@ import logging -from collections.abc import Generator +from typing import Generator from typing import Dict, List, Optional, Type, Union, cast import cohere diff --git a/model-providers/model_providers/core/model_runtime/model_providers/deepseek/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/deepseek/llm/llm.py index de99ed6c..5a68da78 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/deepseek/llm/llm.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/deepseek/llm/llm.py @@ -1,5 +1,5 @@ import logging -from collections.abc import Generator +from typing import Generator from typing import List, Optional, Union, cast from decimal import Decimal diff --git a/model-providers/model_providers/core/model_runtime/model_providers/minimax/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/minimax/llm/llm.py index aea0f9e1..8ff55446 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/minimax/llm/llm.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/minimax/llm/llm.py @@ -1,4 +1,4 @@ -from collections.abc import Generator +from typing import Generator from model_providers.core.model_runtime.entities.llm_entities import ( LLMResult, diff --git a/model-providers/model_providers/core/model_runtime/model_providers/moonshot/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/moonshot/llm/llm.py index 8bd0f500..3300b3cc 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/moonshot/llm/llm.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/moonshot/llm/llm.py @@ -1,4 +1,4 @@ -from collections.abc import Generator +from typing import Generator from typing import List, Optional, Union from model_providers.core.model_runtime.entities.llm_entities import LLMResult diff --git a/model-providers/model_providers/core/model_runtime/model_providers/ollama/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/ollama/llm/llm.py index 1792fa80..328a5e5e 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/ollama/llm/llm.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/ollama/llm/llm.py @@ -1,5 +1,5 @@ import logging -from collections.abc import Generator +from typing import Generator from typing import List, Optional, Union, cast from decimal import Decimal diff --git a/model-providers/model_providers/core/model_runtime/model_providers/openai/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/openai/llm/llm.py index 230ad76e..e32399bd 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/openai/llm/llm.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/openai/llm/llm.py @@ -1,5 +1,5 @@ import logging -from collections.abc import Generator +from typing import Generator from typing import List, Optional, Union, cast import tiktoken diff --git a/model-providers/model_providers/core/model_runtime/model_providers/openai_api_compatible/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/openai_api_compatible/llm/llm.py index 014008dd..1267d01e 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/openai_api_compatible/llm/llm.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/openai_api_compatible/llm/llm.py @@ -1,6 +1,6 @@ import json import logging -from collections.abc import Generator +from typing import Generator from decimal import Decimal from typing import List, Optional, Union, cast from urllib.parse import urljoin diff --git a/model-providers/model_providers/core/model_runtime/model_providers/openllm/llm/openllm_generate.py b/model-providers/model_providers/core/model_runtime/model_providers/openllm/llm/openllm_generate.py index 1c0ab40d..a45aac4b 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/openllm/llm/openllm_generate.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/openllm/llm/openllm_generate.py @@ -1,4 +1,4 @@ -from collections.abc import Generator +from typing import Generator from enum import Enum from json import dumps, loads from typing import Any, Union diff --git a/model-providers/model_providers/core/model_runtime/model_providers/spark/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/spark/llm/llm.py index 02741fd7..cc4ee81e 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/spark/llm/llm.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/spark/llm/llm.py @@ -1,5 +1,5 @@ import threading -from collections.abc import Generator +from typing import Generator from typing import Dict, List, Optional, Type, Union from model_providers.core.model_runtime.entities.llm_entities import ( diff --git a/model-providers/model_providers/core/model_runtime/model_providers/togetherai/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/togetherai/llm/llm.py index 72b4be3c..2010ab34 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/togetherai/llm/llm.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/togetherai/llm/llm.py @@ -1,4 +1,4 @@ -from collections.abc import Generator +from typing import Generator from typing import List, Optional, Union from model_providers.core.model_runtime.entities.llm_entities import LLMResult diff --git a/model-providers/model_providers/core/model_runtime/model_providers/tongyi/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/tongyi/llm/llm.py index 058027cc..2edfb37c 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/tongyi/llm/llm.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/tongyi/llm/llm.py @@ -1,4 +1,4 @@ -from collections.abc import Generator +from typing import Generator from typing import Dict, List, Optional, Type, Union from dashscope import get_tokenizer diff --git a/model-providers/model_providers/core/model_runtime/model_providers/xinference/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/xinference/llm/llm.py index 8f405a35..68b2cf58 100644 --- a/model-providers/model_providers/core/model_runtime/model_providers/xinference/llm/llm.py +++ b/model-providers/model_providers/core/model_runtime/model_providers/xinference/llm/llm.py @@ -1,4 +1,5 @@ -from collections.abc import Generator, Iterator +from typing import Generator, Iterator + from typing import Dict, List, Union, cast, Type from openai import ( @@ -464,7 +465,7 @@ class XinferenceAILargeLanguageModel(LargeLanguageModel): ] if isinstance( - xinference_model, RESTfulChatModelHandle | RESTfulChatglmCppChatModelHandle + xinference_model, (RESTfulChatModelHandle, RESTfulChatglmCppChatModelHandle) ): resp = client.chat.completions.create( model=credentials["model_uid"], diff --git a/model-providers/model_providers/core/model_runtime/utils/encoders.py b/model-providers/model_providers/core/model_runtime/utils/encoders.py index cea96079..fe5836c8 100644 --- a/model-providers/model_providers/core/model_runtime/utils/encoders.py +++ b/model-providers/model_providers/core/model_runtime/utils/encoders.py @@ -15,7 +15,7 @@ from ipaddress import ( from pathlib import Path, PurePath from re import Pattern from types import GeneratorType -from typing import Any, Optional, Union +from typing import Any, Optional, Union, Dict, Type, List, Tuple from uuid import UUID from pydantic import BaseModel @@ -54,7 +54,7 @@ def decimal_encoder(dec_value: Decimal) -> Union[int, float]: return float(dec_value) -ENCODERS_BY_TYPE: Dict[type[Any], Callable[[Any], Any]] = { +ENCODERS_BY_TYPE: Dict[Type[Any], Callable[[Any], Any]] = { bytes: lambda o: o.decode(), Color: str, datetime.date: isoformat, @@ -86,8 +86,8 @@ ENCODERS_BY_TYPE: Dict[type[Any], Callable[[Any], Any]] = { def generate_encoders_by_class_tuples( type_encoder_map: Dict[Any, Callable[[Any], Any]], -) -> Dict[Callable[[Any], Any], tuple[Any, ...]]: - encoders_by_class_tuples: Dict[Callable[[Any], Any], tuple[Any, ...]] = defaultdict( +) -> Dict[Callable[[Any], Any], Tuple[Any, ...]]: + encoders_by_class_tuples: Dict[Callable[[Any], Any], Tuple[Any, ...]] = defaultdict( tuple ) for type_, encoder in type_encoder_map.items(): @@ -104,7 +104,7 @@ def jsonable_encoder( exclude_unset: bool = False, exclude_defaults: bool = False, exclude_none: bool = False, - custom_encoder: Optional[dict[Any, Callable[[Any], Any]]] = None, + custom_encoder: Optional[Dict[Any, Callable[[Any], Any]]] = None, sqlalchemy_safe: bool = True, ) -> Any: custom_encoder = custom_encoder or {} @@ -192,7 +192,7 @@ def jsonable_encoder( ) encoded_dict[encoded_key] = encoded_value return encoded_dict - if isinstance(obj, list | set | frozenset | GeneratorType | tuple | deque): + if isinstance(obj, (list, set, frozenset, GeneratorType, tuple, deque)): encoded_list = [] for item in obj: encoded_list.append( @@ -217,8 +217,7 @@ def jsonable_encoder( try: data = dict(obj) except Exception as e: - errors: List[Exception] = [] - errors.append(e) + errors: List[Exception] = [e] try: data = vars(obj) except Exception as e: diff --git a/model-providers/model_providers/extensions/ext_storage.py b/model-providers/model_providers/extensions/ext_storage.py index 854fa824..fcf863a6 100644 --- a/model-providers/model_providers/extensions/ext_storage.py +++ b/model-providers/model_providers/extensions/ext_storage.py @@ -1,6 +1,6 @@ import os import shutil -from collections.abc import Generator +from typing import Generator from contextlib import closing from typing import Union diff --git a/model-providers/tests/unit_tests/deepseek/test_provider_manager_models.py b/model-providers/tests/unit_tests/deepseek/test_deepseek_provider_manager_models.py similarity index 100% rename from model-providers/tests/unit_tests/deepseek/test_provider_manager_models.py rename to model-providers/tests/unit_tests/deepseek/test_deepseek_provider_manager_models.py diff --git a/model-providers/tests/unit_tests/ollama/test_provider_manager_models.py b/model-providers/tests/unit_tests/ollama/test_ollama_provider_manager_models.py similarity index 100% rename from model-providers/tests/unit_tests/ollama/test_provider_manager_models.py rename to model-providers/tests/unit_tests/ollama/test_ollama_provider_manager_models.py diff --git a/model-providers/tests/unit_tests/openai/test_provider_manager_models.py b/model-providers/tests/unit_tests/openai/test_openai_provider_manager_models.py similarity index 100% rename from model-providers/tests/unit_tests/openai/test_provider_manager_models.py rename to model-providers/tests/unit_tests/openai/test_openai_provider_manager_models.py diff --git a/model-providers/tests/unit_tests/xinference/test_provider_manager_models.py b/model-providers/tests/unit_tests/xinference/test_xinference_provider_manager_models.py similarity index 100% rename from model-providers/tests/unit_tests/xinference/test_provider_manager_models.py rename to model-providers/tests/unit_tests/xinference/test_xinference_provider_manager_models.py diff --git a/model-providers/tests/unit_tests/zhipuai/test_provider_manager_models.py b/model-providers/tests/unit_tests/zhipuai/test_zhipuai_provider_manager_models.py similarity index 100% rename from model-providers/tests/unit_tests/zhipuai/test_provider_manager_models.py rename to model-providers/tests/unit_tests/zhipuai/test_zhipuai_provider_manager_models.py