mito-ai 0.1.37__tar.gz → 0.1.38__tar.gz

{mito_ai-0.1.37 → mito_ai-0.1.38}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mito_ai
-Version: 0.1.37
+Version: 0.1.38
 Summary: AI chat for JupyterLab
 Project-URL: Homepage, https://trymito.io
 Project-URL: Bug Tracker, https://github.com/mito-ds/monorepo/issues

{mito_ai-0.1.37 → mito_ai-0.1.38}/mito_ai/__init__.py

@@ -6,12 +6,14 @@ from jupyter_server.utils import url_path_join
 from mito_ai.completions.handlers import CompletionHandler
 from mito_ai.completions.providers import OpenAIProvider
 from mito_ai.app_builder.handlers import AppBuilderHandler
+from mito_ai.streamlit_preview.handlers import StreamlitPreviewHandler
 from mito_ai.log.urls import get_log_urls
 from mito_ai.version_check import VersionCheckHandler
 from mito_ai.db.urls import get_db_urls
 from mito_ai.settings.urls import get_settings_urls
 from mito_ai.rules.urls import get_rules_urls
 from mito_ai.auth.urls import get_auth_urls
+from mito_ai.streamlit_preview.urls import get_streamlit_preview_urls
 try:
     from _version import __version__
 except ImportError:
@@ -58,6 +60,11 @@ def _load_jupyter_server_extension(server_app) -> None: # type: ignore
             AppBuilderHandler,
             {}
         ),
