refactor package for qbraid

Author: TheGupta2012

.github/workflows/build.yml

@@ -36,10 +36,10 @@ jobs:
         python -m pip install .[test]
 
         jupyter server extension list
-        jupyter server extension list 2>&1 | grep -ie "notebook_intelligence.*OK"
+        jupyter server extension list 2>&1 | grep -ie "lab_notebook_intelligence.*OK"
 
         jupyter labextension list
-        jupyter labextension list 2>&1 | grep -ie "@notebook-intelligence/notebook-intelligence.*OK"
+        jupyter labextension list 2>&1 | grep -ie "@qbraid/lab-notebook-intelligence.*OK"
         python -m jupyterlab.browser_check
 
     - name: Package the extension
@@ -48,13 +48,13 @@ jobs:
 
         pip install build
         python -m build
-        pip uninstall -y "notebook_intelligence" jupyterlab
+        pip uninstall -y "lab_notebook_intelligence" jupyterlab
 
     - name: Upload extension packages
       uses: actions/upload-artifact@v4
       with:
         name: extension-artifacts
-        path: dist/notebook_intelligence*
+        path: dist/lab_notebook_intelligence*
         if-no-files-found: error
 
   test_isolated:
@@ -77,14 +77,14 @@ jobs:
         sudo rm -rf $(which node)
         sudo rm -rf $(which node)
 
-        pip install "jupyterlab>=4.0.0,<5" notebook_intelligence*.whl
+        pip install "jupyterlab>=4.0.0,<5" lab_notebook_intelligence*.whl
 
 
         jupyter server extension list
-        jupyter server extension list 2>&1 | grep -ie "notebook_intelligence.*OK"
+        jupyter server extension list 2>&1 | grep -ie "lab_notebook_intelligence.*OK"
 
         jupyter labextension list
-        jupyter labextension list 2>&1 | grep -ie "@notebook-intelligence/notebook-intelligence.*OK"
+        jupyter labextension list 2>&1 | grep -ie "@qbraid/lab-notebook-intelligence.*OK"
         python -m jupyterlab.browser_check --no-browser-test
 
 

.github/workflows/upload-s3-production.yml

@@ -63,7 +63,7 @@ jobs:
         if: matrix.python-version == '3.11'
         with:
           name: extension
-          path: dist/quantum_jobs*.whl
+          path: dist/lab_notebook_intelligence*.whl
           if-no-files-found: error
 
   deploy:
@@ -87,5 +87,5 @@ jobs:
 
       - name: Copy wheel file to S3 lab-extensions bucket
         run: |
-          aws s3 rm s3://qbraid-lab-extensions/production/ --recursive --exclude "*" --include "notebook_intelligence*.whl"
-          aws s3 cp ./ s3://qbraid-lab-extensions/production/ --recursive --exclude "*" --include "notebook_intelligence*.whl"
+          aws s3 rm s3://qbraid-lab-extensions/production/ --recursive --exclude "*" --include "lab_notebook_intelligence*.whl"
+          aws s3 cp ./ s3://qbraid-lab-extensions/production/ --recursive --exclude "*" --include "lab_notebook_intelligence*.whl"

.github/workflows/upload-s3-staging.yml

@@ -64,7 +64,7 @@ jobs:
         if: matrix.python-version == '3.11'
         with:
           name: extension
-          path: dist/notebook_intelligence*.whl
+          path: dist/lab_notebook_intelligence*.whl
           if-no-files-found: error
 
   deploy:
@@ -88,5 +88,5 @@ jobs:
 
       - name: Copy wheel file to S3 lab-extensions bucket
         run: |
-          aws s3 rm s3://qbraid-lab-extensions/staging/ --recursive --exclude "*" --include "notebook_intelligence*.whl"
-          aws s3 cp ./ s3://qbraid-lab-extensions/staging/ --recursive --exclude "*" --include "notebook_intelligence*.whl"
+          aws s3 rm s3://qbraid-lab-extensions/staging/ --recursive --exclude "*" --include "lab_notebook_intelligence*.whl"
+          aws s3 cp ./ s3://qbraid-lab-extensions/staging/ --recursive --exclude "*" --include "lab_notebook_intelligence*.whl"

.gitignore

@@ -7,9 +7,9 @@ node_modules/
 *.egg-info/
 .ipynb_checkpoints
 *.tsbuildinfo
-notebook_intelligence/labextension
+lab_notebook_intelligence/labextension
 # Version file is handled by hatchling
