Skip to content

Commit

Permalink
Remove qna.yaml from chunker functional tests
Browse files Browse the repository at this point in the history
We no longer need qna.yaml files on disk to chunk documents.

Signed-off-by: Ben Browning <[email protected]>
  • Loading branch information
bbrowning committed Nov 8, 2024
1 parent d8ac253 commit e558258
Showing 1 changed file with 0 additions and 16 deletions.
16 changes: 0 additions & 16 deletions tests/functional/test_chunkers.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,19 +7,8 @@

TEST_DATA_DIR = os.path.join(os.path.dirname(__file__), "testdata")

# TODO: Apparently we don't really need any contents in the qna.yaml?
knowledge_qna = """
version: 3
domain: astronomy
"""


def test_chunk_pdf(tmp_path):
qna_dir = os.path.join(tmp_path, "knowledge")
os.makedirs(qna_dir)
with open(os.path.join(qna_dir, "qna.yaml"), "w", encoding="utf-8") as f:
f.write(knowledge_qna)

leaf_node = [
{
"documents": ["Lorem ipsum"],
Expand All @@ -44,11 +33,6 @@ def test_chunk_pdf(tmp_path):


def test_chunk_md(tmp_path):
qna_dir = os.path.join(tmp_path, "knowledge")
os.makedirs(qna_dir)
with open(os.path.join(qna_dir, "qna.yaml"), "w", encoding="utf-8") as f:
f.write(knowledge_qna)

markdown_path = Path(os.path.join(TEST_DATA_DIR, "phoenix.md"))
leaf_node = [
{
Expand Down

0 comments on commit e558258

Please sign in to comment.