mirror of
https://github.com/VectifyAI/PageIndex.git
synced 2026-06-21 20:18:09 +02:00
refactor(filesystem): make pifs providers configurable
This commit is contained in:
parent
7c021a7dd0
commit
de1992def1
7 changed files with 154 additions and 61 deletions
30
tests/test_metadata_generation.py
Normal file
30
tests/test_metadata_generation.py
Normal file
|
|
@ -0,0 +1,30 @@
|
|||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
REPO_ROOT = Path(__file__).resolve().parents[1]
|
||||
if str(REPO_ROOT) not in sys.path:
|
||||
sys.path.insert(0, str(REPO_ROOT))
|
||||
|
||||
|
||||
def test_metadata_generator_uses_provider_parameter():
|
||||
from pageindex.filesystem.metadata_generation import (
|
||||
MetadataGenerationError,
|
||||
MetadataGenerationInput,
|
||||
MetadataGenerator,
|
||||
)
|
||||
|
||||
generator = MetadataGenerator(provider="unsupported", model="unused")
|
||||
request = MetadataGenerationInput(
|
||||
file_ref="file_a",
|
||||
external_id="doc_a",
|
||||
title="A",
|
||||
source_path="docs/a.txt",
|
||||
content_type="text/plain",
|
||||
source_type=None,
|
||||
text="hello",
|
||||
)
|
||||
|
||||
with pytest.raises(MetadataGenerationError, match="unsupported metadata provider: unsupported"):
|
||||
generator.generate(request, fields=["summary"])
|
||||
Loading…
Add table
Add a link
Reference in a new issue