-notebook_intelligence/_version.py
+lab_notebook_intelligence/_version.py
 
 # Created by https://www.gitignore.io/api/python
 # Edit at https://www.gitignore.io/?templates=python

install.json

@@ -1,5 +1,5 @@
 {
   "packageManager": "python",
-  "packageName": "notebook_intelligence",
-  "uninstallInstructions": "Use your Python package manager (pip, conda, etc.) to uninstall the package notebook_intelligence"
+  "packageName": "lab_notebook_intelligence",
+  "uninstallInstructions": "Use your Python package manager (pip, conda, etc.) to uninstall the package lab_notebook_intelligence"
 }

jupyter-config/server-config/notebook_intelligence.json

@@ -1,7 +1,7 @@
 {
   "ServerApp": {
     "jpserver_extensions": {
-      "notebook_intelligence": true
+      "lab_notebook_intelligence": true
     }
   }
 }

notebook_intelligence/__init__.py lab_notebook_intelligence/__init__.pynotebook_intelligence/__init__.py renamed to lab_notebook_intelligence/__init__.py

@@ -7,7 +7,7 @@
     # in editable mode with pip. It is highly recommended to install
     # the package from a stable release or in editable mode: https://pip.pypa.io/en/stable/topics/local-project-installs/#editable-installs
     import warnings
-    warnings.warn("Importing 'notebook_intelligence' outside a proper installation.")
+    warnings.warn("Importing 'lab_notebook_intelligence' outside a proper installation.")
     __version__ = "dev"
 
 import logging
@@ -19,12 +19,12 @@
 def _jupyter_labextension_paths():
     return [{
         "src": "labextension",
-        "dest": "@notebook-intelligence/notebook-intelligence"
+        "dest": "@qbraid/lab-notebook-intelligence"
     }]
 
 
 def _jupyter_server_extension_points():
     return [{
-        "module": "notebook_intelligence",
+        "module": "lab_notebook_intelligence",
         "app": NotebookIntelligence
     }]

…ebook_intelligence/ai_service_manager.py …ebook_intelligence/ai_service_manager.pynotebook_intelligence/ai_service_manager.py renamed to lab_notebook_intelligence/ai_service_manager.py notebook_intelligence/ai_service_manager.py renamed to lab_notebook_intelligence/ai_service_manager.py

@@ -6,16 +6,16 @@
 import sys
 from typing import Dict
 import logging
-from notebook_intelligence import github_copilot
-from notebook_intelligence.api import ButtonData, ChatModel, EmbeddingModel, InlineCompletionModel, LLMProvider, ChatParticipant, ChatRequest, ChatResponse, CompletionContext, ContextRequest, Host, CompletionContextProvider, MCPServer, MarkdownData, NotebookIntelligenceExtension, TelemetryEvent, TelemetryListener, Tool, Toolset
-from notebook_intelligence.base_chat_participant import BaseChatParticipant
-from notebook_intelligence.config import NBIConfig
-from notebook_intelligence.github_copilot_chat_participant import GithubCopilotChatParticipant
-from notebook_intelligence.llm_providers.github_copilot_llm_provider import GitHubCopilotLLMProvider
-from notebook_intelligence.llm_providers.litellm_compatible_llm_provider import LiteLLMCompatibleLLMProvider
-from notebook_intelligence.llm_providers.ollama_llm_provider import OllamaLLMProvider
-from notebook_intelligence.llm_providers.openai_compatible_llm_provider import OpenAICompatibleLLMProvider
-from notebook_intelligence.mcp_manager import MCPManager
+from lab_notebook_intelligence import github_copilot
+from lab_notebook_intelligence.api import ButtonData, ChatModel, EmbeddingModel, InlineCompletionModel, LLMProvider, ChatParticipant, ChatRequest, ChatResponse, CompletionContext, ContextRequest, Host, CompletionContextProvider, MCPServer, MarkdownData, NotebookIntelligenceExtension, TelemetryEvent, TelemetryListener, Tool, Toolset
+from lab_notebook_intelligence.base_chat_participant import BaseChatParticipant
+from lab_notebook_intelligence.config import NBIConfig
+from lab_notebook_intelligence.github_copilot_chat_participant import GithubCopilotChatParticipant
+from lab_notebook_intelligence.llm_providers.github_copilot_llm_provider import GitHubCopilotLLMProvider
+from lab_notebook_intelligence.llm_providers.litellm_compatible_llm_provider import LiteLLMCompatibleLLMProvider
+from lab_notebook_intelligence.llm_providers.ollama_llm_provider import OllamaLLMProvider
+from lab_notebook_intelligence.llm_providers.openai_compatible_llm_provider import OpenAICompatibleLLMProvider
+from lab_notebook_intelligence.mcp_manager import MCPManager
 
 log = logging.getLogger(__name__)
 
