chore: removing unused imports in tests (#9049)

2024-10-08 11:13:11 +08:00
parent a8b4d1ac2a
commit 7c0b159a81
30 changed files with 21 additions and 52 deletions
--- a/api/tests/integration_tests/model_runtime/azure_ai_studio/test_llm.py
+++ b/api/tests/integration_tests/model_runtime/azure_ai_studio/test_llm.py
@@ -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