Fix ty warnings in lynxkite-app and examples.

examples/fake_data.py

@@ -1,5 +1,5 @@
 from lynxkite.core.ops import op
-from faker import Faker
+from faker import Faker  # ty: ignore[unresolved-import]
 import pandas as pd
 
 faker = Faker()

examples/word2vec.py

@@ -6,7 +6,7 @@ ENV = "LynxKite Graph Analytics"
 
 @op(ENV, "Word2vec for the top 1000 words", slow=True)
 def word2vec_1000():
-    import staticvectors
+    import staticvectors  # ty: ignore[unresolved-import]
 
     model = staticvectors.StaticVectors("neuml/word2vec-quantized")
     df = pd.read_csv(

lynxkite-app/src/lynxkite_app/crdt.py

@@ -7,8 +7,8 @@ import pathlib
 import fastapi
 import os.path
 import pycrdt.websocket
-import pycrdt.store
-import uvicorn
+import pycrdt.store.file
+import uvicorn.protocols.utils
 import builtins
 from lynxkite.core import workspace, ops
 
@@ -59,7 +59,8 @@ class WorkspaceWebsocketServer(pycrdt.websocket.WebsocketServer):
         room = pycrdt.websocket.YRoom(
             ystore=ystore, ydoc=ydoc, exception_handler=ws_exception_handler
         )
-        room.ws = ws
+        # We hang the YDoc pointer on the room, so it only gets garbage collected when the room does.
+        room.ws = ws  # ty: ignore[unresolved-attribute]
 
         def on_change(changes):
             task = asyncio.create_task(workspace_changed(name, changes, ws))
@@ -106,7 +107,8 @@ class CodeWebsocketServer(WorkspaceWebsocketServer):
         room = pycrdt.websocket.YRoom(
             ystore=ystore, ydoc=ydoc, exception_handler=ws_exception_handler
         )
-        room.text = text
+        # We hang the YDoc pointer on the room, so it only gets garbage collected when the room does.
+        room.text = text  # ty: ignore[unresolved-attribute]
 
         def on_change(changes):
             asyncio.create_task(code_changed(name, changes, text))
@@ -163,6 +165,7 @@ def crdt_update(
         ValueError: If the Python object provided is not a dict or list.
     """
     if isinstance(python_obj, dict):
+        assert isinstance(crdt_obj, pycrdt.Map), "CRDT object must be a Map for a dict input"
         for key, value in python_obj.items():
             if key in non_collaborative_fields:
                 crdt_obj[key] = value
@@ -179,6 +182,7 @@ def crdt_update(
             else:
                 crdt_obj[key] = value
     elif isinstance(python_obj, list):
+        assert isinstance(crdt_obj, pycrdt.Array), "CRDT object must be an Array for a list input"
         for i, value in enumerate(python_obj):
             if isinstance(value, dict):
                 if i >= len(crdt_obj):
@@ -220,7 +224,7 @@ last_known_versions = {}
 delayed_executions = {}
 
 
-async def workspace_changed(name: str, changes: pycrdt.MapEvent, ws_crdt: pycrdt.Map):
+async def workspace_changed(name: str, changes: list[pycrdt.MapEvent], ws_crdt: pycrdt.Map):
     """Callback to react to changes in the workspace.
 
     Args:
@@ -295,6 +299,14 @@ async def code_changed(name: str, changes: pycrdt.TextEvent, text: pycrdt.Text):
         f.write(contents)
 
 
+ws_websocket_server: WorkspaceWebsocketServer
+code_websocket_server: CodeWebsocketServer
+
+
+def get_room(name):
+    return ws_websocket_server.get_room(name)
+
+
 @contextlib.asynccontextmanager
 async def lifespan(app):
     global ws_websocket_server

lynxkite-app/src/lynxkite_app/main.py

@@ -8,7 +8,7 @@ import pathlib
 import pkgutil
 from fastapi.staticfiles import StaticFiles
 from fastapi.middleware.gzip import GZipMiddleware
-import starlette
+import starlette.exceptions
 from lynxkite.core import ops
 from lynxkite.core import workspace
 from . import crdt
@@ -136,7 +136,7 @@ async def upload(req: fastapi.Request):
 @app.post("/api/execute_workspace")
 async def execute_workspace(name: str):
     """Trigger and await the execution of a workspace."""
-    room = await crdt.ws_websocket_server.get_room(name)
+    room = await crdt.get_room(name)
     ws_pyd = workspace.Workspace.model_validate(room.ws.to_py())
     await crdt.execute(name, room.ws, ws_pyd)
 

lynxkite-core/src/lynxkite/core/workspace.py

@@ -49,6 +49,13 @@ class WorkspaceNodeData(BaseConfig):
             data["op_id"] = data["title"]
         return data
 
+    @pydantic.model_validator(mode="before")
+    @classmethod
+    def ignore_meta(cls, data: dict) -> dict:
+        """Metadata is never loaded. We will use fresh metadata."""
+        data["meta"] = None
+        return data
+
 
 class WorkspaceNode(BaseConfig):
     # Most of these fields are shared with ReactFlow.