estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

chore: removing unused imports in tests (#9049)

Browse Source
This commit is contained in:
Bowen Liang
2024-10-08 11:13:11 +08:00
committed by GitHub
parent a8b4d1ac2a
commit 7c0b159a81
30 changed files with 21 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -5,7 +5,7 @@ from typing import Any, Literal, Union
import anthropic
import pytest
from _pytest.monkeypatch import MonkeyPatch
from anthropic import Anthropic, Stream
from anthropic import Stream
from anthropic.resources import Messages
from anthropic.types import (
ContentBlock,

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

@@ -1,6 +1,6 @@
import os
from collections.abc import Callable
from typing import Any, Literal, Union
from typing import Any, Literal
import pytest

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

@@ -1,6 +1,6 @@
import re
from collections.abc import Generator
from json import dumps, loads
from json import dumps
from time import time
# import monkeypatch
@@ -11,11 +11,9 @@ from openai._types import NOT_GIVEN, NotGiven
from openai.resources.chat.completions import Completions
from openai.types import Completion as CompletionMessage
from openai.types.chat import (
ChatCompletion,
ChatCompletionChunk,
ChatCompletionMessageParam,
ChatCompletionMessageToolCall,
ChatCompletionToolChoiceOptionParam,
ChatCompletionToolParam,
completion_create_params,
)

1
api/tests/integration_tests/model_runtime/__mock/openai_embeddings.py
View File

@@ -1,7 +1,6 @@
import re
from typing import Any, Literal, Union
from openai import OpenAI
from openai._types import NOT_GIVEN, NotGiven
from openai.resources.embeddings import Embeddings
from openai.types.create_embedding_response import CreateEmbeddingResponse, Usage

1
api/tests/integration_tests/model_runtime/__mock/openai_remote.py
View File

@@ -1,6 +1,5 @@
from time import time
from openai.resources.models import Models
from openai.types.model import Model

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

@@ -5,7 +5,6 @@ from typing import Union
import pytest
from _pytest.monkeypatch import MonkeyPatch
from requests import Response
from requests.exceptions import ConnectionError
from requests.sessions import Session
from xinference_client.client.restful.restful_client import (
Client,

3
api/tests/integration_tests/model_runtime/azure_ai_studio/test_llm.py
View File

@@ -6,10 +6,7 @@ import pytest
from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
from core.model_runtime.entities.message_entities import (
AssistantPromptMessage,
ImagePromptMessageContent,
PromptMessageTool,
SystemPromptMessage,
TextPromptMessageContent,
UserPromptMessage,
)
from core.model_runtime.errors.validate import CredentialsValidateFailedError

1
api/tests/integration_tests/model_runtime/chatglm/test_llm.py
View File

@@ -8,7 +8,6 @@ from core.model_runtime.entities.message_entities import (
AssistantPromptMessage,
PromptMessageTool,
SystemPromptMessage,
TextPromptMessageContent,
UserPromptMessage,
)
from core.model_runtime.entities.model_entities import AIModelEntity

3
api/tests/integration_tests/model_runtime/huggingface_tei/test_rerank.py
View File

@@ -2,8 +2,7 @@ import os
import pytest
from core.model_runtime.entities.rerank_entities import RerankDocument, RerankResult
from core.model_runtime.entities.text_embedding_entities import TextEmbeddingResult
from core.model_runtime.entities.rerank_entities import RerankResult
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.huggingface_tei.rerank.rerank import (
HuggingfaceTeiRerankModel,

2
api/tests/integration_tests/model_runtime/localai/test_llm.py
View File

@@ -8,10 +8,8 @@ from core.model_runtime.entities.message_entities import (
AssistantPromptMessage,
PromptMessageTool,
SystemPromptMessage,
TextPromptMessageContent,
UserPromptMessage,
)
from core.model_runtime.entities.model_entities import ParameterRule
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.localai.llm.llm import LocalAILanguageModel

1
api/tests/integration_tests/model_runtime/nomic/test_provider.py
View File

@@ -4,7 +4,6 @@ import pytest
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.nomic.nomic import NomicAtlasProvider
from core.model_runtime.model_providers.nomic.text_embedding.text_embedding import NomicTextEmbeddingModel
from tests.integration_tests.model_runtime.__mock.nomic_embeddings import setup_nomic_mock

1
api/tests/integration_tests/model_runtime/novita/test_llm.py
View File

@@ -6,7 +6,6 @@ import pytest
from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
from core.model_runtime.entities.message_entities import (
AssistantPromptMessage,
PromptMessageTool,
SystemPromptMessage,
UserPromptMessage,
)

1
api/tests/integration_tests/model_runtime/oci/test_llm.py
View File

@@ -8,7 +8,6 @@ from core.model_runtime.entities.message_entities import (
AssistantPromptMessage,
PromptMessageTool,
SystemPromptMessage,
TextPromptMessageContent,
UserPromptMessage,
)
from core.model_runtime.errors.validate import CredentialsValidateFailedError

1
api/tests/integration_tests/model_runtime/openai/test_llm.py
View File

@@ -14,7 +14,6 @@ from core.model_runtime.entities.message_entities import (
)
from core.model_runtime.entities.model_entities import AIModelEntity, ModelType
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
from core.model_runtime.model_providers.openai.llm.llm import OpenAILargeLanguageModel
"""FOR MOCK FIXTURES, DO NOT REMOVE"""

1
api/tests/integration_tests/model_runtime/openrouter/test_llm.py
View File

@@ -6,7 +6,6 @@ import pytest
from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
from core.model_runtime.entities.message_entities import (
AssistantPromptMessage,
PromptMessageTool,
SystemPromptMessage,
UserPromptMessage,
)

2
api/tests/integration_tests/model_runtime/sagemaker/test_provider.py
View File

@@ -1,5 +1,3 @@
import os
import pytest
from core.model_runtime.errors.validate import CredentialsValidateFailedError

2
api/tests/integration_tests/model_runtime/sagemaker/test_text_embedding.py
View File

@@ -1,5 +1,3 @@
import os
import pytest
from core.model_runtime.entities.text_embedding_entities import TextEmbeddingResult

4
api/tests/integration_tests/model_runtime/stepfun/test_llm.py
View File

@@ -6,13 +6,11 @@ import pytest
from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
from core.model_runtime.entities.message_entities import (
AssistantPromptMessage,
ImagePromptMessageContent,
PromptMessageTool,
SystemPromptMessage,
TextPromptMessageContent,
UserPromptMessage,
)
from core.model_runtime.entities.model_entities import AIModelEntity, ModelType
from core.model_runtime.entities.model_entities import AIModelEntity
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.stepfun.llm.llm import StepfunLargeLanguageModel

1
api/tests/integration_tests/model_runtime/togetherai/test_llm.py
View File

@@ -6,7 +6,6 @@ import pytest
from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
from core.model_runtime.entities.message_entities import (
AssistantPromptMessage,
PromptMessageTool,
SystemPromptMessage,
UserPromptMessage,
)

3
api/tests/integration_tests/model_runtime/upstage/test_llm.py
View File

@@ -10,9 +10,8 @@ from core.model_runtime.entities.message_entities import (
SystemPromptMessage,
UserPromptMessage,
)
from core.model_runtime.entities.model_entities import AIModelEntity, ModelType
from core.model_runtime.entities.model_entities import AIModelEntity
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
from core.model_runtime.model_providers.upstage.llm.llm import UpstageLargeLanguageModel
"""FOR MOCK FIXTURES, DO NOT REMOVE"""

2
api/tests/integration_tests/model_runtime/xinference/test_llm.py
View File

@@ -8,10 +8,8 @@ from core.model_runtime.entities.message_entities import (
AssistantPromptMessage,
PromptMessageTool,
SystemPromptMessage,
TextPromptMessageContent,
UserPromptMessage,
)
from core.model_runtime.entities.model_entities import AIModelEntity
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.xinference.llm.llm import XinferenceAILargeLanguageModel

1
api/tests/integration_tests/vdb/pgvector/test_pgvector.py
View File

@@ -1,5 +1,4 @@
from core.rag.datasource.vdb.pgvector.pgvector import PGVector, PGVectorConfig
from core.rag.models.document import Document
from tests.integration_tests.vdb.test_vector_store import (
AbstractVectorTest,
get_example_text,

1
api/tests/integration_tests/vdb/test_vector_store.py
View File

@@ -1,4 +1,3 @@
import random
import uuid
from unittest.mock import MagicMock

2
api/tests/integration_tests/workflow/nodes/__mock/code_executor.py
View File

@@ -1,5 +1,5 @@
import os
from typing import Literal, Optional
from typing import Literal
import pytest
from _pytest.monkeypatch import MonkeyPatch

1
api/tests/integration_tests/workflow/nodes/test_parameter_extractor.py
View File

@@ -1,4 +1,3 @@
import json
import os
import time
import uuid