sage-agent-internal 0.1.1__tar.gz

sage_agent_internal-0.1.1/.copier-answers.yml

@@ -0,0 +1,14 @@
+# Changes here will be overwritten by Copier; NEVER EDIT MANUALLY
+_commit: v4.3.8
+_src_path: https://github.com/jupyterlab/extension-template
+author_email: danielschaffield@gmail.com
+author_name: Daniel Schaffield
+has_binder: true
+has_settings: true
+kind: frontend
+labextension_name: signalpilot-ai-internal
+project_short_description: Sage Agent - Your Jupyter Notebook Assistant
+python_name: signalpilot_ai_internal
+repository: https://github.com/AlpineX-Labs/jupyter-signalpilot-ai-internal
+test: true
+

sage_agent_internal-0.1.1/.gitignore

@@ -0,0 +1,130 @@
+*.bundle.*
+lib/
+node_modules/
+*.log
+.eslintcache
+.stylelintcache
+*.egg-info/
+.ipynb_checkpoints
+*.tsbuildinfo
+signalpilot_ai_internal/labextension
+# Version file is handled by hatchling
+signalpilot_ai_internal/_version.py
+
+# Integration tests
+ui-tests/test-results/
+ui-tests/playwright-report/
+
+# Created by https://www.gitignore.io/api/python
+# Edit at https://www.gitignore.io/?templates=python
+
+### Python ###
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage/
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# SageMath parsed files
+*.sage.py
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# Mr Developer
+.mr.developer.cfg
+.project
+.pydevproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# End of https://www.gitignore.io/api/python
+
+# OSX files
+.DS_Store
+
+# Yarn cache
+.yarn/
+
+*.ipynb
+
+.env
+/ui-tests/test-results.json

sage_agent_internal-0.1.1/.prettierignore

@@ -0,0 +1,6 @@
+node_modules
+**/node_modules
+**/lib
+**/package.json
+!/package.json
+signalpilot_ai_internal

sage_agent_internal-0.1.1/.vscode/settings.json

@@ -0,0 +1,8 @@
+{
+  "github.copilot.enable": {
+    "*": true,
+    "plaintext": false,
+    "markdown": true,
+    "scminput": false
+  }
+}

sage_agent_internal-0.1.1/.yarnrc.yml

@@ -0,0 +1 @@
+nodeLinker: node-modules

sage_agent_internal-0.1.1/CHANGELOG.md

@@ -0,0 +1,5 @@
+# Changelog
+
+<!-- <START NEW CHANGELOG ENTRY> -->
+
+<!-- <END NEW CHANGELOG ENTRY> -->

sage_agent_internal-0.1.1/CLAUDE.md

