danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix citations typing

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-02-20 14:08:10 +02:00
parent c567842420
commit e6310fc484
No known key found for this signature in database
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
haiku_rag_slim/haiku/rag/skills/rag.py
View file

@ -5,6 +5,7 @@ from typing import Any
from pydantic import BaseModel
from pydantic_ai import RunContext
from haiku.rag.agents.research.models import Citation
from haiku.rag.store.models.chunk import SearchResult
from haiku.rag.tools.document import DocumentInfo
from haiku.rag.tools.qa import QAHistoryEntry
@ -20,7 +21,7 @@ class ResearchEntry(BaseModel):
class RAGState(BaseModel):
citations: list[Any] = []
citations: list[Citation] = []
qa_history: list[QAHistoryEntry] = []
document_filter: str | None = None
searches: dict[str, list[SearchResult]] = {}