+        (
+            url_path_join(base_url, "mito-ai", "streamlit-preview"),
+            StreamlitPreviewHandler,
+            {}
+        ),
         (
             url_path_join(base_url, "mito-ai", "version-check"),
             VersionCheckHandler,
@@ -69,8 +76,9 @@ def _load_jupyter_server_extension(server_app) -> None: # type: ignore
     handlers.extend(get_db_urls(base_url))  # type: ignore
     handlers.extend(get_settings_urls(base_url))  # type: ignore
     handlers.extend(get_rules_urls(base_url))  # type: ignore
-    handlers.extend(get_log_urls(base_url))  # type: ignore
+    handlers.extend(get_log_urls(base_url, open_ai_provider.key_type))  # type: ignore
     handlers.extend(get_auth_urls(base_url))  # type: ignore
+    handlers.extend(get_streamlit_preview_urls(base_url))  # type: ignore
     
     web_app.add_handlers(host_pattern, handlers)
     server_app.log.info("Loaded the mito_ai server extension")

{mito_ai-0.1.37 → mito_ai-0.1.38}/mito_ai/_version.py

@@ -1,4 +1,4 @@
 # This file is auto-generated by Hatchling. As such, do not:
 #   - modify
 #   - track in version control e.g. be sure to add to .gitignore
-__version__ = VERSION = '0.1.37'
+__version__ = VERSION = '0.1.38'

{mito_ai-0.1.37 → mito_ai-0.1.38}/mito_ai/app_builder/handlers.py

@@ -13,6 +13,7 @@ from mito_ai.utils.websocket_base import BaseWebSocketHandler
 from mito_ai.app_builder.models import (
     BuildAppReply,
     AppBuilderError,
+    BuildAppRequest,
     ErrorMessage,
     MessageType
 )
@@ -74,7 +75,8 @@ class AppBuilderHandler(BaseWebSocketHandler):
             
             if message_type == MessageType.BUILD_APP.value:
                 # Handle build app request
-                await self._handle_build_app(parsed_message)
+                build_app_request = BuildAppRequest(**parsed_message)
+                await self._handle_build_app(build_app_request)
             else:
                 self.log.error(f"Unknown message type: {message_type}")
                 error = AppBuilderError(
@@ -98,25 +100,24 @@ class AppBuilderHandler(BaseWebSocketHandler):
         latency_ms = round((time.time() - start) * 1000)
         self.log.info(f"App builder handler processed in {latency_ms} ms.")
     
-    async def _handle_build_app(self, message: dict) -> None:
+    async def _handle_build_app(self, message: BuildAppRequest) -> None:
         """Handle a build app request.
         
         Args:
             message: The parsed message.
         """
-        message_id = message.get('message_id', '')  # Default to empty string if not present
-        notebook_path = message.get('notebook_path')
-        app_path = message.get('app_path')
-        jwt_token = message.get('jwt_token', '')  # Extract JWT token from request, default to empty string
+        message_id = message.message_id
+        notebook_path = message.notebook_path
+        jwt_token = message.jwt_token
 
         if not message_id:
             self.log.error("Missing message_id in request")
             return
         
-        if not app_path:
+        if not notebook_path:
             error = AppBuilderError(
                 error_type="InvalidRequest",
-                title="Missing 'path' parameter"
+                title="Missing 'notebook_path' parameter"
             )
             self.reply(BuildAppReply(
                 parent_id=message_id,
@@ -126,31 +127,30 @@ class AppBuilderHandler(BaseWebSocketHandler):
             return
 
         # Validate JWT token if provided
-        if jwt_token and jwt_token != 'placeholder-jwt-token':
-            self.log.info(f"Validating JWT token: {jwt_token[:20]}...")
-            is_valid = self._validate_jwt_token(jwt_token)
-            if not is_valid:
-                self.log.error("JWT token validation failed")
-                error = AppBuilderError(
-                    error_type="Unauthorized",
-                    title="Invalid authentication token",
-                    hint="Please sign in again to deploy your app."
-                )
-                self.reply(BuildAppReply(
-                    parent_id=message_id,
-                    url="",
-                    error=error
-                ))
-                return
-            else:
-                self.log.info("JWT token validation successful")
+        token_preview = jwt_token[:20] if jwt_token else "No token provided"
+        self.log.info(f"Validating JWT token: {token_preview}...")
+        is_valid = self._validate_jwt_token(jwt_token) if jwt_token else False
+        if not is_valid or not jwt_token:
+            self.log.error("JWT token validation failed")
+            error = AppBuilderError(
+                error_type="Unauthorized",
+                title="Invalid authentication token",
+                hint="Please sign in again to deploy your app."
+            )
+            self.reply(BuildAppReply(
+                parent_id=message_id,
+                url="",
+                error=error
+            ))
+            return
         else:
-            self.log.warning("No JWT token provided or using placeholder token")
-
+            self.log.info("JWT token validation successful")
+            
         try:
 
-            success_flag, result_message = await streamlit_handler(str(notebook_path) if notebook_path else "", app_path)
-            if not success_flag:
+            notebook_path = str(notebook_path) if notebook_path else ""
+            success_flag, app_path, result_message = await streamlit_handler(notebook_path)
+            if not success_flag or app_path is None:
                 raise Exception(result_message)
 
             deploy_url = await self._deploy_app(app_path, jwt_token)

{mito_ai-0.1.37 → mito_ai-0.1.38}/mito_ai/app_builder/models.py

@@ -65,7 +65,7 @@ class BuildAppRequest:
     message_id: str
     
     # Path to the app file.
-    path: str
+    notebook_path: str
     
     # JWT token for authorization.
     jwt_token: Optional[str] = None

{mito_ai-0.1.37/venv/share/jupyter/labextensions/mito_ai → mito_ai-0.1.38/mito_ai/labextension}/build_log.json

@@ -710,7 +710,7 @@
             "semver": {},
             "vscode-diff": {},
             "mito_ai": {
-              "version": "0.1.37",
+              "version": "0.1.38",
               "singleton": true,
               "import": "/home/runner/work/mito/mito/mito-ai/lib/index.js"
             }

{mito_ai-0.1.37/venv/share/jupyter/labextensions/mito_ai → mito_ai-0.1.38/mito_ai/labextension}/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mito_ai",
-  "version": "0.1.37",
+  "version": "0.1.38",
   "description": "AI chat for JupyterLab",
   "keywords": [
     "jupyter",
@@ -138,7 +138,7 @@
     "outputDir": "mito_ai/labextension",
     "schemaDir": "schema",
     "_build": {
-      "load": "static/remoteEntry.93ecc9bc0edba61535cc.js",
+      "load": "static/remoteEntry.bcce4ea34631acf6dbbe.js",
       "extension": "./extension",
       "style": "./style"
     }

{mito_ai-0.1.37/venv/share/jupyter/labextensions/mito_ai → mito_ai-0.1.38/mito_ai/labextension}/schemas/mito_ai/package.json.orig

@@ -1,6 +1,6 @@
 {
   "name": "mito_ai",
-  "version": "0.1.37",
+  "version": "0.1.38",
   "description": "AI chat for JupyterLab",
   "keywords": [
     "jupyter",

{mito_ai-0.1.37/schema → mito_ai-0.1.38/mito_ai/labextension/schemas/mito_ai}/toolbar-buttons.json

@@ -27,10 +27,15 @@
         }
       ],
       "Notebook": [
+        {
+          "name": "preview-app",
+          "command": "toolbar-button:preview-as-streamlit",
+          "rank": 1000
+        },
         {
           "name": "convert-to-streamlit",
           "command": "toolbar-button:convert-to-streamlit",
-          "rank": 1000
+          "rank": 1001
         }
       ]
     }