@@ -0,0 +1,167 @@
+# CLAUDE.md
+
+This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
+
+## Repository Overview
+
+Sage Agent is a JupyterLab extension that provides AI-powered assistance for data analysis and notebook development. It integrates LLM capabilities directly into Jupyter notebooks with chat interface, code generation, and notebook manipulation tools.
+
+## Development Commands
+
+### Build and Development
+- `jlpm build` - Build the extension for development (includes source maps)
+- `jlpm build:prod` - Build for production (optimized, no source maps)
+- `jlpm watch` - Watch source files and rebuild automatically
+- `jlpm watch:src` - Watch TypeScript source files only
+- `jlpm watch:labextension` - Watch labextension files only
+
+### Development Setup
+- `pip install -e "."` - Install package in development mode
+- `jupyter labextension develop . --overwrite` - Link development version with JupyterLab
+- `jupyter lab` - Run JupyterLab with the extension
+
+### Testing and Quality
+- `jlpm test` - Run Jest tests with coverage
+- `jlpm lint` - Run all linters (stylelint, prettier, eslint) with fixes
+- `jlpm lint:check` - Check code quality without fixes
+- `jlpm eslint` - Run ESLint with fixes
+- `jlpm prettier` - Format code with Prettier
+
+### Clean Commands
+- `jlpm clean` - Clean compiled TypeScript files
+- `jlpm clean:all` - Clean all generated files (lib, labextension, cache)
+
+## Architecture Overview
+
+### Core Architecture Pattern
+The extension follows a centralized state management pattern using **AppStateService** as the single source of truth for all application state. This service manages:
+
+- Core services (ToolService, NotebookTracker, etc.)
+- UI containers (ChatContainer, SettingsContainer)
+- Notebook state and tracking
+- Configuration and settings
+
+### Key Architectural Components
+
+**Plugin System (src/plugin.ts:46-750)**
+- Main JupyterLab plugin entry point
+- Initializes all services in dependency order
+- Sets up event handlers for notebook changes
+- Registers commands and UI components
+
+**State Management (src/AppState.ts)**
+- Centralized state using RxJS BehaviorSubject
+- Event-driven architecture for notebook changes
+- Service initialization and dependency injection
+- Settings and configuration management
+
+**Chat System (src/Chat/)**
+- ConversationService: Manages chat interactions with LLMs
+- ChatInputManager: Handles user input and context
+- ActionHistory: Tracks user actions for context
+- RichTextChatInput: Advanced input component with markdown support
+
+**Notebook Management (src/Notebook/)**
+- NotebookTools: Core notebook manipulation utilities
+- NotebookDiffManager: Handles code diff display and approval
+- NotebookContextManager: Manages cell context for chat
+- InlineDiffService: Provides inline diff visualization
+
+**Services (src/Services/)**
+- ToolService: Orchestrates all notebook tools and operations
+- AnthropicService: Handles Claude API integration
+- ConfigService: Manages extension configuration
+- DatabaseMetadataCache: Caches database metadata for tools
+
+### Tool System Architecture
+
+The extension provides structured tools for LLM interaction defined in `src/Config/tools.json`:
+
+**Notebook Tools:**
+- `notebook-edit_plan` - Manages notebook planning workflow
+- `notebook-read_cells` - Reads specific notebook cells  
+- `notebook-add_cell` - Creates new cells with tracking
+- `notebook-edit_cell` - Modifies existing cell content
+- `notebook-run_cell` - Executes code cells
+- `notebook-remove_cells` - Deletes specified cells
+- `notebook-wait_user_reply` - UI component for user interaction
+
+**Data Tools:**
+- `filesystem-read_dataset` - Reads dataset files from data directory
+- `web-search_dataset` - Searches ticker/financial data with fuzzy matching
+
+### Configuration System
+
+**Model Configuration (src/Config/models.json)**
+- Different models for different interaction modes
+- Tool blacklists per model to optimize performance
+- Supports Claude Sonnet 4 and Haiku models
+
+**Prompt System (src/Config/prompts/)**
+- Specialized system prompts for different modes
+- `claude_system_prompt.md` - Main data science assistant prompt
+- Mode-specific prompts for ask, edit, and fast modes
+
+## Important Development Patterns
+
+### Service Initialization Order
+1. Core services (ToolService, NotebookTracker, NotebookTools)
+2. Managers (PlanStateDisplay, WaitingUserReplyBoxManager)  
+3. Additional services (ActionHistory, CellTrackingService)
+4. UI containers (ChatContainer, SettingsContainer)
+
+### Notebook Unique ID System
+- Each notebook gets a unique `sage_ai.unique_id` in metadata
+- Used for tracking chat histories and state across renames
+- Fallback to path-based tracking for compatibility
+
+### Context Management
+- Cells can be added/removed from chat context
+- Visual highlighting shows which cells are in context
+- Context preserved across notebook operations
+
+### Diff Approval Workflow
+- Code changes shown as diffs before execution
+- User approval required for cell modifications
+- Inline diff visualization with syntax highlighting
+
+## Key File Locations
+
+### Source Structure
+- `src/plugin.ts` - Main plugin entry point and initialization
+- `src/AppState.ts` - Centralized state management
+- `src/Config/` - Configuration files (tools, models, prompts)
+- `src/Services/` - Core business logic services
+- `src/Notebook/` - Notebook-specific functionality
+- `src/Chat/` - Chat interface and conversation management
+- `src/Components/` - Reusable UI components
+
+### Build Outputs
+- `lib/` - Compiled TypeScript output
+- `signalpilot_ai_internal/labextension/` - JupyterLab extension package
+- `style/` - CSS stylesheets and icons
+
+## Testing Strategy
+
+- Jest for unit tests (`__tests__/` and `ui-tests/`)
+- Playwright for integration tests
+- Test configuration in `jest.config.js`
+- Coverage reporting enabled
+
+## Extension Development Notes
+
+### JupyterLab Integration
+- Requires JupyterLab >= 4.0.0
+- Uses JupyterLab's plugin system for initialization
+- Integrates with notebook tracker and command palette
+- Supports JupyterLab theming system
+
+### Database Integration
+- PGLite integration for local database operations
+- Database URL management through kernel environment
+- Metadata caching for performance optimization
+
+### Theme Support
+- Automatic dark theme application on first load
+- Diff visualization adapts to JupyterLab theme
+- CSS custom properties for theme consistency

sage_agent_internal-0.1.1/LICENSE

@@ -0,0 +1,29 @@
+BSD 3-Clause License
+
+Copyright (c) 2025, Daniel Schaffield
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

sage_agent_internal-0.1.1/PKG-INFO

