Harrison/fauna loader (#5864)

Co-authored-by: Shadid12 <Shadid12@users.noreply.github.com>
This commit is contained in:
Harrison Chase
2023-06-07 21:32:23 -07:00
committed by GitHub
parent 5518f24ec3
commit 658f8bdee7
4 changed files with 190 additions and 0 deletions

View File

@@ -0,0 +1,41 @@
import unittest
from langchain.document_loaders.fauna import FaunaLoader
try:
import fauna # noqa: F401
fauna_installed = True
except ImportError:
fauna_installed = False
@unittest.skipIf(not fauna_installed, "fauna not installed")
class TestFaunaLoader(unittest.TestCase):
def setUp(self) -> None:
self.fauna_secret = "<enter-valid-fauna-secret>"
self.valid_fql_query = "Item.all()"
self.valid_page_content_field = "text"
self.valid_metadata_fields = ["valid_metadata_fields"]
def test_fauna_loader(self) -> None:
"""Test Fauna loader."""
loader = FaunaLoader(
query=self.valid_fql_query,
page_content_field=self.valid_page_content_field,
secret=self.fauna_secret,
metadata_fields=self.valid_metadata_fields,
)
docs = loader.load()
assert len(docs) > 0 # assuming the query returns at least one document
for doc in docs:
assert (
doc.page_content != ""
) # assuming that every document has page_content
assert (
"id" in doc.metadata and doc.metadata["id"] != ""
) # assuming that every document has 'id'
assert (
"ts" in doc.metadata and doc.metadata["ts"] != ""
) # assuming that every document has 'ts'