Avoid mutable defaults
This commit is contained in:
parent
d1fdbdeb21
commit
492f5cf1e4
2 changed files with 7 additions and 7 deletions
|
|
@ -2,7 +2,7 @@ import os
|
|||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
from pydantic import BaseModel
|
||||
from pydantic import BaseModel, Field
|
||||
from pydantic_ai import RunContext
|
||||
|
||||
from haiku.rag.agents.research.models import Citation
|
||||
|
|
@ -31,12 +31,12 @@ class ResearchEntry(BaseModel):
|
|||
|
||||
|
||||
class RAGState(BaseModel):
|
||||
citations: list[Citation] = []
|
||||
qa_history: list[QAHistoryEntry] = []
|
||||
citations: list[Citation] = Field(default_factory=list)
|
||||
qa_history: list[QAHistoryEntry] = Field(default_factory=list)
|
||||
document_filter: str | None = None
|
||||
searches: dict[str, list[SearchResult]] = {}
|
||||
documents: list[DocumentInfo] = []
|
||||
reports: list[ResearchEntry] = []
|
||||
searches: dict[str, list[SearchResult]] = Field(default_factory=dict)
|
||||
documents: list[DocumentInfo] = Field(default_factory=list)
|
||||
reports: list[ResearchEntry] = Field(default_factory=list)
|
||||
|
||||
|
||||
def create_skill(
|
||||
|
|
|
|||
|
|
@ -11,7 +11,7 @@ class QAHistoryEntry(BaseModel):
|
|||
question: str
|
||||
answer: str
|
||||
confidence: float = 0.9
|
||||
citations: list[Citation] = []
|
||||
citations: list[Citation] = Field(default_factory=list)
|
||||
question_embedding: list[float] | None = Field(default=None, exclude=True)
|
||||
|
||||
@property
|
||||
|
|
|
|||
Loading…
Reference in a new issue