jprdonnelly/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-02-12 22:01:20 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat: mypy for all type check (#10921)

Browse Source
This commit is contained in:
yihong
2024-12-24 18:38:51 +08:00
committed by GitHub
parent c91e8b1737
commit 56e15d09a9
584 changed files with 3975 additions and 2826 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/tests/artifact_tests/dependencies/test_dependencies_sorted.py
View File

@@ -1,6 +1,6 @@
from typing import Any
import toml
import toml # type: ignore
def load_api_poetry_configs() -> dict[str, Any]:
@@ -38,7 +38,7 @@ def test_group_dependencies_version_operator():
)
def test_duplicated_dependency_crossing_groups():
def test_duplicated_dependency_crossing_groups() -> None:
all_dependency_names: list[str] = []
for dependencies in load_all_dependency_groups().values():
dependency_names = list(dependencies.keys())

2
api/tests/integration_tests/controllers/test_controllers.py
View File

@@ -1,6 +1,6 @@
from unittest.mock import patch
from app_fixture import app, mock_user
from app_fixture import mock_user # type: ignore
def test_post_requires_login(app):

4
api/tests/integration_tests/model_runtime/__mock/google.py
View File

@@ -1,7 +1,7 @@
from collections.abc import Generator
from unittest.mock import MagicMock
import google.generativeai.types.generation_types as generation_config_types
import google.generativeai.types.generation_types as generation_config_types # type: ignore
import pytest
from _pytest.monkeypatch import MonkeyPatch
from google.ai import generativelanguage as glm
@@ -45,7 +45,7 @@ class MockGoogleClass:
return GenerateContentResponse(done=True, iterator=None, result=glm.GenerateContentResponse({}), chunks=[])
@staticmethod
def generate_content_stream() -> Generator[GenerateContentResponse, None, None]:
def generate_content_stream() -> MockGoogleResponseClass:
return MockGoogleResponseClass()
def generate_content(

2
api/tests/integration_tests/model_runtime/__mock/huggingface.py
View File

@@ -2,7 +2,7 @@ import os
import pytest
from _pytest.monkeypatch import MonkeyPatch
from huggingface_hub import InferenceClient
from huggingface_hub import InferenceClient # type: ignore
from tests.integration_tests.model_runtime.__mock.huggingface_chat import MockHuggingfaceChatClass

6
api/tests/integration_tests/model_runtime/__mock/huggingface_chat.py
View File

@@ -3,15 +3,15 @@ from collections.abc import Generator
from typing import Any, Literal, Optional, Union
from _pytest.monkeypatch import MonkeyPatch
from huggingface_hub import InferenceClient
from huggingface_hub.inference._text_generation import (
from huggingface_hub import InferenceClient # type: ignore
from huggingface_hub.inference._text_generation import ( # type: ignore
Details,
StreamDetails,
TextGenerationResponse,
TextGenerationStreamResponse,
Token,
)
from huggingface_hub.utils import BadRequestError
from huggingface_hub.utils import BadRequestError # type: ignore
class MockHuggingfaceChatClass:

2
api/tests/integration_tests/model_runtime/__mock/nomic_embeddings.py
View File

@@ -6,7 +6,7 @@ import pytest
# import monkeypatch
from _pytest.monkeypatch import MonkeyPatch
from nomic import embed
from nomic import embed # type: ignore
def create_embedding(texts: list[str], model: str, **kwargs: Any) -> dict:

4
api/tests/integration_tests/model_runtime/__mock/xinference.py
View File

@@ -6,14 +6,14 @@ import pytest
from _pytest.monkeypatch import MonkeyPatch
from requests import Response
from requests.sessions import Session
from xinference_client.client.restful.restful_client import (
from xinference_client.client.restful.restful_client import ( # type: ignore
Client,
RESTfulChatModelHandle,
RESTfulEmbeddingModelHandle,
RESTfulGenerateModelHandle,
RESTfulRerankModelHandle,
)
from xinference_client.types import Embedding, EmbeddingData, EmbeddingUsage
from xinference_client.types import Embedding, EmbeddingData, EmbeddingUsage # type: ignore
class MockXinferenceClass:

2
api/tests/integration_tests/model_runtime/tongyi/test_rerank.py
View File

@@ -1,6 +1,6 @@
import os
import dashscope
import dashscope # type: ignore
import pytest
from core.model_runtime.entities.rerank_entities import RerankResult

2
api/tests/integration_tests/tools/__mock_server/openapi_todo.py
View File

@@ -1,5 +1,5 @@
from flask import Flask, request
from flask_restful import Api, Resource
from flask_restful import Api, Resource # type: ignore
app = Flask(__name__)
api = Api(app)

10
api/tests/integration_tests/vdb/__mock/baiduvectordb.py
View File

@@ -4,11 +4,11 @@ from unittest.mock import MagicMock
import pytest
from _pytest.monkeypatch import MonkeyPatch
from pymochow import MochowClient
from pymochow.model.database import Database
from pymochow.model.enum import IndexState, IndexType, MetricType, ReadConsistency, TableState
from pymochow.model.schema import HNSWParams, VectorIndex
from pymochow.model.table import Table
from pymochow import MochowClient # type: ignore
from pymochow.model.database import Database # type: ignore
from pymochow.model.enum import IndexState, IndexType, MetricType, ReadConsistency, TableState # type: ignore
from pymochow.model.schema import HNSWParams, VectorIndex # type: ignore
from pymochow.model.table import Table # type: ignore
from requests.adapters import HTTPAdapter

12
api/tests/integration_tests/vdb/__mock/tcvectordb.py
View File

@@ -4,12 +4,12 @@ from typing import Optional
import pytest
from _pytest.monkeypatch import MonkeyPatch
from requests.adapters import HTTPAdapter
from tcvectordb import VectorDBClient
from tcvectordb.model.database import Collection, Database
from tcvectordb.model.document import Document, Filter
from tcvectordb.model.enum import ReadConsistency
from tcvectordb.model.index import Index
from xinference_client.types import Embedding
from tcvectordb import VectorDBClient # type: ignore
from tcvectordb.model.database import Collection, Database # type: ignore
from tcvectordb.model.document import Document, Filter # type: ignore
from tcvectordb.model.enum import ReadConsistency # type: ignore
from tcvectordb.model.index import Index # type: ignore
from xinference_client.types import Embedding # type: ignore
class MockTcvectordbClass:

2
api/tests/integration_tests/vdb/__mock/vikingdb.py
View File

@@ -4,7 +4,7 @@ from unittest.mock import MagicMock
import pytest
from _pytest.monkeypatch import MonkeyPatch
from volcengine.viking_db import (
from volcengine.viking_db import ( # type: ignore
Collection,
Data,
DistanceType,

4
api/tests/unit_tests/oss/__mock/aliyun_oss.py
View File

@@ -4,8 +4,8 @@ from unittest.mock import MagicMock
import pytest
from _pytest.monkeypatch import MonkeyPatch
from oss2 import Bucket
from oss2.models import GetObjectResult, PutObjectResult
from oss2 import Bucket # type: ignore
from oss2.models import GetObjectResult, PutObjectResult # type: ignore
from tests.unit_tests.oss.__mock.base import (
get_example_bucket,

4
api/tests/unit_tests/oss/__mock/tencent_cos.py
View File

@@ -3,8 +3,8 @@ from unittest.mock import MagicMock
import pytest
from _pytest.monkeypatch import MonkeyPatch
from qcloud_cos import CosS3Client
from qcloud_cos.streambody import StreamBody
from qcloud_cos import CosS3Client # type: ignore
from qcloud_cos.streambody import StreamBody # type: ignore
from tests.unit_tests.oss.__mock.base import (
get_example_bucket,

4
api/tests/unit_tests/oss/__mock/volcengine_tos.py
View File

@@ -4,8 +4,8 @@ from unittest.mock import MagicMock
import pytest
from _pytest.monkeypatch import MonkeyPatch
from tos import TosClientV2
from tos.clientv2 import DeleteObjectOutput, GetObjectOutput, HeadObjectOutput, PutObjectOutput
from tos import TosClientV2 # type: ignore
from tos.clientv2 import DeleteObjectOutput, GetObjectOutput, HeadObjectOutput, PutObjectOutput # type: ignore
from tests.unit_tests.oss.__mock.base import (
get_example_bucket,

2
api/tests/unit_tests/oss/aliyun_oss/aliyun_oss/test_aliyun_oss.py
View File

@@ -1,7 +1,7 @@
from unittest.mock import MagicMock, patch
import pytest
from oss2 import Auth
from oss2 import Auth # type: ignore
from extensions.storage.aliyun_oss_storage import AliyunOssStorage
from tests.unit_tests.oss.__mock.aliyun_oss import setup_aliyun_oss_mock

2
api/tests/unit_tests/oss/tencent_cos/test_tencent_cos.py
View File

@@ -1,7 +1,7 @@
from unittest.mock import patch
import pytest
from qcloud_cos import CosConfig
from qcloud_cos import CosConfig # type: ignore
from extensions.storage.tencent_cos_storage import TencentCosStorage
from tests.unit_tests.oss.__mock.base import (

2
api/tests/unit_tests/oss/volcengine_tos/test_volcengine_tos.py
View File

@@ -1,5 +1,5 @@
import pytest
from tos import TosClientV2
from tos import TosClientV2 # type: ignore
from extensions.storage.volcengine_tos_storage import VolcengineTosStorage
from tests.unit_tests.oss.__mock.base import (

2
api/tests/unit_tests/utils/yaml/test_yaml_utils.py
View File

@@ -1,7 +1,7 @@
from textwrap import dedent
import pytest
from yaml import YAMLError
from yaml import YAMLError # type: ignore
from core.tools.utils.yaml_utils import load_yaml_file