From e6310fc4844b3f62fbe971b1207715cef5f9eeb2 Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Fri, 20 Feb 2026 14:08:10 +0200 Subject: [PATCH] Fix citations typing --- haiku_rag_slim/haiku/rag/skills/rag.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/haiku_rag_slim/haiku/rag/skills/rag.py b/haiku_rag_slim/haiku/rag/skills/rag.py index ca7f9c71..099a2d8e 100644 --- a/haiku_rag_slim/haiku/rag/skills/rag.py +++ b/haiku_rag_slim/haiku/rag/skills/rag.py @@ -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]] = {}