@@ -0,0 +1,207 @@
+Metadata-Version: 2.4
+Name: sage_agent_internal
+Version: 0.1.1
+Summary: Sage Agent - Your Jupyter Notebook Assistant
+Project-URL: Homepage, https://sagebook.ai/
+Author-email: Daniel Schaffield <danielschaffield@gmail.com>
+License: BSD 3-Clause License
+        
+        Copyright (c) 2025, Daniel Schaffield
+        All rights reserved.
+        
+        Redistribution and use in source and binary forms, with or without
+        modification, are permitted provided that the following conditions are met:
+        
+        1. Redistributions of source code must retain the above copyright notice, this
+           list of conditions and the following disclaimer.
+        
+        2. Redistributions in binary form must reproduce the above copyright notice,
+           this list of conditions and the following disclaimer in the documentation
+           and/or other materials provided with the distribution.
+        
+        3. Neither the name of the copyright holder nor the names of its
+           contributors may be used to endorse or promote products derived from
+           this software without specific prior written permission.
+        
+        THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+        AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+        IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+        DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+        FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+        DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+        SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+        CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+        OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+        OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+License-File: LICENSE
+Keywords: jupyter,jupyterlab,jupyterlab-extension
+Classifier: Framework :: Jupyter
+Classifier: Framework :: Jupyter :: JupyterLab
+Classifier: Framework :: Jupyter :: JupyterLab :: 4
+Classifier: Framework :: Jupyter :: JupyterLab :: Extensions
+Classifier: Framework :: Jupyter :: JupyterLab :: Extensions :: Prebuilt
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.6
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+
+# **Sage Agent — The AI-Powered Assistant for Jupyter Notebooks**
+
+![Logo](https://i.imgur.com/JdA8ilQ.png)
+
+---
+
+## **What is Sage Agent?**
+
+**Sage is an AI-native notebook assistant that supercharges your existing Jupyter workflows.**
+
+Built by leading AI and quant researchers from YC, Harvard, MIT, and Goldman Sachs, Sage brings real-time, context-aware assistance directly into JupyterLab.
+
+Use natural language to clean data, write analysis code, debug errors, explore dataframes, and build models—faster and with fewer mistakes.
+
+**No hallucinated code. No context switching. Just faster insights.**
+
+---
+
+## **Why Use Sage Agent in Jupyter?**
+
+Whether you’re a quant, data scientist, or analyst living in notebooks, Sage helps you:
+
+✅ Clean and transform messy data in seconds
+
+✅ Visualize trends, rollups, and anomalies from a prompt
+
+✅ Connect your custom databases in one click and easily explore from notebooks 
+
+✅ Generate *runnable* Python or SQL that fits your current cell + variable context
+
+✅ Auto-detect schema changes and debug downstream errors
+
+✅ Stay private: run entirely *local-first* or in your own secure VPC
+
+✅ Extend JupyterLab without changing how you work
+
+---
+
+## **Perfect For:**
+
+- Data scientists cleaning huge CSVs
+- Quant researchers testing ML pipelines
+- Product and analytics teams tired of building dashboards and flaky notebooks
+- Anyone tired of LLM tools that break their code
+
+---
+
+## **Installation**
+
+### **📦 Requirements**
+
+- JupyterLab >= 4.0.0
+- NodeJS (for development)
+
+### **🧠 Install Sage Agent:**
+
+```
+pip install jupyterlab signalpilot_ai_internal
+```
+
+### **❌ Uninstall:**
+
+```
+pip uninstall signalpilot_ai_internal
+```
+
+---
+
+## **How to Get Started**
+
+To unlock full functionality, you’ll need Sage API credentials.
+
+👉 [**Request your API key**](https://sagebook.ai/#contact) or email us at [contact@sagebook.ai](mailto:contact@sagebook.ai)
+
+---
+
+## **Why Sage**
+
+- ✅ Context-aware code gen: understands variables, dataframes, imports, and prior cells
+- ✅ AI that *fixes* schema issues and silent join bugs
+- ✅ Inline review + diffs before you run any code
+- ✅ Visualizations via natural language (matplotlib, plotly, seaborn supported)
+- ✅ BYO LLM: Anthropic, OpenAI, vLLM, Ollama, or HF endpoints
+- ✅ Built to run in air-gapped / enterprise environments
+
+---
+
+## **Local Development Instructions**
+
+To contribute or develop locally:
+
+```
+# Clone the repo and enter the directory
+git clone https://github.com/sagebook/signalpilot_ai_internal.git
+cd signalpilot_ai_internal
+
+# Install in editable mode
+pip install -e "."
+
+# Link extension to JupyterLab
+jupyter labextension develop . --overwrite
+
+# Rebuild on changes
+jlpm build
+```
+
+For auto-rebuild while editing:
+
+```
+# Watch source
+jlpm watch
+
+# Run JupyterLab in parallel
+jupyter lab
+```
+
+---
+
+## **Uninstall in Dev Mode**
+
+```
+pip uninstall signalpilot_ai_internal
+# Then manually remove labextension symlink from JupyterLab extensions dir.
+```
+
+---
+
+## **Want to See Sage in Action?**
+
+🎥 Try the demo notebook or explore at [https://sagebook.ai](https://sagebook.ai/)
+
+---
+
+**Built for teams working with sensitive data:**
+
+- Zero data retention by default
+- Optional BYO keys for Claude, OpenAI, or local models
+- Notebook-specific controls for what the model can “see”
+- Fine-grained telemetry settings
+
+---
+
+## **Contact**
+
+Questions? Ideas?
+
+Email: [fahim@sagebook.ai](mailto:fahim@sagebook.ai)
+
+Website: [https://sagebook.ai](https://sagebook.ai/)
+
+---
+
+AI Jupyter Notebook, JupyterLab Extension, Jupyter Assistant, Data Science Assistant, Jupyter LLM, AI code generation, dataframe cleaning, Jupyter AI, Sagebook, Sage Agent, AI for dataframes, Jupyter SQL assistant, notebook extension