From d3a9ac065022f33a4eb35c2f4b844c8e229c950d Mon Sep 17 00:00:00 2001 From: Erick Friis Date: Tue, 19 Mar 2024 12:27:03 -0700 Subject: [PATCH] rename tools folder to community 2 --- libs/community/langchain_google_tools/__init__.py | 10 +++++----- .../integration_tests/test_bigquery_vector_search.py | 2 +- .../test_docai_warehoure_retriever.py | 3 ++- .../tests/integration_tests/test_vertex_ai_search.py | 3 +-- 4 files changed, 9 insertions(+), 9 deletions(-) diff --git a/libs/community/langchain_google_tools/__init__.py b/libs/community/langchain_google_tools/__init__.py index ee1ec9c3..e5196e23 100644 --- a/libs/community/langchain_google_tools/__init__.py +++ b/libs/community/langchain_google_tools/__init__.py @@ -1,8 +1,8 @@ -from langchain_google_tools.bigquery_vector_search import BigQueryVectorSearch -from langchain_google_tools.documentai_warehouse import DocumentAIWarehouseRetriever -from langchain_google_tools.gmail.loader import GMailLoader -from langchain_google_tools.gmail.toolkit import GmailToolkit -from langchain_google_tools.vertex_ai_search import ( +from langchain_google_community.bigquery_vector_search import BigQueryVectorSearch +from langchain_google_community.documentai_warehouse import DocumentAIWarehouseRetriever +from langchain_google_community.gmail.loader import GMailLoader +from langchain_google_community.gmail.toolkit import GmailToolkit +from langchain_google_community.vertex_ai_search import ( VertexAIMultiTurnSearchRetriever, VertexAISearchRetriever, ) diff --git a/libs/community/tests/integration_tests/test_bigquery_vector_search.py b/libs/community/tests/integration_tests/test_bigquery_vector_search.py index ee3ba8f6..cd209e93 100644 --- a/libs/community/tests/integration_tests/test_bigquery_vector_search.py +++ b/libs/community/tests/integration_tests/test_bigquery_vector_search.py @@ -10,7 +10,7 @@ import pytest -from langchain_google_tools import BigQueryVectorSearch +from langchain_google_community import BigQueryVectorSearch from tests.integration_tests.fake import FakeEmbeddings TEST_TABLE_NAME = "langchain_test_table" diff --git a/libs/community/tests/integration_tests/test_docai_warehoure_retriever.py b/libs/community/tests/integration_tests/test_docai_warehoure_retriever.py index 803fd47a..87b94163 100644 --- a/libs/community/tests/integration_tests/test_docai_warehoure_retriever.py +++ b/libs/community/tests/integration_tests/test_docai_warehoure_retriever.py @@ -1,9 +1,10 @@ """Test Google Cloud Document AI Warehouse retriever.""" + import os from langchain_core.documents import Document -from langchain_google_tools import DocumentAIWarehouseRetriever +from langchain_google_community import DocumentAIWarehouseRetriever def test_google_documentai_warehoure_retriever() -> None: diff --git a/libs/community/tests/integration_tests/test_vertex_ai_search.py b/libs/community/tests/integration_tests/test_vertex_ai_search.py index 521cfdbc..82d835ae 100644 --- a/libs/community/tests/integration_tests/test_vertex_ai_search.py +++ b/libs/community/tests/integration_tests/test_vertex_ai_search.py @@ -11,10 +11,9 @@ export DATA_STORE_ID=... - the ID of the search engine to use for the test """ - from langchain_core.documents import Document -from langchain_google_tools import ( +from langchain_google_community import ( VertexAIMultiTurnSearchRetriever, VertexAISearchRetriever, )