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

Avoid mutable defaults

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-02-20 17:59:35 +02:00
parent d1fdbdeb21
commit 492f5cf1e4
No known key found for this signature in database
2 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -2,7 +2,7 @@ import os
from pathlib import Path from pathlib import Path
from typing import Any from typing import Any
from pydantic import BaseModel from pydantic import BaseModel, Field
from pydantic_ai import RunContext from pydantic_ai import RunContext
from haiku.rag.agents.research.models import Citation from haiku.rag.agents.research.models import Citation
@ -31,12 +31,12 @@ class ResearchEntry(BaseModel):
class RAGState(BaseModel): class RAGState(BaseModel):
citations: list[Citation] = [] citations: list[Citation] = Field(default_factory=list)
qa_history: list[QAHistoryEntry] = [] qa_history: list[QAHistoryEntry] = Field(default_factory=list)
document_filter: str | None = None document_filter: str | None = None
searches: dict[str, list[SearchResult]] = {} searches: dict[str, list[SearchResult]] = Field(default_factory=dict)
documents: list[DocumentInfo] = [] documents: list[DocumentInfo] = Field(default_factory=list)
reports: list[ResearchEntry] = [] reports: list[ResearchEntry] = Field(default_factory=list)
def create_skill( def create_skill(

2
haiku_rag_slim/haiku/rag/tools/qa.py
View file

@ -11,7 +11,7 @@ class QAHistoryEntry(BaseModel):
question: str question: str
answer: str answer: str
confidence: float = 0.9 confidence: float = 0.9
citations: list[Citation] = [] citations: list[Citation] = Field(default_factory=list)
question_embedding: list[float] | None = Field(default=None, exclude=True) question_embedding: list[float] | None = Field(default=None, exclude=True)
@property @property