Skip to content

fix: make RAG state model fields optional with defaults #189 #184

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
bahdan111 merged 4 commits into from
Mar 6, 2026
Merged

fix: make RAG state model fields optional with defaults #189 #184

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
5cd126f
upd dial rag state model
Mar 5, 2026
4a88e5b
upd base rag state
Mar 5, 2026
bbf1054
upd defaults
Mar 5, 2026
8ae3a3c
Merge branch 'development' into upd-dial-rag-state-model
navalnica Mar 5, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions statgpt/app/schemas/file_rags/base.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,5 +6,5 @@ class BaseRagState(ToolMessageState):
type: ToolTypes = ToolTypes.FILE_RAG
version: RAGVersion

response: str # This is not needed since we have content field
answered_by: str
response: str = "" # This is not needed since we have content field
answered_by: str = ""
12 changes: 6 additions & 6 deletions statgpt/app/schemas/file_rags/dial_rag.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -292,18 +292,18 @@ def _extract_dimension_values(


class PreFilterResponse(BaseModel):
user_friendly_error: str | None = Field(default=None)
detailed_error: str | None = Field(default=None)
llm_output: RagFilterLLMOutput | None = Field(default=None)
user_friendly_error: str | None = None
detailed_error: str | None = None
llm_output: RagFilterLLMOutput | None = None
# NOTE: Currently None means no filter applied; but RagFilterDial can also contain an empty list;
# NOTE: Should we replace all the None logic with just an empty list?
rag_filter: RagFilterDial | None = Field(default=None)
rag_filter: RagFilterDial | None = None


class DialRagState(BaseRagState):
version: RAGVersion = RAGVersion.DIAL
pre_filter: PreFilterResponse
metadata: DialRagMetadata | None
pre_filter: PreFilterResponse | None = None
metadata: DialRagMetadata | None = None
prefilter_decoder_of_latest: dict[str, str] = Field(
default_factory=dict,
description="Mapping the publication type to a function that generates a time range "
Expand Down
Loading