Skip to content

Commit 71bb951

Browse files
authored
fix: add default value for data_sources in KnowledgeBase pydantic model (#686)
1 parent 24a82d5 commit 71bb951

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

backend/app/models/knowledge_base.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,8 @@ class AdvancedChunkingConfig(BaseChunkingConfig):
118118

119119

120120
class KnowledgeBase(SQLModel, table=True):
121+
__tablename__ = "knowledge_bases"
122+
121123
id: Optional[int] = Field(default=None, primary_key=True)
122124
name: str = Field(max_length=255, nullable=False)
123125
description: Optional[str] = Field(sa_column=Column(MEDIUMTEXT), default=None)
@@ -173,7 +175,9 @@ class KnowledgeBase(SQLModel, table=True):
173175
deleted_by: UUID = Field(foreign_key="users.id", nullable=True)
174176
deleted_at: Optional[datetime] = Field(default=None, sa_column=Column(DateTime()))
175177

176-
__tablename__ = "knowledge_bases"
178+
def __init__(self, **kwargs):
179+
kwargs.setdefault("data_sources", [])
180+
super().__init__(**kwargs)
177181

178182
def __hash__(self):
179183
return hash(self.id)

0 commit comments

Comments
 (0)