@@ -288,7 +288,7 @@ def get_chat_participant(self, prompt: str) -> ChatParticipant:
     async def handle_chat_request(self, request: ChatRequest, response: ChatResponse, options: dict = {}) -> None:
         if self.chat_model is None:
             response.stream(MarkdownData("Chat model is not set!"))
-            response.stream(ButtonData("Configure", "notebook-intelligence:open-configuration-dialog"))
+            response.stream(ButtonData("Configure", "lab-notebook-intelligence:open-configuration-dialog"))
             response.finish()
             return
         request.host = self

notebook_intelligence/api.py lab_notebook_intelligence/api.pynotebook_intelligence/api.py renamed to lab_notebook_intelligence/api.py

@@ -1,7 +1,6 @@
 # Copyright (c) Mehmet Bektas <[email protected]>
 
 import asyncio
-import json
 from typing import Any, Callable, Dict, Union
 from dataclasses import asdict, dataclass
 from enum import Enum
@@ -10,7 +9,7 @@
 import logging
 from mcp.server.fastmcp.tools import Tool as MCPToolClass
 
-from notebook_intelligence.config import NBIConfig
+from lab_notebook_intelligence.config import NBIConfig
 
 log = logging.getLogger(__name__)
 

…ok_intelligence/base_chat_participant.py …ok_intelligence/base_chat_participant.pynotebook_intelligence/base_chat_participant.py renamed to lab_notebook_intelligence/base_chat_participant.py notebook_intelligence/base_chat_participant.py renamed to lab_notebook_intelligence/base_chat_participant.py

@@ -3,13 +3,13 @@
 import os
 from typing import Union
 import json
-from notebook_intelligence.api import ChatCommand, ChatParticipant, ChatRequest, ChatResponse, MarkdownData, ProgressData, Tool, ToolPreInvokeResponse
-from notebook_intelligence.prompts import Prompts
+from lab_notebook_intelligence.api import ChatCommand, ChatParticipant, ChatRequest, ChatResponse, MarkdownData, ProgressData, Tool, ToolPreInvokeResponse
+from lab_notebook_intelligence.prompts import Prompts
 import base64
 import logging
-from notebook_intelligence.built_in_toolsets import built_in_toolsets
+from lab_notebook_intelligence.built_in_toolsets import built_in_toolsets
 
-from notebook_intelligence.util import extract_llm_generated_code
+from lab_notebook_intelligence.util import extract_llm_generated_code
 
 log = logging.getLogger(__name__)
 
@@ -120,17 +120,17 @@ def pre_invoke(self, request: ChatRequest, tool_args: dict) -> Union[ToolPreInvo
     async def handle_tool_call(self, request: ChatRequest, response: ChatResponse, tool_context: dict, tool_args: dict) -> str:
         cell_sources = tool_args.get('cell_sources', [])
 
-        ui_cmd_response = await response.run_ui_command('notebook-intelligence:create-new-notebook-from-py', {'code': ''})
+        ui_cmd_response = await response.run_ui_command('lab-notebook-intelligence:create-new-notebook-from-py', {'code': ''})
         file_path = ui_cmd_response['path']
 
         for cell_source in cell_sources:
             cell_type = cell_source.get('cell_type')
             if cell_type == 'markdown':
                 source = cell_source.get('source', '')
-                ui_cmd_response = await response.run_ui_command('notebook-intelligence:add-markdown-cell-to-notebook', {'markdown': source, 'path': file_path})
+                ui_cmd_response = await response.run_ui_command('lab-notebook-intelligence:add-markdown-cell-to-notebook', {'markdown': source, 'path': file_path})
             elif cell_type == 'code':
                 source = cell_source.get('source', '')
-                ui_cmd_response = await response.run_ui_command('notebook-intelligence:add-code-cell-to-notebook', {'code': source, 'path': file_path})
+                ui_cmd_response = await response.run_ui_command('lab-notebook-intelligence:add-code-cell-to-notebook', {'code': source, 'path': file_path})
 
         return "Notebook created successfully at {file_path}"
 
@@ -195,7 +195,7 @@ async def handle_tool_call(self, request: ChatRequest, response: ChatResponse, t
         if notebook_file_path.startswith(server_root_dir):
             notebook_file_path = os.path.relpath(notebook_file_path, server_root_dir)
         source = tool_args.get('markdown_cell_source')
-        ui_cmd_response = await response.run_ui_command('notebook-intelligence:add-markdown-cell-to-notebook', {'markdown': source, 'path': notebook_file_path})
+        ui_cmd_response = await response.run_ui_command('lab-notebook-intelligence:add-markdown-cell-to-notebook', {'markdown': source, 'path': notebook_file_path})
         return f"Added markdown cell to notebook"
 
 class AddCodeCellTool(Tool):
@@ -259,7 +259,7 @@ async def handle_tool_call(self, request: ChatRequest, response: ChatResponse, t
         if notebook_file_path.startswith(server_root_dir):
             notebook_file_path = os.path.relpath(notebook_file_path, server_root_dir)
         source = tool_args.get('code_cell_source')
-        ui_cmd_response = await response.run_ui_command('notebook-intelligence:add-code-cell-to-notebook', {'code': source, 'path': notebook_file_path})
+        ui_cmd_response = await response.run_ui_command('lab-notebook-intelligence:add-code-cell-to-notebook', {'code': source, 'path': notebook_file_path})
         return "Added code cell added to notebook"
 
 # Fallback tool to handle tool errors
@@ -304,7 +304,7 @@ def schema(self) -> dict:
 
     async def handle_tool_call(self, request: ChatRequest, response: ChatResponse, tool_context: dict, tool_args: dict) -> str:
         code = tool_args.get('code_cell_source')
-        ui_cmd_response = await response.run_ui_command('notebook-intelligence:add-code-cell-to-notebook', {'code': code, 'path': tool_context.get('file_path')})
+        ui_cmd_response = await response.run_ui_command('lab-notebook-intelligence:add-code-cell-to-notebook', {'code': code, 'path': tool_context.get('file_path')})
         return {"result": "Code cell added to notebook"}
 
 class BaseChatParticipant(ChatParticipant):
@@ -426,14 +426,14 @@ async def handle_ask_mode_chat_request(self, request: ChatRequest, response: Cha
         chat_model = request.host.chat_model
         if request.command == 'newNotebook':
             # create a new notebook
-            ui_cmd_response = await response.run_ui_command('notebook-intelligence:create-new-notebook-from-py', {'code': ''})
+            ui_cmd_response = await response.run_ui_command('lab-notebook-intelligence:create-new-notebook-from-py', {'code': ''})
             file_path = ui_cmd_response['path']
 
             code = await self.generate_code_cell(request)
             markdown = await self.generate_markdown_for_code(request, code)
 
-            ui_cmd_response = await response.run_ui_command('notebook-intelligence:add-markdown-cell-to-notebook', {'markdown': markdown, 'path': file_path})
-            ui_cmd_response = await response.run_ui_command('notebook-intelligence:add-code-cell-to-notebook', {'code': code, 'path': file_path})
+            ui_cmd_response = await response.run_ui_command('lab-notebook-intelligence:add-markdown-cell-to-notebook', {'markdown': markdown, 'path': file_path})
+            ui_cmd_response = await response.run_ui_command('lab-notebook-intelligence:add-code-cell-to-notebook', {'code': code, 'path': file_path})
 
             response.stream(MarkdownData(f"Notebook '{file_path}' created and opened successfully"))
             response.finish()
@@ -447,13 +447,13 @@ async def handle_ask_mode_chat_request(self, request: ChatRequest, response: Cha
             generated = chat_model.completions(messages)
             code = generated['choices'][0]['message']['content']
             code = extract_llm_generated_code(code)
-            ui_cmd_response = await response.run_ui_command('notebook-intelligence:create-new-file', {'code': code })
+            ui_cmd_response = await response.run_ui_command('lab-notebook-intelligence:create-new-file', {'code': code })
             file_path = ui_cmd_response['path']
             response.stream(MarkdownData(f"File '{file_path}' created successfully"))
             response.finish()
             return
         elif request.command == 'settings':
-            ui_cmd_response = await response.run_ui_command('notebook-intelligence:open-configuration-dialog')
+            ui_cmd_response = await response.run_ui_command('lab-notebook-intelligence:open-configuration-dialog')
             response.stream(MarkdownData(f"Opened the settings dialog"))
             response.finish()
             return