mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
ed58eeb9c5
Moved the following modules to new package langchain-community in a backwards compatible fashion: ``` mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community ``` Moved the following to core ``` mv langchain/langchain/utils/json_schema.py core/langchain_core/utils mv langchain/langchain/utils/html.py core/langchain_core/utils mv langchain/langchain/utils/strings.py core/langchain_core/utils cat langchain/langchain/utils/env.py >> core/langchain_core/utils/env.py rm langchain/langchain/utils/env.py ``` See .scripts/community_split/script_integrations.sh for all changes
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
from langchain_core.documents import Document
|
|
|
|
from langchain_community.document_loaders import GeoDataFrameLoader
|
|
|
|
if TYPE_CHECKING:
|
|
from geopandas import GeoDataFrame
|
|
else:
|
|
GeoDataFrame = "geopandas.GeoDataFrame"
|
|
|
|
|
|
@pytest.mark.requires("geopandas")
|
|
def sample_gdf() -> GeoDataFrame:
|
|
import geopandas
|
|
|
|
# TODO: geopandas.datasets will be deprecated in 1.0
|
|
path_to_data = geopandas.datasets.get_path("nybb")
|
|
gdf = geopandas.read_file(path_to_data)
|
|
gdf["area"] = gdf.area
|
|
gdf["crs"] = gdf.crs.to_string()
|
|
return gdf.head(2)
|
|
|
|
|
|
@pytest.mark.requires("geopandas")
|
|
def test_load_returns_list_of_documents(sample_gdf: GeoDataFrame) -> None:
|
|
loader = GeoDataFrameLoader(sample_gdf)
|
|
docs = loader.load()
|
|
assert isinstance(docs, list)
|
|
assert all(isinstance(doc, Document) for doc in docs)
|
|
assert len(docs) == 2
|
|
|
|
|
|
@pytest.mark.requires("geopandas")
|
|
def test_load_converts_dataframe_columns_to_document_metadata(
|
|
sample_gdf: GeoDataFrame,
|
|
) -> None:
|
|
loader = GeoDataFrameLoader(sample_gdf)
|
|
docs = loader.load()
|
|
for i, doc in enumerate(docs):
|
|
assert doc.metadata["area"] == sample_gdf.loc[i, "area"]
|
|
assert doc.metadata["crs"] == sample_gdf.loc[i, "crs"]
|