moved imports

langchain-ai · Dec 18, 2023 · 45f2fd5 · 45f2fd5
1 parent cdc32f4
commit 45f2fd5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/libs/community/langchain_community/utils/vertex_functions.py b/libs/community/langchain_community/utils/vertex_functions.py
@@ -1,6 +1,6 @@
 from typing import TYPE_CHECKING
 
-from langchain.agents import Tool
+from langchain_core.tools import Tool
 
 from langchain_community.utils.openai_functions import (
     FunctionDescription,

diff --git a/libs/community/tests/integration_tests/chat_models/test_vertexai.py b/libs/community/tests/integration_tests/chat_models/test_vertexai.py
@@ -11,11 +11,6 @@
 from unittest.mock import MagicMock, Mock, patch
 
 import pytest
-from langchain.agents import AgentExecutor, Tool
-from langchain.agents.format_scratchpad import format_to_openai_function_messages
-from langchain.agents.output_parsers import VertexAIFunctionsAgentOutputParser
-from langchain.chains import LLMMathChain
-from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.messages import (
     AIMessage,
     AIMessageChunk,
@@ -302,6 +297,12 @@ def test_parse_examples_failes_wrong_sequence() -> None:
 
 
 def test_tools() -> None:
+    from langchain.agents import AgentExecutor, Tool
+    from langchain.agents.format_scratchpad import format_to_openai_function_messages
+    from langchain.agents.output_parsers import VertexAIFunctionsAgentOutputParser
+    from langchain.chains import LLMMathChain
+    from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
+
     llm = ChatVertexAI(model_name="gemini-pro")
     math_chain = LLMMathChain.from_llm(llm=llm)
     raw_tools = [