fix(sdk): Move local user to local

python/src/numerous/experimental/marimo/_cookies/fastapi.py

@@ -39,7 +39,7 @@ def add_marimo_cookie_middleware(
     cookies = FileCookieStorage(cookies_dir, session_ident or _ident)
     use_cookie_storage(cookies)
 
-    @app.middleware("http")
+    @app.middleware("http")  # type: ignore[misc, unused-ignore]
     async def middleware(
         request: Request,
         call_next: t.Callable[[Request], t.Awaitable[Response]],

python/src/numerous/frameworks/fastapi.py

@@ -5,7 +5,7 @@
 from fastapi import Request
 
 from numerous import user_session
-from numerous.utils import is_local_mode
+from numerous.local import is_local_mode, local_user
 
 
 class FastapiCookieGetter:
@@ -16,7 +16,7 @@ def cookies(self) -> Dict[str, str]:
         """Get the cookies associated with the current request."""
         if is_local_mode():
             # Update the cookies on the fastapi server
-            self.request.cookies.update(user_session.get_encoded_user_info())
+            self.request.cookies.update(user_session.get_encoded_user_info(local_user))
 
         return {str(key): str(val) for key, val in self.request.cookies.items()}
 

python/src/numerous/frameworks/flask.py

@@ -5,7 +5,7 @@
 from flask import request
 
 from numerous import user_session
-from numerous.utils import is_local_mode
+from numerous.local import is_local_mode, local_user
 
 
 class FlaskCookieGetter:
@@ -14,7 +14,7 @@ def cookies(self) -> Dict[str, str]:
         local_cookies = {}
         if is_local_mode():
             # Update the cookies on the flask server
-            local_cookies = user_session.get_encoded_user_info()
+            local_cookies = user_session.get_encoded_user_info(local_user)
         return {
             **local_cookies,
             **{str(key): str(val) for key, val in request.cookies.items()},

python/src/numerous/frameworks/marimo.py

@@ -4,15 +4,15 @@
 
 from numerous import user_session
 from numerous.experimental import marimo
-from numerous.utils import is_local_mode
+from numerous.local import is_local_mode, local_user
 
 
 class MarimoCookieGetter:
     def cookies(self) -> Dict[str, Any]:
         """Get the cookies associated with the current request."""
         if is_local_mode():
             # Update the cookies on the marimo server
-            marimo.cookies().update(user_session.get_encoded_user_info())
+            marimo.cookies().update(user_session.get_encoded_user_info(local_user))
         return {key: str(val) for key, val in marimo.cookies().items()}
 
 

python/src/numerous/frameworks/panel.py

@@ -5,7 +5,7 @@
 import panel as pn
 
 from numerous import user_session
-from numerous.utils import is_local_mode
+from numerous.local import is_local_mode, local_user
 
 
 class PanelCookieGetter:
@@ -14,7 +14,7 @@ def cookies() -> Dict[str, str]:
         """Get the cookies associated with the current request."""
         if is_local_mode():
             # Add the user info to the cookies on panel server
-            pn.state.cookies.update(user_session.get_encoded_user_info())
+            pn.state.cookies.update(user_session.get_encoded_user_info(local_user))
 
         if pn.state.curdoc and pn.state.curdoc.session_context:
             return {key: str(val) for key, val in pn.state.cookies.items()}

python/src/numerous/frameworks/streamlit.py

@@ -5,7 +5,7 @@
 import streamlit as st
 
 from numerous import user_session
-from numerous.utils import is_local_mode
+from numerous.local import is_local_mode, local_user
 
 
 class StreamlitCookieGetter:
@@ -14,7 +14,7 @@ def cookies(self) -> Dict[str, str]:
         local_cookies = {}
         if is_local_mode():
             # Update the cookies on the streamlit server
-            local_cookies = user_session.get_encoded_user_info()
+            local_cookies = user_session.get_encoded_user_info(local_user)
         return {
             **local_cookies,
             **{key: str(val) for key, val in st.context.cookies.items()},

python/src/numerous/local.py

@@ -0,0 +1,13 @@
+"""Local mode utilities."""
+
+from os import getenv
+
+from numerous.user import User
+
+
+local_user = User(id="local_user", name="Local User")
+
+
+def is_local_mode() -> bool:
+    url = getenv("NUMEROUS_API_URL")
+    return url is None

python/src/numerous/user_session.py

@@ -19,10 +19,8 @@ def encode_user_info(user_id: str, name: str) -> str:
     return base64.b64encode(user_info_json.encode("utf-8")).decode("utf-8")
 
 
-def get_encoded_user_info(
-    user_id: str = "local_user", name: str = "Local User"
-) -> Dict[str, str]:
-    return {"numerous_user_info": encode_user_info(user_id, name)}
+def get_encoded_user_info(user: User) -> Dict[str, str]:
+    return {"numerous_user_info": encode_user_info(user.id, user.name)}
 
 
 class Session:

python/src/numerous/utils.py

@@ -1,6 +1,5 @@
 """Miscellaneous utilities."""
 
-from os import getenv
 from typing import TypeVar
 
 
@@ -11,8 +10,3 @@ class _MissingType:
 MISSING = _MissingType()
 
 AppT = TypeVar("AppT")
-
-
-def is_local_mode() -> bool:
-    url = getenv("NUMEROUS_API_URL")
-    return url is None