notion and website import fix

langgenius · Dec 24, 2024 · a250335 · a250335
1 parent 71950ee
commit a250335
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/api/services/entities/knowledge_entities/knowledge_entities.py b/api/services/entities/knowledge_entities/knowledge_entities.py
@@ -16,10 +16,16 @@ class ParentMode(str, Enum):
     PARAGRAPH = "paragraph"
 
 
+class NotionIcon(BaseModel):
+    type: str
+    url: Optional[str] = None
+    emoji: Optional[str] = None
+
+
 class NotionPage(BaseModel):
     page_id: str
     page_name: str
-    page_icon: Optional[str]
+    page_icon: Optional[NotionIcon] = None
     type: str