fixed linting issues

langchain-ai · Feb 7, 2024 · 2d4135d · 2d4135d
1 parent 0e8a142
commit 2d4135d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/libs/partners/google-vertexai/langchain_google_vertexai/chat_models.py b/libs/partners/google-vertexai/langchain_google_vertexai/chat_models.py
@@ -30,9 +30,6 @@
 )
 from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
 from langchain_core.pydantic_v1 import root_validator
-from langchain_google_vertexai.llms import (
-    _VertexAICommon,
-)
 from vertexai.preview.generative_models import (  # type: ignore
     Candidate,
     Content,
@@ -58,6 +55,9 @@
 from langchain_google_vertexai.functions_utils import (
     _format_tools_to_vertex_tool,
 )
+from langchain_google_vertexai.llms import (
+    _VertexAICommon,
+)
 
 logger = logging.getLogger(__name__)
 

diff --git a/libs/partners/google-vertexai/tests/integration_tests/test_chat_models.py b/libs/partners/google-vertexai/tests/integration_tests/test_chat_models.py
@@ -229,7 +229,7 @@ def test_get_num_tokens_from_messages() -> None:
     assert isinstance(token, int)
     assert token == 3
 
-    #test for exception with chat-bison
+    # test for exception with chat-bison
     model = ChatVertexAI(model_name="chat-bison", temperature=0.0)
     with pytest.raises(Exception):
         token = model.get_num_tokens_from_messages(messages=[message])