fix: remove unnecessary imports and format code.

langchain-ai · Jan 29, 2024 · 6e04433 · 6e04433
1 parent ca6e115
commit 6e04433
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 8 deletions.
diff --git a/libs/community/langchain_community/chat_models/__init__.py b/libs/community/langchain_community/chat_models/__init__.py
@@ -51,8 +51,8 @@
 from langchain_community.chat_models.vertexai import ChatVertexAI
 from langchain_community.chat_models.volcengine_maas import VolcEngineMaasChat
 from langchain_community.chat_models.yandex import ChatYandexGPT
-from langchain_community.chat_models.zhipuai import ChatZhipuAI
 from langchain_community.chat_models.yuan2 import ChatYuan2
+from langchain_community.chat_models.zhipuai import ChatZhipuAI
 
 __all__ = [
     "LlamaEdgeChatService",
@@ -89,6 +89,6 @@
     "GigaChat",
     "VolcEngineMaasChat",
     "GPTRouter",
-    "ChatZhipuAI",
     "ChatYuan2",
+    "ChatZhipuAI",
 ]
diff --git a/libs/community/tests/integration_tests/chat_models/test_yuan2.py b/libs/community/tests/integration_tests/chat_models/test_yuan2.py
@@ -1,16 +1,12 @@
 """Test ChatYuan2 wrapper."""
-from typing import Any, Optional
 
 import pytest
 from langchain_core.callbacks import CallbackManager
-from langchain_core.messages import AIMessage, BaseMessage, HumanMessage, SystemMessage
+from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage
 from langchain_core.outputs import (
     ChatGeneration,
-    ChatResult,
     LLMResult,
 )
-from langchain_core.prompts import ChatPromptTemplate
-from langchain_core.pydantic_v1 import BaseModel, Field
 
 from langchain_community.chat_models.yuan2 import ChatYuan2
 from tests.unit_tests.callbacks.fake_callback_handler import FakeCallbackHandler

diff --git a/libs/community/tests/unit_tests/chat_models/test_imports.py b/libs/community/tests/unit_tests/chat_models/test_imports.py
@@ -35,8 +35,8 @@
     "VolcEngineMaasChat",
     "LlamaEdgeChatService",
     "GPTRouter",
-    "ChatZhipuAI",
     "ChatYuan2",
+    "ChatZhipuAI",
 ]