Display on info pydantic-ai and docling-document schema version
This commit is contained in:
parent
e14499658f
commit
fc1ea82cb1
3 changed files with 13 additions and 0 deletions
|
|
@ -3,6 +3,7 @@
|
||||||
|
|
||||||
### Added
|
### Added
|
||||||
|
|
||||||
|
- **Enhanced Database Info**: `haiku-rag info` now displays `pydantic-ai` version and `docling-document schema` version
|
||||||
- **Keyed State Emission for Chat Agent**: New `state_key` parameter in `ChatDeps` for namespaced AG-UI state snapshots
|
- **Keyed State Emission for Chat Agent**: New `state_key` parameter in `ChatDeps` for namespaced AG-UI state snapshots
|
||||||
- When set, tools emit `{state_key: snapshot}` instead of bare state, enabling state merging when multiple agents share state
|
- When set, tools emit `{state_key: snapshot}` instead of bare state, enabling state merging when multiple agents share state
|
||||||
- Default `None` preserves backwards compatibility (bare state emission)
|
- Default `None` preserves backwards compatibility (bare state emission)
|
||||||
|
|
|
||||||
|
|
@ -98,6 +98,10 @@ class HaikuRAGApp:
|
||||||
docling_version = pkg_version("docling")
|
docling_version = pkg_version("docling")
|
||||||
except Exception:
|
except Exception:
|
||||||
docling_version = "unknown"
|
docling_version = "unknown"
|
||||||
|
pydantic_ai_version = pkg_version("pydantic-ai-slim")
|
||||||
|
from docling_core.types.doc.document import DoclingDocument
|
||||||
|
|
||||||
|
docling_document_version = DoclingDocument.model_construct().version
|
||||||
|
|
||||||
# Get comprehensive table statistics (this also runs migrations)
|
# Get comprehensive table statistics (this also runs migrations)
|
||||||
from haiku.rag.store.engine import Store
|
from haiku.rag.store.engine import Store
|
||||||
|
|
@ -219,6 +223,12 @@ class HaikuRAGApp:
|
||||||
self.console.print(
|
self.console.print(
|
||||||
f" [repr.attrib_name]docling[/repr.attrib_name]: {docling_version}"
|
f" [repr.attrib_name]docling[/repr.attrib_name]: {docling_version}"
|
||||||
)
|
)
|
||||||
|
self.console.print(
|
||||||
|
f" [repr.attrib_name]pydantic-ai[/repr.attrib_name]: {pydantic_ai_version}"
|
||||||
|
)
|
||||||
|
self.console.print(
|
||||||
|
f" [repr.attrib_name]docling-document schema[/repr.attrib_name]: {docling_document_version}"
|
||||||
|
)
|
||||||
|
|
||||||
async def history(self, table: str | None = None, limit: int | None = None):
|
async def history(self, table: str | None = None, limit: int | None = None):
|
||||||
"""Display version history for database tables.
|
"""Display version history for database tables.
|
||||||
|
|
|
||||||
|
|
@ -81,6 +81,8 @@ async def test_app_info_outputs(temp_db_path, capsys):
|
||||||
# Package versions section
|
# Package versions section
|
||||||
assert "lancedb:" in out
|
assert "lancedb:" in out
|
||||||
assert "haiku.rag:" in out
|
assert "haiku.rag:" in out
|
||||||
|
assert "docling-document schema:" in out
|
||||||
|
assert "pydantic-ai:" in out
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue