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