Fix broken resources model

OpenSourceBrain · Feb 20, 2024 · 0f68859 · 0f68859
1 parent 56765f0
commit 0f68859
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/applications/workspaces/server/workspaces/persistence/models.py b/applications/workspaces/server/workspaces/persistence/models.py
@@ -834,6 +834,9 @@ class TWorkspaceResourceEntity(typing.Protocol):
     timestamp_updated: typing.Optional[datetime.datetime]
     timestamp_last_opened: typing.Optional[datetime.datetime]
     resource_type: str
+    origin: typing.Optional[str]
+    workspace_id: typing.Optional[int]
+    folder: typing.Optional[str]
 
     def __init__(
         self,
@@ -844,6 +847,9 @@ def __init__(
         timestamp_created: typing.Optional[datetime.datetime] = None,
         timestamp_updated: typing.Optional[datetime.datetime] = None,
         timestamp_last_opened: typing.Optional[datetime.datetime] = None,
+        origin: typing.Optional[str] = None,
+        workspace_id: typing.Optional[int] = None,
+        folder: typing.Optional[str] = None,
     ) -> None:
         """
         Construct.
@@ -874,6 +880,9 @@ def from_dict(
         timestamp_created: typing.Optional[datetime.datetime] = None,
         timestamp_updated: typing.Optional[datetime.datetime] = None,
         timestamp_last_opened: typing.Optional[datetime.datetime] = None,
+        origin: typing.Optional[str] = None,
+        workspace_id: typing.Optional[int] = None,
+        folder: typing.Optional[str] = None,
     ) -> "TWorkspaceResourceEntity":
         """
         Construct from a dictionary (eg. a POST payload).