mirror of
https://github.com/VectifyAI/PageIndex.git
synced 2026-07-24 21:41:04 +02:00
1054132756
The local SDK path builds IndexConfig directly (not from config.yaml), so retrieve_model defaulting to None made agentic QA silently follow the cheaper indexing model (gpt-4o) instead of the stronger gpt-5.4 the packaged config and CLI use. Set the IndexConfig field default to gpt-5.4; retrieve_model=None still follows model as an explicit escape hatch. Cloud path and the index-only CLI are unaffected.
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
# tests/test_config.py
|
|
import pytest
|
|
from pageindex.config import IndexConfig
|
|
|
|
|
|
def test_defaults():
|
|
config = IndexConfig()
|
|
assert config.model == "gpt-4o-2024-11-20"
|
|
assert config.retrieve_model == "gpt-5.4"
|
|
assert config.toc_check_page_num == 20
|
|
|
|
|
|
def test_overrides():
|
|
config = IndexConfig(model="gpt-5.4", retrieve_model="claude-sonnet")
|
|
assert config.model == "gpt-5.4"
|
|
assert config.retrieve_model == "claude-sonnet"
|
|
|
|
|
|
def test_unknown_key_raises():
|
|
with pytest.raises(Exception):
|
|
IndexConfig(nonexistent_key="value")
|
|
|
|
|
|
def test_model_copy_with_update():
|
|
config = IndexConfig(toc_check_page_num=30)
|
|
updated = config.model_copy(update={"model": "gpt-5.4"})
|
|
assert updated.model == "gpt-5.4"
|
|
assert updated.toc_check_page_num == 30
|
|
|
|
|
|
def test_legacy_yes_no_strings_coerce_to_bool():
|
|
"""Legacy page_index()/run_pageindex callers pass 'yes'/'no' strings;
|
|
pydantic must coerce them to the booleans the pipeline now branches on."""
|
|
config = IndexConfig(if_add_node_id="yes", if_add_node_summary="no")
|
|
assert config.if_add_node_id is True
|
|
assert config.if_add_node_summary is False
|
|
|
|
|
|
def test_llm_params_field_defaults_to_none():
|
|
assert IndexConfig().llm_params is None
|
|
assert IndexConfig(llm_params={"temperature": 1}).llm_params == {"temperature": 1}
|