V2.7.0 (#100)
* feat(memory): replace FalkorDB with LadybugDB embedded database Remove Docker dependency for Graphiti memory integration by switching to LadybugDB, an embedded graph database that works via monkeypatch with graphiti-core's KuzuDriver. Changes: - Remove all FalkorDB configuration and connection code - Simplify config to use GRAPHITI_DB_PATH for local storage - Update requirements to use real_ladybug + graphiti-core - Update documentation for Python 3.12+ requirement This makes the memory system much simpler to set up - no Docker needed. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * chore: remove docker-compose.yml (FalkorDB no longer used) FalkorDB has been replaced with LadybugDB embedded database, so Docker is no longer required for the memory integration. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * feat: add automated release workflow with code signing - Add release.yml workflow triggered on version tags (v*) - Support manual dry-run builds via workflow_dispatch - Build for macOS (arm64 + x64), Windows, and Linux - macOS code signing with Developer ID certificate - macOS notarization support for Gatekeeper approval - Add entitlements.mac.plist for hardened runtime - Generate SHA256 checksums for all release artifacts - Auto-generate changelog from PR labels - Add pnpm caching to CI workflow for faster builds - Add lint, typecheck, and build steps to CI - Update package.json with artifactName for consistent naming - Fix extraResources filter to exclude test venvs Artifacts will be named: Auto-Claude-{version}-{platform}-{arch}.{ext} 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * refactor: replace Docker/FalkorDB with embedded LadybugDB for memory system This major refactoring eliminates the Docker dependency for the memory system by switching from FalkorDB to LadybugDB (embedded graph database). Backend changes: - Add query_memory.py: Python CLI for memory queries via subprocess - Add kuzu_driver_patched.py: Patched Kuzu driver with FTS index support - Add ollama_model_detector.py: Auto-detect Ollama embedding models - Update client.py to use patched driver for proper FTS functionality - Fix parameter handling for LadybugDB compatibility Frontend changes: - Add memory-service.ts: Node.js service wrapping Python subprocess - Add memory-handlers.ts: IPC handlers for memory operations - Add api-validation-service.ts: Validate LLM/embedder API keys - Remove docker-service.ts and falkordb-service.ts (no longer needed) - Update MemoryBackendSection with multi-provider embedder support - Update InfrastructureStatus to show LadybugDB status - Simplify GraphitiStep onboarding (no Docker setup required) Key improvements: - Zero Docker dependency - fully embedded database - Multi-provider embedder support (OpenAI, Gemini, Voyage, Ollama, Azure) - Hybrid RAG with semantic search, FTS, and graph traversal - Automatic FTS index creation via patched driver - Python 3.12 compatibility (LadybugDB requirement) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: correct model name and release workflow conditionals Bug fixes: - Fix model ID: claude-sonnet-4-5-latest → claude-sonnet-4-5-20250514 (using dated version for stability) - Fix release workflow: add github.event_name checks before accessing inputs.dry_run to prevent errors on tag push events 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * feat: add Ollama embedding model support with auto-detected dimensions - Add known model lookup table for popular Ollama embedding models: - embeddinggemma (768 dim) - Google's lightweight model - qwen3-embedding:0.6b/4b/8b (1024/2560/4096 dim) - Qwen3 series - nomic-embed-text, mxbai-embed-large, bge-large, all-minilm - Auto-detect embedding dimensions for known models (no need to set OLLAMA_EMBEDDING_DIM) - Fix config validation to not require OLLAMA_EMBEDDING_DIM for known models - Fix PatchedKuzuDriver to set _database attribute (required by Graphiti) - Fix get_status_summary to use db_path instead of deprecated falkordb_host 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * refactor: rebrand memory system UI and simplify configuration - Rename "Graphiti" to "Memory" throughout UI - Remove LLM provider selection (Claude SDK handles RAG) - Keep embedding provider selection (OpenAI, Ollama, Voyage, Google, Azure) - Add Ollama model pull/download support - Change default storage path from ~/.auto-claude/graphs to ~/.auto-claude/memories - Make embedding provider fields conditional based on selection - Add OllamaModelSelector component with auto-detection - Create simplified MemoryStep for onboarding wizard - Update SecuritySettings with provider-specific field rendering 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix(ui): replace Unix shell syntax with cross-platform git commands Fixes #90 Worktrees were showing "0" for all file changes on Windows because Unix shell constructs (`2>/dev/null || echo`) are invalid in cmd.exe. Changes: - Replace `2>/dev/null || echo` with TypeScript try-catch - Add `stdio: ['pipe', 'pipe', 'pipe']` to capture stderr - Fix 7 locations in worktree-handlers.ts: - git rev-list --count (2 locations) - git diff --stat - git diff --numstat - git diff --name-status - git diff --shortstat - git merge-base --is-ancestor The error "The system cannot find the path specified" was caused by cmd.exe trying to interpret `/dev/null` as a literal file path. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> * fix(cli): update graphiti status display for LadybugDB After migrating from FalkorDB (Docker-based) to LadybugDB (embedded), the validate_environment function tried to access 'host' and 'port' keys that no longer exist in the graphiti status dictionary. Changes: - Replace host:port display with db_path for embedded database - Use .get() for safe key access This fixes a KeyError crash when running auto-claude builds with Graphiti memory enabled. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> * fix: improve Ollama UX in memory settings - Replace manual Ollama config with OllamaModelSelector component - Remove Base URL field (auto-detected from Ollama) - Remove manual embedding dimension input (auto-detected from model) - Show only installed models as selectable options - Add download buttons for recommended models not yet installed - Make embeddinggemma the recommended default model - Remove qwen3-embedding from recommendations (focus on quality models) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * chore: update CI and release workflows, remove changelog config - Removed the obsolete changelog configuration file. - Updated CI workflow to use `pnpm run test` instead of `pnpm test`. - Modified release workflow to use `pnpm install --frozen-lockfile` for consistent dependency installation. - Enhanced artifact handling by validating the presence of build artifacts before proceeding. These changes streamline the workflows and improve reliability in the build process. * fix: resolve all CI failures in PR #100 Python lint fixes (ruff): - Fix import sorting in kuzu_driver_patched.py, ollama_model_detector.py, query_memory.py - Add noqa: F401 for kuzu import used only for availability check Frontend: - Update pnpm-lock.yaml to remove ioredis dependencies Test fixes: - Update test_graphiti.py to reflect new multi-provider architecture - Embedder is now optional (keyword search fallback works) - LLM provider validation removed (Claude SDK handles RAG) - Fix FalkorDB references to LadybugDB (db_path instead of host/port) Config consistency: - Fix get_graphiti_status() to be consistent with is_valid() - Embedder errors are now warnings, not blockers 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * refactor: update memory test suite for LadybugDB Replace FalkorDB-based tests with LadybugDB (embedded database) tests: - test_ladybugdb_connection() - Verify embedded DB works - test_save_episode() - Save test data to graph - test_keyword_search() - Keyword fallback (no embeddings needed) - test_semantic_search() - Vector search with embeddings - test_ollama_embeddings() - Direct Ollama embedding test - test_graphiti_memory_class() - Full wrapper class test - test_database_contents() - Debug view of DB contents Usage: python integrations/graphiti/test_graphiti_memory.py --test ollama python integrations/graphiti/test_graphiti_memory.py --test connection python integrations/graphiti/test_graphiti_memory.py # all tests 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: resolve remaining CI failures Python: - Add missing blank line after imports in query_memory.py (ruff I001) TypeScript: - Fix mock getBestAvailableProfile signature to accept optional parameter 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: remove f-string prefixes from strings without placeholders Fixes ruff F541 errors in test_graphiti_memory.py 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * style: apply ruff formatting to 4 files Auto-formatted kuzu_driver_patched.py and test_graphiti_memory.py 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: address CodeRabbit review comments from PR #100 - release.yml: Fix changelog output property (outputs.body not outputs.changelog) - config.py: Update Anthropic model to generic claude-sonnet-4-5 identifier - api-validation-service.ts: Fix error message to include both sk- and sess- prefixes - ci.yml: Add quotes around command substitution for safety - OllamaModelSelector.tsx: Add error logging to catch block - OllamaModelSelector.tsx: Add AbortController cleanup pattern for unmount - SecuritySettings.tsx: Add useEffect to sync showOpenAIKey prop changes - SecuritySettings.tsx: Fix stale state in toggleShowApiKey - SecuritySettings.tsx: Add aria-labels to password visibility buttons Verified: nomic-embed-text uses 768 dimensions (CodeRabbit was incorrect about 1024) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * docs: add CodeRabbit review response tracking Document rejected CodeRabbit suggestions with justification: - nomic-embed-text uses 768 dims (not 1024 as claimed) - MemoryStep checkmark UX is intentional design 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: sort imports in memory.py for ruff I001 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: address additional CodeRabbit review comments Security fixes: - kuzu_driver_patched.py: Add try/finally blocks for connection cleanup - query_memory.py: Use parameterized queries to prevent Cypher injection - query_memory.py: Use public get_validation_errors() instead of private method - release.yml: Use ./* glob pattern to prevent option injection Code quality: - ollama_model_detector.py: Fix type annotation (str | None) - cleanup-version-branches.sh: Fix empty tag count edge case - config.py: Remove unused _validate_llm_provider method 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: update test and apply ruff formatting - Remove test assertion for deleted _validate_llm_provider method - Apply ruff format to query_memory.py 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: address CodeRabbit review feedback - kuzu_driver_patched.py: Simplify multi-line f-string to single line - config.py: Clarify LadybugDB requires Python 3.12+ in docstrings - ci.yml: Fix pnpm store path echo command quoting - test_graphiti.py: Use public API get_validation_errors() instead of private method 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * feat: add embedding provider change detection and fix import ordering Graphiti improvements: - Add provider change detection in GraphitiMemory.initialize() - Warn users when embedding provider changes to prevent dimension mismatches - Guide users to run migration script or reset state - Add test_provider_naming.py demo script for provider-specific database naming Code quality fixes: - Move inline `import re` to module top in query_memory.py - Alphabetically order standard library imports (PEP8 compliance) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix(ci): quote GITHUB_OUTPUT for shell safety Quote the $GITHUB_OUTPUT variable in the pnpm store path echo command to safely handle paths containing spaces. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: add GH_TOKEN and homepage for release workflow - Add GH_TOKEN to all package steps (required for electron-builder to download native prebuilds) - Add homepage and repository fields to package.json (required for Linux builds) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: add author email for Linux builds electron-builder FPM target requires author email for .deb packages 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: make macOS notarization optional Notarization can fail due to certificate issues - don't block the build. Unsigned DMGs still work, users just see a security warning. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: save notarization logs to private artifact instead of public logs - Captures Apple notarization failure details in downloadable artifact - Keeps sensitive paths and info out of public CI logs - Only repo collaborators can access the notarization-logs artifact 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: respect user's memory enabled flag in query_memory CLI Remove the forced `config.enabled = True` override that ignored user's explicit disable choice. The CLI tool should respect the enabled flag - callers using the semantic-search command are explicitly requesting memory functionality. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: improve migrate_embeddings robustness and correctness - Add proper cleanup of source client when target initialization fails - Fix database name derivation to use source provider's signature instead of incorrectly using current config's signature - Add validation to prevent no-op migration between same providers - Remove unused imports (json, Optional, GraphitiState) - Fix ruff formatting issues (line length) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * style: fix ruff linting errors in graphiti queries - Remove f-string prefix from strings without placeholders - Fix line length formatting issues 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: use shell guard for notarization credentials check The step-level `if: env.APPLE_ID != ''` condition was evaluated before the step's env block was available, causing notarization to always be skipped. Replace with a runtime shell guard that checks the environment variable when the step actually runs. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: allow @lydell/node-pty build scripts in pnpm v10 pnpm v10 blocks dependency lifecycle scripts by default. Add @lydell/node-pty to onlyBuiltDependencies array so its native module build scripts can run during installation. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * feat(ui): add project tab bar from PR #101 Cherry-picked from PR #101 for testing: - Add tab state management to project store - Create ProjectTabBar and SortableProjectTab components - Remove project dropdown from Sidebar - Add drag-and-drop tab reordering - Include unit tests for tab functionality 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * chore: simplify notarization step after successful setup Remove debug logging and notarization-logs artifact upload now that Developer ID Application certificate is properly configured. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: check APPLE_ID in shell instead of workflow if condition secrets context cannot be used directly in if expressions. Check env var in shell script instead. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix(ui): change agent profile fallback from 'Balanced' to 'Auto (Optimized)' The getProfileDisplay() function in AgentProfileSelector had a defensive fallback that incorrectly displayed "Balanced" when no profile was found. Changed to display "Auto (Optimized)" to match the actual default profile configured in DEFAULT_APP_SETTINGS. This ensures consistency with the intended default behavior where fresh users should see "Auto (Optimized)" as their agent profile selection. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * refactor(ui): simplify reference files and images handling in task modal Remove separate reference files and images sections from the new task modal. Images are now displayed as small clickable thumbnails directly below the description field, and files can be referenced via @mentions in the description. - Remove "Reference Images" toggle and dedicated ImageUpload section - Remove "Referenced Files" section and ReferencedFilesSection component usage - Add inline thumbnail display (64x64px) below description with remove buttons - Update hint text to clarify drag & drop and paste functionality - Clean up unused imports and state variables 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * chore: use GitHub noreply email for author field Prevents spam while maintaining traceability for open source project. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: create Python venv in userData for packaged apps On Linux AppImages, the bundled resources directory is read-only, which prevented venv creation. Now packaged apps store the venv in userData (~/.config/auto-claude-ui on Linux) which is always writable. This follows XDG conventions and fixes the Arch Linux AppImage issue. Fixes: venv creation failure on Linux AppImage 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * feat(ui): add keyboard shortcuts and tooltips for project tabs Add standard browser/editor-style keyboard shortcuts for tab navigation: - Cmd/Ctrl+1-9 to switch to specific tabs - Cmd/Ctrl+Tab/Shift+Tab to cycle through tabs - Cmd/Ctrl+W to close current tab Replace native title tooltips with Radix tooltips that: - Show after 200ms instead of ~1s native delay - Display shortcuts in styled kbd badges - Match app design with smooth animations 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * feat: improve task creation UX with @ autocomplete and better drag-drop Fixes three UX issues reported by users: 1. Add @ file autocomplete in task description - Type @ to see file suggestions - Filters files as you type - Keyboard navigation (arrows, Enter, Escape) - Shows file path for disambiguation 2. Fix file browser scroll in project explorer - Remove conflicting ScrollArea wrapper - Let virtualizer handle scrolling directly - Files now visible after expanding deep folders 3. Add auto-scroll during drag-and-drop - Scroll form container when dragging near edges - Makes it possible to drag files to textarea when scrolled out of view 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * feat(agent): enhance task restart functionality with new profile support - Updated `restartTask` method to accept an optional `newProfileId` parameter, allowing for profile swapping during task restarts. - Added logic to set the active profile if a new profile ID is provided. - Adjusted event handling for `auto-swap-restart-task` to pass the new profile ID. fix(agent): correct source type in rate limit info for spec creation - Changed source type from 'task' to 'roadmap' in `createSDKRateLimitInfo` calls for better clarity in spec creation context. 🤖 Generated with [Claude Code](https://claude.com/claude-code) * fix(ui): fix tab persistence and scope terminal shortcuts Two fixes: 1. Tab persistence on restart: Added explicit handling for when no tabs are open after app restart. Now auto-opens the first project tab instead of showing empty tab bar. 2. Keyboard shortcut scoping: Cmd/Ctrl+T now behaves contextually: - On Agent Terminals view: Opens new terminal (existing behavior) - On other views (Kanban, etc.): Opens Add Project dialog Added isActive prop to TerminalGrid to scope shortcuts to when the terminal view is active. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: unify default database path to ~/.auto-claude/memories The default path was inconsistent across files: - utils.ts used 'graphs' - memory-service.ts used 'memories' - .env.example documented 'graphs' Unified all to use 'memories' to match Python backend config.py. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * auto-claude: subtask-1-1 - Add projectPath prop to PreviewPanel and implement custom img component - Added projectPath prop to PreviewPanel interface - Implemented custom img component for ReactMarkdown that converts relative paths (like .github/assets/...) to file:// URLs for Electron - Updated ChangelogDetails to get selected project and pass its path to PreviewPanel - Images now display correctly in preview mode while markdown source remains unchanged 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix(merge): use stored baseBranch from task metadata for merge operations Previously, merge logic hardcoded 'main' as the comparison branch when detecting what files changed in a task. This broke the workflow where: - Tasks branch FROM the configured default (main) or user-specified branch - Tasks merge INTO the user's current working branch Now the merge system: 1. Reads baseBranch from task_metadata.json (set during task creation) 2. Passes --base-branch to Python CLI for merge and merge-preview 3. Uses this for refresh_from_git() comparisons throughout the merge pipeline 4. Falls back to auto-detection (main/master/develop) if not specified Files modified across frontend (TypeScript) and backend (Python) to thread the task_source_branch parameter through the entire merge flow. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix(merge): increase AI merge timeout from 2 to 10 minutes Large merge operations with many conflicting files were timing out before completion. The AI merge resolution was processing files successfully but hitting the 2-minute limit when handling 17+ files. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: use venv Python for terminal name generation TerminalNameGenerator was using system Python which doesn't have claude_agent_sdk installed. Now uses pythonEnvManager to get the venv Python path where dependencies are installed. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: prevent task title from blocking edit/close buttons Two issues were blocking the edit/close buttons in TaskDetailModal: 1. The title element was extending beyond its visual boundaries - Added overflow-hidden to container and truncate to title 2. The Electron window's draggable region was capturing mouse events - Added electron-no-drag class to the button container to allow normal mouse interactions in that area 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix: address CodeRabbit review issues - memory-service.ts: Add app.getAppPath() path resolution for packaged Electron apps. This ensures query_memory.py is found in both dev and production builds, consistent with title-generator.ts pattern. - MemoryStep.tsx: Make CheckCircle2 icon conditional on kuzuAvailable state. Previously showed success checkmark even when database wasn't available, misleading users. - GitHubSetupModal.tsx: Fix incorrect API method name from getStoredGitHubToken to getGitHubToken. Also adds existing auth detection to skip already-completed authentication steps. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * Readme for installors * Project tab persistence and github org init on project creation * fix(ui): address CodeRabbit PR review issues Fix actual bugs identified in PR #100 review: - Fix race condition in memory-handlers.ts timeout handling by using single timeout with proper cleanup and resolved flag - Fix API key resolution in GraphitiStep.tsx to handle groq, azure_openai, and ollama providers - Add TabState interface and getTabState/saveTabState to ElectronAPI types - Add mock implementations for browser development Also includes: - Exclude pnpm-lock.yaml from check-yaml (uses URLs with colons as keys) - Auto-fixes from pre-commit hooks (trailing whitespace, EOF, ruff-format) Note: nomic-embed-text dimension (768) and import ordering were verified as correct - CodeRabbit suggestions were false positives. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> * fix(tests): update tab management tests for IPC-based persistence The project store now uses IPC (saveTabState/getTabState) instead of localStorage for tab state persistence. Updated tests to: - Remove localStorage.setItem assertions (no longer used) - Update restoreTabState test to reflect it's now a no-op (actual loading happens via loadProjects → getTabState) - Add getTabState/saveTabState mocks to test setup This fixes the CI test failures where tests expected localStorage calls but the implementation uses IPC. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> --------- Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Andy
Generated
+2544
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,39 @@
|
||||
/**
|
||||
* Centralized Icon Exports for Design System
|
||||
*
|
||||
* This file serves as the single source of truth for all lucide-react icons used
|
||||
* throughout the design system demo app. By consolidating imports here, we enable:
|
||||
*
|
||||
* 1. Better tracking of which icons are actually used
|
||||
* 2. Potential code-splitting opportunities
|
||||
* 3. Easier future migration to alternative icon solutions
|
||||
* 4. Reduced bundle size through optimized tree-shaking
|
||||
*
|
||||
* Usage:
|
||||
* import { Check, ChevronLeft, X } from '../lib/icons';
|
||||
*
|
||||
* When adding new icons:
|
||||
* 1. Import the icon from 'lucide-react'
|
||||
* 2. Add it to the export statement in alphabetical order
|
||||
*/
|
||||
|
||||
export {
|
||||
Check,
|
||||
ChevronLeft,
|
||||
ChevronRight,
|
||||
Github,
|
||||
Heart,
|
||||
MessageSquare,
|
||||
Minus,
|
||||
Moon,
|
||||
MoreVertical,
|
||||
Plus,
|
||||
RotateCcw,
|
||||
Slack,
|
||||
Sparkles,
|
||||
Star,
|
||||
Sun,
|
||||
Video,
|
||||
X,
|
||||
Zap,
|
||||
} from 'lucide-react';
|
||||
@@ -0,0 +1,35 @@
|
||||
name-template: 'v$RESOLVED_VERSION'
|
||||
tag-template: 'v$RESOLVED_VERSION'
|
||||
|
||||
categories:
|
||||
- title: '## New Features'
|
||||
labels:
|
||||
- 'feature'
|
||||
- 'enhancement'
|
||||
- title: '## Bug Fixes'
|
||||
labels:
|
||||
- 'bug'
|
||||
- 'fix'
|
||||
- title: '## Improvements'
|
||||
labels:
|
||||
- 'improvement'
|
||||
- 'refactor'
|
||||
- title: '## Documentation'
|
||||
labels:
|
||||
- 'documentation'
|
||||
- title: '## Other Changes'
|
||||
labels:
|
||||
- '*'
|
||||
|
||||
change-template: '* $TITLE (#$NUMBER) @$AUTHOR'
|
||||
|
||||
sort-by: merged_at
|
||||
sort-direction: ascending
|
||||
|
||||
template: |
|
||||
$CHANGES
|
||||
|
||||
**Full Changelog**: https://github.com/$OWNER/$REPOSITORY/compare/$PREVIOUS_TAG...$RESOLVED_VERSION
|
||||
|
||||
## Contributors
|
||||
$CONTRIBUTORS
|
||||
@@ -57,7 +57,7 @@ jobs:
|
||||
env:
|
||||
CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
|
||||
|
||||
# Frontend tests
|
||||
# Frontend lint, typecheck, test, and build
|
||||
test-frontend:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
@@ -72,12 +72,34 @@ jobs:
|
||||
- name: Setup pnpm
|
||||
uses: pnpm/action-setup@v4
|
||||
with:
|
||||
version: 9
|
||||
version: 10
|
||||
|
||||
- name: Get pnpm store directory
|
||||
id: pnpm-cache
|
||||
run: echo "dir=$(pnpm store path)" >> "$GITHUB_OUTPUT"
|
||||
|
||||
- uses: actions/cache@v4
|
||||
with:
|
||||
path: ${{ steps.pnpm-cache.outputs.dir }}
|
||||
key: ${{ runner.os }}-pnpm-${{ hashFiles('**/pnpm-lock.yaml') }}
|
||||
restore-keys: ${{ runner.os }}-pnpm-
|
||||
|
||||
- name: Install dependencies
|
||||
working-directory: auto-claude-ui
|
||||
run: pnpm install --frozen-lockfile --ignore-scripts
|
||||
|
||||
- name: Lint
|
||||
working-directory: auto-claude-ui
|
||||
run: pnpm run lint
|
||||
|
||||
- name: Type check
|
||||
working-directory: auto-claude-ui
|
||||
run: pnpm run typecheck
|
||||
|
||||
- name: Run tests
|
||||
working-directory: auto-claude-ui
|
||||
run: pnpm test
|
||||
run: pnpm run test
|
||||
|
||||
- name: Build
|
||||
working-directory: auto-claude-ui
|
||||
run: pnpm run build
|
||||
|
||||
@@ -22,4 +22,3 @@ jobs:
|
||||
footer_timestamp: true
|
||||
reduce_headings: true
|
||||
remove_github_reference_links: true
|
||||
|
||||
|
||||
@@ -0,0 +1,344 @@
|
||||
name: Release
|
||||
|
||||
on:
|
||||
push:
|
||||
tags:
|
||||
- 'v*'
|
||||
workflow_dispatch:
|
||||
inputs:
|
||||
dry_run:
|
||||
description: 'Test build without creating release'
|
||||
required: false
|
||||
default: true
|
||||
type: boolean
|
||||
|
||||
jobs:
|
||||
build-macos:
|
||||
runs-on: macos-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
|
||||
- name: Setup Node.js
|
||||
uses: actions/setup-node@v4
|
||||
with:
|
||||
node-version: '20'
|
||||
|
||||
- name: Setup pnpm
|
||||
uses: pnpm/action-setup@v4
|
||||
with:
|
||||
version: 10
|
||||
|
||||
- name: Get pnpm store directory
|
||||
id: pnpm-cache
|
||||
run: echo "dir=$(pnpm store path)" >> $GITHUB_OUTPUT
|
||||
|
||||
- uses: actions/cache@v4
|
||||
with:
|
||||
path: ${{ steps.pnpm-cache.outputs.dir }}
|
||||
key: ${{ runner.os }}-pnpm-${{ hashFiles('**/pnpm-lock.yaml') }}
|
||||
restore-keys: ${{ runner.os }}-pnpm-
|
||||
|
||||
- name: Install dependencies
|
||||
run: cd auto-claude-ui && pnpm install --frozen-lockfile
|
||||
|
||||
- name: Build application
|
||||
run: cd auto-claude-ui && pnpm run build
|
||||
|
||||
- name: Package macOS (Intel)
|
||||
run: cd auto-claude-ui && pnpm run package:mac -- --arch=x64
|
||||
env:
|
||||
GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
CSC_LINK: ${{ secrets.MAC_CERTIFICATE }}
|
||||
CSC_KEY_PASSWORD: ${{ secrets.MAC_CERTIFICATE_PASSWORD }}
|
||||
|
||||
- name: Package macOS (Apple Silicon)
|
||||
run: cd auto-claude-ui && pnpm run package:mac -- --arch=arm64
|
||||
env:
|
||||
GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
CSC_LINK: ${{ secrets.MAC_CERTIFICATE }}
|
||||
CSC_KEY_PASSWORD: ${{ secrets.MAC_CERTIFICATE_PASSWORD }}
|
||||
|
||||
- name: Notarize macOS apps
|
||||
env:
|
||||
APPLE_ID: ${{ secrets.APPLE_ID }}
|
||||
APPLE_APP_SPECIFIC_PASSWORD: ${{ secrets.APPLE_APP_SPECIFIC_PASSWORD }}
|
||||
APPLE_TEAM_ID: ${{ secrets.APPLE_TEAM_ID }}
|
||||
run: |
|
||||
if [ -z "$APPLE_ID" ]; then
|
||||
echo "Skipping notarization: APPLE_ID not configured"
|
||||
exit 0
|
||||
fi
|
||||
cd auto-claude-ui
|
||||
for dmg in dist/*.dmg; do
|
||||
echo "Notarizing $dmg..."
|
||||
xcrun notarytool submit "$dmg" \
|
||||
--apple-id "$APPLE_ID" \
|
||||
--password "$APPLE_APP_SPECIFIC_PASSWORD" \
|
||||
--team-id "$APPLE_TEAM_ID" \
|
||||
--wait
|
||||
xcrun stapler staple "$dmg"
|
||||
echo "Successfully notarized and stapled $dmg"
|
||||
done
|
||||
|
||||
- name: Upload artifacts
|
||||
uses: actions/upload-artifact@v4
|
||||
with:
|
||||
name: macos-builds
|
||||
path: |
|
||||
auto-claude-ui/dist/*.dmg
|
||||
auto-claude-ui/dist/*.zip
|
||||
|
||||
build-windows:
|
||||
runs-on: windows-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
|
||||
- name: Setup Node.js
|
||||
uses: actions/setup-node@v4
|
||||
with:
|
||||
node-version: '20'
|
||||
|
||||
- name: Setup pnpm
|
||||
uses: pnpm/action-setup@v4
|
||||
with:
|
||||
version: 10
|
||||
|
||||
- name: Get pnpm store directory
|
||||
id: pnpm-cache
|
||||
shell: bash
|
||||
run: echo "dir=$(pnpm store path)" >> $GITHUB_OUTPUT
|
||||
|
||||
- uses: actions/cache@v4
|
||||
with:
|
||||
path: ${{ steps.pnpm-cache.outputs.dir }}
|
||||
key: ${{ runner.os }}-pnpm-${{ hashFiles('**/pnpm-lock.yaml') }}
|
||||
restore-keys: ${{ runner.os }}-pnpm-
|
||||
|
||||
- name: Install dependencies
|
||||
run: cd auto-claude-ui && pnpm install --frozen-lockfile
|
||||
|
||||
- name: Build application
|
||||
run: cd auto-claude-ui && pnpm run build
|
||||
|
||||
- name: Package Windows
|
||||
run: cd auto-claude-ui && pnpm run package:win
|
||||
env:
|
||||
GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
CSC_LINK: ${{ secrets.WIN_CERTIFICATE }}
|
||||
CSC_KEY_PASSWORD: ${{ secrets.WIN_CERTIFICATE_PASSWORD }}
|
||||
|
||||
- name: Upload artifacts
|
||||
uses: actions/upload-artifact@v4
|
||||
with:
|
||||
name: windows-builds
|
||||
path: |
|
||||
auto-claude-ui/dist/*.exe
|
||||
|
||||
build-linux:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
|
||||
- name: Setup Node.js
|
||||
uses: actions/setup-node@v4
|
||||
with:
|
||||
node-version: '20'
|
||||
|
||||
- name: Setup pnpm
|
||||
uses: pnpm/action-setup@v4
|
||||
with:
|
||||
version: 10
|
||||
|
||||
- name: Get pnpm store directory
|
||||
id: pnpm-cache
|
||||
run: echo "dir=$(pnpm store path)" >> $GITHUB_OUTPUT
|
||||
|
||||
- uses: actions/cache@v4
|
||||
with:
|
||||
path: ${{ steps.pnpm-cache.outputs.dir }}
|
||||
key: ${{ runner.os }}-pnpm-${{ hashFiles('**/pnpm-lock.yaml') }}
|
||||
restore-keys: ${{ runner.os }}-pnpm-
|
||||
|
||||
- name: Install dependencies
|
||||
run: cd auto-claude-ui && pnpm install --frozen-lockfile
|
||||
|
||||
- name: Build application
|
||||
run: cd auto-claude-ui && pnpm run build
|
||||
|
||||
- name: Package Linux
|
||||
run: cd auto-claude-ui && pnpm run package:linux
|
||||
env:
|
||||
GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
|
||||
- name: Upload artifacts
|
||||
uses: actions/upload-artifact@v4
|
||||
with:
|
||||
name: linux-builds
|
||||
path: |
|
||||
auto-claude-ui/dist/*.AppImage
|
||||
auto-claude-ui/dist/*.deb
|
||||
|
||||
create-release:
|
||||
needs: [build-macos, build-windows, build-linux]
|
||||
runs-on: ubuntu-latest
|
||||
permissions:
|
||||
contents: write
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
|
||||
- name: Download all artifacts
|
||||
uses: actions/download-artifact@v4
|
||||
with:
|
||||
path: dist
|
||||
|
||||
- name: Flatten and validate artifacts
|
||||
run: |
|
||||
mkdir -p release-assets
|
||||
find dist -type f \( -name "*.dmg" -o -name "*.zip" -o -name "*.exe" -o -name "*.AppImage" -o -name "*.deb" \) -exec cp {} release-assets/ \;
|
||||
|
||||
# Validate that at least one artifact was copied
|
||||
artifact_count=$(find release-assets -type f \( -name "*.dmg" -o -name "*.zip" -o -name "*.exe" -o -name "*.AppImage" -o -name "*.deb" \) | wc -l)
|
||||
if [ "$artifact_count" -eq 0 ]; then
|
||||
echo "::error::No build artifacts found! Expected .dmg, .zip, .exe, .AppImage, or .deb files."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo "Found $artifact_count artifact(s):"
|
||||
ls -la release-assets/
|
||||
|
||||
- name: Generate checksums
|
||||
run: |
|
||||
cd release-assets
|
||||
sha256sum ./* > checksums.sha256
|
||||
cat checksums.sha256
|
||||
|
||||
- name: Scan with VirusTotal
|
||||
id: virustotal
|
||||
if: ${{ github.event_name == 'push' || (github.event_name == 'workflow_dispatch' && inputs.dry_run != true) }}
|
||||
env:
|
||||
VT_API_KEY: ${{ secrets.VIRUSTOTAL_API_KEY }}
|
||||
run: |
|
||||
if [ -z "$VT_API_KEY" ]; then
|
||||
echo "::warning::VIRUSTOTAL_API_KEY not configured, skipping scan"
|
||||
echo "vt_results=" >> $GITHUB_OUTPUT
|
||||
exit 0
|
||||
fi
|
||||
|
||||
echo "## VirusTotal Scan Results" > vt_results.md
|
||||
echo "" >> vt_results.md
|
||||
|
||||
scan_failed=false
|
||||
|
||||
for file in release-assets/*.{exe,dmg,AppImage,deb}; do
|
||||
[ -f "$file" ] || continue
|
||||
filename=$(basename "$file")
|
||||
echo "Scanning $filename..."
|
||||
|
||||
# Upload file to VirusTotal
|
||||
response=$(curl -s --request POST \
|
||||
--url "https://www.virustotal.com/api/v3/files" \
|
||||
--header "x-apikey: $VT_API_KEY" \
|
||||
--form "file=@$file")
|
||||
|
||||
# Extract analysis ID
|
||||
analysis_id=$(echo "$response" | jq -r '.data.id // empty')
|
||||
|
||||
if [ -z "$analysis_id" ]; then
|
||||
echo "::error::Failed to upload $filename to VirusTotal"
|
||||
echo "Response: $response"
|
||||
scan_failed=true
|
||||
continue
|
||||
fi
|
||||
|
||||
echo "Uploaded $filename, analysis ID: $analysis_id"
|
||||
|
||||
# Wait for analysis to complete (max 5 minutes per file)
|
||||
for i in {1..30}; do
|
||||
sleep 10
|
||||
analysis=$(curl -s --request GET \
|
||||
--url "https://www.virustotal.com/api/v3/analyses/$analysis_id" \
|
||||
--header "x-apikey: $VT_API_KEY")
|
||||
|
||||
status=$(echo "$analysis" | jq -r '.data.attributes.status')
|
||||
echo " Status: $status (attempt $i/30)"
|
||||
|
||||
if [ "$status" = "completed" ]; then
|
||||
break
|
||||
fi
|
||||
done
|
||||
|
||||
# Get file hash for permanent URL
|
||||
file_hash=$(echo "$analysis" | jq -r '.meta.file_info.sha256 // empty')
|
||||
|
||||
if [ -z "$file_hash" ]; then
|
||||
# Fallback: calculate hash locally
|
||||
file_hash=$(sha256sum "$file" | cut -d' ' -f1)
|
||||
fi
|
||||
|
||||
# Get detection stats
|
||||
malicious=$(echo "$analysis" | jq -r '.data.attributes.stats.malicious // 0')
|
||||
suspicious=$(echo "$analysis" | jq -r '.data.attributes.stats.suspicious // 0')
|
||||
undetected=$(echo "$analysis" | jq -r '.data.attributes.stats.undetected // 0')
|
||||
|
||||
vt_url="https://www.virustotal.com/gui/file/$file_hash"
|
||||
|
||||
if [ "$malicious" -gt 0 ] || [ "$suspicious" -gt 0 ]; then
|
||||
echo "::error::$filename has $malicious malicious and $suspicious suspicious detections!"
|
||||
echo "- [$filename]($vt_url) - ⚠️ **$malicious malicious, $suspicious suspicious** detections" >> vt_results.md
|
||||
scan_failed=true
|
||||
else
|
||||
echo "$filename is clean ($undetected engines, 0 detections)"
|
||||
echo "- [$filename]($vt_url) - ✅ Clean ($undetected engines, 0 detections)" >> vt_results.md
|
||||
fi
|
||||
done
|
||||
|
||||
echo "" >> vt_results.md
|
||||
|
||||
# Save results for release notes
|
||||
cat vt_results.md
|
||||
echo "vt_results<<EOF" >> $GITHUB_OUTPUT
|
||||
cat vt_results.md >> $GITHUB_OUTPUT
|
||||
echo "EOF" >> $GITHUB_OUTPUT
|
||||
|
||||
if [ "$scan_failed" = true ]; then
|
||||
echo "::error::VirusTotal scan found issues. Review the results above."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
- name: Dry run summary
|
||||
if: ${{ github.event_name == 'workflow_dispatch' && inputs.dry_run == true }}
|
||||
run: |
|
||||
echo "## Dry Run Complete" >> $GITHUB_STEP_SUMMARY
|
||||
echo "Build artifacts created successfully:" >> $GITHUB_STEP_SUMMARY
|
||||
echo "\`\`\`" >> $GITHUB_STEP_SUMMARY
|
||||
ls -la release-assets/ >> $GITHUB_STEP_SUMMARY
|
||||
echo "\`\`\`" >> $GITHUB_STEP_SUMMARY
|
||||
echo "### Checksums" >> $GITHUB_STEP_SUMMARY
|
||||
echo "\`\`\`" >> $GITHUB_STEP_SUMMARY
|
||||
cat release-assets/checksums.sha256 >> $GITHUB_STEP_SUMMARY
|
||||
echo "\`\`\`" >> $GITHUB_STEP_SUMMARY
|
||||
|
||||
- name: Generate changelog
|
||||
if: ${{ github.event_name == 'push' || (github.event_name == 'workflow_dispatch' && inputs.dry_run != true) }}
|
||||
id: changelog
|
||||
uses: release-drafter/release-drafter@v6
|
||||
with:
|
||||
config-name: release-drafter.yml
|
||||
env:
|
||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
|
||||
- name: Create Release
|
||||
if: ${{ github.event_name == 'push' || (github.event_name == 'workflow_dispatch' && inputs.dry_run != true) }}
|
||||
uses: softprops/action-gh-release@v2
|
||||
with:
|
||||
body: |
|
||||
${{ steps.changelog.outputs.body }}
|
||||
|
||||
${{ steps.virustotal.outputs.vt_results }}
|
||||
files: release-assets/*
|
||||
draft: false
|
||||
prerelease: ${{ contains(github.ref, 'beta') || contains(github.ref, 'alpha') }}
|
||||
env:
|
||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
@@ -33,4 +33,5 @@ repos:
|
||||
- id: trailing-whitespace
|
||||
- id: end-of-file-fixer
|
||||
- id: check-yaml
|
||||
exclude: pnpm-lock\.yaml$
|
||||
- id: check-added-large-files
|
||||
|
||||
@@ -189,12 +189,18 @@ Dual-layer memory architecture:
|
||||
- Session insights, patterns, gotchas, codebase map
|
||||
|
||||
**Graphiti Memory (Optional Enhancement)** - `graphiti_memory.py`
|
||||
- Graph database with semantic search (FalkorDB)
|
||||
- Graph database with semantic search (LadybugDB - embedded, no Docker)
|
||||
- Cross-session context retrieval
|
||||
- Multi-provider support (V2):
|
||||
- Requires Python 3.12+
|
||||
- Multi-provider support:
|
||||
- LLM: OpenAI, Anthropic, Azure OpenAI, Ollama, Google AI (Gemini)
|
||||
- Embedders: OpenAI, Voyage AI, Azure OpenAI, Ollama, Google AI
|
||||
|
||||
```bash
|
||||
# Setup (requires Python 3.12+)
|
||||
pip install real_ladybug graphiti-core
|
||||
```
|
||||
|
||||
Enable with: `GRAPHITI_ENABLED=true` + provider credentials. See `.env.example`.
|
||||
|
||||
## Project Structure
|
||||
|
||||
+53
-1
@@ -8,6 +8,7 @@ Thank you for your interest in contributing to Auto Claude! This document provid
|
||||
- [Development Setup](#development-setup)
|
||||
- [Python Backend](#python-backend)
|
||||
- [Electron Frontend](#electron-frontend)
|
||||
- [Running from Source](#running-from-source)
|
||||
- [Pre-commit Hooks](#pre-commit-hooks)
|
||||
- [Code Style](#code-style)
|
||||
- [Testing](#testing)
|
||||
@@ -28,7 +29,6 @@ Before contributing, ensure you have the following installed:
|
||||
- **pnpm** - Package manager for the frontend (`npm install -g pnpm`)
|
||||
- **uv** (recommended) or **pip** - Python package manager
|
||||
- **Git** - Version control
|
||||
- **Docker** (optional) - For running FalkorDB if using Graphiti memory
|
||||
|
||||
## Development Setup
|
||||
|
||||
@@ -80,6 +80,58 @@ pnpm build
|
||||
pnpm package
|
||||
```
|
||||
|
||||
## Running from Source
|
||||
|
||||
If you want to run Auto Claude from source (for development or testing unreleased features), follow these steps:
|
||||
|
||||
### Step 1: Clone and Set Up Python Backend
|
||||
|
||||
```bash
|
||||
git clone https://github.com/AndyMik90/Auto-Claude.git
|
||||
cd Auto-Claude/auto-claude
|
||||
|
||||
# Using uv (recommended)
|
||||
uv venv && uv pip install -r requirements.txt
|
||||
|
||||
# Or using standard Python
|
||||
python3 -m venv .venv
|
||||
source .venv/bin/activate # On Windows: .venv\Scripts\activate
|
||||
pip install -r requirements.txt
|
||||
|
||||
# Set up environment
|
||||
cp .env.example .env
|
||||
# Edit .env and add your CLAUDE_CODE_OAUTH_TOKEN (get it via: claude setup-token)
|
||||
```
|
||||
|
||||
### Step 2: Run the Desktop UI
|
||||
|
||||
```bash
|
||||
cd ../auto-claude-ui
|
||||
|
||||
# Install dependencies
|
||||
pnpm install
|
||||
|
||||
# Development mode (hot reload)
|
||||
pnpm dev
|
||||
|
||||
# Or production build
|
||||
pnpm run build && pnpm run start
|
||||
```
|
||||
|
||||
<details>
|
||||
<summary><b>Windows users:</b> If installation fails with node-gyp errors, click here</summary>
|
||||
|
||||
Auto Claude automatically downloads prebuilt binaries for Windows. If prebuilts aren't available for your Electron version yet, you'll need Visual Studio Build Tools:
|
||||
|
||||
1. Download [Visual Studio Build Tools 2022](https://visualstudio.microsoft.com/visual-cpp-build-tools/)
|
||||
2. Select "Desktop development with C++" workload
|
||||
3. In "Individual Components", add "MSVC v143 - VS 2022 C++ x64/x86 Spectre-mitigated libs"
|
||||
4. Restart terminal and run `pnpm install` again
|
||||
|
||||
</details>
|
||||
|
||||
> **Note:** For regular usage, we recommend downloading the pre-built releases from [GitHub Releases](https://github.com/AndyMik90/Auto-Claude/releases). Running from source is primarily for contributors and those testing unreleased features.
|
||||
|
||||
## Pre-commit Hooks
|
||||
|
||||
We use [pre-commit](https://pre-commit.com/) to run linting and formatting checks before each commit. This ensures code quality and consistency across the project.
|
||||
|
||||
@@ -24,114 +24,57 @@ Your AI coding companion. Build features, fix bugs, and ship faster — with aut
|
||||
- **Self-Validating**: Built-in QA loop catches issues before you review
|
||||
- **Isolated Workspaces**: All work happens in git worktrees — your code stays safe
|
||||
- **AI Merge Resolution**: Intelligent conflict resolution when merging back to main — no manual conflict fixing
|
||||
- **Memory Layer**: Agents remember insights across sessions for smarter decisions
|
||||
- **Cross-Platform**: Desktop app runs on Mac, Windows, and Linux
|
||||
- **Any Project Type**: Build web apps, APIs, CLIs — works with any software project
|
||||
|
||||
## 🚀 Quick Start (Desktop UI)
|
||||
## Quick Start
|
||||
|
||||
The Desktop UI is the recommended way to use Auto Claude. It provides visual task management, real-time progress tracking, and a Kanban board interface.
|
||||
### Download Auto Claude
|
||||
|
||||
Download the latest release for your platform from [GitHub Releases](https://github.com/AndyMik90/Auto-Claude/releases/latest):
|
||||
|
||||
| Platform | Download |
|
||||
|----------|----------|
|
||||
| **macOS (Apple Silicon M1-M4)** | `Auto-Claude-X.X.X-arm64.dmg` |
|
||||
| **macOS (Intel)** | `Auto-Claude-X.X.X-x64.dmg` |
|
||||
| **Windows** | `Auto-Claude-X.X.X.exe` |
|
||||
| **Linux** | `Auto-Claude-X.X.X.AppImage` or `.deb` |
|
||||
|
||||
> **Not sure which Mac?** Click the Apple menu () > "About This Mac". Look for "Chip" - M1/M2/M3/M4 = Apple Silicon, otherwise Intel.
|
||||
|
||||
### Prerequisites
|
||||
|
||||
1. **Node.js 18+** - [Download Node.js](https://nodejs.org/)
|
||||
2. **Python 3.10+** - [Download Python](https://www.python.org/downloads/)
|
||||
3. **Docker Desktop** - Required for the Memory Layer
|
||||
4. **Claude Code CLI** - `npm install -g @anthropic-ai/claude-code`
|
||||
5. **Claude Subscription** - Requires [Claude Pro or Max](https://claude.ai/upgrade) for Claude Code access
|
||||
6. **Git Repository** - Your project must be initialized as a git repository
|
||||
Before using Auto Claude, you need:
|
||||
|
||||
### Git Initialization
|
||||
1. **Claude Subscription** - Requires [Claude Pro or Max](https://claude.ai/upgrade) for Claude Code access
|
||||
2. **Claude Code CLI** - Install with: `npm install -g @anthropic-ai/claude-code`
|
||||
|
||||
**Auto Claude requires a git repository** to create isolated worktrees for safe parallel development. If your project isn't a git repo yet:
|
||||
### Install and Run
|
||||
|
||||
```bash
|
||||
cd your-project
|
||||
git init
|
||||
git add .
|
||||
git commit -m "Initial commit"
|
||||
```
|
||||
|
||||
> **Why git?** Auto Claude uses git branches and worktrees to isolate each task in its own workspace, keeping your main branch clean until you're ready to merge. This allows you to work on multiple features simultaneously without conflicts.
|
||||
|
||||
---
|
||||
|
||||
### Installing Docker Desktop
|
||||
|
||||
Docker runs the FalkorDB database that powers Auto Claude's cross-session memory.
|
||||
|
||||
| Operating System | Download Link |
|
||||
|------------------|---------------|
|
||||
| **Mac (Apple Silicon M1/M2/M3/M4)** | [Download for Apple Chip](https://desktop.docker.com/mac/main/arm64/Docker.dmg) |
|
||||
| **Mac (Intel)** | [Download for Intel Chip](https://desktop.docker.com/mac/main/amd64/Docker.dmg) |
|
||||
| **Windows** | [Download for Windows](https://desktop.docker.com/win/main/amd64/Docker%20Desktop%20Installer.exe) |
|
||||
| **Linux** | [Installation Guide](https://docs.docker.com/desktop/install/linux-install/) |
|
||||
|
||||
> **Not sure which Mac?** Click the Apple menu (🍎) → "About This Mac". Look for "Chip" - M1/M2/M3/M4 = Apple Silicon, otherwise Intel.
|
||||
|
||||
**After installing:** Open Docker Desktop and wait for the whale icon (🐳) to appear in your menu bar/system tray.
|
||||
|
||||
> **Using the Desktop UI?** It automatically detects Docker status and offers one-click FalkorDB setup. No terminal commands needed!
|
||||
|
||||
📚 **For detailed installation steps, troubleshooting, and advanced configuration, see [guides/DOCKER-SETUP.md](guides/DOCKER-SETUP.md)**
|
||||
|
||||
---
|
||||
|
||||
### Step 1: Set Up the Python Backend
|
||||
|
||||
The Desktop UI runs Python scripts behind the scenes. Set up the Python environment:
|
||||
|
||||
```bash
|
||||
cd auto-claude
|
||||
|
||||
# Using uv (recommended)
|
||||
uv venv && uv pip install -r requirements.txt
|
||||
|
||||
# Or using standard Python
|
||||
python3 -m venv .venv && source .venv/bin/activate && pip install -r requirements.txt
|
||||
```
|
||||
|
||||
### Step 2: Start the Memory Layer
|
||||
|
||||
The Auto Claude Memory Layer provides cross-session context retention using a graph database:
|
||||
|
||||
```bash
|
||||
# Make sure Docker Desktop is running, then:
|
||||
docker-compose up -d falkordb
|
||||
```
|
||||
|
||||
### Step 3: Install and Launch the Desktop UI
|
||||
|
||||
```bash
|
||||
cd auto-claude-ui
|
||||
|
||||
# Install dependencies (pnpm recommended, npm works too)
|
||||
pnpm install
|
||||
# or: npm install
|
||||
|
||||
# Build and start the application
|
||||
pnpm run build && pnpm run start
|
||||
# or: npm run build && npm run start
|
||||
```
|
||||
1. **Download** the installer for your platform from the table above
|
||||
2. **Install**:
|
||||
- **macOS**: Open the `.dmg`, drag Auto Claude to Applications
|
||||
- **Windows**: Run the `.exe` installer (see note below about security warning)
|
||||
- **Linux**: Make the AppImage executable (`chmod +x`) and run it, or install the `.deb`
|
||||
3. **Launch** Auto Claude
|
||||
4. **Add your project** and start building!
|
||||
|
||||
<details>
|
||||
<summary><b>Windows users:</b> If installation fails with node-gyp errors, click here</summary>
|
||||
<summary><b>Windows users:</b> Security warning when installing</summary>
|
||||
|
||||
Auto Claude automatically downloads prebuilt binaries for Windows. If prebuilts aren't available for your Electron version yet, you'll need Visual Studio Build Tools:
|
||||
The Windows installer is not yet code-signed, so you may see a "Windows protected your PC" warning from Microsoft Defender SmartScreen.
|
||||
|
||||
1. Download [Visual Studio Build Tools 2022](https://visualstudio.microsoft.com/visual-cpp-build-tools/)
|
||||
2. Select "Desktop development with C++" workload
|
||||
3. In "Individual Components", add "MSVC v143 - VS 2022 C++ x64/x86 Spectre-mitigated libs"
|
||||
4. Restart terminal and run `npm install` again
|
||||
**To proceed:**
|
||||
1. Click "More info"
|
||||
2. Click "Run anyway"
|
||||
|
||||
This is safe — all releases are automatically scanned with VirusTotal before publishing. You can verify any installer by checking the **VirusTotal Scan Results** section in each [release's notes](https://github.com/AndyMik90/Auto-Claude/releases).
|
||||
|
||||
We're working on obtaining a code signing certificate for future releases.
|
||||
|
||||
</details>
|
||||
|
||||
### Step 4: Start Building
|
||||
|
||||
1. Add your project in the UI
|
||||
2. Create a new task describing what you want to build
|
||||
3. Watch as Auto Claude creates a spec, plans, and implements your feature
|
||||
4. Review changes and merge when satisfied
|
||||
> **Want to build from source?** See [CONTRIBUTING.md](CONTRIBUTING.md#running-from-source) for development setup.
|
||||
|
||||
---
|
||||
|
||||
@@ -241,23 +184,6 @@ Three-layer defense keeps your code safe:
|
||||
- **Filesystem Restrictions** — Operations limited to project directory
|
||||
- **Command Allowlist** — Only approved commands based on your project's stack
|
||||
|
||||
### 🧠 Memory Layer
|
||||
|
||||
The Memory Layer is a **hybrid RAG system** combining graph nodes with semantic search to deliver the best possible context during AI coding. Agents remember insights from previous sessions, discovered codebase patterns persist and are reusable, and historical context helps agents make smarter decisions.
|
||||
|
||||
**Architecture:**
|
||||
- **Backend**: FalkorDB (graph database) via Docker
|
||||
- **Library**: Graphiti for knowledge graph operations
|
||||
- **Providers**: OpenAI, Anthropic, Azure OpenAI, Google AI, or Ollama (local/offline)
|
||||
|
||||
| Setup | LLM | Embeddings | Notes |
|
||||
|-------|-----|------------|-------|
|
||||
| **OpenAI** | OpenAI | OpenAI | Simplest - single API key |
|
||||
| **Anthropic + Voyage** | Anthropic | Voyage AI | High quality |
|
||||
| **Google AI** | Gemini | Google | Single API key, fast inference |
|
||||
| **Ollama** | Ollama | Ollama | Fully offline |
|
||||
| **Azure** | Azure OpenAI | Azure OpenAI | Enterprise |
|
||||
|
||||
## Project Structure
|
||||
|
||||
```
|
||||
@@ -273,9 +199,8 @@ your-project/
|
||||
│ ├── spec_runner.py # Spec creation orchestrator
|
||||
│ ├── prompts/ # Agent prompt templates
|
||||
│ └── ...
|
||||
├── auto-claude-ui/ # Electron desktop application
|
||||
│ └── ...
|
||||
└── docker-compose.yml # FalkorDB for Memory Layer
|
||||
└── auto-claude-ui/ # Electron desktop application
|
||||
└── ...
|
||||
```
|
||||
|
||||
### Understanding the Folders
|
||||
@@ -309,13 +234,6 @@ The `.auto-claude/` directory is gitignored and project-specific - you'll have o
|
||||
|----------|----------|-------------|
|
||||
| `CLAUDE_CODE_OAUTH_TOKEN` | Yes | OAuth token from `claude setup-token` |
|
||||
| `AUTO_BUILD_MODEL` | No | Model override (default: claude-opus-4-5-20251101) |
|
||||
| `GRAPHITI_ENABLED` | Recommended | Set to `true` to enable Memory Layer |
|
||||
| `GRAPHITI_LLM_PROVIDER` | For Memory | LLM provider: openai, anthropic, azure_openai, ollama, google |
|
||||
| `GRAPHITI_EMBEDDER_PROVIDER` | For Memory | Embedder: openai, voyage, azure_openai, ollama, google |
|
||||
| `OPENAI_API_KEY` | For OpenAI | Required for OpenAI provider |
|
||||
| `ANTHROPIC_API_KEY` | For Anthropic | Required for Anthropic LLM |
|
||||
| `VOYAGE_API_KEY` | For Voyage | Required for Voyage embeddings |
|
||||
| `GOOGLE_API_KEY` | For Google | Required for Google AI (Gemini) provider |
|
||||
|
||||
See `auto-claude/.env.example` for complete configuration options.
|
||||
|
||||
|
||||
@@ -9,7 +9,7 @@ export default defineConfig({
|
||||
exclude: [
|
||||
'uuid',
|
||||
'chokidar',
|
||||
'ioredis',
|
||||
'kuzu',
|
||||
'electron-updater',
|
||||
'@electron-toolkit/utils'
|
||||
]
|
||||
|
||||
Generated
+343
-166
File diff suppressed because it is too large
Load Diff
@@ -2,8 +2,16 @@
|
||||
"name": "auto-claude-ui",
|
||||
"version": "2.6.5",
|
||||
"description": "Desktop UI for Auto Claude autonomous coding framework",
|
||||
"homepage": "https://github.com/AndyMik90/Auto-Claude",
|
||||
"repository": {
|
||||
"type": "git",
|
||||
"url": "https://github.com/AndyMik90/Auto-Claude.git"
|
||||
},
|
||||
"main": "./out/main/index.js",
|
||||
"author": "Auto Claude Team",
|
||||
"author": {
|
||||
"name": "Auto Claude Team",
|
||||
"email": "119136210+AndyMik90@users.noreply.github.com"
|
||||
},
|
||||
"license": "AGPL-3.0",
|
||||
"scripts": {
|
||||
"postinstall": "node scripts/postinstall.js",
|
||||
@@ -59,7 +67,6 @@
|
||||
"class-variance-authority": "^0.7.1",
|
||||
"clsx": "^2.1.1",
|
||||
"electron-updater": "^6.6.2",
|
||||
"ioredis": "^5.8.2",
|
||||
"lucide-react": "^0.560.0",
|
||||
"motion": "^12.23.26",
|
||||
"react": "^19.2.3",
|
||||
@@ -79,7 +86,6 @@
|
||||
"@playwright/test": "^1.52.0",
|
||||
"@tailwindcss/postcss": "^4.1.17",
|
||||
"@testing-library/react": "^16.1.0",
|
||||
"@types/ioredis": "^4.28.10",
|
||||
"@types/node": "^25.0.0",
|
||||
"@types/react": "^19.2.7",
|
||||
"@types/react-dom": "^19.2.3",
|
||||
@@ -110,6 +116,7 @@
|
||||
"node-pty": "npm:@lydell/node-pty@^1.1.0"
|
||||
},
|
||||
"onlyBuiltDependencies": [
|
||||
"@lydell/node-pty",
|
||||
"electron",
|
||||
"electron-winstaller",
|
||||
"esbuild"
|
||||
@@ -118,6 +125,7 @@
|
||||
"build": {
|
||||
"appId": "com.autoclaude.ui",
|
||||
"productName": "Auto-Claude",
|
||||
"artifactName": "${productName}-${version}-${platform}-${arch}.${ext}",
|
||||
"publish": [
|
||||
{
|
||||
"provider": "github",
|
||||
@@ -151,13 +159,23 @@
|
||||
"!**/*.pyc",
|
||||
"!**/specs",
|
||||
"!**/.venv",
|
||||
"!**/.env"
|
||||
"!**/.venv-*",
|
||||
"!**/venv",
|
||||
"!**/.env",
|
||||
"!**/tests",
|
||||
"!**/*.egg-info",
|
||||
"!**/.pytest_cache",
|
||||
"!**/.mypy_cache"
|
||||
]
|
||||
}
|
||||
],
|
||||
"mac": {
|
||||
"category": "public.app-category.developer-tools",
|
||||
"icon": "resources/icon.icns",
|
||||
"hardenedRuntime": true,
|
||||
"gatekeeperAssess": false,
|
||||
"entitlements": "resources/entitlements.mac.plist",
|
||||
"entitlementsInherit": "resources/entitlements.mac.plist",
|
||||
"target": [
|
||||
"dmg",
|
||||
"zip"
|
||||
|
||||
Generated
-77
@@ -103,9 +103,6 @@ importers:
|
||||
electron-updater:
|
||||
specifier: ^6.6.2
|
||||
version: 6.6.2
|
||||
ioredis:
|
||||
specifier: ^5.8.2
|
||||
version: 5.8.2
|
||||
lucide-react:
|
||||
specifier: ^0.560.0
|
||||
version: 0.560.0(react@19.2.3)
|
||||
@@ -158,9 +155,6 @@ importers:
|
||||
'@testing-library/react':
|
||||
specifier: ^16.1.0
|
||||
version: 16.3.1(@testing-library/dom@10.4.1)(@types/react-dom@19.2.3(@types/react@19.2.7))(@types/react@19.2.7)(react-dom@19.2.3(react@19.2.3))(react@19.2.3)
|
||||
'@types/ioredis':
|
||||
specifier: ^4.28.10
|
||||
version: 4.28.10
|
||||
'@types/node':
|
||||
specifier: ^25.0.0
|
||||
version: 25.0.3
|
||||
@@ -833,9 +827,6 @@ packages:
|
||||
resolution: {integrity: sha512-bV0Tgo9K4hfPCek+aMAn81RppFKv2ySDQeMoSZuvTASywNTnVJCArCZE2FWqpvIatKu7VMRLWlR1EazvVhDyhQ==}
|
||||
engines: {node: '>=18.18'}
|
||||
|
||||
'@ioredis/commands@1.4.0':
|
||||
resolution: {integrity: sha512-aFT2yemJJo+TZCmieA7qnYGQooOS7QfNmYrzGtsYd3g9j5iDP8AimYYAesf79ohjbLG12XxC4nG5DyEnC88AsQ==}
|
||||
|
||||
'@isaacs/balanced-match@4.0.1':
|
||||
resolution: {integrity: sha512-yzMTt9lEb8Gv7zRioUilSglI0c0smZ9k5D65677DLWLtWJaXIS3CqcGyUFByYKlnUj6TkjLVs54fBl6+TiGQDQ==}
|
||||
engines: {node: 20 || >=22}
|
||||
@@ -1705,9 +1696,6 @@ packages:
|
||||
'@types/http-cache-semantics@4.0.4':
|
||||
resolution: {integrity: sha512-1m0bIFVc7eJWyve9S0RnuRgcQqF/Xd5QsUZAZeQFr1Q3/p9JWoQQEqmVy+DPTNpGXwhgIetAoYF8JSc33q29QA==}
|
||||
|
||||
'@types/ioredis@4.28.10':
|
||||
resolution: {integrity: sha512-69LyhUgrXdgcNDv7ogs1qXZomnfOEnSmrmMFqKgt1XMJxmoOSG/u3wYy13yACIfKuMJ8IhKgHafDO3sx19zVQQ==}
|
||||
|
||||
'@types/json-schema@7.0.15':
|
||||
resolution: {integrity: sha512-5+fP8P8MFNC+AyZCDxrB2pkZFPGzqQWUzpSeuuVLvm8VMcorNYavBqoFcxK8bQz4Qsbn4oUEEem4wDLfcysGHA==}
|
||||
|
||||
@@ -2195,10 +2183,6 @@ packages:
|
||||
resolution: {integrity: sha512-eYm0QWBtUrBWZWG0d386OGAw16Z995PiOVo2B7bjWSbHedGl5e0ZWaq65kOGgUSNesEIDkB9ISbTg/JK9dhCZA==}
|
||||
engines: {node: '>=6'}
|
||||
|
||||
cluster-key-slot@1.1.2:
|
||||
resolution: {integrity: sha512-RMr0FhtfXemyinomL4hrWcYJxmX6deFdCxpJzhDttxgO1+bcCnkk+9drydLVDmAMG7NE6aN/fl4F7ucU/90gAA==}
|
||||
engines: {node: '>=0.10.0'}
|
||||
|
||||
color-convert@2.0.1:
|
||||
resolution: {integrity: sha512-RRECPsj7iu/xb5oKYcsFHSppFNnsj/52OVTRKb4zP5onXwVF3zVmmToNcOfGC+CRDpfK/U584fMg38ZHCaElKQ==}
|
||||
engines: {node: '>=7.0.0'}
|
||||
@@ -2323,10 +2307,6 @@ packages:
|
||||
resolution: {integrity: sha512-ZySD7Nf91aLB0RxL4KGrKHBXl7Eds1DAmEdcoVawXnLD7SDhpNgtuII2aAkg7a7QS41jxPSZ17p4VdGnMHk3MQ==}
|
||||
engines: {node: '>=0.4.0'}
|
||||
|
||||
denque@2.1.0:
|
||||
resolution: {integrity: sha512-HVQE3AAb/pxF8fQAoiqpvg9i3evqug3hoiwakOyZAwJm+6vZehbkYXZ0l4JxS+I3QxM97v5aaRNhj8v5oBhekw==}
|
||||
engines: {node: '>=0.10'}
|
||||
|
||||
dequal@2.0.3:
|
||||
resolution: {integrity: sha512-0je+qPKHEMohvfRTCEo3CrPG6cAzAYgmzKyxRiYSSDkS6eGJdyVJm7WaYA5ECaAD9wLB2T4EEeymA5aFVcYXCA==}
|
||||
engines: {node: '>=6'}
|
||||
@@ -2925,10 +2905,6 @@ packages:
|
||||
resolution: {integrity: sha512-4gd7VpWNQNB4UKKCFFVcp1AVv+FMOgs9NKzjHKusc8jTMhd5eL1NqQqOpE0KzMds804/yHlglp3uxgluOqAPLw==}
|
||||
engines: {node: '>= 0.4'}
|
||||
|
||||
ioredis@5.8.2:
|
||||
resolution: {integrity: sha512-C6uC+kleiIMmjViJINWk80sOQw5lEzse1ZmvD+S/s8p8CWapftSaC+kocGTx6xrbrJ4WmYQGC08ffHLr6ToR6Q==}
|
||||
engines: {node: '>=12.22.0'}
|
||||
|
||||
ip-address@10.1.0:
|
||||
resolution: {integrity: sha512-XXADHxXmvT9+CRxhXg56LJovE+bmWnEWB78LB83VZTprKTmaC5QfruXocxzTZ2Kl0DNwKuBdlIhjL8LeY8Sf8Q==}
|
||||
engines: {node: '>= 12'}
|
||||
@@ -3246,15 +3222,9 @@ packages:
|
||||
resolution: {integrity: sha512-iPZK6eYjbxRu3uB4/WZ3EsEIMJFMqAoopl3R+zuq0UjcAm/MO6KCweDgPfP3elTztoKP3KtnVHxTn2NHBSDVUw==}
|
||||
engines: {node: '>=10'}
|
||||
|
||||
lodash.defaults@4.2.0:
|
||||
resolution: {integrity: sha512-qjxPLHd3r5DnsdGacqOMU6pb/avJzdh9tFX2ymgoZE27BmjXrNy/y4LoaiTeAb+O3gL8AfpJGtqfX/ae2leYYQ==}
|
||||
|
||||
lodash.escaperegexp@4.1.2:
|
||||
resolution: {integrity: sha512-TM9YBvyC84ZxE3rgfefxUWiQKLilstD6k7PTGt6wfbtXF8ixIJLOL3VYyV/z+ZiPLsVxAsKAFVwWlWeb2Y8Yyw==}
|
||||
|
||||
lodash.isarguments@3.1.0:
|
||||
resolution: {integrity: sha512-chi4NHZlZqZD18a0imDHnZPrDeBbTtVN7GXMwuGdRH9qotxAjYs3aVLKc7zNOG9eddR5Ksd8rvFEBc9SsggPpg==}
|
||||
|
||||
lodash.isequal@4.5.0:
|
||||
resolution: {integrity: sha512-pDo3lu8Jhfjqls6GkMgpahsF9kCyayhgykjyLMNFTKWrpVdAQtYyB4muAMWozBB4ig/dtWAmsMxLEI8wuz+DYQ==}
|
||||
deprecated: This package is deprecated. Use require('node:util').isDeepStrictEqual instead.
|
||||
@@ -3894,14 +3864,6 @@ packages:
|
||||
resolution: {integrity: sha512-9u/XQ1pvrQtYyMpZe7DXKv2p5CNvyVwzUB6uhLAnQwHMSgKMBR62lc7AHljaeteeHXn11XTAaLLUVZYVZyuRBQ==}
|
||||
engines: {node: '>= 20.19.0'}
|
||||
|
||||
redis-errors@1.2.0:
|
||||
resolution: {integrity: sha512-1qny3OExCf0UvUV/5wpYKf2YwPcOqXzkwKKSmKHiE6ZMQs5heeE/c8eXK+PNllPvmjgAbfnsbpkGZWy8cBpn9w==}
|
||||
engines: {node: '>=4'}
|
||||
|
||||
redis-parser@3.0.0:
|
||||
resolution: {integrity: sha512-DJnGAeenTdpMEH6uAJRK/uiyEIH9WVsUmoLwzudwGJUwZPp80PDBWPHXSAGNPwNvIXAbe7MSUB1zQFugFml66A==}
|
||||
engines: {node: '>=4'}
|
||||
|
||||
reflect.getprototypeof@1.0.10:
|
||||
resolution: {integrity: sha512-00o4I+DVrefhv+nX0ulyi3biSHCPDe+yLv5o/p6d/UVlirijB8E16FtfwSAi4g3tcqrQ4lRAqQSoFEZJehYEcw==}
|
||||
engines: {node: '>= 0.4'}
|
||||
@@ -4126,9 +4088,6 @@ packages:
|
||||
stackback@0.0.2:
|
||||
resolution: {integrity: sha512-1XMJE5fQo1jGH6Y/7ebnwPOBEkIEnT4QF32d5R1+VXdXveM0IBMJt8zfaxX1P3QhVwrYe+576+jkANtSS2mBbw==}
|
||||
|
||||
standard-as-callback@2.1.0:
|
||||
resolution: {integrity: sha512-qoRRSyROncaz1z0mvYqIE4lCd9p2R90i6GxW3uZv5ucSu8tU7B5HXUP1gG8pVZsYNVaXjk8ClXHPttLyxAL48A==}
|
||||
|
||||
stat-mode@1.0.0:
|
||||
resolution: {integrity: sha512-jH9EhtKIjuXZ2cWxmXS8ZP80XyC3iasQxMDV8jzhNJpfDb7VbQLVW4Wvsxz9QZvzV+G4YoSfBUVKDOyxLzi/sg==}
|
||||
engines: {node: '>= 6'}
|
||||
@@ -5211,8 +5170,6 @@ snapshots:
|
||||
|
||||
'@humanwhocodes/retry@0.4.3': {}
|
||||
|
||||
'@ioredis/commands@1.4.0': {}
|
||||
|
||||
'@isaacs/balanced-match@4.0.1': {}
|
||||
|
||||
'@isaacs/brace-expansion@5.0.0':
|
||||
@@ -6043,10 +6000,6 @@ snapshots:
|
||||
|
||||
'@types/http-cache-semantics@4.0.4': {}
|
||||
|
||||
'@types/ioredis@4.28.10':
|
||||
dependencies:
|
||||
'@types/node': 25.0.3
|
||||
|
||||
'@types/json-schema@7.0.15': {}
|
||||
|
||||
'@types/keyv@3.1.4':
|
||||
@@ -6658,8 +6611,6 @@ snapshots:
|
||||
|
||||
clsx@2.1.1: {}
|
||||
|
||||
cluster-key-slot@1.1.2: {}
|
||||
|
||||
color-convert@2.0.1:
|
||||
dependencies:
|
||||
color-name: 1.1.4
|
||||
@@ -6777,8 +6728,6 @@ snapshots:
|
||||
|
||||
delayed-stream@1.0.0: {}
|
||||
|
||||
denque@2.1.0: {}
|
||||
|
||||
dequal@2.0.3: {}
|
||||
|
||||
detect-libc@2.1.2: {}
|
||||
@@ -7634,20 +7583,6 @@ snapshots:
|
||||
hasown: 2.0.2
|
||||
side-channel: 1.1.0
|
||||
|
||||
ioredis@5.8.2:
|
||||
dependencies:
|
||||
'@ioredis/commands': 1.4.0
|
||||
cluster-key-slot: 1.1.2
|
||||
debug: 4.4.3
|
||||
denque: 2.1.0
|
||||
lodash.defaults: 4.2.0
|
||||
lodash.isarguments: 3.1.0
|
||||
redis-errors: 1.2.0
|
||||
redis-parser: 3.0.0
|
||||
standard-as-callback: 2.1.0
|
||||
transitivePeerDependencies:
|
||||
- supports-color
|
||||
|
||||
ip-address@10.1.0: {}
|
||||
|
||||
is-alphabetical@2.0.1: {}
|
||||
@@ -7966,12 +7901,8 @@ snapshots:
|
||||
dependencies:
|
||||
p-locate: 5.0.0
|
||||
|
||||
lodash.defaults@4.2.0: {}
|
||||
|
||||
lodash.escaperegexp@4.1.2: {}
|
||||
|
||||
lodash.isarguments@3.1.0: {}
|
||||
|
||||
lodash.isequal@4.5.0: {}
|
||||
|
||||
lodash.merge@4.6.2: {}
|
||||
@@ -8808,12 +8739,6 @@ snapshots:
|
||||
|
||||
readdirp@5.0.0: {}
|
||||
|
||||
redis-errors@1.2.0: {}
|
||||
|
||||
redis-parser@3.0.0:
|
||||
dependencies:
|
||||
redis-errors: 1.2.0
|
||||
|
||||
reflect.getprototypeof@1.0.10:
|
||||
dependencies:
|
||||
call-bind: 1.0.8
|
||||
@@ -9112,8 +9037,6 @@ snapshots:
|
||||
|
||||
stackback@0.0.2: {}
|
||||
|
||||
standard-as-callback@2.1.0: {}
|
||||
|
||||
stat-mode@1.0.0: {}
|
||||
|
||||
std-env@3.10.0: {}
|
||||
|
||||
@@ -0,0 +1,24 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
|
||||
<plist version="1.0">
|
||||
<dict>
|
||||
<!-- Allow the app to be debugged -->
|
||||
<key>com.apple.security.cs.allow-jit</key>
|
||||
<true/>
|
||||
<!-- Allow loading unsigned libraries (needed for native modules) -->
|
||||
<key>com.apple.security.cs.allow-unsigned-executable-memory</key>
|
||||
<true/>
|
||||
<!-- Allow dyld environment variables (needed for Electron) -->
|
||||
<key>com.apple.security.cs.disable-library-validation</key>
|
||||
<true/>
|
||||
<!-- Allow spawning child processes (needed for Python backend and terminals) -->
|
||||
<key>com.apple.security.cs.allow-dyld-environment-variables</key>
|
||||
<true/>
|
||||
<!-- Network access for API calls -->
|
||||
<key>com.apple.security.network.client</key>
|
||||
<true/>
|
||||
<!-- File access for user documents -->
|
||||
<key>com.apple.security.files.user-selected.read-write</key>
|
||||
<true/>
|
||||
</dict>
|
||||
</plist>
|
||||
@@ -5,11 +5,37 @@ import { vi, beforeEach, afterEach } from 'vitest';
|
||||
import { mkdirSync, rmSync, existsSync } from 'fs';
|
||||
import path from 'path';
|
||||
|
||||
// Mock localStorage for tests that need it
|
||||
const localStorageMock = (() => {
|
||||
let store: Record<string, string> = {};
|
||||
|
||||
return {
|
||||
getItem: vi.fn((key: string) => store[key] || null),
|
||||
setItem: vi.fn((key: string, value: string) => {
|
||||
store[key] = value;
|
||||
}),
|
||||
removeItem: vi.fn((key: string) => {
|
||||
delete store[key];
|
||||
}),
|
||||
clear: vi.fn(() => {
|
||||
store = {};
|
||||
})
|
||||
};
|
||||
})();
|
||||
|
||||
// Make localStorage available globally
|
||||
Object.defineProperty(global, 'localStorage', {
|
||||
value: localStorageMock
|
||||
});
|
||||
|
||||
// Test data directory for isolated file operations
|
||||
export const TEST_DATA_DIR = '/tmp/auto-claude-ui-tests';
|
||||
|
||||
// Create fresh test directory before each test
|
||||
beforeEach(() => {
|
||||
// Clear localStorage
|
||||
localStorageMock.clear();
|
||||
|
||||
// Use a unique subdirectory per test to avoid race conditions in parallel tests
|
||||
const testId = `test-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
|
||||
const _testDir = path.join(TEST_DATA_DIR, testId);
|
||||
@@ -56,7 +82,13 @@ if (typeof window !== 'undefined') {
|
||||
getSettings: vi.fn(),
|
||||
saveSettings: vi.fn(),
|
||||
selectDirectory: vi.fn(),
|
||||
getAppVersion: vi.fn()
|
||||
getAppVersion: vi.fn(),
|
||||
// Tab state persistence (IPC-based)
|
||||
getTabState: vi.fn().mockResolvedValue({
|
||||
success: true,
|
||||
data: { openProjectIds: [], activeProjectId: null, tabOrder: [] }
|
||||
}),
|
||||
saveTabState: vi.fn().mockResolvedValue({ success: true })
|
||||
};
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,605 @@
|
||||
/**
|
||||
* Integration tests for Rate Limit Auto-Recovery System
|
||||
* Tests the complete flow: rate limit detection → account swap → task restart
|
||||
*/
|
||||
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
||||
import { EventEmitter } from 'events';
|
||||
|
||||
// Mock data
|
||||
const mockProfiles = {
|
||||
mai: {
|
||||
id: 'profile-mai',
|
||||
name: 'MAI',
|
||||
email: 'mai@example.com',
|
||||
isDefault: true,
|
||||
oauthToken: 'encrypted-token-mai',
|
||||
createdAt: new Date(),
|
||||
rateLimitEvents: []
|
||||
},
|
||||
mu: {
|
||||
id: 'profile-mu',
|
||||
name: 'MU',
|
||||
email: 'mu@example.com',
|
||||
isDefault: false,
|
||||
oauthToken: 'encrypted-token-mu',
|
||||
createdAt: new Date(),
|
||||
rateLimitEvents: []
|
||||
}
|
||||
};
|
||||
|
||||
const mockAutoSwitchSettings = {
|
||||
enabled: true,
|
||||
proactiveSwapEnabled: true,
|
||||
sessionThreshold: 95,
|
||||
weeklyThreshold: 99,
|
||||
autoSwitchOnRateLimit: true,
|
||||
usageCheckInterval: 30000
|
||||
};
|
||||
|
||||
// Create mock profile manager
|
||||
function createMockProfileManager(options: {
|
||||
activeProfileId?: string;
|
||||
profiles?: typeof mockProfiles;
|
||||
autoSwitchSettings?: typeof mockAutoSwitchSettings;
|
||||
bestAvailableProfile?: typeof mockProfiles.mai | null;
|
||||
} = {}) {
|
||||
const activeId = options.activeProfileId || 'profile-mai';
|
||||
const profiles = options.profiles || mockProfiles;
|
||||
const settings = options.autoSwitchSettings || mockAutoSwitchSettings;
|
||||
const bestProfile = options.bestAvailableProfile !== undefined
|
||||
? options.bestAvailableProfile
|
||||
: profiles.mu;
|
||||
|
||||
return {
|
||||
getActiveProfile: vi.fn(() => profiles[activeId === 'profile-mai' ? 'mai' : 'mu']),
|
||||
getProfile: vi.fn((id: string) => {
|
||||
if (id === 'profile-mai') return profiles.mai;
|
||||
if (id === 'profile-mu') return profiles.mu;
|
||||
return null;
|
||||
}),
|
||||
getBestAvailableProfile: vi.fn((_excludeProfileId?: string) => bestProfile),
|
||||
setActiveProfile: vi.fn(),
|
||||
recordRateLimitEvent: vi.fn(),
|
||||
getAutoSwitchSettings: vi.fn(() => settings),
|
||||
getProfileToken: vi.fn(() => 'decrypted-token'),
|
||||
getActiveProfileToken: vi.fn(() => 'decrypted-token')
|
||||
};
|
||||
}
|
||||
|
||||
describe('Rate Limit Auto-Recovery Integration', () => {
|
||||
let mockProfileManager: ReturnType<typeof createMockProfileManager>;
|
||||
|
||||
beforeEach(() => {
|
||||
vi.resetModules();
|
||||
mockProfileManager = createMockProfileManager();
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
describe('Rate Limit Detection Patterns', () => {
|
||||
beforeEach(() => {
|
||||
vi.doMock('../claude-profile-manager', () => ({
|
||||
getClaudeProfileManager: vi.fn(() => mockProfileManager)
|
||||
}));
|
||||
});
|
||||
|
||||
it('should detect standard Claude rate limit message', async () => {
|
||||
const { detectRateLimit } = await import('../rate-limit-detector');
|
||||
|
||||
const output = 'Limit reached · resets Dec 17 at 6am (Europe/Oslo)';
|
||||
const result = detectRateLimit(output);
|
||||
|
||||
expect(result.isRateLimited).toBe(true);
|
||||
expect(result.resetTime).toBe('Dec 17 at 6am (Europe/Oslo)');
|
||||
expect(result.limitType).toBe('weekly');
|
||||
});
|
||||
|
||||
it('should detect session limit (time only reset)', async () => {
|
||||
const { detectRateLimit } = await import('../rate-limit-detector');
|
||||
|
||||
const output = 'Limit reached • resets 11:59pm';
|
||||
const result = detectRateLimit(output);
|
||||
|
||||
expect(result.isRateLimited).toBe(true);
|
||||
expect(result.limitType).toBe('session');
|
||||
});
|
||||
|
||||
it('should detect rate limit in multiline output', async () => {
|
||||
const { detectRateLimit } = await import('../rate-limit-detector');
|
||||
|
||||
const output = `Processing task...
|
||||
Some output here
|
||||
Limit reached · resets Dec 20 at 3pm (America/New_York)
|
||||
Stack trace follows`;
|
||||
|
||||
const result = detectRateLimit(output);
|
||||
|
||||
expect(result.isRateLimited).toBe(true);
|
||||
expect(result.resetTime).toBe('Dec 20 at 3pm (America/New_York)');
|
||||
});
|
||||
|
||||
it('should suggest alternative profile when rate limited', async () => {
|
||||
const { detectRateLimit } = await import('../rate-limit-detector');
|
||||
|
||||
const output = 'Limit reached · resets Dec 17 at 6am';
|
||||
const result = detectRateLimit(output, 'profile-mai');
|
||||
|
||||
expect(result.isRateLimited).toBe(true);
|
||||
expect(result.suggestedProfile).toBeDefined();
|
||||
expect(result.suggestedProfile?.id).toBe('profile-mu');
|
||||
});
|
||||
|
||||
it('should record rate limit event in profile manager', async () => {
|
||||
const { detectRateLimit } = await import('../rate-limit-detector');
|
||||
|
||||
detectRateLimit('Limit reached · resets Dec 17 at 6am', 'profile-mai');
|
||||
|
||||
expect(mockProfileManager.recordRateLimitEvent).toHaveBeenCalledWith(
|
||||
'profile-mai',
|
||||
'Dec 17 at 6am'
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
describe('Auto-Switch Settings Verification', () => {
|
||||
it('should respect enabled flag', async () => {
|
||||
const disabledManager = createMockProfileManager({
|
||||
autoSwitchSettings: { ...mockAutoSwitchSettings, enabled: false }
|
||||
});
|
||||
|
||||
vi.doMock('../claude-profile-manager', () => ({
|
||||
getClaudeProfileManager: vi.fn(() => disabledManager)
|
||||
}));
|
||||
|
||||
const settings = disabledManager.getAutoSwitchSettings();
|
||||
|
||||
expect(settings.enabled).toBe(false);
|
||||
// When enabled is false, auto-swap should NOT happen even if autoSwitchOnRateLimit is true
|
||||
});
|
||||
|
||||
it('should respect autoSwitchOnRateLimit flag', async () => {
|
||||
const manualManager = createMockProfileManager({
|
||||
autoSwitchSettings: { ...mockAutoSwitchSettings, autoSwitchOnRateLimit: false }
|
||||
});
|
||||
|
||||
const settings = manualManager.getAutoSwitchSettings();
|
||||
|
||||
expect(settings.enabled).toBe(true);
|
||||
expect(settings.autoSwitchOnRateLimit).toBe(false);
|
||||
// When autoSwitchOnRateLimit is false, should show manual modal instead
|
||||
});
|
||||
|
||||
it('should have both enabled and autoSwitchOnRateLimit for auto-recovery', () => {
|
||||
const settings = mockProfileManager.getAutoSwitchSettings();
|
||||
|
||||
// Both must be true for automatic recovery
|
||||
const shouldAutoRecover = settings.enabled && settings.autoSwitchOnRateLimit;
|
||||
expect(shouldAutoRecover).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
describe('Profile Scoring and Selection', () => {
|
||||
it('should return alternative profile when one is available', () => {
|
||||
const bestProfile = mockProfileManager.getBestAvailableProfile('profile-mai');
|
||||
|
||||
expect(bestProfile).toBeDefined();
|
||||
expect(bestProfile?.id).toBe('profile-mu');
|
||||
});
|
||||
|
||||
it('should return null when no alternative profile is available', () => {
|
||||
const noAlternativeManager = createMockProfileManager({
|
||||
bestAvailableProfile: null
|
||||
});
|
||||
|
||||
const bestProfile = noAlternativeManager.getBestAvailableProfile('profile-mai');
|
||||
|
||||
expect(bestProfile).toBeNull();
|
||||
});
|
||||
|
||||
it('should not return the same profile that hit the limit', () => {
|
||||
const bestProfile = mockProfileManager.getBestAvailableProfile('profile-mai');
|
||||
|
||||
// Best profile should be different from the one that hit the limit
|
||||
expect(bestProfile?.id).not.toBe('profile-mai');
|
||||
});
|
||||
});
|
||||
|
||||
describe('Auto-Recovery Flow Simulation', () => {
|
||||
/**
|
||||
* Simulates the flow in agent-process.ts lines 274-327
|
||||
*/
|
||||
function simulateRateLimitRecovery(
|
||||
output: string,
|
||||
exitCode: number,
|
||||
profileManager: ReturnType<typeof createMockProfileManager>
|
||||
): {
|
||||
rateLimitDetected: boolean;
|
||||
autoSwapped: boolean;
|
||||
taskRestarted: boolean;
|
||||
modalShown: boolean;
|
||||
swappedToProfile?: { id: string; name: string };
|
||||
} {
|
||||
const result = {
|
||||
rateLimitDetected: false,
|
||||
autoSwapped: false,
|
||||
taskRestarted: false,
|
||||
modalShown: false,
|
||||
swappedToProfile: undefined as { id: string; name: string } | undefined
|
||||
};
|
||||
|
||||
// Only check rate limit if process failed
|
||||
if (exitCode !== 0) {
|
||||
// Simulate detectRateLimit
|
||||
const rateLimitPattern = /Limit reached\s*[·•]\s*resets\s+(.+?)(?:\s*$|\n)/im;
|
||||
const rateIndicators = [/rate\s*limit/i, /usage\s*limit/i, /limit\s*reached/i];
|
||||
|
||||
const isRateLimited = rateLimitPattern.test(output) ||
|
||||
rateIndicators.some(p => p.test(output));
|
||||
|
||||
if (isRateLimited) {
|
||||
result.rateLimitDetected = true;
|
||||
|
||||
const settings = profileManager.getAutoSwitchSettings();
|
||||
|
||||
if (settings.enabled && settings.autoSwitchOnRateLimit) {
|
||||
const bestProfile = profileManager.getBestAvailableProfile('current-profile');
|
||||
|
||||
if (bestProfile) {
|
||||
// Auto-swap
|
||||
profileManager.setActiveProfile(bestProfile.id);
|
||||
result.autoSwapped = true;
|
||||
result.swappedToProfile = { id: bestProfile.id, name: bestProfile.name };
|
||||
result.taskRestarted = true;
|
||||
result.modalShown = true; // Notification modal
|
||||
} else {
|
||||
// No alternative - show manual modal
|
||||
result.modalShown = true;
|
||||
}
|
||||
} else {
|
||||
// Auto-switch disabled - show manual modal
|
||||
result.modalShown = true;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
it('should auto-swap and restart when all conditions met', () => {
|
||||
const result = simulateRateLimitRecovery(
|
||||
'Limit reached · resets Dec 17 at 6am',
|
||||
1, // non-zero exit
|
||||
mockProfileManager
|
||||
);
|
||||
|
||||
expect(result.rateLimitDetected).toBe(true);
|
||||
expect(result.autoSwapped).toBe(true);
|
||||
expect(result.taskRestarted).toBe(true);
|
||||
expect(result.modalShown).toBe(true);
|
||||
expect(result.swappedToProfile?.id).toBe('profile-mu');
|
||||
});
|
||||
|
||||
it('should NOT auto-swap when exit code is 0', () => {
|
||||
const result = simulateRateLimitRecovery(
|
||||
'Limit reached · resets Dec 17 at 6am',
|
||||
0, // success exit
|
||||
mockProfileManager
|
||||
);
|
||||
|
||||
expect(result.rateLimitDetected).toBe(false);
|
||||
expect(result.autoSwapped).toBe(false);
|
||||
expect(result.taskRestarted).toBe(false);
|
||||
});
|
||||
|
||||
it('should NOT auto-swap when enabled is false', () => {
|
||||
const disabledManager = createMockProfileManager({
|
||||
autoSwitchSettings: { ...mockAutoSwitchSettings, enabled: false }
|
||||
});
|
||||
|
||||
const result = simulateRateLimitRecovery(
|
||||
'Limit reached · resets Dec 17 at 6am',
|
||||
1,
|
||||
disabledManager
|
||||
);
|
||||
|
||||
expect(result.rateLimitDetected).toBe(true);
|
||||
expect(result.autoSwapped).toBe(false);
|
||||
expect(result.modalShown).toBe(true); // Manual modal
|
||||
});
|
||||
|
||||
it('should NOT auto-swap when autoSwitchOnRateLimit is false', () => {
|
||||
const manualManager = createMockProfileManager({
|
||||
autoSwitchSettings: { ...mockAutoSwitchSettings, autoSwitchOnRateLimit: false }
|
||||
});
|
||||
|
||||
const result = simulateRateLimitRecovery(
|
||||
'Limit reached · resets Dec 17 at 6am',
|
||||
1,
|
||||
manualManager
|
||||
);
|
||||
|
||||
expect(result.rateLimitDetected).toBe(true);
|
||||
expect(result.autoSwapped).toBe(false);
|
||||
expect(result.modalShown).toBe(true); // Manual modal
|
||||
});
|
||||
|
||||
it('should show manual modal when no alternative profile available', () => {
|
||||
const noAlternativeManager = createMockProfileManager({
|
||||
bestAvailableProfile: null
|
||||
});
|
||||
|
||||
const result = simulateRateLimitRecovery(
|
||||
'Limit reached · resets Dec 17 at 6am',
|
||||
1,
|
||||
noAlternativeManager
|
||||
);
|
||||
|
||||
expect(result.rateLimitDetected).toBe(true);
|
||||
expect(result.autoSwapped).toBe(false);
|
||||
expect(result.taskRestarted).toBe(false);
|
||||
expect(result.modalShown).toBe(true); // Manual modal because no alternative
|
||||
});
|
||||
|
||||
it('should NOT detect rate limit for normal errors', () => {
|
||||
const result = simulateRateLimitRecovery(
|
||||
'Error: File not found',
|
||||
1,
|
||||
mockProfileManager
|
||||
);
|
||||
|
||||
expect(result.rateLimitDetected).toBe(false);
|
||||
expect(result.autoSwapped).toBe(false);
|
||||
expect(result.modalShown).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
describe('Task Restart Context Preservation', () => {
|
||||
it('should preserve task context for restart', () => {
|
||||
// Simulate task execution context
|
||||
const taskContext = {
|
||||
taskId: 'task-123',
|
||||
projectPath: '/path/to/project',
|
||||
specId: 'spec-001',
|
||||
options: { qa: false },
|
||||
swapCount: 0,
|
||||
isSpecCreation: false
|
||||
};
|
||||
|
||||
// After swap, swapCount should increment
|
||||
taskContext.swapCount++;
|
||||
|
||||
expect(taskContext.swapCount).toBe(1);
|
||||
});
|
||||
|
||||
it('should limit swap retries to prevent infinite loops', () => {
|
||||
const MAX_SWAPS = 2;
|
||||
let swapCount = 0;
|
||||
|
||||
// Simulate multiple rate limits
|
||||
for (let i = 0; i < 5; i++) {
|
||||
if (swapCount >= MAX_SWAPS) {
|
||||
break; // Should stop after 2 swaps
|
||||
}
|
||||
swapCount++;
|
||||
}
|
||||
|
||||
expect(swapCount).toBe(MAX_SWAPS);
|
||||
});
|
||||
});
|
||||
|
||||
describe('Event Emission Verification', () => {
|
||||
it('should emit sdk-rate-limit event on rate limit', () => {
|
||||
const emitter = new EventEmitter();
|
||||
const sdkRateLimitHandler = vi.fn();
|
||||
|
||||
emitter.on('sdk-rate-limit', sdkRateLimitHandler);
|
||||
|
||||
// Simulate rate limit detected with auto-swap
|
||||
const rateLimitInfo = {
|
||||
source: 'task' as const,
|
||||
taskId: 'task-123',
|
||||
resetTime: 'Dec 17 at 6am',
|
||||
limitType: 'weekly' as const,
|
||||
profileId: 'profile-mai',
|
||||
profileName: 'MAI',
|
||||
wasAutoSwapped: true,
|
||||
swappedToProfile: { id: 'profile-mu', name: 'MU' },
|
||||
swapReason: 'reactive' as const,
|
||||
detectedAt: new Date()
|
||||
};
|
||||
|
||||
emitter.emit('sdk-rate-limit', rateLimitInfo);
|
||||
|
||||
expect(sdkRateLimitHandler).toHaveBeenCalledWith(rateLimitInfo);
|
||||
expect(sdkRateLimitHandler).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it('should emit auto-swap-restart-task event for task restart', () => {
|
||||
const emitter = new EventEmitter();
|
||||
const restartHandler = vi.fn();
|
||||
|
||||
emitter.on('auto-swap-restart-task', restartHandler);
|
||||
|
||||
emitter.emit('auto-swap-restart-task', 'task-123', 'profile-mu');
|
||||
|
||||
expect(restartHandler).toHaveBeenCalledWith('task-123', 'profile-mu');
|
||||
});
|
||||
|
||||
it('should handle event chain: rate-limit → swap → restart', () => {
|
||||
const emitter = new EventEmitter();
|
||||
const events: string[] = [];
|
||||
|
||||
emitter.on('sdk-rate-limit', () => events.push('sdk-rate-limit'));
|
||||
emitter.on('auto-swap-restart-task', () => events.push('auto-swap-restart-task'));
|
||||
|
||||
// Simulate the flow
|
||||
emitter.emit('sdk-rate-limit', { /* info */ });
|
||||
emitter.emit('auto-swap-restart-task', 'task-123', 'profile-mu');
|
||||
|
||||
expect(events).toEqual(['sdk-rate-limit', 'auto-swap-restart-task']);
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('Rate Limit Edge Cases', () => {
|
||||
beforeEach(() => {
|
||||
vi.resetModules();
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
describe('Pattern Matching Edge Cases', () => {
|
||||
const mockManager = createMockProfileManager();
|
||||
|
||||
beforeEach(() => {
|
||||
vi.doMock('../claude-profile-manager', () => ({
|
||||
getClaudeProfileManager: vi.fn(() => mockManager)
|
||||
}));
|
||||
});
|
||||
|
||||
it('should handle different bullet characters', async () => {
|
||||
const { detectRateLimit } = await import('../rate-limit-detector');
|
||||
|
||||
// Middle dot (·)
|
||||
expect(detectRateLimit('Limit reached · resets 5pm').isRateLimited).toBe(true);
|
||||
|
||||
// Bullet (•)
|
||||
expect(detectRateLimit('Limit reached • resets 5pm').isRateLimited).toBe(true);
|
||||
});
|
||||
|
||||
it('should handle different timezone formats', async () => {
|
||||
const { detectRateLimit } = await import('../rate-limit-detector');
|
||||
|
||||
const timezones = [
|
||||
'Dec 17 at 6am (Europe/Oslo)',
|
||||
'Dec 17 at 6am (America/New_York)',
|
||||
'Dec 17 at 6am (Asia/Tokyo)',
|
||||
'Dec 17 at 6am (UTC)',
|
||||
'Dec 17 at 6am' // No timezone
|
||||
];
|
||||
|
||||
for (const tz of timezones) {
|
||||
const result = detectRateLimit(`Limit reached · resets ${tz}`);
|
||||
expect(result.isRateLimited).toBe(true);
|
||||
}
|
||||
});
|
||||
|
||||
it('should handle 12-hour and 24-hour time formats', async () => {
|
||||
const { detectRateLimit } = await import('../rate-limit-detector');
|
||||
|
||||
expect(detectRateLimit('Limit reached · resets 11:59pm').isRateLimited).toBe(true);
|
||||
expect(detectRateLimit('Limit reached · resets 6am').isRateLimited).toBe(true);
|
||||
expect(detectRateLimit('Limit reached · resets 18:00').isRateLimited).toBe(true);
|
||||
});
|
||||
|
||||
it('should NOT false-positive on similar messages', async () => {
|
||||
const { detectRateLimit } = await import('../rate-limit-detector');
|
||||
|
||||
// These should NOT trigger rate limit detection
|
||||
const falsePositives = [
|
||||
'Limit your requests to avoid issues', // Contains 'limit' but not rate limit
|
||||
'The speed limit is 60mph', // Unrelated limit
|
||||
'Character limit reached for input field' // Different kind of limit
|
||||
];
|
||||
|
||||
for (const msg of falsePositives) {
|
||||
const result = detectRateLimit(msg);
|
||||
// Note: Some may still match secondary indicators - that's intentional
|
||||
// The primary pattern should NOT match these
|
||||
const primaryPattern = /Limit reached\s*[·•]\s*resets/i;
|
||||
expect(primaryPattern.test(msg)).toBe(false);
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
describe('Both Profiles Rate Limited', () => {
|
||||
it('should return null when all profiles are rate limited', () => {
|
||||
const bothLimitedManager = createMockProfileManager({
|
||||
bestAvailableProfile: null
|
||||
});
|
||||
|
||||
const best = bothLimitedManager.getBestAvailableProfile('profile-mai');
|
||||
expect(best).toBeNull();
|
||||
});
|
||||
|
||||
it('should show manual modal when no profiles available', () => {
|
||||
// User must either wait or add a new account
|
||||
const bothLimitedManager = createMockProfileManager({
|
||||
bestAvailableProfile: null
|
||||
});
|
||||
|
||||
const settings = bothLimitedManager.getAutoSwitchSettings();
|
||||
const bestProfile = bothLimitedManager.getBestAvailableProfile('profile-mai');
|
||||
|
||||
// Even with auto-switch enabled, should show modal since no alternative
|
||||
const shouldShowManualModal = settings.enabled && settings.autoSwitchOnRateLimit && !bestProfile;
|
||||
expect(shouldShowManualModal).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
describe('Rapid Rate Limit Succession', () => {
|
||||
it('should enforce max swap count', () => {
|
||||
const MAX_SWAP_COUNT = 2;
|
||||
const context = { swapCount: 0 };
|
||||
|
||||
// First swap
|
||||
context.swapCount++;
|
||||
expect(context.swapCount < MAX_SWAP_COUNT).toBe(true);
|
||||
|
||||
// Second swap
|
||||
context.swapCount++;
|
||||
expect(context.swapCount >= MAX_SWAP_COUNT).toBe(true);
|
||||
|
||||
// Third swap should be blocked
|
||||
const shouldAllowSwap = context.swapCount < MAX_SWAP_COUNT;
|
||||
expect(shouldAllowSwap).toBe(false);
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('Modal Behavior with Reactive Recovery', () => {
|
||||
describe('Modal Content Variations', () => {
|
||||
it('should show notification-style modal when auto-swapped', () => {
|
||||
const rateLimitInfo = {
|
||||
source: 'task' as const,
|
||||
wasAutoSwapped: true,
|
||||
swappedToProfile: { id: 'profile-mu', name: 'MU' },
|
||||
swapReason: 'reactive' as const
|
||||
};
|
||||
|
||||
// When wasAutoSwapped is true, modal should be informational
|
||||
expect(rateLimitInfo.wasAutoSwapped).toBe(true);
|
||||
expect(rateLimitInfo.swapReason).toBe('reactive');
|
||||
});
|
||||
|
||||
it('should show action-required modal when NOT auto-swapped', () => {
|
||||
const rateLimitInfo = {
|
||||
source: 'task' as const,
|
||||
wasAutoSwapped: false,
|
||||
suggestedProfile: { id: 'profile-mu', name: 'MU' }
|
||||
};
|
||||
|
||||
// When wasAutoSwapped is false, user needs to take action
|
||||
expect(rateLimitInfo.wasAutoSwapped).toBe(false);
|
||||
});
|
||||
|
||||
it('should distinguish proactive vs reactive swaps', () => {
|
||||
const proactiveSwap = {
|
||||
wasAutoSwapped: true,
|
||||
swapReason: 'proactive' as const // Before limit hit
|
||||
};
|
||||
|
||||
const reactiveSwap = {
|
||||
wasAutoSwapped: true,
|
||||
swapReason: 'reactive' as const // After limit hit
|
||||
};
|
||||
|
||||
expect(proactiveSwap.swapReason).toBe('proactive');
|
||||
expect(reactiveSwap.swapReason).toBe('reactive');
|
||||
});
|
||||
});
|
||||
});
|
||||
@@ -43,8 +43,10 @@ export class AgentManager extends EventEmitter {
|
||||
this.queueManager = new AgentQueueManager(this.state, this.events, this.processManager, this);
|
||||
|
||||
// Listen for auto-swap restart events
|
||||
this.on('auto-swap-restart-task', (taskId: string, _newProfileId: string) => {
|
||||
this.restartTask(taskId);
|
||||
this.on('auto-swap-restart-task', (taskId: string, newProfileId: string) => {
|
||||
console.log('[AgentManager] Received auto-swap-restart-task event:', { taskId, newProfileId });
|
||||
const success = this.restartTask(taskId, newProfileId);
|
||||
console.log('[AgentManager] Task restart result:', success ? 'SUCCESS' : 'FAILED');
|
||||
});
|
||||
|
||||
// Listen for task completion to clean up context (prevent memory leak)
|
||||
@@ -350,28 +352,56 @@ export class AgentManager extends EventEmitter {
|
||||
|
||||
/**
|
||||
* Restart task after profile swap
|
||||
* @param taskId - The task to restart
|
||||
* @param newProfileId - Optional new profile ID to apply (from auto-swap)
|
||||
*/
|
||||
restartTask(taskId: string): boolean {
|
||||
restartTask(taskId: string, newProfileId?: string): boolean {
|
||||
console.log('[AgentManager] restartTask called for:', taskId, 'with newProfileId:', newProfileId);
|
||||
|
||||
const context = this.taskExecutionContext.get(taskId);
|
||||
if (!context) {
|
||||
console.error('[AgentManager] No context for task:', taskId);
|
||||
console.log('[AgentManager] Available task contexts:', Array.from(this.taskExecutionContext.keys()));
|
||||
return false;
|
||||
}
|
||||
|
||||
console.log('[AgentManager] Task context found:', {
|
||||
taskId,
|
||||
projectPath: context.projectPath,
|
||||
specId: context.specId,
|
||||
isSpecCreation: context.isSpecCreation,
|
||||
swapCount: context.swapCount
|
||||
});
|
||||
|
||||
// Prevent infinite swap loops
|
||||
if (context.swapCount >= 2) {
|
||||
console.error('[AgentManager] Max swap count reached for task:', taskId);
|
||||
console.error('[AgentManager] Max swap count reached for task:', taskId, '- stopping restart loop');
|
||||
return false;
|
||||
}
|
||||
|
||||
context.swapCount++;
|
||||
console.log('[AgentManager] Incremented swap count to:', context.swapCount);
|
||||
|
||||
// If a new profile was specified, ensure it's set as active before restart
|
||||
if (newProfileId) {
|
||||
const profileManager = getClaudeProfileManager();
|
||||
const currentActiveId = profileManager.getActiveProfile()?.id;
|
||||
if (currentActiveId !== newProfileId) {
|
||||
console.log('[AgentManager] Setting active profile to:', newProfileId);
|
||||
profileManager.setActiveProfile(newProfileId);
|
||||
}
|
||||
}
|
||||
|
||||
// Kill current process
|
||||
console.log('[AgentManager] Killing current process for task:', taskId);
|
||||
this.killTask(taskId);
|
||||
|
||||
// Wait for cleanup, then restart
|
||||
console.log('[AgentManager] Scheduling task restart in 500ms');
|
||||
setTimeout(() => {
|
||||
console.log('[AgentManager] Restarting task now:', taskId);
|
||||
if (context.isSpecCreation) {
|
||||
console.log('[AgentManager] Restarting as spec creation');
|
||||
this.startSpecCreation(
|
||||
taskId,
|
||||
context.projectPath,
|
||||
@@ -380,6 +410,7 @@ export class AgentManager extends EventEmitter {
|
||||
context.metadata
|
||||
);
|
||||
} else {
|
||||
console.log('[AgentManager] Restarting as task execution');
|
||||
this.startTaskExecution(
|
||||
taskId,
|
||||
context.projectPath,
|
||||
|
||||
@@ -273,22 +273,45 @@ export class AgentProcessManager {
|
||||
|
||||
// Check for rate limit if process failed
|
||||
if (code !== 0) {
|
||||
console.log('[AgentProcess] Process failed with code:', code, 'for task:', taskId);
|
||||
console.log('[AgentProcess] Checking for rate limit in output (last 500 chars):', allOutput.slice(-500));
|
||||
|
||||
const rateLimitDetection = detectRateLimit(allOutput);
|
||||
console.log('[AgentProcess] Rate limit detection result:', {
|
||||
isRateLimited: rateLimitDetection.isRateLimited,
|
||||
resetTime: rateLimitDetection.resetTime,
|
||||
limitType: rateLimitDetection.limitType,
|
||||
profileId: rateLimitDetection.profileId,
|
||||
suggestedProfile: rateLimitDetection.suggestedProfile
|
||||
});
|
||||
|
||||
if (rateLimitDetection.isRateLimited) {
|
||||
// Check if auto-swap is enabled
|
||||
const profileManager = getClaudeProfileManager();
|
||||
const autoSwitchSettings = profileManager.getAutoSwitchSettings();
|
||||
|
||||
console.log('[AgentProcess] Auto-switch settings:', {
|
||||
enabled: autoSwitchSettings.enabled,
|
||||
autoSwitchOnRateLimit: autoSwitchSettings.autoSwitchOnRateLimit,
|
||||
proactiveSwapEnabled: autoSwitchSettings.proactiveSwapEnabled
|
||||
});
|
||||
|
||||
if (autoSwitchSettings.enabled && autoSwitchSettings.autoSwitchOnRateLimit) {
|
||||
const currentProfileId = rateLimitDetection.profileId;
|
||||
const bestProfile = profileManager.getBestAvailableProfile(currentProfileId);
|
||||
|
||||
console.log('[AgentProcess] Best available profile:', bestProfile ? {
|
||||
id: bestProfile.id,
|
||||
name: bestProfile.name
|
||||
} : 'NONE');
|
||||
|
||||
if (bestProfile) {
|
||||
// Switch active profile
|
||||
console.log('[AgentProcess] AUTO-SWAP: Switching from', currentProfileId, 'to', bestProfile.id);
|
||||
profileManager.setActiveProfile(bestProfile.id);
|
||||
|
||||
// Emit swap info (for modal)
|
||||
const source = processType === 'spec-creation' ? 'task' : 'task';
|
||||
const source = processType === 'spec-creation' ? 'roadmap' : 'task';
|
||||
const rateLimitInfo = createSDKRateLimitInfo(source, rateLimitDetection, {
|
||||
taskId
|
||||
});
|
||||
@@ -298,30 +321,42 @@ export class AgentProcessManager {
|
||||
name: bestProfile.name
|
||||
};
|
||||
rateLimitInfo.swapReason = 'reactive';
|
||||
|
||||
console.log('[AgentProcess] Emitting sdk-rate-limit event (auto-swapped):', rateLimitInfo);
|
||||
this.emitter.emit('sdk-rate-limit', rateLimitInfo);
|
||||
|
||||
// Restart task
|
||||
console.log('[AgentProcess] Emitting auto-swap-restart-task event for task:', taskId);
|
||||
this.emitter.emit('auto-swap-restart-task', taskId, bestProfile.id);
|
||||
return;
|
||||
} else {
|
||||
console.log('[AgentProcess] No alternative profile available - falling back to manual modal');
|
||||
}
|
||||
} else {
|
||||
console.log('[AgentProcess] Auto-switch disabled - showing manual modal');
|
||||
}
|
||||
|
||||
// Fall back to manual modal (no auto-swap or no alternative profile)
|
||||
const source = processType === 'spec-creation' ? 'task' : 'task';
|
||||
const source = processType === 'spec-creation' ? 'roadmap' : 'task';
|
||||
const rateLimitInfo = createSDKRateLimitInfo(source, rateLimitDetection, {
|
||||
taskId
|
||||
});
|
||||
console.log('[AgentProcess] Emitting sdk-rate-limit event (manual):', rateLimitInfo);
|
||||
this.emitter.emit('sdk-rate-limit', rateLimitInfo);
|
||||
} else {
|
||||
console.log('[AgentProcess] No rate limit detected - checking for auth failure');
|
||||
// Not rate limited - check for authentication failure
|
||||
const authFailureDetection = detectAuthFailure(allOutput);
|
||||
if (authFailureDetection.isAuthFailure) {
|
||||
console.log('[AgentProcess] Auth failure detected:', authFailureDetection);
|
||||
this.emitter.emit('auth-failure', taskId, {
|
||||
profileId: authFailureDetection.profileId,
|
||||
failureType: authFailureDetection.failureType,
|
||||
message: authFailureDetection.message,
|
||||
originalError: authFailureDetection.originalError
|
||||
});
|
||||
} else {
|
||||
console.log('[AgentProcess] Process failed but no rate limit or auth failure detected');
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,243 @@
|
||||
/**
|
||||
* API Validation Service
|
||||
*
|
||||
* Provides validation for external LLM API providers (OpenAI, Anthropic, Google, etc.)
|
||||
* Used by the Graphiti memory integration for embedding and LLM operations.
|
||||
*/
|
||||
|
||||
export interface ApiValidationResult {
|
||||
success: boolean;
|
||||
message: string;
|
||||
details?: {
|
||||
provider?: string;
|
||||
model?: string;
|
||||
latencyMs?: number;
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Validate OpenAI API key by attempting to list models
|
||||
* @param apiKey - OpenAI API key
|
||||
*/
|
||||
export async function validateOpenAIApiKey(
|
||||
apiKey: string
|
||||
): Promise<ApiValidationResult> {
|
||||
if (!apiKey || !apiKey.trim()) {
|
||||
return {
|
||||
success: false,
|
||||
message: 'API key is required',
|
||||
};
|
||||
}
|
||||
|
||||
// Basic format validation
|
||||
const trimmedKey = apiKey.trim();
|
||||
if (!trimmedKey.startsWith('sk-') && !trimmedKey.startsWith('sess-')) {
|
||||
return {
|
||||
success: false,
|
||||
message: 'Invalid API key format. OpenAI API keys should start with "sk-" or "sess-"',
|
||||
};
|
||||
}
|
||||
|
||||
try {
|
||||
const startTime = Date.now();
|
||||
|
||||
// Use native https module to avoid additional dependencies
|
||||
const result = await new Promise<ApiValidationResult>((resolve) => {
|
||||
const https = require('https');
|
||||
|
||||
const options = {
|
||||
hostname: 'api.openai.com',
|
||||
port: 443,
|
||||
path: '/v1/models',
|
||||
method: 'GET',
|
||||
headers: {
|
||||
Authorization: `Bearer ${trimmedKey}`,
|
||||
'Content-Type': 'application/json',
|
||||
},
|
||||
timeout: 15000,
|
||||
};
|
||||
|
||||
const req = https.request(options, (res: { statusCode: number; on: (event: string, callback: (chunk: Buffer) => void) => void }) => {
|
||||
let data = '';
|
||||
|
||||
res.on('data', (chunk: Buffer) => {
|
||||
data += chunk;
|
||||
});
|
||||
|
||||
res.on('end', () => {
|
||||
const latencyMs = Date.now() - startTime;
|
||||
|
||||
if (res.statusCode === 200) {
|
||||
resolve({
|
||||
success: true,
|
||||
message: 'OpenAI API key is valid',
|
||||
details: {
|
||||
provider: 'openai',
|
||||
latencyMs,
|
||||
},
|
||||
});
|
||||
} else if (res.statusCode === 401) {
|
||||
resolve({
|
||||
success: false,
|
||||
message: 'Invalid API key. Please check your OpenAI API key.',
|
||||
});
|
||||
} else if (res.statusCode === 429) {
|
||||
// Rate limited but key is valid
|
||||
resolve({
|
||||
success: true,
|
||||
message: 'OpenAI API key is valid (rate limited, please wait)',
|
||||
details: {
|
||||
provider: 'openai',
|
||||
latencyMs,
|
||||
},
|
||||
});
|
||||
} else {
|
||||
try {
|
||||
const errorData = JSON.parse(data);
|
||||
resolve({
|
||||
success: false,
|
||||
message: errorData.error?.message || `API error: ${res.statusCode}`,
|
||||
});
|
||||
} catch {
|
||||
resolve({
|
||||
success: false,
|
||||
message: `API error: ${res.statusCode}`,
|
||||
});
|
||||
}
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
req.on('error', (error: Error) => {
|
||||
resolve({
|
||||
success: false,
|
||||
message: `Connection error: ${error.message}`,
|
||||
});
|
||||
});
|
||||
|
||||
req.on('timeout', () => {
|
||||
req.destroy();
|
||||
resolve({
|
||||
success: false,
|
||||
message: 'Connection timeout. Please check your network connection.',
|
||||
});
|
||||
});
|
||||
|
||||
req.end();
|
||||
});
|
||||
|
||||
return result;
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
message: error instanceof Error ? error.message : 'Unknown error occurred',
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Validate Anthropic API key
|
||||
* @param apiKey - Anthropic API key
|
||||
*/
|
||||
export async function validateAnthropicApiKey(
|
||||
apiKey: string
|
||||
): Promise<ApiValidationResult> {
|
||||
if (!apiKey || !apiKey.trim()) {
|
||||
return {
|
||||
success: false,
|
||||
message: 'API key is required',
|
||||
};
|
||||
}
|
||||
|
||||
const trimmedKey = apiKey.trim();
|
||||
if (!trimmedKey.startsWith('sk-ant-')) {
|
||||
return {
|
||||
success: false,
|
||||
message: 'Invalid API key format. Anthropic API keys should start with "sk-ant-"',
|
||||
};
|
||||
}
|
||||
|
||||
// For now, just validate format - full validation would require an API call
|
||||
return {
|
||||
success: true,
|
||||
message: 'Anthropic API key format is valid',
|
||||
details: {
|
||||
provider: 'anthropic',
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Validate Google AI API key
|
||||
* @param apiKey - Google AI API key
|
||||
*/
|
||||
export async function validateGoogleApiKey(
|
||||
apiKey: string
|
||||
): Promise<ApiValidationResult> {
|
||||
if (!apiKey || !apiKey.trim()) {
|
||||
return {
|
||||
success: false,
|
||||
message: 'API key is required',
|
||||
};
|
||||
}
|
||||
|
||||
const trimmedKey = apiKey.trim();
|
||||
if (!trimmedKey.startsWith('AIza')) {
|
||||
return {
|
||||
success: false,
|
||||
message: 'Invalid API key format. Google AI API keys should start with "AIza"',
|
||||
};
|
||||
}
|
||||
|
||||
return {
|
||||
success: true,
|
||||
message: 'Google AI API key format is valid',
|
||||
details: {
|
||||
provider: 'google',
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Validate an LLM provider API key based on provider type
|
||||
* @param provider - The LLM provider (openai, anthropic, google, etc.)
|
||||
* @param apiKey - The API key to validate
|
||||
*/
|
||||
export async function validateLLMApiKey(
|
||||
provider: string,
|
||||
apiKey: string
|
||||
): Promise<ApiValidationResult> {
|
||||
switch (provider) {
|
||||
case 'openai':
|
||||
return validateOpenAIApiKey(apiKey);
|
||||
case 'anthropic':
|
||||
return validateAnthropicApiKey(apiKey);
|
||||
case 'google':
|
||||
return validateGoogleApiKey(apiKey);
|
||||
case 'ollama':
|
||||
// Ollama is local, no API key needed
|
||||
return {
|
||||
success: true,
|
||||
message: 'Ollama runs locally, no API key required',
|
||||
details: { provider: 'ollama' },
|
||||
};
|
||||
case 'azure_openai':
|
||||
// Azure OpenAI uses different auth, just validate presence
|
||||
if (!apiKey || !apiKey.trim()) {
|
||||
return {
|
||||
success: false,
|
||||
message: 'Azure OpenAI API key is required',
|
||||
};
|
||||
}
|
||||
return {
|
||||
success: true,
|
||||
message: 'Azure OpenAI API key format accepted',
|
||||
details: { provider: 'azure_openai' },
|
||||
};
|
||||
default:
|
||||
return {
|
||||
success: false,
|
||||
message: `Unknown provider: ${provider}`,
|
||||
};
|
||||
}
|
||||
}
|
||||
@@ -1,586 +0,0 @@
|
||||
/**
|
||||
* Docker & FalkorDB Service
|
||||
*
|
||||
* Provides automatic detection and management of Docker and FalkorDB
|
||||
* for non-technical users. This eliminates the need for manual
|
||||
* "docker --version" verification steps.
|
||||
*/
|
||||
|
||||
import { exec, spawn } from 'child_process';
|
||||
import { promisify } from 'util';
|
||||
|
||||
const execAsync = promisify(exec);
|
||||
|
||||
// FalkorDB container configuration
|
||||
const FALKORDB_CONTAINER_NAME = 'auto-claude-falkordb';
|
||||
const FALKORDB_IMAGE = 'falkordb/falkordb:latest';
|
||||
const FALKORDB_DEFAULT_PORT = 6380;
|
||||
|
||||
export interface DockerStatus {
|
||||
installed: boolean;
|
||||
running: boolean;
|
||||
version?: string;
|
||||
error?: string;
|
||||
}
|
||||
|
||||
export interface FalkorDBStatus {
|
||||
containerExists: boolean;
|
||||
containerRunning: boolean;
|
||||
containerName: string;
|
||||
port: number;
|
||||
healthy: boolean;
|
||||
error?: string;
|
||||
}
|
||||
|
||||
export interface InfrastructureStatus {
|
||||
docker: DockerStatus;
|
||||
falkordb: FalkorDBStatus;
|
||||
ready: boolean; // True if both Docker is running and FalkorDB is healthy
|
||||
}
|
||||
|
||||
/**
|
||||
* Check if Docker is installed and running
|
||||
*/
|
||||
export async function checkDockerStatus(): Promise<DockerStatus> {
|
||||
try {
|
||||
// Check if Docker CLI is available
|
||||
const { stdout: versionOutput } = await execAsync('docker --version', {
|
||||
timeout: 5000,
|
||||
});
|
||||
|
||||
const version = versionOutput.trim();
|
||||
|
||||
// Check if Docker daemon is running by trying to ping it
|
||||
try {
|
||||
await execAsync('docker info', { timeout: 10000 });
|
||||
return {
|
||||
installed: true,
|
||||
running: true,
|
||||
version,
|
||||
};
|
||||
} catch {
|
||||
return {
|
||||
installed: true,
|
||||
running: false,
|
||||
version,
|
||||
error: 'Docker is installed but not running. Please start Docker Desktop.',
|
||||
};
|
||||
}
|
||||
} catch (error) {
|
||||
const errorMsg = error instanceof Error ? error.message : String(error);
|
||||
|
||||
// Check if it's a "command not found" type error
|
||||
if (
|
||||
errorMsg.includes('not found') ||
|
||||
errorMsg.includes('ENOENT') ||
|
||||
errorMsg.includes('not recognized')
|
||||
) {
|
||||
return {
|
||||
installed: false,
|
||||
running: false,
|
||||
error: 'Docker is not installed. Please install Docker Desktop.',
|
||||
};
|
||||
}
|
||||
|
||||
return {
|
||||
installed: false,
|
||||
running: false,
|
||||
error: `Docker check failed: ${errorMsg}`,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the actual port mapping for the FalkorDB container from Docker
|
||||
*/
|
||||
async function getContainerPortMapping(): Promise<number | null> {
|
||||
try {
|
||||
// Get the port mapping from Docker - format: "0.0.0.0:6380->6379/tcp"
|
||||
const { stdout } = await execAsync(
|
||||
`docker port ${FALKORDB_CONTAINER_NAME} 6379`,
|
||||
{ timeout: 5000 }
|
||||
);
|
||||
|
||||
const portMatch = stdout.trim().match(/:(\d+)/);
|
||||
if (portMatch) {
|
||||
return parseInt(portMatch[1], 10);
|
||||
}
|
||||
return null;
|
||||
} catch {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Check FalkorDB container status
|
||||
*/
|
||||
export async function checkFalkorDBStatus(port: number = FALKORDB_DEFAULT_PORT): Promise<FalkorDBStatus> {
|
||||
const status: FalkorDBStatus = {
|
||||
containerExists: false,
|
||||
containerRunning: false,
|
||||
containerName: FALKORDB_CONTAINER_NAME,
|
||||
port,
|
||||
healthy: false,
|
||||
};
|
||||
|
||||
try {
|
||||
// Check if container exists and get its status
|
||||
const { stdout } = await execAsync(
|
||||
`docker ps -a --filter "name=${FALKORDB_CONTAINER_NAME}" --format "{{.Status}}"`,
|
||||
{ timeout: 5000 }
|
||||
);
|
||||
|
||||
const containerStatus = stdout.trim();
|
||||
|
||||
if (containerStatus) {
|
||||
status.containerExists = true;
|
||||
status.containerRunning = containerStatus.toLowerCase().startsWith('up');
|
||||
|
||||
if (status.containerRunning) {
|
||||
// Get the actual port mapping from Docker
|
||||
const actualPort = await getContainerPortMapping();
|
||||
if (actualPort) {
|
||||
status.port = actualPort;
|
||||
}
|
||||
|
||||
// Check if FalkorDB is responding
|
||||
status.healthy = await checkFalkorDBHealth(status.port);
|
||||
}
|
||||
}
|
||||
|
||||
return status;
|
||||
} catch (error) {
|
||||
status.error = error instanceof Error ? error.message : String(error);
|
||||
return status;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Check if FalkorDB is responding to connections
|
||||
*/
|
||||
async function checkFalkorDBHealth(_port: number): Promise<boolean> {
|
||||
try {
|
||||
// Try to ping FalkorDB using redis-cli (FalkorDB uses Redis protocol)
|
||||
// Since we may not have redis-cli, we'll check if the port is listening
|
||||
await execAsync(`docker exec ${FALKORDB_CONTAINER_NAME} redis-cli PING`, {
|
||||
timeout: 5000,
|
||||
});
|
||||
return true;
|
||||
} catch {
|
||||
// Fallback: just check if container is running (less accurate)
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get combined infrastructure status
|
||||
*/
|
||||
export async function getInfrastructureStatus(
|
||||
falkordbPort: number = FALKORDB_DEFAULT_PORT
|
||||
): Promise<InfrastructureStatus> {
|
||||
const [docker, falkordb] = await Promise.all([
|
||||
checkDockerStatus(),
|
||||
checkFalkorDBStatus(falkordbPort),
|
||||
]);
|
||||
|
||||
return {
|
||||
docker,
|
||||
falkordb,
|
||||
ready: docker.running && falkordb.containerRunning && falkordb.healthy,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Start FalkorDB container
|
||||
* Creates a new container if it doesn't exist, or starts the existing one
|
||||
*/
|
||||
export async function startFalkorDB(
|
||||
port: number = FALKORDB_DEFAULT_PORT
|
||||
): Promise<{ success: boolean; error?: string }> {
|
||||
try {
|
||||
// First, check Docker status
|
||||
const dockerStatus = await checkDockerStatus();
|
||||
if (!dockerStatus.running) {
|
||||
return {
|
||||
success: false,
|
||||
error: dockerStatus.error || 'Docker is not running',
|
||||
};
|
||||
}
|
||||
|
||||
// Check if container already exists
|
||||
const falkordbStatus = await checkFalkorDBStatus(port);
|
||||
|
||||
if (falkordbStatus.containerExists) {
|
||||
if (falkordbStatus.containerRunning) {
|
||||
// Already running
|
||||
return { success: true };
|
||||
}
|
||||
|
||||
// Start existing container
|
||||
await execAsync(`docker start ${FALKORDB_CONTAINER_NAME}`, { timeout: 30000 });
|
||||
} else {
|
||||
// Create and start new container
|
||||
await execAsync(
|
||||
`docker run -d --name ${FALKORDB_CONTAINER_NAME} -p ${port}:6379 ${FALKORDB_IMAGE}`,
|
||||
{ timeout: 60000 }
|
||||
);
|
||||
}
|
||||
|
||||
// Wait for FalkorDB to be ready (up to 30 seconds)
|
||||
const ready = await waitForFalkorDB(port, 30000);
|
||||
|
||||
if (!ready) {
|
||||
return {
|
||||
success: false,
|
||||
error: 'FalkorDB container started but is not responding. Please check Docker logs.',
|
||||
};
|
||||
}
|
||||
|
||||
return { success: true };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : String(error),
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Stop FalkorDB container
|
||||
*/
|
||||
export async function stopFalkorDB(): Promise<{ success: boolean; error?: string }> {
|
||||
try {
|
||||
await execAsync(`docker stop ${FALKORDB_CONTAINER_NAME}`, { timeout: 30000 });
|
||||
return { success: true };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : String(error),
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Wait for FalkorDB to be ready
|
||||
*/
|
||||
async function waitForFalkorDB(port: number, timeoutMs: number): Promise<boolean> {
|
||||
const startTime = Date.now();
|
||||
const checkInterval = 1000; // Check every second
|
||||
|
||||
while (Date.now() - startTime < timeoutMs) {
|
||||
const status = await checkFalkorDBStatus(port);
|
||||
if (status.containerRunning && status.healthy) {
|
||||
return true;
|
||||
}
|
||||
// If container is running but not healthy yet, wait
|
||||
if (status.containerRunning) {
|
||||
await new Promise((resolve) => setTimeout(resolve, checkInterval));
|
||||
} else {
|
||||
// Container stopped unexpectedly
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* Open Docker Desktop application (macOS/Windows)
|
||||
*/
|
||||
export async function openDockerDesktop(): Promise<{ success: boolean; error?: string }> {
|
||||
try {
|
||||
if (process.platform === 'darwin') {
|
||||
// macOS
|
||||
await execAsync('open -a Docker', { timeout: 5000 });
|
||||
} else if (process.platform === 'win32') {
|
||||
// Windows
|
||||
spawn('cmd', ['/c', 'start', '', 'Docker Desktop'], {
|
||||
detached: true,
|
||||
stdio: 'ignore',
|
||||
});
|
||||
} else {
|
||||
// Linux - Docker doesn't have a GUI, suggest starting daemon
|
||||
return {
|
||||
success: false,
|
||||
error: 'On Linux, start Docker with: sudo systemctl start docker',
|
||||
};
|
||||
}
|
||||
|
||||
return { success: true };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : String(error),
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get download URL for Docker Desktop
|
||||
*/
|
||||
export function getDockerDownloadUrl(): string {
|
||||
if (process.platform === 'darwin') {
|
||||
return 'https://www.docker.com/products/docker-desktop/';
|
||||
} else if (process.platform === 'win32') {
|
||||
return 'https://www.docker.com/products/docker-desktop/';
|
||||
}
|
||||
return 'https://docs.docker.com/engine/install/';
|
||||
}
|
||||
|
||||
// ============================================
|
||||
// Graphiti Validation Functions
|
||||
// ============================================
|
||||
|
||||
export interface GraphitiValidationResult {
|
||||
success: boolean;
|
||||
message: string;
|
||||
details?: {
|
||||
provider?: string;
|
||||
model?: string;
|
||||
latencyMs?: number;
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Validate FalkorDB connection by attempting to connect and ping
|
||||
* @param uri - FalkorDB URI (e.g., "bolt://localhost:6380" or "redis://localhost:6380")
|
||||
*/
|
||||
export async function validateFalkorDBConnection(
|
||||
uri: string
|
||||
): Promise<GraphitiValidationResult> {
|
||||
try {
|
||||
// Parse the URI to extract host and port
|
||||
let host = 'localhost';
|
||||
let port = FALKORDB_DEFAULT_PORT;
|
||||
|
||||
// Support both bolt:// and redis:// protocols
|
||||
const uriMatch = uri.match(/^(?:bolt|redis):\/\/([^:]+):(\d+)/);
|
||||
if (uriMatch) {
|
||||
host = uriMatch[1];
|
||||
port = parseInt(uriMatch[2], 10);
|
||||
} else {
|
||||
// Try simple host:port format
|
||||
const simpleMatch = uri.match(/^([^:]+):(\d+)/);
|
||||
if (simpleMatch) {
|
||||
host = simpleMatch[1];
|
||||
port = parseInt(simpleMatch[2], 10);
|
||||
}
|
||||
}
|
||||
|
||||
const startTime = Date.now();
|
||||
|
||||
// First, check the actual FalkorDB container status to get the correct port
|
||||
const falkorStatus = await checkFalkorDBStatus(port);
|
||||
|
||||
// If container exists but user specified wrong port, try to detect the actual port
|
||||
if (!falkorStatus.containerRunning) {
|
||||
// Check if container is running on default port
|
||||
const defaultStatus = await checkFalkorDBStatus(FALKORDB_DEFAULT_PORT);
|
||||
if (defaultStatus.containerRunning && defaultStatus.healthy) {
|
||||
return {
|
||||
success: false,
|
||||
message: `FalkorDB is running on port ${FALKORDB_DEFAULT_PORT}, but you specified port ${port}. Please update the URI to bolt://localhost:${FALKORDB_DEFAULT_PORT}`,
|
||||
};
|
||||
}
|
||||
|
||||
return {
|
||||
success: false,
|
||||
message: `FalkorDB container is not running. Please start FalkorDB first using Docker.`,
|
||||
};
|
||||
}
|
||||
|
||||
// Try to ping FalkorDB using redis-cli in Docker container
|
||||
try {
|
||||
const { stdout } = await execAsync(
|
||||
`docker exec ${FALKORDB_CONTAINER_NAME} redis-cli PING`,
|
||||
{ timeout: 10000 }
|
||||
);
|
||||
|
||||
if (stdout.trim().toUpperCase() === 'PONG') {
|
||||
const latencyMs = Date.now() - startTime;
|
||||
return {
|
||||
success: true,
|
||||
message: `Connected to FalkorDB at ${host}:${port}`,
|
||||
details: { latencyMs },
|
||||
};
|
||||
}
|
||||
} catch {
|
||||
// redis-cli failed, try port check as fallback
|
||||
}
|
||||
|
||||
// Fallback: check if the port is open using nc or direct connection
|
||||
try {
|
||||
// Check if we can connect to the mapped port from the host
|
||||
await execAsync(`nc -z -w 5 ${host} ${port}`, { timeout: 10000 });
|
||||
const latencyMs = Date.now() - startTime;
|
||||
return {
|
||||
success: true,
|
||||
message: `FalkorDB port ${port} is reachable at ${host}`,
|
||||
details: { latencyMs },
|
||||
};
|
||||
} catch {
|
||||
// Port check failed, but container is running - might be a different port mapping
|
||||
if (falkorStatus.containerRunning) {
|
||||
return {
|
||||
success: false,
|
||||
message: `FalkorDB container is running but port ${port} is not reachable. The container may be mapped to a different port.`,
|
||||
};
|
||||
}
|
||||
|
||||
return {
|
||||
success: false,
|
||||
message: `Cannot connect to FalkorDB at ${host}:${port}. Make sure FalkorDB is running.`,
|
||||
};
|
||||
}
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
message: error instanceof Error ? error.message : 'Unknown error occurred',
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Validate OpenAI API key by attempting to list models
|
||||
* @param apiKey - OpenAI API key
|
||||
*/
|
||||
export async function validateOpenAIApiKey(
|
||||
apiKey: string
|
||||
): Promise<GraphitiValidationResult> {
|
||||
if (!apiKey || !apiKey.trim()) {
|
||||
return {
|
||||
success: false,
|
||||
message: 'API key is required',
|
||||
};
|
||||
}
|
||||
|
||||
// Basic format validation
|
||||
const trimmedKey = apiKey.trim();
|
||||
if (!trimmedKey.startsWith('sk-') && !trimmedKey.startsWith('sess-')) {
|
||||
return {
|
||||
success: false,
|
||||
message: 'Invalid API key format. OpenAI API keys should start with "sk-"',
|
||||
};
|
||||
}
|
||||
|
||||
try {
|
||||
const startTime = Date.now();
|
||||
|
||||
// Use native https module to avoid additional dependencies
|
||||
const result = await new Promise<GraphitiValidationResult>((resolve) => {
|
||||
const https = require('https');
|
||||
|
||||
const options = {
|
||||
hostname: 'api.openai.com',
|
||||
port: 443,
|
||||
path: '/v1/models',
|
||||
method: 'GET',
|
||||
headers: {
|
||||
Authorization: `Bearer ${trimmedKey}`,
|
||||
'Content-Type': 'application/json',
|
||||
},
|
||||
timeout: 15000,
|
||||
};
|
||||
|
||||
const req = https.request(options, (res: { statusCode: number; on: (event: string, callback: (chunk: Buffer) => void) => void }) => {
|
||||
let data = '';
|
||||
|
||||
res.on('data', (chunk: Buffer) => {
|
||||
data += chunk;
|
||||
});
|
||||
|
||||
res.on('end', () => {
|
||||
const latencyMs = Date.now() - startTime;
|
||||
|
||||
if (res.statusCode === 200) {
|
||||
resolve({
|
||||
success: true,
|
||||
message: 'OpenAI API key is valid',
|
||||
details: {
|
||||
provider: 'openai',
|
||||
latencyMs,
|
||||
},
|
||||
});
|
||||
} else if (res.statusCode === 401) {
|
||||
resolve({
|
||||
success: false,
|
||||
message: 'Invalid API key. Please check your OpenAI API key.',
|
||||
});
|
||||
} else if (res.statusCode === 429) {
|
||||
// Rate limited but key is valid
|
||||
resolve({
|
||||
success: true,
|
||||
message: 'OpenAI API key is valid (rate limited, please wait)',
|
||||
details: {
|
||||
provider: 'openai',
|
||||
latencyMs,
|
||||
},
|
||||
});
|
||||
} else {
|
||||
try {
|
||||
const errorData = JSON.parse(data);
|
||||
resolve({
|
||||
success: false,
|
||||
message: errorData.error?.message || `API error: ${res.statusCode}`,
|
||||
});
|
||||
} catch {
|
||||
resolve({
|
||||
success: false,
|
||||
message: `API error: ${res.statusCode}`,
|
||||
});
|
||||
}
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
req.on('error', (error: Error) => {
|
||||
resolve({
|
||||
success: false,
|
||||
message: `Connection error: ${error.message}`,
|
||||
});
|
||||
});
|
||||
|
||||
req.on('timeout', () => {
|
||||
req.destroy();
|
||||
resolve({
|
||||
success: false,
|
||||
message: 'Connection timeout. Please check your network connection.',
|
||||
});
|
||||
});
|
||||
|
||||
req.end();
|
||||
});
|
||||
|
||||
return result;
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
message: error instanceof Error ? error.message : 'Unknown error occurred',
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Test the full Graphiti connection (FalkorDB + OpenAI)
|
||||
* @param falkorDbUri - FalkorDB URI
|
||||
* @param openAiApiKey - OpenAI API key
|
||||
*/
|
||||
export async function testGraphitiConnection(
|
||||
falkorDbUri: string,
|
||||
openAiApiKey: string
|
||||
): Promise<{
|
||||
falkordb: GraphitiValidationResult;
|
||||
openai: GraphitiValidationResult;
|
||||
ready: boolean;
|
||||
}> {
|
||||
const [falkordb, openai] = await Promise.all([
|
||||
validateFalkorDBConnection(falkorDbUri),
|
||||
validateOpenAIApiKey(openAiApiKey),
|
||||
]);
|
||||
|
||||
return {
|
||||
falkordb,
|
||||
openai,
|
||||
ready: falkordb.success && openai.success,
|
||||
};
|
||||
}
|
||||
@@ -1,365 +0,0 @@
|
||||
/**
|
||||
* FalkorDB Service
|
||||
*
|
||||
* Queries the FalkorDB graph database for memories stored by Graphiti.
|
||||
* Uses ioredis to communicate with FalkorDB via Redis protocol.
|
||||
*/
|
||||
|
||||
import Redis from 'ioredis';
|
||||
import type { MemoryEpisode } from '../shared/types';
|
||||
|
||||
interface FalkorDBConfig {
|
||||
host: string;
|
||||
port: number;
|
||||
password?: string;
|
||||
}
|
||||
|
||||
interface EpisodicNode {
|
||||
uuid: string;
|
||||
name: string;
|
||||
created_at: string;
|
||||
content?: string;
|
||||
source_description?: string;
|
||||
}
|
||||
|
||||
interface EntityNode {
|
||||
uuid: string;
|
||||
name: string;
|
||||
summary?: string;
|
||||
}
|
||||
|
||||
/**
|
||||
* Parse FalkorDB GRAPH.QUERY results into structured data
|
||||
*/
|
||||
function parseGraphResult(result: unknown[]): Record<string, unknown>[] {
|
||||
if (!Array.isArray(result) || result.length < 2) {
|
||||
return [];
|
||||
}
|
||||
|
||||
// Result format: [headers, [row1, row2, ...], stats]
|
||||
const headers = result[0] as string[];
|
||||
const rows = result[1] as unknown[][];
|
||||
|
||||
if (!Array.isArray(headers) || !Array.isArray(rows)) {
|
||||
return [];
|
||||
}
|
||||
|
||||
return rows.map(row => {
|
||||
const obj: Record<string, unknown> = {};
|
||||
headers.forEach((header, idx) => {
|
||||
obj[header] = row[idx];
|
||||
});
|
||||
return obj;
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* FalkorDB Service for querying graph memories
|
||||
*/
|
||||
export class FalkorDBService {
|
||||
private config: FalkorDBConfig;
|
||||
private redis: Redis | null = null;
|
||||
|
||||
constructor(config: FalkorDBConfig) {
|
||||
this.config = config;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a Redis connection (lazy initialization)
|
||||
*/
|
||||
private async getConnection(): Promise<Redis> {
|
||||
if (this.redis) {
|
||||
return this.redis;
|
||||
}
|
||||
|
||||
this.redis = new Redis({
|
||||
host: this.config.host,
|
||||
port: this.config.port,
|
||||
password: this.config.password,
|
||||
lazyConnect: true,
|
||||
connectTimeout: 5000,
|
||||
maxRetriesPerRequest: 1,
|
||||
});
|
||||
|
||||
await this.redis.connect();
|
||||
return this.redis;
|
||||
}
|
||||
|
||||
/**
|
||||
* Close the Redis connection
|
||||
*/
|
||||
async close(): Promise<void> {
|
||||
if (this.redis) {
|
||||
await this.redis.quit();
|
||||
this.redis = null;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* List all available graphs in the database
|
||||
*/
|
||||
async listGraphs(): Promise<string[]> {
|
||||
try {
|
||||
const redis = await this.getConnection();
|
||||
const result = await redis.call('GRAPH.LIST') as string[];
|
||||
return result || [];
|
||||
} catch (error) {
|
||||
console.error('Failed to list graphs:', error);
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Query episodic memories from a specific graph
|
||||
*/
|
||||
async getEpisodicMemories(graphName: string, limit: number = 20): Promise<MemoryEpisode[]> {
|
||||
try {
|
||||
const redis = await this.getConnection();
|
||||
|
||||
// Query episodic nodes with their details
|
||||
const query = `
|
||||
MATCH (e:Episodic)
|
||||
RETURN e.uuid as uuid, e.name as name, e.created_at as created_at,
|
||||
e.content as content, e.source_description as description
|
||||
ORDER BY e.created_at DESC
|
||||
LIMIT ${limit}
|
||||
`;
|
||||
|
||||
const result = await redis.call('GRAPH.QUERY', graphName, query) as unknown[];
|
||||
const episodes = parseGraphResult(result) as unknown as EpisodicNode[];
|
||||
|
||||
return episodes.map(ep => ({
|
||||
id: ep.uuid || ep.name,
|
||||
type: this.inferEpisodeType(ep.name, ep.content),
|
||||
timestamp: ep.created_at || new Date().toISOString(),
|
||||
content: ep.content || ep.source_description || ep.name,
|
||||
session_number: this.extractSessionNumber(ep.name),
|
||||
}));
|
||||
} catch (error) {
|
||||
console.error(`Failed to get episodic memories from ${graphName}:`, error);
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Query entity memories (patterns, gotchas, etc.) from a graph
|
||||
*/
|
||||
async getEntityMemories(graphName: string, limit: number = 20): Promise<MemoryEpisode[]> {
|
||||
try {
|
||||
const redis = await this.getConnection();
|
||||
|
||||
// Query entity nodes
|
||||
const query = `
|
||||
MATCH (e:Entity)
|
||||
RETURN e.uuid as uuid, e.name as name, e.summary as summary, e.created_at as created_at
|
||||
ORDER BY e.created_at DESC
|
||||
LIMIT ${limit}
|
||||
`;
|
||||
|
||||
const result = await redis.call('GRAPH.QUERY', graphName, query) as unknown[];
|
||||
const entities = parseGraphResult(result) as unknown as EntityNode[];
|
||||
|
||||
return entities
|
||||
.filter(ent => ent.summary) // Only include entities with summaries
|
||||
.map(ent => ({
|
||||
id: ent.uuid || ent.name,
|
||||
type: this.inferEntityType(ent.name),
|
||||
timestamp: new Date().toISOString(),
|
||||
content: ent.summary || ent.name,
|
||||
}));
|
||||
} catch (error) {
|
||||
console.error(`Failed to get entity memories from ${graphName}:`, error);
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get all memories from all spec-related graphs
|
||||
*/
|
||||
async getAllMemories(limit: number = 20): Promise<MemoryEpisode[]> {
|
||||
const graphs = await this.listGraphs();
|
||||
const memories: MemoryEpisode[] = [];
|
||||
|
||||
// Filter to spec-related graphs (exclude auto_build_memory and project_ prefixed)
|
||||
const specGraphs = graphs.filter(g =>
|
||||
!g.startsWith('project_') &&
|
||||
g !== 'auto_build_memory' &&
|
||||
g !== 'default_db'
|
||||
);
|
||||
|
||||
for (const graph of specGraphs) {
|
||||
const episodic = await this.getEpisodicMemories(graph, Math.ceil(limit / specGraphs.length));
|
||||
memories.push(...episodic.map(m => ({ ...m, id: `${graph}:${m.id}` })));
|
||||
}
|
||||
|
||||
// Sort by timestamp descending
|
||||
memories.sort((a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime());
|
||||
|
||||
return memories.slice(0, limit);
|
||||
}
|
||||
|
||||
/**
|
||||
* Search memories across all graphs
|
||||
*/
|
||||
async searchMemories(query: string, limit: number = 20): Promise<MemoryEpisode[]> {
|
||||
const graphs = await this.listGraphs();
|
||||
const results: MemoryEpisode[] = [];
|
||||
const queryLower = query.toLowerCase();
|
||||
|
||||
// Filter to spec-related graphs
|
||||
const specGraphs = graphs.filter(g =>
|
||||
!g.startsWith('project_') &&
|
||||
g !== 'auto_build_memory' &&
|
||||
g !== 'default_db'
|
||||
);
|
||||
|
||||
for (const graph of specGraphs) {
|
||||
try {
|
||||
const redis = await this.getConnection();
|
||||
|
||||
// Search in episodic nodes
|
||||
const episodicQuery = `
|
||||
MATCH (e:Episodic)
|
||||
WHERE toLower(e.name) CONTAINS '${queryLower}' OR toLower(e.content) CONTAINS '${queryLower}'
|
||||
RETURN e.uuid as uuid, e.name as name, e.created_at as created_at,
|
||||
e.content as content, e.source_description as description
|
||||
LIMIT ${Math.ceil(limit / specGraphs.length)}
|
||||
`;
|
||||
|
||||
const episodicResult = await redis.call('GRAPH.QUERY', graph, episodicQuery) as unknown[];
|
||||
const episodes = parseGraphResult(episodicResult) as unknown as EpisodicNode[];
|
||||
|
||||
results.push(...episodes.map(ep => ({
|
||||
id: `${graph}:${ep.uuid || ep.name}`,
|
||||
type: this.inferEpisodeType(ep.name, ep.content),
|
||||
timestamp: ep.created_at || new Date().toISOString(),
|
||||
content: ep.content || ep.source_description || ep.name,
|
||||
session_number: this.extractSessionNumber(ep.name),
|
||||
score: 1.0,
|
||||
})));
|
||||
|
||||
// Search in entity nodes
|
||||
const entityQuery = `
|
||||
MATCH (e:Entity)
|
||||
WHERE toLower(e.name) CONTAINS '${queryLower}' OR toLower(e.summary) CONTAINS '${queryLower}'
|
||||
RETURN e.uuid as uuid, e.name as name, e.summary as summary
|
||||
LIMIT ${Math.ceil(limit / specGraphs.length)}
|
||||
`;
|
||||
|
||||
const entityResult = await redis.call('GRAPH.QUERY', graph, entityQuery) as unknown[];
|
||||
const entities = parseGraphResult(entityResult) as unknown as EntityNode[];
|
||||
|
||||
results.push(...entities
|
||||
.filter(ent => ent.summary)
|
||||
.map(ent => ({
|
||||
id: `${graph}:${ent.uuid || ent.name}`,
|
||||
type: this.inferEntityType(ent.name),
|
||||
timestamp: new Date().toISOString(),
|
||||
content: ent.summary || ent.name,
|
||||
score: 1.0,
|
||||
})));
|
||||
} catch (error) {
|
||||
console.error(`Failed to search memories in ${graph}:`, error);
|
||||
}
|
||||
}
|
||||
|
||||
return results.slice(0, limit);
|
||||
}
|
||||
|
||||
/**
|
||||
* Test connection to FalkorDB
|
||||
*/
|
||||
async testConnection(): Promise<{ success: boolean; message: string }> {
|
||||
try {
|
||||
const redis = await this.getConnection();
|
||||
await redis.ping();
|
||||
const graphs = await this.listGraphs();
|
||||
return {
|
||||
success: true,
|
||||
message: `Connected to FalkorDB with ${graphs.length} graphs`,
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
message: error instanceof Error ? error.message : 'Connection failed',
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Infer the episode type from its name
|
||||
*/
|
||||
private inferEpisodeType(name: string, content?: string): MemoryEpisode['type'] {
|
||||
const nameLower = (name || '').toLowerCase();
|
||||
const contentLower = (content || '').toLowerCase();
|
||||
|
||||
if (nameLower.includes('session_') || contentLower.includes('"type": "session_insight"')) {
|
||||
return 'session_insight';
|
||||
}
|
||||
if (nameLower.includes('pattern') || contentLower.includes('"type": "pattern"')) {
|
||||
return 'pattern';
|
||||
}
|
||||
if (nameLower.includes('gotcha') || contentLower.includes('"type": "gotcha"')) {
|
||||
return 'gotcha';
|
||||
}
|
||||
if (nameLower.includes('codebase') || contentLower.includes('"type": "codebase_discovery"')) {
|
||||
return 'codebase_discovery';
|
||||
}
|
||||
return 'session_insight';
|
||||
}
|
||||
|
||||
/**
|
||||
* Infer the entity type from its name
|
||||
*/
|
||||
private inferEntityType(name: string): MemoryEpisode['type'] {
|
||||
const nameLower = (name || '').toLowerCase();
|
||||
|
||||
if (nameLower.includes('pattern')) {
|
||||
return 'pattern';
|
||||
}
|
||||
if (nameLower.includes('gotcha')) {
|
||||
return 'gotcha';
|
||||
}
|
||||
if (nameLower.includes('file_insight') || nameLower.includes('codebase')) {
|
||||
return 'codebase_discovery';
|
||||
}
|
||||
return 'session_insight';
|
||||
}
|
||||
|
||||
/**
|
||||
* Extract session number from episode name
|
||||
*/
|
||||
private extractSessionNumber(name: string): number | undefined {
|
||||
const match = name.match(/session_(\d+)/i);
|
||||
return match ? parseInt(match[1], 10) : undefined;
|
||||
}
|
||||
}
|
||||
|
||||
// Singleton instance for reuse
|
||||
let serviceInstance: FalkorDBService | null = null;
|
||||
|
||||
/**
|
||||
* Get or create a FalkorDB service instance
|
||||
*/
|
||||
export function getFalkorDBService(config: FalkorDBConfig): FalkorDBService {
|
||||
if (!serviceInstance ||
|
||||
serviceInstance['config'].host !== config.host ||
|
||||
serviceInstance['config'].port !== config.port) {
|
||||
// Close existing connection if config changed
|
||||
if (serviceInstance) {
|
||||
serviceInstance.close().catch(() => {});
|
||||
}
|
||||
serviceInstance = new FalkorDBService(config);
|
||||
}
|
||||
return serviceInstance;
|
||||
}
|
||||
|
||||
/**
|
||||
* Close the singleton service instance
|
||||
*/
|
||||
export async function closeFalkorDBService(): Promise<void> {
|
||||
if (serviceInstance) {
|
||||
await serviceInstance.close();
|
||||
serviceInstance = null;
|
||||
}
|
||||
}
|
||||
@@ -1,7 +1,7 @@
|
||||
import { ipcMain } from 'electron';
|
||||
import type { BrowserWindow } from 'electron';
|
||||
import path from 'path';
|
||||
import { existsSync, mkdirSync, writeFileSync } from 'fs';
|
||||
import { existsSync, mkdirSync, writeFileSync, readFileSync } from 'fs';
|
||||
import { IPC_CHANNELS, getSpecsDir } from '../../shared/constants';
|
||||
import type {
|
||||
IPCResult,
|
||||
@@ -375,6 +375,46 @@ export function registerChangelogHandlers(
|
||||
}
|
||||
);
|
||||
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.CHANGELOG_READ_LOCAL_IMAGE,
|
||||
async (_, projectPath: string, relativePath: string): Promise<IPCResult<string>> => {
|
||||
try {
|
||||
// Construct full path from project path and relative path
|
||||
const fullPath = path.join(projectPath, relativePath);
|
||||
|
||||
// Verify the file exists
|
||||
if (!existsSync(fullPath)) {
|
||||
return { success: false, error: `Image not found: ${relativePath}` };
|
||||
}
|
||||
|
||||
// Read the file and convert to base64
|
||||
const buffer = readFileSync(fullPath);
|
||||
const base64 = buffer.toString('base64');
|
||||
|
||||
// Determine MIME type from extension
|
||||
const ext = path.extname(relativePath).toLowerCase();
|
||||
const mimeTypes: Record<string, string> = {
|
||||
'.png': 'image/png',
|
||||
'.jpg': 'image/jpeg',
|
||||
'.jpeg': 'image/jpeg',
|
||||
'.gif': 'image/gif',
|
||||
'.webp': 'image/webp',
|
||||
'.svg': 'image/svg+xml'
|
||||
};
|
||||
const mimeType = mimeTypes[ext] || 'image/png';
|
||||
|
||||
// Return as data URL
|
||||
const dataUrl = `data:${mimeType};base64,${base64}`;
|
||||
return { success: true, data: dataUrl };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to read image'
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// ============================================
|
||||
// Changelog Agent Events → Renderer
|
||||
}
|
||||
|
||||
@@ -9,11 +9,11 @@ import type {
|
||||
ContextSearchResult
|
||||
} from '../../../shared/types';
|
||||
import { projectStore } from '../../project-store';
|
||||
import { getFalkorDBService } from '../../falkordb-service';
|
||||
import { getMemoryService, isKuzuAvailable } from '../../memory-service';
|
||||
import {
|
||||
loadProjectEnvVars,
|
||||
isGraphitiEnabled,
|
||||
getGraphitiConnectionDetails
|
||||
getGraphitiDatabaseDetails
|
||||
} from './utils';
|
||||
|
||||
/**
|
||||
@@ -169,20 +169,20 @@ export function registerMemoryDataHandlers(
|
||||
const projectEnvVars = loadProjectEnvVars(project.path, project.autoBuildPath);
|
||||
const graphitiEnabled = isGraphitiEnabled(projectEnvVars);
|
||||
|
||||
// Try FalkorDB first if available
|
||||
if (graphitiEnabled) {
|
||||
// Try LadybugDB first if available
|
||||
if (graphitiEnabled && isKuzuAvailable()) {
|
||||
try {
|
||||
const connDetails = getGraphitiConnectionDetails(projectEnvVars);
|
||||
const falkorService = getFalkorDBService({
|
||||
host: connDetails.host,
|
||||
port: connDetails.port,
|
||||
const dbDetails = getGraphitiDatabaseDetails(projectEnvVars);
|
||||
const memoryService = getMemoryService({
|
||||
dbPath: dbDetails.dbPath,
|
||||
database: dbDetails.database,
|
||||
});
|
||||
const falkorMemories = await falkorService.getAllMemories(limit);
|
||||
if (falkorMemories.length > 0) {
|
||||
return { success: true, data: falkorMemories };
|
||||
const graphMemories = await memoryService.getEpisodicMemories(limit);
|
||||
if (graphMemories.length > 0) {
|
||||
return { success: true, data: graphMemories };
|
||||
}
|
||||
} catch (error) {
|
||||
console.warn('Failed to get memories from FalkorDB, falling back to file-based:', error);
|
||||
console.warn('Failed to get memories from LadybugDB, falling back to file-based:', error);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -207,19 +207,19 @@ export function registerMemoryDataHandlers(
|
||||
const projectEnvVars = loadProjectEnvVars(project.path, project.autoBuildPath);
|
||||
const graphitiEnabled = isGraphitiEnabled(projectEnvVars);
|
||||
|
||||
// Try FalkorDB search if available
|
||||
if (graphitiEnabled) {
|
||||
// Try LadybugDB search if available
|
||||
if (graphitiEnabled && isKuzuAvailable()) {
|
||||
try {
|
||||
const connDetails = getGraphitiConnectionDetails(projectEnvVars);
|
||||
const falkorService = getFalkorDBService({
|
||||
host: connDetails.host,
|
||||
port: connDetails.port,
|
||||
const dbDetails = getGraphitiDatabaseDetails(projectEnvVars);
|
||||
const memoryService = getMemoryService({
|
||||
dbPath: dbDetails.dbPath,
|
||||
database: dbDetails.database,
|
||||
});
|
||||
const falkorResults = await falkorService.searchMemories(query, 20);
|
||||
if (falkorResults.length > 0) {
|
||||
const graphResults = await memoryService.searchMemories(query, 20);
|
||||
if (graphResults.length > 0) {
|
||||
return {
|
||||
success: true,
|
||||
data: falkorResults.map(r => ({
|
||||
data: graphResults.map(r => ({
|
||||
content: r.content,
|
||||
score: r.score || 1.0,
|
||||
type: r.type
|
||||
@@ -227,7 +227,7 @@ export function registerMemoryDataHandlers(
|
||||
};
|
||||
}
|
||||
} catch (error) {
|
||||
console.warn('Failed to search FalkorDB, falling back to file-based:', error);
|
||||
console.warn('Failed to search LadybugDB, falling back to file-based:', error);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -10,7 +10,7 @@ import {
|
||||
loadGlobalSettings,
|
||||
isGraphitiEnabled,
|
||||
hasOpenAIKey,
|
||||
getGraphitiConnectionDetails
|
||||
getGraphitiDatabaseDetails
|
||||
} from './utils';
|
||||
|
||||
/**
|
||||
@@ -65,13 +65,12 @@ export function buildMemoryStatus(
|
||||
|
||||
// If we have initialized state from specs, use it
|
||||
if (memoryState?.initialized) {
|
||||
const connDetails = getGraphitiConnectionDetails(projectEnvVars);
|
||||
const dbDetails = getGraphitiDatabaseDetails(projectEnvVars);
|
||||
return {
|
||||
enabled: true,
|
||||
available: true,
|
||||
database: memoryState.database || 'auto_claude_memory',
|
||||
host: connDetails.host,
|
||||
port: connDetails.port
|
||||
dbPath: dbDetails.dbPath
|
||||
};
|
||||
}
|
||||
|
||||
@@ -95,13 +94,12 @@ export function buildMemoryStatus(
|
||||
};
|
||||
}
|
||||
|
||||
const connDetails = getGraphitiConnectionDetails(projectEnvVars);
|
||||
const dbDetails = getGraphitiDatabaseDetails(projectEnvVars);
|
||||
return {
|
||||
enabled: true,
|
||||
available: true,
|
||||
host: connDetails.host,
|
||||
port: connDetails.port,
|
||||
database: connDetails.database
|
||||
dbPath: dbDetails.dbPath,
|
||||
database: dbDetails.database
|
||||
};
|
||||
}
|
||||
|
||||
|
||||
@@ -11,10 +11,11 @@ import type {
|
||||
MemoryEpisode
|
||||
} from '../../../shared/types';
|
||||
import { projectStore } from '../../project-store';
|
||||
import { getFalkorDBService } from '../../falkordb-service';
|
||||
import { getMemoryService, isKuzuAvailable } from '../../memory-service';
|
||||
import {
|
||||
getAutoBuildSourcePath
|
||||
getGraphitiDatabaseDetails
|
||||
} from './utils';
|
||||
import { getEffectiveSourcePath } from '../../updater/path-resolver';
|
||||
import {
|
||||
loadGraphitiStateFromSpecs,
|
||||
buildMemoryStatus
|
||||
@@ -39,34 +40,34 @@ function loadProjectIndex(projectPath: string): ProjectIndex | null {
|
||||
}
|
||||
|
||||
/**
|
||||
* Load recent memories with FalkorDB fallback
|
||||
* Load recent memories from LadybugDB with file-based fallback
|
||||
*/
|
||||
async function loadRecentMemories(
|
||||
projectPath: string,
|
||||
autoBuildPath: string | undefined,
|
||||
memoryStatusAvailable: boolean,
|
||||
memoryHost?: string,
|
||||
memoryPort?: number
|
||||
dbPath?: string,
|
||||
database?: string
|
||||
): Promise<MemoryEpisode[]> {
|
||||
let recentMemories: MemoryEpisode[] = [];
|
||||
|
||||
// Try to load from FalkorDB first if Graphiti is available
|
||||
if (memoryStatusAvailable && memoryHost && memoryPort) {
|
||||
// Try to load from LadybugDB first if Graphiti is available and Kuzu is installed
|
||||
if (memoryStatusAvailable && isKuzuAvailable() && dbPath && database) {
|
||||
try {
|
||||
const falkorService = getFalkorDBService({
|
||||
host: memoryHost,
|
||||
port: memoryPort,
|
||||
const memoryService = getMemoryService({
|
||||
dbPath,
|
||||
database,
|
||||
});
|
||||
const falkorMemories = await falkorService.getAllMemories(20);
|
||||
if (falkorMemories.length > 0) {
|
||||
recentMemories = falkorMemories;
|
||||
const graphMemories = await memoryService.getEpisodicMemories(20);
|
||||
if (graphMemories.length > 0) {
|
||||
recentMemories = graphMemories;
|
||||
}
|
||||
} catch (error) {
|
||||
console.warn('Failed to load memories from FalkorDB, falling back to file-based:', error);
|
||||
console.warn('Failed to load memories from LadybugDB, falling back to file-based:', error);
|
||||
}
|
||||
}
|
||||
|
||||
// Fall back to file-based memory if no FalkorDB memories found
|
||||
// Fall back to file-based memory if no graph memories found
|
||||
if (recentMemories.length === 0) {
|
||||
const specsBaseDir = getSpecsDir(autoBuildPath);
|
||||
const specsDir = path.join(projectPath, specsBaseDir);
|
||||
@@ -110,8 +111,8 @@ export function registerProjectContextHandlers(
|
||||
project.path,
|
||||
project.autoBuildPath,
|
||||
memoryStatus.available,
|
||||
memoryStatus.host,
|
||||
memoryStatus.port
|
||||
memoryStatus.dbPath,
|
||||
memoryStatus.database
|
||||
);
|
||||
|
||||
return {
|
||||
@@ -144,7 +145,7 @@ export function registerProjectContextHandlers(
|
||||
|
||||
try {
|
||||
// Run the analyzer script to regenerate project_index.json
|
||||
const autoBuildSource = getAutoBuildSourcePath();
|
||||
const autoBuildSource = getEffectiveSourcePath();
|
||||
|
||||
if (!autoBuildSource) {
|
||||
return {
|
||||
|
||||
@@ -120,29 +120,21 @@ export function hasOpenAIKey(projectEnvVars: EnvironmentVars, globalSettings: Gl
|
||||
}
|
||||
|
||||
/**
|
||||
* Get Graphiti connection details
|
||||
* Get Graphiti database details (LadybugDB - embedded database)
|
||||
*/
|
||||
export interface GraphitiConnectionDetails {
|
||||
host: string;
|
||||
port: number;
|
||||
export interface GraphitiDatabaseDetails {
|
||||
dbPath: string;
|
||||
database: string;
|
||||
}
|
||||
|
||||
export function getGraphitiConnectionDetails(projectEnvVars: EnvironmentVars): GraphitiConnectionDetails {
|
||||
const host = projectEnvVars['GRAPHITI_FALKORDB_HOST'] ||
|
||||
process.env.GRAPHITI_FALKORDB_HOST ||
|
||||
'localhost';
|
||||
|
||||
const port = parseInt(
|
||||
projectEnvVars['GRAPHITI_FALKORDB_PORT'] ||
|
||||
process.env.GRAPHITI_FALKORDB_PORT ||
|
||||
'6380',
|
||||
10
|
||||
);
|
||||
export function getGraphitiDatabaseDetails(projectEnvVars: EnvironmentVars): GraphitiDatabaseDetails {
|
||||
const dbPath = projectEnvVars['GRAPHITI_DB_PATH'] ||
|
||||
process.env.GRAPHITI_DB_PATH ||
|
||||
require('path').join(require('os').homedir(), '.auto-claude', 'memories');
|
||||
|
||||
const database = projectEnvVars['GRAPHITI_DATABASE'] ||
|
||||
process.env.GRAPHITI_DATABASE ||
|
||||
'auto_claude_memory';
|
||||
|
||||
return { host, port, database };
|
||||
return { dbPath, database };
|
||||
}
|
||||
|
||||
@@ -1,156 +1,20 @@
|
||||
/**
|
||||
* Docker & FalkorDB IPC Handlers
|
||||
* Docker & Infrastructure IPC Handlers
|
||||
*
|
||||
* Provides automatic infrastructure detection for non-technical users.
|
||||
* When Graphiti is enabled, the UI can check Docker/FalkorDB status
|
||||
* and offer one-click solutions instead of manual terminal commands.
|
||||
* DEPRECATED: This file is kept for backward compatibility.
|
||||
* Memory infrastructure has moved to LadybugDB (no Docker required).
|
||||
* See memory-handlers.ts for the new implementation.
|
||||
*
|
||||
* This file now re-exports from memory-handlers.ts
|
||||
*/
|
||||
|
||||
import { ipcMain } from 'electron';
|
||||
import { IPC_CHANNELS } from '../../shared/constants';
|
||||
import type { IPCResult, InfrastructureStatus } from '../../shared/types';
|
||||
import {
|
||||
getInfrastructureStatus,
|
||||
startFalkorDB,
|
||||
stopFalkorDB,
|
||||
openDockerDesktop,
|
||||
getDockerDownloadUrl,
|
||||
validateFalkorDBConnection,
|
||||
validateOpenAIApiKey,
|
||||
testGraphitiConnection,
|
||||
type GraphitiValidationResult,
|
||||
} from '../docker-service';
|
||||
import { registerMemoryHandlers } from './memory-handlers';
|
||||
|
||||
/**
|
||||
* Register all Docker-related IPC handlers
|
||||
* @deprecated Use registerMemoryHandlers() instead
|
||||
*/
|
||||
export function registerDockerHandlers(): void {
|
||||
// Get infrastructure status (Docker + FalkorDB)
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.DOCKER_STATUS,
|
||||
async (_, port?: number): Promise<IPCResult<InfrastructureStatus>> => {
|
||||
try {
|
||||
const status = await getInfrastructureStatus(port);
|
||||
return { success: true, data: status };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to check Docker status',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// Start FalkorDB container
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.DOCKER_START_FALKORDB,
|
||||
async (_, port?: number): Promise<IPCResult<{ success: boolean; error?: string }>> => {
|
||||
try {
|
||||
const result = await startFalkorDB(port);
|
||||
return { success: true, data: result };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to start FalkorDB',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// Stop FalkorDB container
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.DOCKER_STOP_FALKORDB,
|
||||
async (): Promise<IPCResult<{ success: boolean; error?: string }>> => {
|
||||
try {
|
||||
const result = await stopFalkorDB();
|
||||
return { success: true, data: result };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to stop FalkorDB',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// Open Docker Desktop application
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.DOCKER_OPEN_DESKTOP,
|
||||
async (): Promise<IPCResult<{ success: boolean; error?: string }>> => {
|
||||
try {
|
||||
const result = await openDockerDesktop();
|
||||
return { success: true, data: result };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to open Docker Desktop',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// Get Docker download URL
|
||||
ipcMain.handle(IPC_CHANNELS.DOCKER_GET_DOWNLOAD_URL, async (): Promise<string> => {
|
||||
return getDockerDownloadUrl();
|
||||
});
|
||||
|
||||
// ============================================
|
||||
// Graphiti Validation Handlers
|
||||
// ============================================
|
||||
|
||||
// Validate FalkorDB connection
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.GRAPHITI_VALIDATE_FALKORDB,
|
||||
async (_, uri: string): Promise<IPCResult<GraphitiValidationResult>> => {
|
||||
try {
|
||||
const result = await validateFalkorDBConnection(uri);
|
||||
return { success: true, data: result };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to validate FalkorDB connection',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// Validate OpenAI API key
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.GRAPHITI_VALIDATE_OPENAI,
|
||||
async (_, apiKey: string): Promise<IPCResult<GraphitiValidationResult>> => {
|
||||
try {
|
||||
const result = await validateOpenAIApiKey(apiKey);
|
||||
return { success: true, data: result };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to validate OpenAI API key',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// Test full Graphiti connection (FalkorDB + OpenAI)
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.GRAPHITI_TEST_CONNECTION,
|
||||
async (
|
||||
_,
|
||||
falkorDbUri: string,
|
||||
openAiApiKey: string
|
||||
): Promise<IPCResult<{
|
||||
falkordb: GraphitiValidationResult;
|
||||
openai: GraphitiValidationResult;
|
||||
ready: boolean;
|
||||
}>> => {
|
||||
try {
|
||||
const result = await testGraphitiConnection(falkorDbUri, openAiApiKey);
|
||||
return { success: true, data: result };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to test Graphiti connection',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
// Register the new memory handlers instead
|
||||
registerMemoryHandlers();
|
||||
}
|
||||
|
||||
@@ -69,56 +69,42 @@ export function registerEnvHandlers(
|
||||
if (config.graphitiEnabled !== undefined) {
|
||||
existingVars['GRAPHITI_ENABLED'] = config.graphitiEnabled ? 'true' : 'false';
|
||||
}
|
||||
// Graphiti Provider Configuration
|
||||
// Memory Provider Configuration (embeddings only - LLM uses Claude SDK)
|
||||
if (config.graphitiProviderConfig) {
|
||||
const pc = config.graphitiProviderConfig;
|
||||
if (pc.llmProvider) existingVars['GRAPHITI_LLM_PROVIDER'] = pc.llmProvider;
|
||||
// Embedding provider only (LLM provider removed - Claude SDK handles RAG)
|
||||
if (pc.embeddingProvider) existingVars['GRAPHITI_EMBEDDER_PROVIDER'] = pc.embeddingProvider;
|
||||
// OpenAI
|
||||
// OpenAI Embeddings
|
||||
if (pc.openaiApiKey) existingVars['OPENAI_API_KEY'] = pc.openaiApiKey;
|
||||
if (pc.openaiModel) existingVars['OPENAI_MODEL'] = pc.openaiModel;
|
||||
if (pc.openaiEmbeddingModel) existingVars['OPENAI_EMBEDDING_MODEL'] = pc.openaiEmbeddingModel;
|
||||
// Anthropic
|
||||
if (pc.anthropicApiKey) existingVars['ANTHROPIC_API_KEY'] = pc.anthropicApiKey;
|
||||
if (pc.anthropicModel) existingVars['GRAPHITI_ANTHROPIC_MODEL'] = pc.anthropicModel;
|
||||
// Azure OpenAI
|
||||
// Azure OpenAI Embeddings
|
||||
if (pc.azureOpenaiApiKey) existingVars['AZURE_OPENAI_API_KEY'] = pc.azureOpenaiApiKey;
|
||||
if (pc.azureOpenaiBaseUrl) existingVars['AZURE_OPENAI_BASE_URL'] = pc.azureOpenaiBaseUrl;
|
||||
if (pc.azureOpenaiLlmDeployment) existingVars['AZURE_OPENAI_LLM_DEPLOYMENT'] = pc.azureOpenaiLlmDeployment;
|
||||
if (pc.azureOpenaiEmbeddingDeployment) existingVars['AZURE_OPENAI_EMBEDDING_DEPLOYMENT'] = pc.azureOpenaiEmbeddingDeployment;
|
||||
// Voyage
|
||||
// Voyage Embeddings
|
||||
if (pc.voyageApiKey) existingVars['VOYAGE_API_KEY'] = pc.voyageApiKey;
|
||||
if (pc.voyageEmbeddingModel) existingVars['VOYAGE_EMBEDDING_MODEL'] = pc.voyageEmbeddingModel;
|
||||
// Google
|
||||
// Google Embeddings
|
||||
if (pc.googleApiKey) existingVars['GOOGLE_API_KEY'] = pc.googleApiKey;
|
||||
if (pc.googleLlmModel) existingVars['GOOGLE_LLM_MODEL'] = pc.googleLlmModel;
|
||||
if (pc.googleEmbeddingModel) existingVars['GOOGLE_EMBEDDING_MODEL'] = pc.googleEmbeddingModel;
|
||||
// Ollama
|
||||
// Ollama Embeddings
|
||||
if (pc.ollamaBaseUrl) existingVars['OLLAMA_BASE_URL'] = pc.ollamaBaseUrl;
|
||||
if (pc.ollamaLlmModel) existingVars['OLLAMA_LLM_MODEL'] = pc.ollamaLlmModel;
|
||||
if (pc.ollamaEmbeddingModel) existingVars['OLLAMA_EMBEDDING_MODEL'] = pc.ollamaEmbeddingModel;
|
||||
if (pc.ollamaEmbeddingDim) existingVars['OLLAMA_EMBEDDING_DIM'] = String(pc.ollamaEmbeddingDim);
|
||||
// FalkorDB
|
||||
if (pc.falkorDbHost) existingVars['GRAPHITI_FALKORDB_HOST'] = pc.falkorDbHost;
|
||||
if (pc.falkorDbPort) existingVars['GRAPHITI_FALKORDB_PORT'] = String(pc.falkorDbPort);
|
||||
if (pc.falkorDbPassword) existingVars['GRAPHITI_FALKORDB_PASSWORD'] = pc.falkorDbPassword;
|
||||
// LadybugDB (embedded database)
|
||||
if (pc.dbPath) existingVars['GRAPHITI_DB_PATH'] = pc.dbPath;
|
||||
if (pc.database) existingVars['GRAPHITI_DATABASE'] = pc.database;
|
||||
}
|
||||
// Legacy fields (still supported)
|
||||
if (config.openaiApiKey !== undefined) {
|
||||
existingVars['OPENAI_API_KEY'] = config.openaiApiKey;
|
||||
}
|
||||
if (config.graphitiFalkorDbHost !== undefined) {
|
||||
existingVars['GRAPHITI_FALKORDB_HOST'] = config.graphitiFalkorDbHost;
|
||||
}
|
||||
if (config.graphitiFalkorDbPort !== undefined) {
|
||||
existingVars['GRAPHITI_FALKORDB_PORT'] = String(config.graphitiFalkorDbPort);
|
||||
}
|
||||
if (config.graphitiFalkorDbPassword !== undefined) {
|
||||
existingVars['GRAPHITI_FALKORDB_PASSWORD'] = config.graphitiFalkorDbPassword;
|
||||
}
|
||||
if (config.graphitiDatabase !== undefined) {
|
||||
existingVars['GRAPHITI_DATABASE'] = config.graphitiDatabase;
|
||||
}
|
||||
if (config.graphitiDbPath !== undefined) {
|
||||
existingVars['GRAPHITI_DB_PATH'] = config.graphitiDbPath;
|
||||
}
|
||||
if (config.enableFancyUi !== undefined) {
|
||||
existingVars['ENABLE_FANCY_UI'] = config.enableFancyUi ? 'true' : 'false';
|
||||
}
|
||||
@@ -161,50 +147,39 @@ ${existingVars['DEFAULT_BRANCH'] ? `DEFAULT_BRANCH=${existingVars['DEFAULT_BRANC
|
||||
${existingVars['ENABLE_FANCY_UI'] !== undefined ? `ENABLE_FANCY_UI=${existingVars['ENABLE_FANCY_UI']}` : '# ENABLE_FANCY_UI=true'}
|
||||
|
||||
# =============================================================================
|
||||
# GRAPHITI MEMORY INTEGRATION (OPTIONAL)
|
||||
# Multi-provider support: OpenAI, Anthropic, Google AI, Azure OpenAI, Ollama, Voyage
|
||||
# MEMORY INTEGRATION
|
||||
# Embedding providers: OpenAI, Google AI, Azure OpenAI, Ollama, Voyage
|
||||
# =============================================================================
|
||||
${existingVars['GRAPHITI_ENABLED'] ? `GRAPHITI_ENABLED=${existingVars['GRAPHITI_ENABLED']}` : '# GRAPHITI_ENABLED=false'}
|
||||
${existingVars['GRAPHITI_ENABLED'] ? `GRAPHITI_ENABLED=${existingVars['GRAPHITI_ENABLED']}` : '# GRAPHITI_ENABLED=true'}
|
||||
|
||||
# Provider Selection
|
||||
${existingVars['GRAPHITI_LLM_PROVIDER'] ? `GRAPHITI_LLM_PROVIDER=${existingVars['GRAPHITI_LLM_PROVIDER']}` : '# GRAPHITI_LLM_PROVIDER=openai'}
|
||||
${existingVars['GRAPHITI_EMBEDDER_PROVIDER'] ? `GRAPHITI_EMBEDDER_PROVIDER=${existingVars['GRAPHITI_EMBEDDER_PROVIDER']}` : '# GRAPHITI_EMBEDDER_PROVIDER=openai'}
|
||||
# Embedding Provider (for semantic search - optional, keyword search works without)
|
||||
${existingVars['GRAPHITI_EMBEDDER_PROVIDER'] ? `GRAPHITI_EMBEDDER_PROVIDER=${existingVars['GRAPHITI_EMBEDDER_PROVIDER']}` : '# GRAPHITI_EMBEDDER_PROVIDER=ollama'}
|
||||
|
||||
# OpenAI Settings
|
||||
# OpenAI Embeddings
|
||||
${existingVars['OPENAI_API_KEY'] ? `OPENAI_API_KEY=${existingVars['OPENAI_API_KEY']}` : '# OPENAI_API_KEY='}
|
||||
${existingVars['OPENAI_MODEL'] ? `OPENAI_MODEL=${existingVars['OPENAI_MODEL']}` : '# OPENAI_MODEL=gpt-4o-mini'}
|
||||
${existingVars['OPENAI_EMBEDDING_MODEL'] ? `OPENAI_EMBEDDING_MODEL=${existingVars['OPENAI_EMBEDDING_MODEL']}` : '# OPENAI_EMBEDDING_MODEL=text-embedding-3-small'}
|
||||
|
||||
# Anthropic Settings (LLM only - use with Voyage or OpenAI for embeddings)
|
||||
${existingVars['ANTHROPIC_API_KEY'] ? `ANTHROPIC_API_KEY=${existingVars['ANTHROPIC_API_KEY']}` : '# ANTHROPIC_API_KEY='}
|
||||
${existingVars['GRAPHITI_ANTHROPIC_MODEL'] ? `GRAPHITI_ANTHROPIC_MODEL=${existingVars['GRAPHITI_ANTHROPIC_MODEL']}` : '# GRAPHITI_ANTHROPIC_MODEL=claude-sonnet-4-5-latest'}
|
||||
|
||||
# Azure OpenAI Settings
|
||||
# Azure OpenAI Embeddings
|
||||
${existingVars['AZURE_OPENAI_API_KEY'] ? `AZURE_OPENAI_API_KEY=${existingVars['AZURE_OPENAI_API_KEY']}` : '# AZURE_OPENAI_API_KEY='}
|
||||
${existingVars['AZURE_OPENAI_BASE_URL'] ? `AZURE_OPENAI_BASE_URL=${existingVars['AZURE_OPENAI_BASE_URL']}` : '# AZURE_OPENAI_BASE_URL='}
|
||||
${existingVars['AZURE_OPENAI_LLM_DEPLOYMENT'] ? `AZURE_OPENAI_LLM_DEPLOYMENT=${existingVars['AZURE_OPENAI_LLM_DEPLOYMENT']}` : '# AZURE_OPENAI_LLM_DEPLOYMENT='}
|
||||
${existingVars['AZURE_OPENAI_EMBEDDING_DEPLOYMENT'] ? `AZURE_OPENAI_EMBEDDING_DEPLOYMENT=${existingVars['AZURE_OPENAI_EMBEDDING_DEPLOYMENT']}` : '# AZURE_OPENAI_EMBEDDING_DEPLOYMENT='}
|
||||
|
||||
# Voyage AI Settings (Embeddings only - great with Anthropic)
|
||||
# Voyage AI Embeddings
|
||||
${existingVars['VOYAGE_API_KEY'] ? `VOYAGE_API_KEY=${existingVars['VOYAGE_API_KEY']}` : '# VOYAGE_API_KEY='}
|
||||
${existingVars['VOYAGE_EMBEDDING_MODEL'] ? `VOYAGE_EMBEDDING_MODEL=${existingVars['VOYAGE_EMBEDDING_MODEL']}` : '# VOYAGE_EMBEDDING_MODEL=voyage-3'}
|
||||
|
||||
# Google AI Settings (LLM and Embeddings - Gemini)
|
||||
# Google AI Embeddings
|
||||
${existingVars['GOOGLE_API_KEY'] ? `GOOGLE_API_KEY=${existingVars['GOOGLE_API_KEY']}` : '# GOOGLE_API_KEY='}
|
||||
${existingVars['GOOGLE_LLM_MODEL'] ? `GOOGLE_LLM_MODEL=${existingVars['GOOGLE_LLM_MODEL']}` : '# GOOGLE_LLM_MODEL=gemini-2.0-flash'}
|
||||
${existingVars['GOOGLE_EMBEDDING_MODEL'] ? `GOOGLE_EMBEDDING_MODEL=${existingVars['GOOGLE_EMBEDDING_MODEL']}` : '# GOOGLE_EMBEDDING_MODEL=text-embedding-004'}
|
||||
|
||||
# Ollama Settings (Local - free)
|
||||
# Ollama Embeddings (Local - free)
|
||||
${existingVars['OLLAMA_BASE_URL'] ? `OLLAMA_BASE_URL=${existingVars['OLLAMA_BASE_URL']}` : '# OLLAMA_BASE_URL=http://localhost:11434'}
|
||||
${existingVars['OLLAMA_LLM_MODEL'] ? `OLLAMA_LLM_MODEL=${existingVars['OLLAMA_LLM_MODEL']}` : '# OLLAMA_LLM_MODEL='}
|
||||
${existingVars['OLLAMA_EMBEDDING_MODEL'] ? `OLLAMA_EMBEDDING_MODEL=${existingVars['OLLAMA_EMBEDDING_MODEL']}` : '# OLLAMA_EMBEDDING_MODEL='}
|
||||
${existingVars['OLLAMA_EMBEDDING_MODEL'] ? `OLLAMA_EMBEDDING_MODEL=${existingVars['OLLAMA_EMBEDDING_MODEL']}` : '# OLLAMA_EMBEDDING_MODEL=embeddinggemma'}
|
||||
${existingVars['OLLAMA_EMBEDDING_DIM'] ? `OLLAMA_EMBEDDING_DIM=${existingVars['OLLAMA_EMBEDDING_DIM']}` : '# OLLAMA_EMBEDDING_DIM=768'}
|
||||
|
||||
# FalkorDB Connection
|
||||
${existingVars['GRAPHITI_FALKORDB_HOST'] ? `GRAPHITI_FALKORDB_HOST=${existingVars['GRAPHITI_FALKORDB_HOST']}` : '# GRAPHITI_FALKORDB_HOST=localhost'}
|
||||
${existingVars['GRAPHITI_FALKORDB_PORT'] ? `GRAPHITI_FALKORDB_PORT=${existingVars['GRAPHITI_FALKORDB_PORT']}` : '# GRAPHITI_FALKORDB_PORT=6380'}
|
||||
${existingVars['GRAPHITI_FALKORDB_PASSWORD'] ? `GRAPHITI_FALKORDB_PASSWORD=${existingVars['GRAPHITI_FALKORDB_PASSWORD']}` : '# GRAPHITI_FALKORDB_PASSWORD='}
|
||||
# LadybugDB Database (embedded - no Docker required)
|
||||
${existingVars['GRAPHITI_DATABASE'] ? `GRAPHITI_DATABASE=${existingVars['GRAPHITI_DATABASE']}` : '# GRAPHITI_DATABASE=auto_claude_memory'}
|
||||
${existingVars['GRAPHITI_DB_PATH'] ? `GRAPHITI_DB_PATH=${existingVars['GRAPHITI_DB_PATH']}` : '# GRAPHITI_DB_PATH=~/.auto-claude/memories'}
|
||||
`;
|
||||
|
||||
return content;
|
||||
@@ -316,59 +291,43 @@ ${existingVars['GRAPHITI_DATABASE'] ? `GRAPHITI_DATABASE=${existingVars['GRAPHIT
|
||||
config.openaiKeyIsGlobal = true;
|
||||
}
|
||||
|
||||
if (vars['GRAPHITI_FALKORDB_HOST']) {
|
||||
config.graphitiFalkorDbHost = vars['GRAPHITI_FALKORDB_HOST'];
|
||||
}
|
||||
if (vars['GRAPHITI_FALKORDB_PORT']) {
|
||||
config.graphitiFalkorDbPort = parseInt(vars['GRAPHITI_FALKORDB_PORT'], 10);
|
||||
}
|
||||
if (vars['GRAPHITI_FALKORDB_PASSWORD']) {
|
||||
config.graphitiFalkorDbPassword = vars['GRAPHITI_FALKORDB_PASSWORD'];
|
||||
}
|
||||
if (vars['GRAPHITI_DATABASE']) {
|
||||
config.graphitiDatabase = vars['GRAPHITI_DATABASE'];
|
||||
}
|
||||
if (vars['GRAPHITI_DB_PATH']) {
|
||||
config.graphitiDbPath = vars['GRAPHITI_DB_PATH'];
|
||||
}
|
||||
|
||||
if (vars['ENABLE_FANCY_UI']?.toLowerCase() === 'false') {
|
||||
config.enableFancyUi = false;
|
||||
}
|
||||
|
||||
// Populate graphitiProviderConfig from .env file
|
||||
const llmProvider = vars['GRAPHITI_LLM_PROVIDER'];
|
||||
// Populate graphitiProviderConfig from .env file (embeddings only - no LLM provider)
|
||||
const embeddingProvider = vars['GRAPHITI_EMBEDDER_PROVIDER'];
|
||||
if (llmProvider || embeddingProvider || vars['ANTHROPIC_API_KEY'] || vars['AZURE_OPENAI_API_KEY'] ||
|
||||
if (embeddingProvider || vars['AZURE_OPENAI_API_KEY'] ||
|
||||
vars['VOYAGE_API_KEY'] || vars['GOOGLE_API_KEY'] || vars['OLLAMA_BASE_URL']) {
|
||||
config.graphitiProviderConfig = {
|
||||
llmProvider: (llmProvider as 'openai' | 'anthropic' | 'azure_openai' | 'ollama' | 'google' | 'groq') || 'openai',
|
||||
embeddingProvider: (embeddingProvider as 'openai' | 'voyage' | 'azure_openai' | 'ollama' | 'google' | 'huggingface') || 'openai',
|
||||
// OpenAI
|
||||
embeddingProvider: (embeddingProvider as 'openai' | 'voyage' | 'azure_openai' | 'ollama' | 'google') || 'ollama',
|
||||
// OpenAI Embeddings
|
||||
openaiApiKey: vars['OPENAI_API_KEY'],
|
||||
openaiModel: vars['OPENAI_MODEL'],
|
||||
openaiEmbeddingModel: vars['OPENAI_EMBEDDING_MODEL'],
|
||||
// Anthropic
|
||||
anthropicApiKey: vars['ANTHROPIC_API_KEY'],
|
||||
anthropicModel: vars['GRAPHITI_ANTHROPIC_MODEL'],
|
||||
// Azure OpenAI
|
||||
// Azure OpenAI Embeddings
|
||||
azureOpenaiApiKey: vars['AZURE_OPENAI_API_KEY'],
|
||||
azureOpenaiBaseUrl: vars['AZURE_OPENAI_BASE_URL'],
|
||||
azureOpenaiLlmDeployment: vars['AZURE_OPENAI_LLM_DEPLOYMENT'],
|
||||
azureOpenaiEmbeddingDeployment: vars['AZURE_OPENAI_EMBEDDING_DEPLOYMENT'],
|
||||
// Voyage
|
||||
// Voyage Embeddings
|
||||
voyageApiKey: vars['VOYAGE_API_KEY'],
|
||||
voyageEmbeddingModel: vars['VOYAGE_EMBEDDING_MODEL'],
|
||||
// Google
|
||||
// Google Embeddings
|
||||
googleApiKey: vars['GOOGLE_API_KEY'],
|
||||
googleLlmModel: vars['GOOGLE_LLM_MODEL'],
|
||||
googleEmbeddingModel: vars['GOOGLE_EMBEDDING_MODEL'],
|
||||
// Ollama
|
||||
// Ollama Embeddings
|
||||
ollamaBaseUrl: vars['OLLAMA_BASE_URL'],
|
||||
ollamaLlmModel: vars['OLLAMA_LLM_MODEL'],
|
||||
ollamaEmbeddingModel: vars['OLLAMA_EMBEDDING_MODEL'],
|
||||
ollamaEmbeddingDim: vars['OLLAMA_EMBEDDING_DIM'] ? parseInt(vars['OLLAMA_EMBEDDING_DIM'], 10) : undefined,
|
||||
// FalkorDB
|
||||
falkorDbHost: vars['GRAPHITI_FALKORDB_HOST'],
|
||||
falkorDbPort: vars['GRAPHITI_FALKORDB_PORT'] ? parseInt(vars['GRAPHITI_FALKORDB_PORT'], 10) : undefined,
|
||||
falkorDbPassword: vars['GRAPHITI_FALKORDB_PASSWORD'],
|
||||
// LadybugDB
|
||||
database: vars['GRAPHITI_DATABASE'],
|
||||
dbPath: vars['GRAPHITI_DB_PATH'],
|
||||
};
|
||||
}
|
||||
|
||||
|
||||
@@ -573,6 +573,203 @@ export function registerGetGitHubBranches(): void {
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new GitHub repository using gh CLI
|
||||
*/
|
||||
export function registerCreateGitHubRepo(): void {
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.GITHUB_CREATE_REPO,
|
||||
async (
|
||||
_event: Electron.IpcMainInvokeEvent,
|
||||
repoName: string,
|
||||
options: { description?: string; isPrivate?: boolean; projectPath: string; owner?: string }
|
||||
): Promise<IPCResult<{ fullName: string; url: string }>> => {
|
||||
debugLog('createGitHubRepo handler called', { repoName, options });
|
||||
|
||||
// Validate repo name - only alphanumeric, hyphens, underscores
|
||||
if (!/^[A-Za-z0-9_.-]+$/.test(repoName)) {
|
||||
return {
|
||||
success: false,
|
||||
error: 'Invalid repository name. Use only letters, numbers, hyphens, underscores, and periods.'
|
||||
};
|
||||
}
|
||||
|
||||
try {
|
||||
// Get the authenticated username
|
||||
const username = execSync('gh api user --jq .login', {
|
||||
encoding: 'utf-8',
|
||||
stdio: 'pipe'
|
||||
}).trim();
|
||||
|
||||
// Determine the owner (personal account or organization)
|
||||
const owner = options.owner || username;
|
||||
const isOrgRepo = owner !== username;
|
||||
|
||||
// Build the full repo name (owner/repo format for orgs)
|
||||
const repoFullName = isOrgRepo ? `${owner}/${repoName}` : repoName;
|
||||
|
||||
// Build gh repo create command arguments
|
||||
const args = ['repo', 'create', repoFullName, '--source', options.projectPath];
|
||||
|
||||
if (options.isPrivate) {
|
||||
args.push('--private');
|
||||
} else {
|
||||
args.push('--public');
|
||||
}
|
||||
|
||||
if (options.description) {
|
||||
args.push('--description', options.description);
|
||||
}
|
||||
|
||||
// Push to remote after creation
|
||||
args.push('--push');
|
||||
|
||||
debugLog('Running: gh', args);
|
||||
const output = execFileSync('gh', args, {
|
||||
encoding: 'utf-8',
|
||||
cwd: options.projectPath,
|
||||
stdio: 'pipe'
|
||||
});
|
||||
|
||||
debugLog('gh repo create output:', output);
|
||||
|
||||
const fullName = `${owner}/${repoName}`;
|
||||
const url = `https://github.com/${fullName}`;
|
||||
|
||||
debugLog('Created repo:', { fullName, url });
|
||||
|
||||
return {
|
||||
success: true,
|
||||
data: { fullName, url }
|
||||
};
|
||||
} catch (error) {
|
||||
const errorMessage = error instanceof Error ? error.message : 'Failed to create repository';
|
||||
debugLog('Failed to create repo:', errorMessage);
|
||||
return {
|
||||
success: false,
|
||||
error: errorMessage
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Add a remote origin to a local git repository
|
||||
*/
|
||||
export function registerAddGitRemote(): void {
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.GITHUB_ADD_REMOTE,
|
||||
async (
|
||||
_event: Electron.IpcMainInvokeEvent,
|
||||
projectPath: string,
|
||||
repoFullName: string
|
||||
): Promise<IPCResult<{ remoteUrl: string }>> => {
|
||||
debugLog('addGitRemote handler called', { projectPath, repoFullName });
|
||||
|
||||
// Validate repo format
|
||||
if (!isValidGitHubRepo(repoFullName)) {
|
||||
return {
|
||||
success: false,
|
||||
error: 'Invalid repository format. Expected: owner/repo'
|
||||
};
|
||||
}
|
||||
|
||||
const remoteUrl = `https://github.com/${repoFullName}.git`;
|
||||
|
||||
try {
|
||||
// Check if origin already exists
|
||||
try {
|
||||
execSync('git remote get-url origin', {
|
||||
cwd: projectPath,
|
||||
encoding: 'utf-8',
|
||||
stdio: 'pipe'
|
||||
});
|
||||
// Origin exists, remove it first
|
||||
debugLog('Removing existing origin remote');
|
||||
execSync('git remote remove origin', {
|
||||
cwd: projectPath,
|
||||
encoding: 'utf-8',
|
||||
stdio: 'pipe'
|
||||
});
|
||||
} catch {
|
||||
// No origin exists, which is fine
|
||||
}
|
||||
|
||||
// Add the remote
|
||||
debugLog('Adding remote origin:', remoteUrl);
|
||||
execFileSync('git', ['remote', 'add', 'origin', remoteUrl], {
|
||||
cwd: projectPath,
|
||||
encoding: 'utf-8',
|
||||
stdio: 'pipe'
|
||||
});
|
||||
|
||||
debugLog('Remote added successfully');
|
||||
return {
|
||||
success: true,
|
||||
data: { remoteUrl }
|
||||
};
|
||||
} catch (error) {
|
||||
const errorMessage = error instanceof Error ? error.message : 'Failed to add remote';
|
||||
debugLog('Failed to add remote:', errorMessage);
|
||||
return {
|
||||
success: false,
|
||||
error: errorMessage
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* List user's GitHub organizations
|
||||
*/
|
||||
export function registerListGitHubOrgs(): void {
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.GITHUB_LIST_ORGS,
|
||||
async (): Promise<IPCResult<{ orgs: Array<{ login: string; avatarUrl?: string }> }>> => {
|
||||
debugLog('listGitHubOrgs handler called');
|
||||
|
||||
try {
|
||||
// Get user's organizations
|
||||
const output = execSync('gh api user/orgs --jq \'.[] | {login: .login, avatarUrl: .avatar_url}\'', {
|
||||
encoding: 'utf-8',
|
||||
stdio: 'pipe'
|
||||
});
|
||||
|
||||
// Parse the JSON lines output
|
||||
const orgs: Array<{ login: string; avatarUrl?: string }> = [];
|
||||
const lines = output.trim().split('\n').filter(line => line.trim());
|
||||
|
||||
for (const line of lines) {
|
||||
try {
|
||||
const org = JSON.parse(line);
|
||||
orgs.push({
|
||||
login: org.login,
|
||||
avatarUrl: org.avatarUrl
|
||||
});
|
||||
} catch {
|
||||
// Skip invalid JSON lines
|
||||
}
|
||||
}
|
||||
|
||||
debugLog('Found organizations:', orgs.length);
|
||||
return {
|
||||
success: true,
|
||||
data: { orgs }
|
||||
};
|
||||
} catch (error) {
|
||||
const errorMessage = error instanceof Error ? error.message : 'Failed to list organizations';
|
||||
debugLog('Failed to list orgs:', errorMessage);
|
||||
return {
|
||||
success: true, // Return success with empty array - user might not have any orgs
|
||||
data: { orgs: [] }
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Register all GitHub OAuth handlers
|
||||
*/
|
||||
@@ -586,5 +783,8 @@ export function registerGithubOAuthHandlers(): void {
|
||||
registerListUserRepos();
|
||||
registerDetectGitHubRepo();
|
||||
registerGetGitHubBranches();
|
||||
registerCreateGitHubRepo();
|
||||
registerAddGitRemote();
|
||||
registerListGitHubOrgs();
|
||||
debugLog('GitHub OAuth handlers registered');
|
||||
}
|
||||
|
||||
@@ -92,7 +92,7 @@ export function setupIpcHandlers(
|
||||
// Insights handlers
|
||||
registerInsightsHandlers(getMainWindow);
|
||||
|
||||
// Docker & infrastructure handlers (for Graphiti/FalkorDB)
|
||||
// Memory & infrastructure handlers (for Graphiti/LadybugDB)
|
||||
registerDockerHandlers();
|
||||
|
||||
// App auto-update handlers
|
||||
|
||||
@@ -0,0 +1,532 @@
|
||||
/**
|
||||
* Memory Infrastructure IPC Handlers
|
||||
*
|
||||
* Provides memory database status and validation for the Graphiti integration.
|
||||
* Uses LadybugDB (embedded Kuzu-based database) - no Docker required.
|
||||
*/
|
||||
|
||||
import { ipcMain } from 'electron';
|
||||
import { spawn } from 'child_process';
|
||||
import * as path from 'path';
|
||||
import * as fs from 'fs';
|
||||
import { IPC_CHANNELS } from '../../shared/constants';
|
||||
import type {
|
||||
IPCResult,
|
||||
InfrastructureStatus,
|
||||
GraphitiValidationResult,
|
||||
GraphitiConnectionTestResult,
|
||||
} from '../../shared/types';
|
||||
import {
|
||||
getMemoryServiceStatus,
|
||||
getMemoryService,
|
||||
getDefaultDbPath,
|
||||
isKuzuAvailable,
|
||||
} from '../memory-service';
|
||||
import { validateOpenAIApiKey } from '../api-validation-service';
|
||||
import { findPythonCommand, parsePythonCommand } from '../python-detector';
|
||||
|
||||
// Ollama types
|
||||
interface OllamaStatus {
|
||||
running: boolean;
|
||||
url: string;
|
||||
version?: string;
|
||||
message?: string;
|
||||
}
|
||||
|
||||
interface OllamaModel {
|
||||
name: string;
|
||||
size_bytes: number;
|
||||
size_gb: number;
|
||||
modified_at: string;
|
||||
is_embedding: boolean;
|
||||
embedding_dim?: number | null;
|
||||
description?: string;
|
||||
}
|
||||
|
||||
interface OllamaEmbeddingModel {
|
||||
name: string;
|
||||
embedding_dim: number | null;
|
||||
description: string;
|
||||
size_bytes: number;
|
||||
size_gb: number;
|
||||
}
|
||||
|
||||
interface OllamaRecommendedModel {
|
||||
name: string;
|
||||
description: string;
|
||||
size_estimate: string;
|
||||
dim: number;
|
||||
installed: boolean;
|
||||
}
|
||||
|
||||
interface OllamaPullResult {
|
||||
model: string;
|
||||
status: 'completed' | 'failed';
|
||||
output: string[];
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute the ollama_model_detector.py script
|
||||
*/
|
||||
async function executeOllamaDetector(
|
||||
command: string,
|
||||
baseUrl?: string
|
||||
): Promise<{ success: boolean; data?: unknown; error?: string }> {
|
||||
const pythonCmd = findPythonCommand();
|
||||
if (!pythonCmd) {
|
||||
return { success: false, error: 'Python not found' };
|
||||
}
|
||||
|
||||
// Find the ollama_model_detector.py script
|
||||
const possiblePaths = [
|
||||
path.resolve(__dirname, '..', '..', '..', 'auto-claude', 'ollama_model_detector.py'),
|
||||
path.resolve(process.cwd(), 'auto-claude', 'ollama_model_detector.py'),
|
||||
path.resolve(process.cwd(), '..', 'auto-claude', 'ollama_model_detector.py'),
|
||||
];
|
||||
|
||||
let scriptPath: string | null = null;
|
||||
for (const p of possiblePaths) {
|
||||
if (fs.existsSync(p)) {
|
||||
scriptPath = p;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (!scriptPath) {
|
||||
return { success: false, error: 'ollama_model_detector.py script not found' };
|
||||
}
|
||||
|
||||
const [pythonExe, baseArgs] = parsePythonCommand(pythonCmd);
|
||||
const args = [...baseArgs, scriptPath, command];
|
||||
if (baseUrl) {
|
||||
args.push('--base-url', baseUrl);
|
||||
}
|
||||
|
||||
return new Promise((resolve) => {
|
||||
let resolved = false;
|
||||
const proc = spawn(pythonExe, args, {
|
||||
stdio: ['ignore', 'pipe', 'pipe'],
|
||||
});
|
||||
|
||||
let stdout = '';
|
||||
let stderr = '';
|
||||
|
||||
proc.stdout.on('data', (data) => {
|
||||
stdout += data.toString();
|
||||
});
|
||||
|
||||
proc.stderr.on('data', (data) => {
|
||||
stderr += data.toString();
|
||||
});
|
||||
|
||||
// Single timeout mechanism to avoid race condition
|
||||
const timeoutId = setTimeout(() => {
|
||||
if (!resolved) {
|
||||
resolved = true;
|
||||
proc.kill();
|
||||
resolve({ success: false, error: 'Timeout' });
|
||||
}
|
||||
}, 10000);
|
||||
|
||||
proc.on('close', (code) => {
|
||||
if (resolved) return;
|
||||
resolved = true;
|
||||
clearTimeout(timeoutId);
|
||||
if (code === 0 && stdout) {
|
||||
try {
|
||||
resolve(JSON.parse(stdout));
|
||||
} catch {
|
||||
resolve({ success: false, error: `Invalid JSON: ${stdout}` });
|
||||
}
|
||||
} else {
|
||||
resolve({ success: false, error: stderr || `Exit code ${code}` });
|
||||
}
|
||||
});
|
||||
|
||||
proc.on('error', (err) => {
|
||||
if (resolved) return;
|
||||
resolved = true;
|
||||
clearTimeout(timeoutId);
|
||||
resolve({ success: false, error: err.message });
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Register all memory-related IPC handlers
|
||||
*/
|
||||
export function registerMemoryHandlers(): void {
|
||||
// Get memory infrastructure status
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.MEMORY_STATUS,
|
||||
async (_): Promise<IPCResult<InfrastructureStatus>> => {
|
||||
try {
|
||||
const status = getMemoryServiceStatus();
|
||||
return {
|
||||
success: true,
|
||||
data: {
|
||||
memory: status,
|
||||
ready: status.kuzuInstalled && status.databaseExists,
|
||||
},
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to check memory status',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// List available databases
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.MEMORY_LIST_DATABASES,
|
||||
async (_, dbPath?: string): Promise<IPCResult<string[]>> => {
|
||||
try {
|
||||
const status = getMemoryServiceStatus(dbPath);
|
||||
return { success: true, data: status.databases };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to list databases',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// Test memory database connection
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.MEMORY_TEST_CONNECTION,
|
||||
async (_, dbPath?: string, database?: string): Promise<IPCResult<GraphitiValidationResult>> => {
|
||||
try {
|
||||
if (!isKuzuAvailable()) {
|
||||
return {
|
||||
success: true,
|
||||
data: {
|
||||
success: false,
|
||||
message: 'kuzu-node is not installed. Memory features require Python 3.12+ with LadybugDB.',
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
const service = getMemoryService({
|
||||
dbPath: dbPath || getDefaultDbPath(),
|
||||
database: database || 'auto_claude_memory',
|
||||
});
|
||||
|
||||
const result = await service.testConnection();
|
||||
return { success: true, data: result };
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to test connection',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// ============================================
|
||||
// Graphiti Validation Handlers
|
||||
// ============================================
|
||||
|
||||
// Validate LLM provider API key (OpenAI, Anthropic, etc.)
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.GRAPHITI_VALIDATE_LLM,
|
||||
async (_, provider: string, apiKey: string): Promise<IPCResult<GraphitiValidationResult>> => {
|
||||
try {
|
||||
// For now, we only validate OpenAI - other providers can be added later
|
||||
if (provider === 'openai') {
|
||||
const result = await validateOpenAIApiKey(apiKey);
|
||||
return { success: true, data: result };
|
||||
}
|
||||
|
||||
// For other providers, do basic validation
|
||||
if (!apiKey || !apiKey.trim()) {
|
||||
return {
|
||||
success: true,
|
||||
data: {
|
||||
success: false,
|
||||
message: 'API key is required',
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
return {
|
||||
success: true,
|
||||
data: {
|
||||
success: true,
|
||||
message: `${provider} API key format appears valid`,
|
||||
details: { provider },
|
||||
},
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to validate API key',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// Test full Graphiti connection (Database + LLM provider)
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.GRAPHITI_TEST_CONNECTION,
|
||||
async (
|
||||
_,
|
||||
config: {
|
||||
dbPath?: string;
|
||||
database?: string;
|
||||
llmProvider: string;
|
||||
apiKey: string;
|
||||
}
|
||||
): Promise<IPCResult<GraphitiConnectionTestResult>> => {
|
||||
try {
|
||||
// Test database connection
|
||||
let databaseResult: GraphitiValidationResult;
|
||||
|
||||
if (!isKuzuAvailable()) {
|
||||
databaseResult = {
|
||||
success: false,
|
||||
message: 'kuzu-node is not installed. Memory features require Python 3.12+ with LadybugDB.',
|
||||
};
|
||||
} else {
|
||||
const service = getMemoryService({
|
||||
dbPath: config.dbPath || getDefaultDbPath(),
|
||||
database: config.database || 'auto_claude_memory',
|
||||
});
|
||||
databaseResult = await service.testConnection();
|
||||
}
|
||||
|
||||
// Test LLM provider
|
||||
let llmResult: GraphitiValidationResult;
|
||||
|
||||
if (config.llmProvider === 'openai') {
|
||||
llmResult = await validateOpenAIApiKey(config.apiKey);
|
||||
} else if (config.llmProvider === 'ollama') {
|
||||
// Ollama doesn't need API key validation
|
||||
llmResult = {
|
||||
success: true,
|
||||
message: 'Ollama (local) does not require API key validation',
|
||||
details: { provider: 'ollama' },
|
||||
};
|
||||
} else {
|
||||
// Basic validation for other providers
|
||||
llmResult = config.apiKey && config.apiKey.trim()
|
||||
? {
|
||||
success: true,
|
||||
message: `${config.llmProvider} API key format appears valid`,
|
||||
details: { provider: config.llmProvider },
|
||||
}
|
||||
: {
|
||||
success: false,
|
||||
message: 'API key is required',
|
||||
};
|
||||
}
|
||||
|
||||
return {
|
||||
success: true,
|
||||
data: {
|
||||
database: databaseResult,
|
||||
llmProvider: llmResult,
|
||||
ready: databaseResult.success && llmResult.success,
|
||||
},
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to test Graphiti connection',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// ============================================
|
||||
// Ollama Model Detection Handlers
|
||||
// ============================================
|
||||
|
||||
// Check if Ollama is running
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.OLLAMA_CHECK_STATUS,
|
||||
async (_, baseUrl?: string): Promise<IPCResult<OllamaStatus>> => {
|
||||
try {
|
||||
const result = await executeOllamaDetector('check-status', baseUrl);
|
||||
|
||||
if (!result.success) {
|
||||
return {
|
||||
success: false,
|
||||
error: result.error || 'Failed to check Ollama status',
|
||||
};
|
||||
}
|
||||
|
||||
return {
|
||||
success: true,
|
||||
data: result.data as OllamaStatus,
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to check Ollama status',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// List all Ollama models
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.OLLAMA_LIST_MODELS,
|
||||
async (_, baseUrl?: string): Promise<IPCResult<{ models: OllamaModel[]; count: number }>> => {
|
||||
try {
|
||||
const result = await executeOllamaDetector('list-models', baseUrl);
|
||||
|
||||
if (!result.success) {
|
||||
return {
|
||||
success: false,
|
||||
error: result.error || 'Failed to list Ollama models',
|
||||
};
|
||||
}
|
||||
|
||||
const data = result.data as { models: OllamaModel[]; count: number; url: string };
|
||||
return {
|
||||
success: true,
|
||||
data: {
|
||||
models: data.models,
|
||||
count: data.count,
|
||||
},
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to list Ollama models',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// List only embedding models from Ollama
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.OLLAMA_LIST_EMBEDDING_MODELS,
|
||||
async (
|
||||
_,
|
||||
baseUrl?: string
|
||||
): Promise<IPCResult<{ embedding_models: OllamaEmbeddingModel[]; count: number }>> => {
|
||||
try {
|
||||
const result = await executeOllamaDetector('list-embedding-models', baseUrl);
|
||||
|
||||
if (!result.success) {
|
||||
return {
|
||||
success: false,
|
||||
error: result.error || 'Failed to list Ollama embedding models',
|
||||
};
|
||||
}
|
||||
|
||||
const data = result.data as {
|
||||
embedding_models: OllamaEmbeddingModel[];
|
||||
count: number;
|
||||
url: string;
|
||||
};
|
||||
return {
|
||||
success: true,
|
||||
data: {
|
||||
embedding_models: data.embedding_models,
|
||||
count: data.count,
|
||||
},
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to list embedding models',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
// Pull (download) an Ollama model
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.OLLAMA_PULL_MODEL,
|
||||
async (
|
||||
_,
|
||||
modelName: string,
|
||||
baseUrl?: string
|
||||
): Promise<IPCResult<OllamaPullResult>> => {
|
||||
try {
|
||||
const pythonCmd = findPythonCommand();
|
||||
if (!pythonCmd) {
|
||||
return { success: false, error: 'Python not found' };
|
||||
}
|
||||
|
||||
// Find the ollama_model_detector.py script
|
||||
const possiblePaths = [
|
||||
path.resolve(__dirname, '..', '..', '..', 'auto-claude', 'ollama_model_detector.py'),
|
||||
path.resolve(process.cwd(), 'auto-claude', 'ollama_model_detector.py'),
|
||||
path.resolve(process.cwd(), '..', 'auto-claude', 'ollama_model_detector.py'),
|
||||
];
|
||||
|
||||
let scriptPath: string | null = null;
|
||||
for (const p of possiblePaths) {
|
||||
if (fs.existsSync(p)) {
|
||||
scriptPath = p;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (!scriptPath) {
|
||||
return { success: false, error: 'ollama_model_detector.py script not found' };
|
||||
}
|
||||
|
||||
const [pythonExe, baseArgs] = parsePythonCommand(pythonCmd);
|
||||
const args = [...baseArgs, scriptPath, 'pull-model', modelName];
|
||||
|
||||
return new Promise((resolve) => {
|
||||
const proc = spawn(pythonExe, args, {
|
||||
stdio: ['ignore', 'pipe', 'pipe'],
|
||||
timeout: 600000, // 10 minute timeout for large models
|
||||
});
|
||||
|
||||
let stdout = '';
|
||||
let stderr = '';
|
||||
|
||||
proc.stdout.on('data', (data) => {
|
||||
stdout += data.toString();
|
||||
});
|
||||
|
||||
proc.stderr.on('data', (data) => {
|
||||
stderr += data.toString();
|
||||
// Could emit progress events here in the future
|
||||
});
|
||||
|
||||
proc.on('close', (code) => {
|
||||
if (code === 0 && stdout) {
|
||||
try {
|
||||
const result = JSON.parse(stdout);
|
||||
if (result.success) {
|
||||
resolve({
|
||||
success: true,
|
||||
data: result.data as OllamaPullResult,
|
||||
});
|
||||
} else {
|
||||
resolve({
|
||||
success: false,
|
||||
error: result.error || 'Failed to pull model',
|
||||
});
|
||||
}
|
||||
} catch {
|
||||
resolve({ success: false, error: `Invalid JSON: ${stdout}` });
|
||||
}
|
||||
} else {
|
||||
resolve({ success: false, error: stderr || `Exit code ${code}` });
|
||||
}
|
||||
});
|
||||
|
||||
proc.on('error', (err) => {
|
||||
resolve({ success: false, error: err.message });
|
||||
});
|
||||
});
|
||||
} catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error instanceof Error ? error.message : 'Failed to pull model',
|
||||
};
|
||||
}
|
||||
}
|
||||
);
|
||||
}
|
||||
@@ -26,6 +26,7 @@ import { changelogService } from '../changelog-service';
|
||||
import { insightsService } from '../insights-service';
|
||||
import { titleGenerator } from '../title-generator';
|
||||
import type { BrowserWindow } from 'electron';
|
||||
import { getEffectiveSourcePath } from '../updater/path-resolver';
|
||||
|
||||
// ============================================
|
||||
// Git Helper Functions
|
||||
@@ -102,93 +103,6 @@ function detectMainBranch(projectPath: string): string | null {
|
||||
|
||||
const settingsPath = path.join(app.getPath('userData'), 'settings.json');
|
||||
|
||||
/**
|
||||
* Auto-detect the auto-claude source path relative to the app location.
|
||||
* Works across platforms (macOS, Windows, Linux) in both dev and production modes.
|
||||
*/
|
||||
const detectAutoBuildSourcePath = (): string | null => {
|
||||
const possiblePaths: string[] = [];
|
||||
|
||||
// Development mode paths
|
||||
if (is.dev) {
|
||||
// In dev, __dirname is typically auto-claude-ui/out/main
|
||||
// We need to go up to the project root to find auto-claude/
|
||||
possiblePaths.push(
|
||||
path.resolve(__dirname, '..', '..', '..', 'auto-claude'), // From out/main up 3 levels
|
||||
path.resolve(__dirname, '..', '..', 'auto-claude'), // From out/main up 2 levels
|
||||
path.resolve(process.cwd(), 'auto-claude'), // From cwd (project root)
|
||||
path.resolve(process.cwd(), '..', 'auto-claude') // From cwd parent (if running from auto-claude-ui/)
|
||||
);
|
||||
} else {
|
||||
// Production mode paths (packaged app)
|
||||
// On Windows/Linux/macOS, the app might be installed anywhere
|
||||
// We check common locations relative to the app bundle
|
||||
const appPath = app.getAppPath();
|
||||
possiblePaths.push(
|
||||
path.resolve(appPath, '..', 'auto-claude'), // Sibling to app
|
||||
path.resolve(appPath, '..', '..', 'auto-claude'), // Up 2 from app
|
||||
path.resolve(appPath, '..', '..', '..', 'auto-claude'), // Up 3 from app
|
||||
path.resolve(process.resourcesPath, '..', 'auto-claude'), // Relative to resources
|
||||
path.resolve(process.resourcesPath, '..', '..', 'auto-claude')
|
||||
);
|
||||
}
|
||||
|
||||
// Add process.cwd() as last resort on all platforms
|
||||
possiblePaths.push(path.resolve(process.cwd(), 'auto-claude'));
|
||||
|
||||
// Enable debug logging with DEBUG=1
|
||||
const debug = process.env.DEBUG === '1' || process.env.DEBUG === 'true';
|
||||
|
||||
if (debug) {
|
||||
console.warn('[project-handlers:detectAutoBuildSourcePath] Platform:', process.platform);
|
||||
console.warn('[project-handlers:detectAutoBuildSourcePath] Is dev:', is.dev);
|
||||
console.warn('[project-handlers:detectAutoBuildSourcePath] __dirname:', __dirname);
|
||||
console.warn('[project-handlers:detectAutoBuildSourcePath] app.getAppPath():', app.getAppPath());
|
||||
console.warn('[project-handlers:detectAutoBuildSourcePath] process.cwd():', process.cwd());
|
||||
console.warn('[project-handlers:detectAutoBuildSourcePath] Checking paths:', possiblePaths);
|
||||
}
|
||||
|
||||
for (const p of possiblePaths) {
|
||||
// Use requirements.txt as marker - it always exists in auto-claude source
|
||||
const markerPath = path.join(p, 'requirements.txt');
|
||||
const exists = existsSync(p) && existsSync(markerPath);
|
||||
|
||||
if (debug) {
|
||||
console.warn(`[project-handlers:detectAutoBuildSourcePath] Checking ${p}: ${exists ? '✓ FOUND' : '✗ not found'}`);
|
||||
}
|
||||
|
||||
if (exists) {
|
||||
console.warn(`[project-handlers:detectAutoBuildSourcePath] Auto-detected source path: ${p}`);
|
||||
return p;
|
||||
}
|
||||
}
|
||||
|
||||
console.warn('[project-handlers:detectAutoBuildSourcePath] Could not auto-detect Auto Claude source path.');
|
||||
console.warn('[project-handlers:detectAutoBuildSourcePath] Set DEBUG=1 environment variable for detailed path checking.');
|
||||
return null;
|
||||
};
|
||||
|
||||
/**
|
||||
* Get the configured auto-claude source path from settings, or auto-detect
|
||||
*/
|
||||
const getAutoBuildSourcePath = (): string | null => {
|
||||
// First check if manually configured
|
||||
if (existsSync(settingsPath)) {
|
||||
try {
|
||||
const content = readFileSync(settingsPath, 'utf-8');
|
||||
const settings = JSON.parse(content);
|
||||
if (settings.autoBuildPath && existsSync(settings.autoBuildPath)) {
|
||||
return settings.autoBuildPath;
|
||||
}
|
||||
} catch {
|
||||
// Fall through to auto-detect
|
||||
}
|
||||
}
|
||||
|
||||
// Auto-detect from app location
|
||||
return detectAutoBuildSourcePath();
|
||||
};
|
||||
|
||||
/**
|
||||
* Configure all Python-dependent services with the managed Python path
|
||||
*/
|
||||
@@ -211,7 +125,7 @@ const initializePythonEnvironment = async (
|
||||
pythonEnvManager: PythonEnvManager,
|
||||
agentManager: AgentManager
|
||||
): Promise<PythonEnvStatus> => {
|
||||
const autoBuildSource = getAutoBuildSourcePath();
|
||||
const autoBuildSource = getEffectiveSourcePath();
|
||||
if (!autoBuildSource) {
|
||||
console.warn('[IPC] Auto-build source not found, skipping Python env init');
|
||||
return {
|
||||
@@ -304,6 +218,31 @@ export function registerProjectHandlers(
|
||||
}
|
||||
);
|
||||
|
||||
// ============================================
|
||||
// Tab State Operations (persisted in main process)
|
||||
// ============================================
|
||||
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.TAB_STATE_GET,
|
||||
async (): Promise<IPCResult<{ openProjectIds: string[]; activeProjectId: string | null; tabOrder: string[] }>> => {
|
||||
const tabState = projectStore.getTabState();
|
||||
console.log('[IPC] TAB_STATE_GET returning:', tabState);
|
||||
return { success: true, data: tabState };
|
||||
}
|
||||
);
|
||||
|
||||
ipcMain.handle(
|
||||
IPC_CHANNELS.TAB_STATE_SAVE,
|
||||
async (
|
||||
_,
|
||||
tabState: { openProjectIds: string[]; activeProjectId: string | null; tabOrder: string[] }
|
||||
): Promise<IPCResult> => {
|
||||
console.log('[IPC] TAB_STATE_SAVE called with:', tabState);
|
||||
projectStore.saveTabState(tabState);
|
||||
return { success: true };
|
||||
}
|
||||
);
|
||||
|
||||
// ============================================
|
||||
// Project Initialization Operations
|
||||
// ============================================
|
||||
|
||||
@@ -874,4 +874,3 @@ ${(feature.acceptance_criteria || []).map((c: string) => `- [ ] ${c}`).join('\n'
|
||||
return { success: true, data: memories };
|
||||
}
|
||||
);
|
||||
|
||||
|
||||
@@ -1172,4 +1172,3 @@ ${idea.rationale}
|
||||
return { success: false, error: 'Failed to rename session' };
|
||||
}
|
||||
);
|
||||
|
||||
|
||||
@@ -1650,4 +1650,3 @@ ${issue.body || 'No description provided.'}
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
|
||||
@@ -1494,4 +1494,3 @@
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
|
||||
@@ -2,7 +2,7 @@ import { ipcMain, BrowserWindow } from 'electron';
|
||||
import { IPC_CHANNELS, AUTO_BUILD_PATHS } from '../../../shared/constants';
|
||||
import type { IPCResult, WorktreeStatus, WorktreeDiff, WorktreeDiffFile, WorktreeMergeResult, WorktreeDiscardResult, WorktreeListResult, WorktreeListItem } from '../../../shared/types';
|
||||
import path from 'path';
|
||||
import { existsSync, readdirSync, statSync } from 'fs';
|
||||
import { existsSync, readdirSync, statSync, readFileSync } from 'fs';
|
||||
import { execSync, spawn, spawnSync } from 'child_process';
|
||||
import { projectStore } from '../../project-store';
|
||||
import { PythonEnvManager } from '../../python-env-manager';
|
||||
@@ -11,6 +11,26 @@ import { getProfileEnv } from '../../rate-limit-detector';
|
||||
import { findTaskAndProject } from './shared';
|
||||
import { findPythonCommand, parsePythonCommand } from '../../python-detector';
|
||||
|
||||
/**
|
||||
* Read the stored base branch from task_metadata.json
|
||||
* This is the branch the task was created from (set by user during task creation)
|
||||
*/
|
||||
function getTaskBaseBranch(specDir: string): string | undefined {
|
||||
try {
|
||||
const metadataPath = path.join(specDir, 'task_metadata.json');
|
||||
if (existsSync(metadataPath)) {
|
||||
const metadata = JSON.parse(readFileSync(metadataPath, 'utf-8'));
|
||||
// Return baseBranch if explicitly set (not the __project_default__ marker)
|
||||
if (metadata.baseBranch && metadata.baseBranch !== '__project_default__') {
|
||||
return metadata.baseBranch;
|
||||
}
|
||||
}
|
||||
} catch (e) {
|
||||
console.warn('[getTaskBaseBranch] Failed to read task metadata:', e);
|
||||
}
|
||||
return undefined;
|
||||
}
|
||||
|
||||
/**
|
||||
* Register worktree management handlers
|
||||
*/
|
||||
@@ -61,12 +81,13 @@ export function registerWorktreeHandlers(
|
||||
baseBranch = 'main';
|
||||
}
|
||||
|
||||
// Get commit count
|
||||
// Get commit count (cross-platform - no shell syntax)
|
||||
let commitCount = 0;
|
||||
try {
|
||||
const countOutput = execSync(`git rev-list --count ${baseBranch}..HEAD 2>/dev/null || echo 0`, {
|
||||
const countOutput = execSync(`git rev-list --count ${baseBranch}..HEAD`, {
|
||||
cwd: worktreePath,
|
||||
encoding: 'utf-8'
|
||||
encoding: 'utf-8',
|
||||
stdio: ['pipe', 'pipe', 'pipe']
|
||||
}).trim();
|
||||
commitCount = parseInt(countOutput, 10) || 0;
|
||||
} catch {
|
||||
@@ -78,10 +99,12 @@ export function registerWorktreeHandlers(
|
||||
let additions = 0;
|
||||
let deletions = 0;
|
||||
|
||||
let diffStat = '';
|
||||
try {
|
||||
const diffStat = execSync(`git diff --stat ${baseBranch}...HEAD 2>/dev/null || echo ""`, {
|
||||
diffStat = execSync(`git diff --stat ${baseBranch}...HEAD`, {
|
||||
cwd: worktreePath,
|
||||
encoding: 'utf-8'
|
||||
encoding: 'utf-8',
|
||||
stdio: ['pipe', 'pipe', 'pipe']
|
||||
}).trim();
|
||||
|
||||
// Parse the summary line (e.g., "3 files changed, 50 insertions(+), 10 deletions(-)")
|
||||
@@ -159,17 +182,21 @@ export function registerWorktreeHandlers(
|
||||
// Get the diff with file stats
|
||||
const files: WorktreeDiffFile[] = [];
|
||||
|
||||
let numstat = '';
|
||||
let nameStatus = '';
|
||||
try {
|
||||
// Get numstat for additions/deletions per file
|
||||
const numstat = execSync(`git diff --numstat ${baseBranch}...HEAD 2>/dev/null || echo ""`, {
|
||||
// Get numstat for additions/deletions per file (cross-platform)
|
||||
numstat = execSync(`git diff --numstat ${baseBranch}...HEAD`, {
|
||||
cwd: worktreePath,
|
||||
encoding: 'utf-8'
|
||||
encoding: 'utf-8',
|
||||
stdio: ['pipe', 'pipe', 'pipe']
|
||||
}).trim();
|
||||
|
||||
// Get name-status for file status
|
||||
const nameStatus = execSync(`git diff --name-status ${baseBranch}...HEAD 2>/dev/null || echo ""`, {
|
||||
// Get name-status for file status (cross-platform)
|
||||
nameStatus = execSync(`git diff --name-status ${baseBranch}...HEAD`, {
|
||||
cwd: worktreePath,
|
||||
encoding: 'utf-8'
|
||||
encoding: 'utf-8',
|
||||
stdio: ['pipe', 'pipe', 'pipe']
|
||||
}).trim();
|
||||
|
||||
// Parse name-status to get file statuses
|
||||
@@ -320,6 +347,13 @@ export function registerWorktreeHandlers(
|
||||
args.push('--no-commit');
|
||||
}
|
||||
|
||||
// Add --base-branch if task was created with a specific base branch
|
||||
const taskBaseBranch = getTaskBaseBranch(specDir);
|
||||
if (taskBaseBranch) {
|
||||
args.push('--base-branch', taskBaseBranch);
|
||||
debug('Using stored base branch:', taskBaseBranch);
|
||||
}
|
||||
|
||||
const pythonPath = pythonEnvManager.getPythonPath() || findPythonCommand() || 'python';
|
||||
debug('Running command:', pythonPath, args.join(' '));
|
||||
debug('Working directory:', sourcePath);
|
||||
@@ -332,7 +366,7 @@ export function registerWorktreeHandlers(
|
||||
});
|
||||
|
||||
return new Promise((resolve) => {
|
||||
const MERGE_TIMEOUT_MS = 120000; // 2 minutes timeout for merge operations
|
||||
const MERGE_TIMEOUT_MS = 600000; // 10 minutes timeout for AI merge operations with many files
|
||||
let timeoutId: NodeJS.Timeout | null = null;
|
||||
let resolved = false;
|
||||
|
||||
@@ -446,14 +480,17 @@ export function registerWorktreeHandlers(
|
||||
const specBranch = `auto-claude/${task.specId}`;
|
||||
try {
|
||||
// Check if current branch contains all commits from spec branch
|
||||
const mergeBaseResult = execSync(
|
||||
`git merge-base --is-ancestor ${specBranch} HEAD 2>/dev/null && echo "merged" || echo "not-merged"`,
|
||||
{ cwd: project.path, encoding: 'utf-8' }
|
||||
).trim();
|
||||
mergeAlreadyCommitted = mergeBaseResult === 'merged';
|
||||
// git merge-base --is-ancestor returns exit code 0 if true, 1 if false
|
||||
execSync(
|
||||
`git merge-base --is-ancestor ${specBranch} HEAD`,
|
||||
{ cwd: project.path, encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] }
|
||||
);
|
||||
// If we reach here, the command succeeded (exit code 0) - branch is merged
|
||||
mergeAlreadyCommitted = true;
|
||||
debug('Merge already committed check:', mergeAlreadyCommitted);
|
||||
} catch {
|
||||
// Branch may not exist or other error - assume not merged
|
||||
// Exit code 1 means not merged, or branch may not exist
|
||||
mergeAlreadyCommitted = false;
|
||||
debug('Could not check merge status, assuming not merged');
|
||||
}
|
||||
}
|
||||
@@ -659,6 +696,7 @@ export function registerWorktreeHandlers(
|
||||
}
|
||||
|
||||
const runScript = path.join(sourcePath, 'run.py');
|
||||
const specDir = path.join(project.path, project.autoBuildPath || '.auto-claude', 'specs', task.specId);
|
||||
const args = [
|
||||
runScript,
|
||||
'--spec', task.specId,
|
||||
@@ -666,6 +704,13 @@ export function registerWorktreeHandlers(
|
||||
'--merge-preview'
|
||||
];
|
||||
|
||||
// Add --base-branch if task was created with a specific base branch
|
||||
const taskBaseBranch = getTaskBaseBranch(specDir);
|
||||
if (taskBaseBranch) {
|
||||
args.push('--base-branch', taskBaseBranch);
|
||||
console.warn('[IPC] Using stored base branch for preview:', taskBaseBranch);
|
||||
}
|
||||
|
||||
const pythonPath = pythonEnvManager.getPythonPath() || findPythonCommand() || 'python';
|
||||
console.warn('[IPC] Running merge preview:', pythonPath, args.join(' '));
|
||||
|
||||
@@ -892,27 +937,30 @@ export function registerWorktreeHandlers(
|
||||
baseBranch = 'main';
|
||||
}
|
||||
|
||||
// Get commit count
|
||||
// Get commit count (cross-platform - no shell syntax)
|
||||
let commitCount = 0;
|
||||
try {
|
||||
const countOutput = execSync(`git rev-list --count ${baseBranch}..HEAD 2>/dev/null || echo 0`, {
|
||||
const countOutput = execSync(`git rev-list --count ${baseBranch}..HEAD`, {
|
||||
cwd: entryPath,
|
||||
encoding: 'utf-8'
|
||||
encoding: 'utf-8',
|
||||
stdio: ['pipe', 'pipe', 'pipe']
|
||||
}).trim();
|
||||
commitCount = parseInt(countOutput, 10) || 0;
|
||||
} catch {
|
||||
commitCount = 0;
|
||||
}
|
||||
|
||||
// Get diff stats
|
||||
// Get diff stats (cross-platform - no shell syntax)
|
||||
let filesChanged = 0;
|
||||
let additions = 0;
|
||||
let deletions = 0;
|
||||
let diffStat = '';
|
||||
|
||||
try {
|
||||
const diffStat = execSync(`git diff --shortstat ${baseBranch}...HEAD 2>/dev/null || echo ""`, {
|
||||
diffStat = execSync(`git diff --shortstat ${baseBranch}...HEAD`, {
|
||||
cwd: entryPath,
|
||||
encoding: 'utf-8'
|
||||
encoding: 'utf-8',
|
||||
stdio: ['pipe', 'pipe', 'pipe']
|
||||
}).trim();
|
||||
|
||||
const filesMatch = diffStat.match(/(\d+) files? changed/);
|
||||
|
||||
@@ -0,0 +1,631 @@
|
||||
/**
|
||||
* Memory Service
|
||||
*
|
||||
* Queries the LadybugDB graph database for memories stored by Graphiti.
|
||||
* Uses Python subprocess to communicate with the embedded database.
|
||||
*
|
||||
* LadybugDB stores data in Kuzu format at ~/.auto-claude/memories/<database>/
|
||||
*/
|
||||
|
||||
import { spawn } from 'child_process';
|
||||
import * as path from 'path';
|
||||
import * as os from 'os';
|
||||
import * as fs from 'fs';
|
||||
import { app } from 'electron';
|
||||
import { findPythonCommand, parsePythonCommand } from './python-detector';
|
||||
import type { MemoryEpisode } from '../shared/types';
|
||||
|
||||
interface MemoryServiceConfig {
|
||||
dbPath: string;
|
||||
database: string;
|
||||
}
|
||||
|
||||
// Embedder configuration for semantic search
|
||||
export interface EmbedderConfig {
|
||||
provider: 'openai' | 'google' | 'ollama' | 'voyage' | 'azure_openai';
|
||||
// OpenAI
|
||||
openaiApiKey?: string;
|
||||
openaiEmbeddingModel?: string;
|
||||
// Google AI
|
||||
googleApiKey?: string;
|
||||
googleEmbeddingModel?: string;
|
||||
// Ollama
|
||||
ollamaBaseUrl?: string;
|
||||
ollamaEmbeddingModel?: string;
|
||||
ollamaEmbeddingDim?: number;
|
||||
// Voyage AI
|
||||
voyageApiKey?: string;
|
||||
voyageEmbeddingModel?: string;
|
||||
// Azure OpenAI
|
||||
azureOpenaiApiKey?: string;
|
||||
azureOpenaiBaseUrl?: string;
|
||||
azureOpenaiEmbeddingDeployment?: string;
|
||||
}
|
||||
|
||||
interface SemanticSearchResult extends MemoryQueryResult {
|
||||
search_type: 'semantic' | 'keyword';
|
||||
embedder?: string;
|
||||
}
|
||||
|
||||
interface QueryResult {
|
||||
success: boolean;
|
||||
data?: unknown;
|
||||
error?: string;
|
||||
}
|
||||
|
||||
interface MemoryQueryResult {
|
||||
memories: Array<{
|
||||
id: string;
|
||||
name: string;
|
||||
type: string;
|
||||
timestamp: string;
|
||||
content: string;
|
||||
description?: string;
|
||||
group_id?: string;
|
||||
session_number?: number;
|
||||
score?: number;
|
||||
}>;
|
||||
count: number;
|
||||
query?: string;
|
||||
}
|
||||
|
||||
interface StatusResult {
|
||||
available: boolean;
|
||||
ladybugInstalled: boolean;
|
||||
databasePath: string;
|
||||
database: string;
|
||||
databaseExists: boolean;
|
||||
connected?: boolean;
|
||||
databases?: string[];
|
||||
error?: string | null;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the default database path
|
||||
*/
|
||||
export function getDefaultDbPath(): string {
|
||||
return path.join(os.homedir(), '.auto-claude', 'memories');
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the path to the query_memory.py script
|
||||
*/
|
||||
function getQueryScriptPath(): string | null {
|
||||
// Look for the script in auto-claude directory (sibling to auto-claude-ui)
|
||||
const possiblePaths = [
|
||||
// Dev mode: from dist/main -> ../../auto-claude
|
||||
path.resolve(__dirname, '..', '..', '..', 'auto-claude', 'query_memory.py'),
|
||||
// Packaged app: from app.getAppPath() (handles asar and resources correctly)
|
||||
path.resolve(app.getAppPath(), '..', 'auto-claude', 'query_memory.py'),
|
||||
// Alternative: from app root
|
||||
path.resolve(process.cwd(), 'auto-claude', 'query_memory.py'),
|
||||
// If running from repo root
|
||||
path.resolve(process.cwd(), '..', 'auto-claude', 'query_memory.py'),
|
||||
];
|
||||
|
||||
for (const p of possiblePaths) {
|
||||
if (fs.existsSync(p)) {
|
||||
return p;
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute a Python memory query command
|
||||
*/
|
||||
async function executeQuery(
|
||||
command: string,
|
||||
args: string[],
|
||||
timeout: number = 10000
|
||||
): Promise<QueryResult> {
|
||||
const pythonCmd = findPythonCommand();
|
||||
if (!pythonCmd) {
|
||||
return { success: false, error: 'Python not found' };
|
||||
}
|
||||
|
||||
const scriptPath = getQueryScriptPath();
|
||||
if (!scriptPath) {
|
||||
return { success: false, error: 'query_memory.py script not found' };
|
||||
}
|
||||
|
||||
const [pythonExe, baseArgs] = parsePythonCommand(pythonCmd);
|
||||
|
||||
return new Promise((resolve) => {
|
||||
const fullArgs = [...baseArgs, scriptPath, command, ...args];
|
||||
const proc = spawn(pythonExe, fullArgs, {
|
||||
stdio: ['ignore', 'pipe', 'pipe'],
|
||||
timeout,
|
||||
});
|
||||
|
||||
let stdout = '';
|
||||
let stderr = '';
|
||||
|
||||
proc.stdout.on('data', (data) => {
|
||||
stdout += data.toString();
|
||||
});
|
||||
|
||||
proc.stderr.on('data', (data) => {
|
||||
stderr += data.toString();
|
||||
});
|
||||
|
||||
proc.on('close', (code) => {
|
||||
if (code === 0 && stdout) {
|
||||
try {
|
||||
const result = JSON.parse(stdout);
|
||||
resolve(result);
|
||||
} catch {
|
||||
resolve({ success: false, error: `Invalid JSON response: ${stdout}` });
|
||||
}
|
||||
} else {
|
||||
resolve({
|
||||
success: false,
|
||||
error: stderr || `Process exited with code ${code}`,
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
proc.on('error', (err) => {
|
||||
resolve({ success: false, error: err.message });
|
||||
});
|
||||
|
||||
// Handle timeout
|
||||
setTimeout(() => {
|
||||
proc.kill();
|
||||
resolve({ success: false, error: 'Query timed out' });
|
||||
}, timeout);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute semantic search with embedder configuration passed via environment
|
||||
*/
|
||||
async function executeSemanticQuery(
|
||||
args: string[],
|
||||
embedderConfig: EmbedderConfig,
|
||||
timeout: number = 30000 // Longer timeout for embedding operations
|
||||
): Promise<QueryResult> {
|
||||
const pythonCmd = findPythonCommand();
|
||||
if (!pythonCmd) {
|
||||
return { success: false, error: 'Python not found' };
|
||||
}
|
||||
|
||||
const scriptPath = getQueryScriptPath();
|
||||
if (!scriptPath) {
|
||||
return { success: false, error: 'query_memory.py script not found' };
|
||||
}
|
||||
|
||||
const [pythonExe, baseArgs] = parsePythonCommand(pythonCmd);
|
||||
|
||||
// Build environment with embedder configuration
|
||||
const env: Record<string, string | undefined> = { ...process.env };
|
||||
|
||||
// Set the embedder provider
|
||||
env.GRAPHITI_EMBEDDER_PROVIDER = embedderConfig.provider;
|
||||
|
||||
// Provider-specific configuration
|
||||
switch (embedderConfig.provider) {
|
||||
case 'openai':
|
||||
if (embedderConfig.openaiApiKey) {
|
||||
env.OPENAI_API_KEY = embedderConfig.openaiApiKey;
|
||||
}
|
||||
if (embedderConfig.openaiEmbeddingModel) {
|
||||
env.OPENAI_EMBEDDING_MODEL = embedderConfig.openaiEmbeddingModel;
|
||||
}
|
||||
break;
|
||||
|
||||
case 'google':
|
||||
if (embedderConfig.googleApiKey) {
|
||||
env.GOOGLE_API_KEY = embedderConfig.googleApiKey;
|
||||
}
|
||||
if (embedderConfig.googleEmbeddingModel) {
|
||||
env.GOOGLE_EMBEDDING_MODEL = embedderConfig.googleEmbeddingModel;
|
||||
}
|
||||
break;
|
||||
|
||||
case 'ollama':
|
||||
if (embedderConfig.ollamaBaseUrl) {
|
||||
env.OLLAMA_BASE_URL = embedderConfig.ollamaBaseUrl;
|
||||
}
|
||||
if (embedderConfig.ollamaEmbeddingModel) {
|
||||
env.OLLAMA_EMBEDDING_MODEL = embedderConfig.ollamaEmbeddingModel;
|
||||
}
|
||||
if (embedderConfig.ollamaEmbeddingDim) {
|
||||
env.OLLAMA_EMBEDDING_DIM = String(embedderConfig.ollamaEmbeddingDim);
|
||||
}
|
||||
break;
|
||||
|
||||
case 'voyage':
|
||||
if (embedderConfig.voyageApiKey) {
|
||||
env.VOYAGE_API_KEY = embedderConfig.voyageApiKey;
|
||||
}
|
||||
if (embedderConfig.voyageEmbeddingModel) {
|
||||
env.VOYAGE_EMBEDDING_MODEL = embedderConfig.voyageEmbeddingModel;
|
||||
}
|
||||
break;
|
||||
|
||||
case 'azure_openai':
|
||||
if (embedderConfig.azureOpenaiApiKey) {
|
||||
env.AZURE_OPENAI_API_KEY = embedderConfig.azureOpenaiApiKey;
|
||||
}
|
||||
if (embedderConfig.azureOpenaiBaseUrl) {
|
||||
env.AZURE_OPENAI_BASE_URL = embedderConfig.azureOpenaiBaseUrl;
|
||||
}
|
||||
if (embedderConfig.azureOpenaiEmbeddingDeployment) {
|
||||
env.AZURE_OPENAI_EMBEDDING_DEPLOYMENT = embedderConfig.azureOpenaiEmbeddingDeployment;
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
return new Promise((resolve) => {
|
||||
const fullArgs = [...baseArgs, scriptPath, 'semantic-search', ...args];
|
||||
const proc = spawn(pythonExe, fullArgs, {
|
||||
stdio: ['ignore', 'pipe', 'pipe'],
|
||||
env,
|
||||
timeout,
|
||||
});
|
||||
|
||||
let stdout = '';
|
||||
let stderr = '';
|
||||
|
||||
proc.stdout.on('data', (data) => {
|
||||
stdout += data.toString();
|
||||
});
|
||||
|
||||
proc.stderr.on('data', (data) => {
|
||||
stderr += data.toString();
|
||||
});
|
||||
|
||||
proc.on('close', (code) => {
|
||||
if (code === 0 && stdout) {
|
||||
try {
|
||||
const result = JSON.parse(stdout);
|
||||
resolve(result);
|
||||
} catch {
|
||||
resolve({ success: false, error: `Invalid JSON response: ${stdout}` });
|
||||
}
|
||||
} else {
|
||||
resolve({
|
||||
success: false,
|
||||
error: stderr || `Process exited with code ${code}`,
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
proc.on('error', (err) => {
|
||||
resolve({ success: false, error: err.message });
|
||||
});
|
||||
|
||||
setTimeout(() => {
|
||||
proc.kill();
|
||||
resolve({ success: false, error: 'Semantic search timed out' });
|
||||
}, timeout);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Memory Service for querying graph memories from LadybugDB
|
||||
*/
|
||||
export class MemoryService {
|
||||
private config: MemoryServiceConfig;
|
||||
|
||||
constructor(config: MemoryServiceConfig) {
|
||||
this.config = config;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the full path to the database
|
||||
*/
|
||||
private getDbFullPath(): string {
|
||||
return path.join(this.config.dbPath, this.config.database);
|
||||
}
|
||||
|
||||
/**
|
||||
* Check if the database exists
|
||||
*/
|
||||
databaseExists(): boolean {
|
||||
const dbPath = this.getDbFullPath();
|
||||
return fs.existsSync(dbPath);
|
||||
}
|
||||
|
||||
/**
|
||||
* List all available databases
|
||||
*/
|
||||
listDatabases(): string[] {
|
||||
try {
|
||||
const basePath = this.config.dbPath;
|
||||
if (!fs.existsSync(basePath)) {
|
||||
return [];
|
||||
}
|
||||
|
||||
return fs.readdirSync(basePath).filter((name) => {
|
||||
if (name.startsWith('.')) return false;
|
||||
return true; // Include both files and directories
|
||||
});
|
||||
} catch (error) {
|
||||
console.error('Failed to list databases:', error);
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Query episodic memories from the database
|
||||
*/
|
||||
async getEpisodicMemories(limit: number = 20): Promise<MemoryEpisode[]> {
|
||||
const result = await executeQuery('get-memories', [
|
||||
this.config.dbPath,
|
||||
this.config.database,
|
||||
'--limit',
|
||||
String(limit),
|
||||
]);
|
||||
|
||||
if (!result.success || !result.data) {
|
||||
console.error('Failed to get memories:', result.error);
|
||||
return [];
|
||||
}
|
||||
|
||||
const data = result.data as MemoryQueryResult;
|
||||
return data.memories.map((m) => ({
|
||||
id: m.id,
|
||||
type: this.mapMemoryType(m.type),
|
||||
timestamp: m.timestamp,
|
||||
content: m.content,
|
||||
session_number: m.session_number,
|
||||
}));
|
||||
}
|
||||
|
||||
/**
|
||||
* Query entity memories (patterns, gotchas, etc.) from the database
|
||||
*/
|
||||
async getEntityMemories(limit: number = 20): Promise<MemoryEpisode[]> {
|
||||
const result = await executeQuery('get-entities', [
|
||||
this.config.dbPath,
|
||||
this.config.database,
|
||||
'--limit',
|
||||
String(limit),
|
||||
]);
|
||||
|
||||
if (!result.success || !result.data) {
|
||||
console.error('Failed to get entities:', result.error);
|
||||
return [];
|
||||
}
|
||||
|
||||
const data = result.data as { entities: MemoryQueryResult['memories']; count: number };
|
||||
return data.entities.map((e) => ({
|
||||
id: e.id,
|
||||
type: this.mapMemoryType(e.type),
|
||||
timestamp: e.timestamp,
|
||||
content: e.content,
|
||||
}));
|
||||
}
|
||||
|
||||
/**
|
||||
* Get all memories from the database
|
||||
*/
|
||||
async getAllMemories(limit: number = 20): Promise<MemoryEpisode[]> {
|
||||
const [episodic, entities] = await Promise.all([
|
||||
this.getEpisodicMemories(limit),
|
||||
this.getEntityMemories(limit),
|
||||
]);
|
||||
|
||||
const memories = [...episodic, ...entities];
|
||||
|
||||
// Sort by timestamp descending
|
||||
memories.sort((a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime());
|
||||
|
||||
return memories.slice(0, limit);
|
||||
}
|
||||
|
||||
/**
|
||||
* Search memories in the database (keyword search)
|
||||
*/
|
||||
async searchMemories(searchQuery: string, limit: number = 20): Promise<MemoryEpisode[]> {
|
||||
const result = await executeQuery('search', [
|
||||
this.config.dbPath,
|
||||
this.config.database,
|
||||
searchQuery,
|
||||
'--limit',
|
||||
String(limit),
|
||||
]);
|
||||
|
||||
if (!result.success || !result.data) {
|
||||
console.error('Failed to search memories:', result.error);
|
||||
return [];
|
||||
}
|
||||
|
||||
const data = result.data as MemoryQueryResult;
|
||||
return data.memories.map((m) => ({
|
||||
id: m.id,
|
||||
type: this.mapMemoryType(m.type),
|
||||
timestamp: m.timestamp,
|
||||
content: m.content,
|
||||
session_number: m.session_number,
|
||||
score: m.score,
|
||||
}));
|
||||
}
|
||||
|
||||
/**
|
||||
* Semantic search using embeddings
|
||||
*
|
||||
* Uses the configured embedder to create vector embeddings and perform
|
||||
* similarity search. Falls back to keyword search if embedder fails.
|
||||
*
|
||||
* @param searchQuery The search query
|
||||
* @param embedderConfig Configuration for the embedding provider
|
||||
* @param limit Maximum number of results
|
||||
* @returns Memories with relevance scores
|
||||
*/
|
||||
async searchMemoriesSemantic(
|
||||
searchQuery: string,
|
||||
embedderConfig: EmbedderConfig,
|
||||
limit: number = 20
|
||||
): Promise<{ memories: MemoryEpisode[]; searchType: 'semantic' | 'keyword' }> {
|
||||
const result = await executeSemanticQuery(
|
||||
[this.config.dbPath, this.config.database, searchQuery, '--limit', String(limit)],
|
||||
embedderConfig
|
||||
);
|
||||
|
||||
if (!result.success || !result.data) {
|
||||
console.error('Semantic search failed, falling back to keyword:', result.error);
|
||||
// Fall back to keyword search
|
||||
const memories = await this.searchMemories(searchQuery, limit);
|
||||
return { memories, searchType: 'keyword' };
|
||||
}
|
||||
|
||||
const data = result.data as SemanticSearchResult;
|
||||
const memories = data.memories.map((m) => ({
|
||||
id: m.id,
|
||||
type: this.mapMemoryType(m.type),
|
||||
timestamp: m.timestamp,
|
||||
content: m.content,
|
||||
session_number: m.session_number,
|
||||
score: m.score,
|
||||
}));
|
||||
|
||||
return {
|
||||
memories,
|
||||
searchType: data.search_type || 'semantic',
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Test connection to the database
|
||||
*/
|
||||
async testConnection(): Promise<{ success: boolean; message: string }> {
|
||||
const result = await executeQuery('get-status', [this.config.dbPath, this.config.database]);
|
||||
|
||||
if (!result.success) {
|
||||
return {
|
||||
success: false,
|
||||
message: result.error || 'Failed to check database status',
|
||||
};
|
||||
}
|
||||
|
||||
const data = result.data as StatusResult;
|
||||
|
||||
if (!data.available) {
|
||||
return {
|
||||
success: false,
|
||||
message: 'LadybugDB (real_ladybug) not installed. Requires Python 3.12+',
|
||||
};
|
||||
}
|
||||
|
||||
if (!data.databaseExists) {
|
||||
return {
|
||||
success: false,
|
||||
message: `Database not found at ${data.databasePath}/${data.database}`,
|
||||
};
|
||||
}
|
||||
|
||||
if (!data.connected) {
|
||||
return {
|
||||
success: false,
|
||||
message: data.error || 'Failed to connect to database',
|
||||
};
|
||||
}
|
||||
|
||||
const dbCount = data.databases?.length || 0;
|
||||
return {
|
||||
success: true,
|
||||
message: `Connected to LadybugDB with ${dbCount} databases`,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Close the database connection (no-op for subprocess model)
|
||||
*/
|
||||
async close(): Promise<void> {
|
||||
// No persistent connection to close with subprocess model
|
||||
}
|
||||
|
||||
/**
|
||||
* Map string type to MemoryEpisode type
|
||||
*/
|
||||
private mapMemoryType(type: string): MemoryEpisode['type'] {
|
||||
switch (type) {
|
||||
case 'session_insight':
|
||||
return 'session_insight';
|
||||
case 'pattern':
|
||||
return 'pattern';
|
||||
case 'gotcha':
|
||||
return 'gotcha';
|
||||
case 'codebase_discovery':
|
||||
return 'codebase_discovery';
|
||||
case 'task_outcome':
|
||||
return 'task_outcome';
|
||||
default:
|
||||
return 'session_insight';
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Singleton instance for reuse
|
||||
let serviceInstance: MemoryService | null = null;
|
||||
|
||||
/**
|
||||
* Get or create a Memory service instance
|
||||
*/
|
||||
export function getMemoryService(config: MemoryServiceConfig): MemoryService {
|
||||
if (
|
||||
!serviceInstance ||
|
||||
serviceInstance['config'].dbPath !== config.dbPath ||
|
||||
serviceInstance['config'].database !== config.database
|
||||
) {
|
||||
serviceInstance = new MemoryService(config);
|
||||
}
|
||||
return serviceInstance;
|
||||
}
|
||||
|
||||
/**
|
||||
* Close the singleton service instance
|
||||
*/
|
||||
export async function closeMemoryService(): Promise<void> {
|
||||
if (serviceInstance) {
|
||||
await serviceInstance.close();
|
||||
serviceInstance = null;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Check if Python with LadybugDB is available
|
||||
*/
|
||||
export function isKuzuAvailable(): boolean {
|
||||
// Check if Python is available
|
||||
const pythonCmd = findPythonCommand();
|
||||
if (!pythonCmd) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// Check if query script exists
|
||||
const scriptPath = getQueryScriptPath();
|
||||
return scriptPath !== null;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get memory service status
|
||||
*/
|
||||
export interface MemoryServiceStatus {
|
||||
kuzuInstalled: boolean;
|
||||
databasePath: string;
|
||||
databaseExists: boolean;
|
||||
databases: string[];
|
||||
}
|
||||
|
||||
export function getMemoryServiceStatus(dbPath?: string): MemoryServiceStatus {
|
||||
const basePath = dbPath || getDefaultDbPath();
|
||||
|
||||
const databases = fs.existsSync(basePath)
|
||||
? fs.readdirSync(basePath).filter((name) => !name.startsWith('.'))
|
||||
: [];
|
||||
|
||||
// Check if Python and script are available
|
||||
const pythonAvailable = findPythonCommand() !== null;
|
||||
const scriptAvailable = getQueryScriptPath() !== null;
|
||||
|
||||
return {
|
||||
kuzuInstalled: pythonAvailable && scriptAvailable,
|
||||
databasePath: basePath,
|
||||
databaseExists: databases.length > 0,
|
||||
databases,
|
||||
};
|
||||
}
|
||||
@@ -162,4 +162,3 @@ class NotificationService {
|
||||
|
||||
// Export singleton instance
|
||||
export const notificationService = new NotificationService();
|
||||
|
||||
|
||||
@@ -6,9 +6,16 @@ import type { Project, ProjectSettings, Task, TaskStatus, TaskMetadata, Implemen
|
||||
import { DEFAULT_PROJECT_SETTINGS, AUTO_BUILD_PATHS, getSpecsDir } from '../shared/constants';
|
||||
import { getAutoBuildPath, isInitialized } from './project-initializer';
|
||||
|
||||
interface TabState {
|
||||
openProjectIds: string[];
|
||||
activeProjectId: string | null;
|
||||
tabOrder: string[];
|
||||
}
|
||||
|
||||
interface StoreData {
|
||||
projects: Project[];
|
||||
settings: Record<string, unknown>;
|
||||
tabState?: TabState;
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -68,6 +75,14 @@ export class ProjectStore {
|
||||
// Check if project already exists
|
||||
const existing = this.data.projects.find((p) => p.path === projectPath);
|
||||
if (existing) {
|
||||
// Validate that .auto-claude folder still exists for existing project
|
||||
// If manually deleted, reset autoBuildPath so UI prompts for reinitialization
|
||||
if (existing.autoBuildPath && !isInitialized(existing.path)) {
|
||||
console.warn(`[ProjectStore] .auto-claude folder was deleted for project "${existing.name}" - resetting autoBuildPath`);
|
||||
existing.autoBuildPath = '';
|
||||
existing.updatedAt = new Date();
|
||||
this.save();
|
||||
}
|
||||
return existing;
|
||||
}
|
||||
|
||||
@@ -126,6 +141,34 @@ export class ProjectStore {
|
||||
return this.data.projects;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get tab state
|
||||
*/
|
||||
getTabState(): TabState {
|
||||
return this.data.tabState || {
|
||||
openProjectIds: [],
|
||||
activeProjectId: null,
|
||||
tabOrder: []
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Save tab state
|
||||
*/
|
||||
saveTabState(tabState: TabState): void {
|
||||
// Filter out any project IDs that no longer exist
|
||||
const validProjectIds = this.data.projects.map(p => p.id);
|
||||
this.data.tabState = {
|
||||
openProjectIds: tabState.openProjectIds.filter(id => validProjectIds.includes(id)),
|
||||
activeProjectId: tabState.activeProjectId && validProjectIds.includes(tabState.activeProjectId)
|
||||
? tabState.activeProjectId
|
||||
: null,
|
||||
tabOrder: tabState.tabOrder.filter(id => validProjectIds.includes(id))
|
||||
};
|
||||
console.log('[ProjectStore] Saving tab state:', this.data.tabState);
|
||||
this.save();
|
||||
}
|
||||
|
||||
/**
|
||||
* Validate all projects to ensure their .auto-claude folders still exist.
|
||||
* If a project has autoBuildPath set but the folder was deleted,
|
||||
|
||||
@@ -2,6 +2,7 @@ import { spawn, execSync } from 'child_process';
|
||||
import { existsSync } from 'fs';
|
||||
import path from 'path';
|
||||
import { EventEmitter } from 'events';
|
||||
import { app } from 'electron';
|
||||
|
||||
export interface PythonEnvStatus {
|
||||
ready: boolean;
|
||||
@@ -14,6 +15,9 @@ export interface PythonEnvStatus {
|
||||
/**
|
||||
* Manages the Python virtual environment for the auto-claude backend.
|
||||
* Automatically creates venv and installs dependencies if needed.
|
||||
*
|
||||
* On packaged apps (especially Linux AppImages), the bundled source is read-only,
|
||||
* so we create the venv in userData instead of inside the source directory.
|
||||
*/
|
||||
export class PythonEnvManager extends EventEmitter {
|
||||
private autoBuildSourcePath: string | null = null;
|
||||
@@ -21,16 +25,35 @@ export class PythonEnvManager extends EventEmitter {
|
||||
private isInitializing = false;
|
||||
private isReady = false;
|
||||
|
||||
/**
|
||||
* Get the path where the venv should be created.
|
||||
* For packaged apps, this is in userData to avoid read-only filesystem issues.
|
||||
* For development, this is inside the source directory.
|
||||
*/
|
||||
private getVenvBasePath(): string | null {
|
||||
if (!this.autoBuildSourcePath) return null;
|
||||
|
||||
// For packaged apps, put venv in userData (writable location)
|
||||
// This fixes Linux AppImage where resources are read-only
|
||||
if (app.isPackaged) {
|
||||
return path.join(app.getPath('userData'), 'python-venv');
|
||||
}
|
||||
|
||||
// Development mode - use source directory
|
||||
return path.join(this.autoBuildSourcePath, '.venv');
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the path to the venv Python executable
|
||||
*/
|
||||
private getVenvPythonPath(): string | null {
|
||||
if (!this.autoBuildSourcePath) return null;
|
||||
const venvPath = this.getVenvBasePath();
|
||||
if (!venvPath) return null;
|
||||
|
||||
const venvPython =
|
||||
process.platform === 'win32'
|
||||
? path.join(this.autoBuildSourcePath, '.venv', 'Scripts', 'python.exe')
|
||||
: path.join(this.autoBuildSourcePath, '.venv', 'bin', 'python');
|
||||
? path.join(venvPath, 'Scripts', 'python.exe')
|
||||
: path.join(venvPath, 'bin', 'python');
|
||||
|
||||
return venvPython;
|
||||
}
|
||||
@@ -142,10 +165,10 @@ export class PythonEnvManager extends EventEmitter {
|
||||
}
|
||||
|
||||
this.emit('status', 'Creating Python virtual environment...');
|
||||
console.warn('[PythonEnvManager] Creating venv with:', systemPython);
|
||||
const venvPath = this.getVenvBasePath()!;
|
||||
console.warn('[PythonEnvManager] Creating venv at:', venvPath, 'with:', systemPython);
|
||||
|
||||
return new Promise((resolve) => {
|
||||
const venvPath = path.join(this.autoBuildSourcePath!, '.venv');
|
||||
const proc = spawn(systemPython, ['-m', 'venv', venvPath], {
|
||||
cwd: this.autoBuildSourcePath!,
|
||||
stdio: 'pipe'
|
||||
|
||||
@@ -4,7 +4,8 @@ import { spawn } from 'child_process';
|
||||
import { app } from 'electron';
|
||||
import { EventEmitter } from 'events';
|
||||
import { detectRateLimit, createSDKRateLimitInfo, getProfileEnv } from './rate-limit-detector';
|
||||
import { findPythonCommand, parsePythonCommand } from './python-detector';
|
||||
import { parsePythonCommand } from './python-detector';
|
||||
import { pythonEnvManager } from './python-env-manager';
|
||||
|
||||
/**
|
||||
* Debug logging - only logs when DEBUG=true or in development mode
|
||||
@@ -21,8 +22,6 @@ function debug(...args: unknown[]): void {
|
||||
* Service for generating terminal names from commands using Claude AI
|
||||
*/
|
||||
export class TerminalNameGenerator extends EventEmitter {
|
||||
// Auto-detect Python command on initialization
|
||||
private pythonPath: string = findPythonCommand() || 'python';
|
||||
private autoBuildSourcePath: string = '';
|
||||
|
||||
constructor() {
|
||||
@@ -31,12 +30,9 @@ export class TerminalNameGenerator extends EventEmitter {
|
||||
}
|
||||
|
||||
/**
|
||||
* Configure paths for Python and auto-claude source
|
||||
* Configure the auto-claude source path
|
||||
*/
|
||||
configure(pythonPath?: string, autoBuildSourcePath?: string): void {
|
||||
if (pythonPath) {
|
||||
this.pythonPath = pythonPath;
|
||||
}
|
||||
configure(autoBuildSourcePath?: string): void {
|
||||
if (autoBuildSourcePath) {
|
||||
this.autoBuildSourcePath = autoBuildSourcePath;
|
||||
}
|
||||
@@ -118,6 +114,23 @@ export class TerminalNameGenerator extends EventEmitter {
|
||||
return null;
|
||||
}
|
||||
|
||||
// Check if Python environment is ready (has claude_agent_sdk installed)
|
||||
if (!pythonEnvManager.isEnvReady()) {
|
||||
debug('Python environment not ready, initializing...');
|
||||
const status = await pythonEnvManager.initialize(autoBuildSource);
|
||||
if (!status.ready) {
|
||||
debug('Python environment initialization failed:', status.error);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
// Get the venv Python path (where claude_agent_sdk is installed)
|
||||
const venvPythonPath = pythonEnvManager.getPythonPath();
|
||||
if (!venvPythonPath) {
|
||||
debug('Venv Python path not available');
|
||||
return null;
|
||||
}
|
||||
|
||||
const prompt = this.createNamePrompt(command, cwd);
|
||||
const script = this.createGenerationScript(prompt);
|
||||
|
||||
@@ -132,8 +145,8 @@ export class TerminalNameGenerator extends EventEmitter {
|
||||
const profileEnv = getProfileEnv();
|
||||
|
||||
return new Promise((resolve) => {
|
||||
// Parse Python command to handle space-separated commands like "py -3"
|
||||
const [pythonCommand, pythonBaseArgs] = parsePythonCommand(this.pythonPath);
|
||||
// Use the venv Python where claude_agent_sdk is installed
|
||||
const [pythonCommand, pythonBaseArgs] = parsePythonCommand(venvPythonPath);
|
||||
const childProcess = spawn(pythonCommand, [...pythonBaseArgs, '-c', script], {
|
||||
cwd: autoBuildSource,
|
||||
env: {
|
||||
|
||||
@@ -48,6 +48,10 @@ export interface ChangelogAPI {
|
||||
imageData: string,
|
||||
filename: string
|
||||
) => Promise<IPCResult<{ relativePath: string; url: string }>>;
|
||||
readLocalImage: (
|
||||
projectPath: string,
|
||||
relativePath: string
|
||||
) => Promise<IPCResult<string>>;
|
||||
|
||||
// Event Listeners
|
||||
onChangelogGenerationProgress: (
|
||||
@@ -113,6 +117,12 @@ export const createChangelogAPI = (): ChangelogAPI => ({
|
||||
): Promise<IPCResult<{ relativePath: string; url: string }>> =>
|
||||
invokeIpc(IPC_CHANNELS.CHANGELOG_SAVE_IMAGE, projectId, imageData, filename),
|
||||
|
||||
readLocalImage: (
|
||||
projectPath: string,
|
||||
relativePath: string
|
||||
): Promise<IPCResult<string>> =>
|
||||
invokeIpc(IPC_CHANNELS.CHANGELOG_READ_LOCAL_IMAGE, projectPath, relativePath),
|
||||
|
||||
// Event Listeners
|
||||
onChangelogGenerationProgress: (
|
||||
callback: (projectId: string, progress: ChangelogGenerationProgress) => void
|
||||
|
||||
@@ -41,9 +41,18 @@ export interface GitHubAPI {
|
||||
getGitHubUser: () => Promise<IPCResult<{ username: string; name?: string }>>;
|
||||
listGitHubUserRepos: () => Promise<IPCResult<{ repos: Array<{ fullName: string; description: string | null; isPrivate: boolean }> }>>;
|
||||
|
||||
// Repository detection
|
||||
// Repository detection and management
|
||||
detectGitHubRepo: (projectPath: string) => Promise<IPCResult<string>>;
|
||||
getGitHubBranches: (repo: string, token: string) => Promise<IPCResult<string[]>>;
|
||||
createGitHubRepo: (
|
||||
repoName: string,
|
||||
options: { description?: string; isPrivate?: boolean; projectPath: string; owner?: string }
|
||||
) => Promise<IPCResult<{ fullName: string; url: string }>>;
|
||||
addGitRemote: (
|
||||
projectPath: string,
|
||||
repoFullName: string
|
||||
) => Promise<IPCResult<{ remoteUrl: string }>>;
|
||||
listGitHubOrgs: () => Promise<IPCResult<{ orgs: Array<{ login: string; avatarUrl?: string }> }>>;
|
||||
|
||||
// Event Listeners
|
||||
onGitHubInvestigationProgress: (
|
||||
@@ -113,13 +122,28 @@ export const createGitHubAPI = (): GitHubAPI => ({
|
||||
listGitHubUserRepos: (): Promise<IPCResult<{ repos: Array<{ fullName: string; description: string | null; isPrivate: boolean }> }>> =>
|
||||
invokeIpc(IPC_CHANNELS.GITHUB_LIST_USER_REPOS),
|
||||
|
||||
// Repository detection
|
||||
// Repository detection and management
|
||||
detectGitHubRepo: (projectPath: string): Promise<IPCResult<string>> =>
|
||||
invokeIpc(IPC_CHANNELS.GITHUB_DETECT_REPO, projectPath),
|
||||
|
||||
getGitHubBranches: (repo: string, token: string): Promise<IPCResult<string[]>> =>
|
||||
invokeIpc(IPC_CHANNELS.GITHUB_GET_BRANCHES, repo, token),
|
||||
|
||||
createGitHubRepo: (
|
||||
repoName: string,
|
||||
options: { description?: string; isPrivate?: boolean; projectPath: string; owner?: string }
|
||||
): Promise<IPCResult<{ fullName: string; url: string }>> =>
|
||||
invokeIpc(IPC_CHANNELS.GITHUB_CREATE_REPO, repoName, options),
|
||||
|
||||
addGitRemote: (
|
||||
projectPath: string,
|
||||
repoFullName: string
|
||||
): Promise<IPCResult<{ remoteUrl: string }>> =>
|
||||
invokeIpc(IPC_CHANNELS.GITHUB_ADD_REMOTE, projectPath, repoFullName),
|
||||
|
||||
listGitHubOrgs: (): Promise<IPCResult<{ orgs: Array<{ login: string; avatarUrl?: string }> }>> =>
|
||||
invokeIpc(IPC_CHANNELS.GITHUB_LIST_ORGS),
|
||||
|
||||
// Event Listeners
|
||||
onGitHubInvestigationProgress: (
|
||||
callback: (projectId: string, status: GitHubInvestigationStatus) => void
|
||||
|
||||
@@ -14,6 +14,13 @@ import type {
|
||||
GitStatus
|
||||
} from '../../shared/types';
|
||||
|
||||
// Tab state interface (persisted in main process)
|
||||
export interface TabState {
|
||||
openProjectIds: string[];
|
||||
activeProjectId: string | null;
|
||||
tabOrder: string[];
|
||||
}
|
||||
|
||||
export interface ProjectAPI {
|
||||
// Project Management
|
||||
addProject: (projectPath: string) => Promise<IPCResult<Project>>;
|
||||
@@ -27,6 +34,10 @@ export interface ProjectAPI {
|
||||
updateProjectAutoBuild: (projectId: string) => Promise<IPCResult<InitializationResult>>;
|
||||
checkProjectVersion: (projectId: string) => Promise<IPCResult<AutoBuildVersionInfo>>;
|
||||
|
||||
// Tab State (persisted in main process for reliability)
|
||||
getTabState: () => Promise<IPCResult<TabState>>;
|
||||
saveTabState: (tabState: TabState) => Promise<IPCResult>;
|
||||
|
||||
// Context Operations
|
||||
getProjectContext: (projectId: string) => Promise<IPCResult<unknown>>;
|
||||
refreshProjectIndex: (projectId: string) => Promise<IPCResult<unknown>>;
|
||||
@@ -49,20 +60,19 @@ export interface ProjectAPI {
|
||||
) => Promise<IPCResult<import('../../shared/types').CreateProjectFolderResult>>;
|
||||
getDefaultProjectLocation: () => Promise<string | null>;
|
||||
|
||||
// Docker & Infrastructure Operations (for Graphiti/FalkorDB)
|
||||
getInfrastructureStatus: (port?: number) => Promise<IPCResult<InfrastructureStatus>>;
|
||||
startFalkorDB: (port?: number) => Promise<IPCResult<{ success: boolean; error?: string }>>;
|
||||
stopFalkorDB: () => Promise<IPCResult<{ success: boolean; error?: string }>>;
|
||||
openDockerDesktop: () => Promise<IPCResult<{ success: boolean; error?: string }>>;
|
||||
getDockerDownloadUrl: () => Promise<string>;
|
||||
// Memory Infrastructure Operations (LadybugDB - no Docker required)
|
||||
getMemoryInfrastructureStatus: (dbPath?: string) => Promise<IPCResult<InfrastructureStatus>>;
|
||||
listMemoryDatabases: (dbPath?: string) => Promise<IPCResult<string[]>>;
|
||||
testMemoryConnection: (dbPath?: string, database?: string) => Promise<IPCResult<GraphitiValidationResult>>;
|
||||
|
||||
// Graphiti Validation Operations
|
||||
validateFalkorDBConnection: (uri: string) => Promise<IPCResult<GraphitiValidationResult>>;
|
||||
validateOpenAIApiKey: (apiKey: string) => Promise<IPCResult<GraphitiValidationResult>>;
|
||||
testGraphitiConnection: (
|
||||
falkorDbUri: string,
|
||||
openAiApiKey: string
|
||||
) => Promise<IPCResult<GraphitiConnectionTestResult>>;
|
||||
validateLLMApiKey: (provider: string, apiKey: string) => Promise<IPCResult<GraphitiValidationResult>>;
|
||||
testGraphitiConnection: (config: {
|
||||
dbPath?: string;
|
||||
database?: string;
|
||||
llmProvider: string;
|
||||
apiKey: string;
|
||||
}) => Promise<IPCResult<GraphitiConnectionTestResult>>;
|
||||
|
||||
// Git Operations
|
||||
getGitBranches: (projectPath: string) => Promise<IPCResult<string[]>>;
|
||||
@@ -70,6 +80,41 @@ export interface ProjectAPI {
|
||||
detectMainBranch: (projectPath: string) => Promise<IPCResult<string | null>>;
|
||||
checkGitStatus: (projectPath: string) => Promise<IPCResult<GitStatus>>;
|
||||
initializeGit: (projectPath: string) => Promise<IPCResult<InitializationResult>>;
|
||||
|
||||
// Ollama Model Detection
|
||||
checkOllamaStatus: (baseUrl?: string) => Promise<IPCResult<{
|
||||
running: boolean;
|
||||
url: string;
|
||||
version?: string;
|
||||
message?: string;
|
||||
}>>;
|
||||
listOllamaModels: (baseUrl?: string) => Promise<IPCResult<{
|
||||
models: Array<{
|
||||
name: string;
|
||||
size_bytes: number;
|
||||
size_gb: number;
|
||||
modified_at: string;
|
||||
is_embedding: boolean;
|
||||
embedding_dim?: number | null;
|
||||
description?: string;
|
||||
}>;
|
||||
count: number;
|
||||
}>>;
|
||||
listOllamaEmbeddingModels: (baseUrl?: string) => Promise<IPCResult<{
|
||||
embedding_models: Array<{
|
||||
name: string;
|
||||
embedding_dim: number | null;
|
||||
description: string;
|
||||
size_bytes: number;
|
||||
size_gb: number;
|
||||
}>;
|
||||
count: number;
|
||||
}>>;
|
||||
pullOllamaModel: (modelName: string, baseUrl?: string) => Promise<IPCResult<{
|
||||
model: string;
|
||||
status: 'completed' | 'failed';
|
||||
output: string[];
|
||||
}>>;
|
||||
}
|
||||
|
||||
export const createProjectAPI = (): ProjectAPI => ({
|
||||
@@ -98,6 +143,13 @@ export const createProjectAPI = (): ProjectAPI => ({
|
||||
checkProjectVersion: (projectId: string): Promise<IPCResult<AutoBuildVersionInfo>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.PROJECT_CHECK_VERSION, projectId),
|
||||
|
||||
// Tab State (persisted in main process for reliability)
|
||||
getTabState: (): Promise<IPCResult<TabState>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.TAB_STATE_GET),
|
||||
|
||||
saveTabState: (tabState: TabState): Promise<IPCResult> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.TAB_STATE_SAVE, tabState),
|
||||
|
||||
// Context Operations
|
||||
getProjectContext: (projectId: string) =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.CONTEXT_GET, projectId),
|
||||
@@ -141,34 +193,27 @@ export const createProjectAPI = (): ProjectAPI => ({
|
||||
getDefaultProjectLocation: (): Promise<string | null> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.DIALOG_GET_DEFAULT_PROJECT_LOCATION),
|
||||
|
||||
// Docker & Infrastructure Operations
|
||||
getInfrastructureStatus: (port?: number): Promise<IPCResult<InfrastructureStatus>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.DOCKER_STATUS, port),
|
||||
// Memory Infrastructure Operations (LadybugDB - no Docker required)
|
||||
getMemoryInfrastructureStatus: (dbPath?: string): Promise<IPCResult<InfrastructureStatus>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.MEMORY_STATUS, dbPath),
|
||||
|
||||
startFalkorDB: (port?: number): Promise<IPCResult<{ success: boolean; error?: string }>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.DOCKER_START_FALKORDB, port),
|
||||
listMemoryDatabases: (dbPath?: string): Promise<IPCResult<string[]>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.MEMORY_LIST_DATABASES, dbPath),
|
||||
|
||||
stopFalkorDB: (): Promise<IPCResult<{ success: boolean; error?: string }>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.DOCKER_STOP_FALKORDB),
|
||||
|
||||
openDockerDesktop: (): Promise<IPCResult<{ success: boolean; error?: string }>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.DOCKER_OPEN_DESKTOP),
|
||||
|
||||
getDockerDownloadUrl: (): Promise<string> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.DOCKER_GET_DOWNLOAD_URL),
|
||||
testMemoryConnection: (dbPath?: string, database?: string): Promise<IPCResult<GraphitiValidationResult>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.MEMORY_TEST_CONNECTION, dbPath, database),
|
||||
|
||||
// Graphiti Validation Operations
|
||||
validateFalkorDBConnection: (uri: string): Promise<IPCResult<GraphitiValidationResult>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.GRAPHITI_VALIDATE_FALKORDB, uri),
|
||||
validateLLMApiKey: (provider: string, apiKey: string): Promise<IPCResult<GraphitiValidationResult>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.GRAPHITI_VALIDATE_LLM, provider, apiKey),
|
||||
|
||||
validateOpenAIApiKey: (apiKey: string): Promise<IPCResult<GraphitiValidationResult>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.GRAPHITI_VALIDATE_OPENAI, apiKey),
|
||||
|
||||
testGraphitiConnection: (
|
||||
falkorDbUri: string,
|
||||
openAiApiKey: string
|
||||
): Promise<IPCResult<GraphitiConnectionTestResult>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.GRAPHITI_TEST_CONNECTION, falkorDbUri, openAiApiKey),
|
||||
testGraphitiConnection: (config: {
|
||||
dbPath?: string;
|
||||
database?: string;
|
||||
llmProvider: string;
|
||||
apiKey: string;
|
||||
}): Promise<IPCResult<GraphitiConnectionTestResult>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.GRAPHITI_TEST_CONNECTION, config),
|
||||
|
||||
// Git Operations
|
||||
getGitBranches: (projectPath: string): Promise<IPCResult<string[]>> =>
|
||||
@@ -184,5 +229,18 @@ export const createProjectAPI = (): ProjectAPI => ({
|
||||
ipcRenderer.invoke(IPC_CHANNELS.GIT_CHECK_STATUS, projectPath),
|
||||
|
||||
initializeGit: (projectPath: string): Promise<IPCResult<InitializationResult>> =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.GIT_INITIALIZE, projectPath)
|
||||
ipcRenderer.invoke(IPC_CHANNELS.GIT_INITIALIZE, projectPath),
|
||||
|
||||
// Ollama Model Detection
|
||||
checkOllamaStatus: (baseUrl?: string) =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.OLLAMA_CHECK_STATUS, baseUrl),
|
||||
|
||||
listOllamaModels: (baseUrl?: string) =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.OLLAMA_LIST_MODELS, baseUrl),
|
||||
|
||||
listOllamaEmbeddingModels: (baseUrl?: string) =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.OLLAMA_LIST_EMBEDDING_MODELS, baseUrl),
|
||||
|
||||
pullOllamaModel: (modelName: string, baseUrl?: string) =>
|
||||
ipcRenderer.invoke(IPC_CHANNELS.OLLAMA_PULL_MODEL, modelName, baseUrl)
|
||||
});
|
||||
|
||||
@@ -1,5 +1,18 @@
|
||||
import { useState, useEffect } from 'react';
|
||||
import { Settings2, Download, RefreshCw, AlertCircle } from 'lucide-react';
|
||||
import {
|
||||
DndContext,
|
||||
DragOverlay,
|
||||
closestCenter,
|
||||
PointerSensor,
|
||||
useSensor,
|
||||
useSensors,
|
||||
type DragEndEvent
|
||||
} from '@dnd-kit/core';
|
||||
import {
|
||||
SortableContext,
|
||||
horizontalListSortingStrategy
|
||||
} from '@dnd-kit/sortable';
|
||||
import { TooltipProvider } from './components/ui/tooltip';
|
||||
import { Button } from './components/ui/button';
|
||||
import {
|
||||
@@ -44,6 +57,7 @@ import { useTerminalStore, restoreTerminalSessions } from './stores/terminal-sto
|
||||
import { useIpcListeners } from './hooks/useIpc';
|
||||
import { COLOR_THEMES } from '../shared/constants';
|
||||
import type { Task, Project, ColorTheme } from '../shared/types';
|
||||
import { ProjectTabBar } from './components/ProjectTabBar';
|
||||
|
||||
export function App() {
|
||||
// Load IPC listeners for real-time updates
|
||||
@@ -52,6 +66,13 @@ export function App() {
|
||||
// Stores
|
||||
const projects = useProjectStore((state) => state.projects);
|
||||
const selectedProjectId = useProjectStore((state) => state.selectedProjectId);
|
||||
const activeProjectId = useProjectStore((state) => state.activeProjectId);
|
||||
const getProjectTabs = useProjectStore((state) => state.getProjectTabs);
|
||||
const openProjectIds = useProjectStore((state) => state.openProjectIds);
|
||||
const openProjectTab = useProjectStore((state) => state.openProjectTab);
|
||||
const closeProjectTab = useProjectStore((state) => state.closeProjectTab);
|
||||
const setActiveProject = useProjectStore((state) => state.setActiveProject);
|
||||
const reorderTabs = useProjectStore((state) => state.reorderTabs);
|
||||
const tasks = useTaskStore((state) => state.tasks);
|
||||
const settings = useSettingsStore((state) => state.settings);
|
||||
const settingsLoading = useSettingsStore((state) => state.isLoading);
|
||||
@@ -77,8 +98,21 @@ export function App() {
|
||||
const [showGitHubSetup, setShowGitHubSetup] = useState(false);
|
||||
const [gitHubSetupProject, setGitHubSetupProject] = useState<Project | null>(null);
|
||||
|
||||
// Get selected project
|
||||
const selectedProject = projects.find((p) => p.id === selectedProjectId);
|
||||
// Setup drag sensors
|
||||
const sensors = useSensors(
|
||||
useSensor(PointerSensor, {
|
||||
activationConstraint: {
|
||||
distance: 8, // 8px movement required before drag starts
|
||||
},
|
||||
})
|
||||
);
|
||||
|
||||
// Track dragging state for overlay
|
||||
const [activeDragProject, setActiveDragProject] = useState<Project | null>(null);
|
||||
|
||||
// Get tabs and selected project
|
||||
const projectTabs = getProjectTabs();
|
||||
const selectedProject = projects.find((p) => p.id === (activeProjectId || selectedProjectId));
|
||||
|
||||
// Initial load
|
||||
useEffect(() => {
|
||||
@@ -86,6 +120,53 @@ export function App() {
|
||||
loadSettings();
|
||||
}, []);
|
||||
|
||||
// Restore tab state and open tabs for loaded projects
|
||||
useEffect(() => {
|
||||
console.log('[App] Tab restore useEffect triggered:', {
|
||||
projectsCount: projects.length,
|
||||
openProjectIds,
|
||||
activeProjectId,
|
||||
selectedProjectId,
|
||||
projectTabsCount: projectTabs.length,
|
||||
projectTabIds: projectTabs.map(p => p.id)
|
||||
});
|
||||
|
||||
if (projects.length > 0) {
|
||||
// Check openProjectIds (persisted state) instead of projectTabs (computed)
|
||||
// to avoid race condition where projectTabs is empty before projects load
|
||||
if (openProjectIds.length === 0) {
|
||||
// No tabs persisted at all, open the first available project
|
||||
const projectToOpen = activeProjectId || selectedProjectId || projects[0].id;
|
||||
console.log('[App] No tabs persisted, opening project:', projectToOpen);
|
||||
// Verify the project exists before opening
|
||||
if (projects.some(p => p.id === projectToOpen)) {
|
||||
openProjectTab(projectToOpen);
|
||||
setActiveProject(projectToOpen);
|
||||
} else {
|
||||
// Fallback to first project if stored IDs are invalid
|
||||
console.log('[App] Project not found, falling back to first project:', projects[0].id);
|
||||
openProjectTab(projects[0].id);
|
||||
setActiveProject(projects[0].id);
|
||||
}
|
||||
return;
|
||||
}
|
||||
console.log('[App] Tabs already persisted, checking active project');
|
||||
// If there's an active project but no tabs open for it, open a tab
|
||||
if (activeProjectId && !projectTabs.some(tab => tab.id === activeProjectId)) {
|
||||
console.log('[App] Active project has no tab, opening:', activeProjectId);
|
||||
openProjectTab(activeProjectId);
|
||||
}
|
||||
// If there's a selected project but no active project, make it active
|
||||
else if (selectedProjectId && !activeProjectId) {
|
||||
console.log('[App] No active project, using selected:', selectedProjectId);
|
||||
setActiveProject(selectedProjectId);
|
||||
openProjectTab(selectedProjectId);
|
||||
} else {
|
||||
console.log('[App] Tab state is valid, no action needed');
|
||||
}
|
||||
}
|
||||
}, [projects, activeProjectId, selectedProjectId, openProjectIds, projectTabs, openProjectTab, setActiveProject]);
|
||||
|
||||
// Track if settings have been loaded at least once
|
||||
const [settingsHaveLoaded, setSettingsHaveLoaded] = useState(false);
|
||||
|
||||
@@ -135,11 +216,22 @@ export function App() {
|
||||
};
|
||||
}, []);
|
||||
|
||||
// Reset init success flag when selected project changes
|
||||
// This allows the init dialog to show for new/different projects
|
||||
useEffect(() => {
|
||||
setInitSuccess(false);
|
||||
setInitError(null);
|
||||
}, [selectedProjectId]);
|
||||
|
||||
// Check if selected project needs initialization (e.g., .auto-claude folder was deleted)
|
||||
useEffect(() => {
|
||||
// Don't show dialog while initialization is in progress
|
||||
if (isInitializing) return;
|
||||
|
||||
// Don't reopen dialog after successful initialization
|
||||
// (project update with autoBuildPath may not have propagated yet)
|
||||
if (initSuccess) return;
|
||||
|
||||
if (selectedProject && !selectedProject.autoBuildPath && skippedInitProjectId !== selectedProject.id) {
|
||||
// Project exists but isn't initialized - show init dialog
|
||||
setPendingProject(selectedProject);
|
||||
@@ -147,12 +239,52 @@ export function App() {
|
||||
setInitSuccess(false); // Reset success flag
|
||||
setShowInitDialog(true);
|
||||
}
|
||||
}, [selectedProject, skippedInitProjectId, isInitializing]);
|
||||
}, [selectedProject, skippedInitProjectId, isInitializing, initSuccess]);
|
||||
|
||||
// Global keyboard shortcut: Cmd/Ctrl+T to add project (when not on terminals view)
|
||||
useEffect(() => {
|
||||
const handleKeyDown = async (e: KeyboardEvent) => {
|
||||
// Skip if in input fields
|
||||
if (
|
||||
e.target instanceof HTMLInputElement ||
|
||||
e.target instanceof HTMLTextAreaElement ||
|
||||
(e.target as HTMLElement)?.isContentEditable
|
||||
) {
|
||||
return;
|
||||
}
|
||||
|
||||
// Cmd/Ctrl+T: Add new project (only when not on terminals view)
|
||||
if ((e.ctrlKey || e.metaKey) && e.key === 't' && activeView !== 'terminals') {
|
||||
e.preventDefault();
|
||||
try {
|
||||
const path = await window.electronAPI.selectDirectory();
|
||||
if (path) {
|
||||
const project = await addProject(path);
|
||||
if (project) {
|
||||
openProjectTab(project.id);
|
||||
if (!project.autoBuildPath) {
|
||||
setPendingProject(project);
|
||||
setInitError(null);
|
||||
setInitSuccess(false);
|
||||
setShowInitDialog(true);
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('Failed to add project:', error);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
window.addEventListener('keydown', handleKeyDown);
|
||||
return () => window.removeEventListener('keydown', handleKeyDown);
|
||||
}, [activeView, openProjectTab]);
|
||||
|
||||
// Load tasks when project changes
|
||||
useEffect(() => {
|
||||
if (selectedProjectId) {
|
||||
loadTasks(selectedProjectId);
|
||||
const currentProjectId = activeProjectId || selectedProjectId;
|
||||
if (currentProjectId) {
|
||||
loadTasks(currentProjectId);
|
||||
setSelectedTask(null); // Clear selection on project change
|
||||
} else {
|
||||
useTaskStore.getState().clearTasks();
|
||||
@@ -173,7 +305,7 @@ export function App() {
|
||||
console.error('[App] Failed to restore sessions:', err);
|
||||
});
|
||||
}
|
||||
}, [selectedProjectId, selectedProject?.path, selectedProject?.name]);
|
||||
}, [activeProjectId, selectedProjectId, selectedProject?.path, selectedProject?.name]);
|
||||
|
||||
// Apply theme on load
|
||||
useEffect(() => {
|
||||
@@ -250,7 +382,11 @@ export function App() {
|
||||
const path = await window.electronAPI.selectDirectory();
|
||||
if (path) {
|
||||
const project = await addProject(path);
|
||||
if (project && !project.autoBuildPath) {
|
||||
if (project) {
|
||||
// Open a tab for the new project
|
||||
openProjectTab(project.id);
|
||||
|
||||
if (!project.autoBuildPath) {
|
||||
// Project doesn't have Auto Claude initialized, show init dialog
|
||||
setPendingProject(project);
|
||||
setInitError(null); // Clear any previous errors
|
||||
@@ -258,11 +394,44 @@ export function App() {
|
||||
setShowInitDialog(true);
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('Failed to add project:', error);
|
||||
}
|
||||
};
|
||||
|
||||
const handleProjectTabSelect = (projectId: string) => {
|
||||
setActiveProject(projectId);
|
||||
};
|
||||
|
||||
const handleProjectTabClose = (projectId: string) => {
|
||||
closeProjectTab(projectId);
|
||||
};
|
||||
|
||||
// Handle drag start - set the active dragged project
|
||||
const handleDragStart = (event: any) => {
|
||||
const { active } = event;
|
||||
const draggedProject = projectTabs.find(p => p.id === active.id);
|
||||
if (draggedProject) {
|
||||
setActiveDragProject(draggedProject);
|
||||
}
|
||||
};
|
||||
|
||||
// Handle drag end - reorder tabs if dropped over another tab
|
||||
const handleDragEnd = (event: DragEndEvent) => {
|
||||
const { active, over } = event;
|
||||
setActiveDragProject(null);
|
||||
|
||||
if (!over) return;
|
||||
|
||||
const oldIndex = projectTabs.findIndex(p => p.id === active.id);
|
||||
const newIndex = projectTabs.findIndex(p => p.id === over.id);
|
||||
|
||||
if (oldIndex !== newIndex && oldIndex !== -1 && newIndex !== -1) {
|
||||
reorderTabs(oldIndex, newIndex);
|
||||
}
|
||||
};
|
||||
|
||||
const handleInitialize = async () => {
|
||||
if (!pendingProject) return;
|
||||
|
||||
@@ -386,6 +555,38 @@ export function App() {
|
||||
|
||||
{/* Main content */}
|
||||
<div className="flex flex-1 flex-col overflow-hidden">
|
||||
{/* Project Tabs */}
|
||||
{projectTabs.length > 0 && (
|
||||
<DndContext
|
||||
sensors={sensors}
|
||||
collisionDetection={closestCenter}
|
||||
onDragStart={handleDragStart}
|
||||
onDragEnd={handleDragEnd}
|
||||
>
|
||||
<SortableContext items={projectTabs.map(p => p.id)} strategy={horizontalListSortingStrategy}>
|
||||
<ProjectTabBar
|
||||
projects={projectTabs}
|
||||
activeProjectId={activeProjectId}
|
||||
onProjectSelect={handleProjectTabSelect}
|
||||
onProjectClose={handleProjectTabClose}
|
||||
onAddProject={handleAddProject}
|
||||
/>
|
||||
</SortableContext>
|
||||
|
||||
{/* Drag overlay - shows what's being dragged */}
|
||||
<DragOverlay>
|
||||
{activeDragProject && (
|
||||
<div className="flex items-center gap-2 bg-card border border-border rounded-md px-4 py-2.5 shadow-lg max-w-[200px]">
|
||||
<div className="w-1 h-4 bg-muted-foreground rounded-full" />
|
||||
<span className="truncate font-medium text-sm">
|
||||
{activeDragProject.name}
|
||||
</span>
|
||||
</div>
|
||||
)}
|
||||
</DragOverlay>
|
||||
</DndContext>
|
||||
)}
|
||||
|
||||
{/* Header */}
|
||||
<header className="electron-drag flex h-14 items-center justify-between border-b border-border bg-card/50 backdrop-blur-sm px-6">
|
||||
<div className="electron-no-drag">
|
||||
@@ -432,21 +633,22 @@ export function App() {
|
||||
<TerminalGrid
|
||||
projectPath={selectedProject?.path}
|
||||
onNewTaskClick={() => setIsNewTaskDialogOpen(true)}
|
||||
isActive={activeView === 'terminals'}
|
||||
/>
|
||||
</div>
|
||||
{activeView === 'roadmap' && selectedProjectId && (
|
||||
<Roadmap projectId={selectedProjectId} onGoToTask={handleGoToTask} />
|
||||
{activeView === 'roadmap' && (activeProjectId || selectedProjectId) && (
|
||||
<Roadmap projectId={activeProjectId || selectedProjectId!} onGoToTask={handleGoToTask} />
|
||||
)}
|
||||
{activeView === 'context' && selectedProjectId && (
|
||||
<Context projectId={selectedProjectId} />
|
||||
{activeView === 'context' && (activeProjectId || selectedProjectId) && (
|
||||
<Context projectId={activeProjectId || selectedProjectId!} />
|
||||
)}
|
||||
{activeView === 'ideation' && selectedProjectId && (
|
||||
<Ideation projectId={selectedProjectId} onGoToTask={handleGoToTask} />
|
||||
{activeView === 'ideation' && (activeProjectId || selectedProjectId) && (
|
||||
<Ideation projectId={activeProjectId || selectedProjectId!} onGoToTask={handleGoToTask} />
|
||||
)}
|
||||
{activeView === 'insights' && selectedProjectId && (
|
||||
<Insights projectId={selectedProjectId} />
|
||||
{activeView === 'insights' && (activeProjectId || selectedProjectId) && (
|
||||
<Insights projectId={activeProjectId || selectedProjectId!} />
|
||||
)}
|
||||
{activeView === 'github-issues' && selectedProjectId && (
|
||||
{activeView === 'github-issues' && (activeProjectId || selectedProjectId) && (
|
||||
<GitHubIssues
|
||||
onOpenSettings={() => {
|
||||
setSettingsInitialProjectSection('github');
|
||||
@@ -455,11 +657,11 @@ export function App() {
|
||||
onNavigateToTask={handleGoToTask}
|
||||
/>
|
||||
)}
|
||||
{activeView === 'changelog' && selectedProjectId && (
|
||||
{activeView === 'changelog' && (activeProjectId || selectedProjectId) && (
|
||||
<Changelog />
|
||||
)}
|
||||
{activeView === 'worktrees' && selectedProjectId && (
|
||||
<Worktrees projectId={selectedProjectId} />
|
||||
{activeView === 'worktrees' && (activeProjectId || selectedProjectId) && (
|
||||
<Worktrees projectId={activeProjectId || selectedProjectId!} />
|
||||
)}
|
||||
{activeView === 'agent-tools' && (
|
||||
<div className="flex h-full items-center justify-center">
|
||||
@@ -477,7 +679,9 @@ export function App() {
|
||||
projects={projects}
|
||||
onNewProject={handleAddProject}
|
||||
onOpenProject={handleAddProject}
|
||||
onSelectProject={(projectId) => useProjectStore.getState().selectProject(projectId)}
|
||||
onSelectProject={(projectId) => {
|
||||
openProjectTab(projectId);
|
||||
}}
|
||||
/>
|
||||
)}
|
||||
</main>
|
||||
@@ -491,9 +695,9 @@ export function App() {
|
||||
/>
|
||||
|
||||
{/* Dialogs */}
|
||||
{selectedProjectId && (
|
||||
{(activeProjectId || selectedProjectId) && (
|
||||
<TaskCreationWizard
|
||||
projectId={selectedProjectId}
|
||||
projectId={activeProjectId || selectedProjectId!}
|
||||
open={isNewTaskDialogOpen}
|
||||
onOpenChange={setIsNewTaskDialogOpen}
|
||||
/>
|
||||
|
||||
@@ -0,0 +1,478 @@
|
||||
/**
|
||||
* Unit tests for Project Store Tab Management
|
||||
* Tests Zustand store for project tab state management
|
||||
*
|
||||
* Note: Tab state persistence is now handled via IPC (saveTabState/getTabState)
|
||||
* rather than localStorage. The saveTabState calls are debounced, so we don't
|
||||
* assert on them directly in these unit tests.
|
||||
*/
|
||||
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
||||
import { useProjectStore } from '../stores/project-store';
|
||||
import type { Project, ProjectSettings } from '../../shared/types';
|
||||
|
||||
// Helper to create test projects
|
||||
function createTestProject(overrides: Partial<Project> = {}): Project {
|
||||
const defaultSettings: ProjectSettings = {
|
||||
model: 'claude-3-opus',
|
||||
memoryBackend: 'graphiti',
|
||||
linearSync: false,
|
||||
notifications: {
|
||||
onTaskComplete: true,
|
||||
onTaskFailed: true,
|
||||
onReviewNeeded: true,
|
||||
sound: false
|
||||
},
|
||||
graphitiMcpEnabled: false
|
||||
};
|
||||
|
||||
return {
|
||||
id: `project-${Date.now()}-${Math.random().toString(36).substring(7)}`,
|
||||
name: 'Test Project',
|
||||
path: '/path/to/test-project',
|
||||
autoBuildPath: '.auto-claude',
|
||||
settings: defaultSettings,
|
||||
createdAt: new Date(),
|
||||
updatedAt: new Date(),
|
||||
...overrides
|
||||
};
|
||||
}
|
||||
|
||||
|
||||
describe('Project Store Tab Management', () => {
|
||||
beforeEach(() => {
|
||||
// Reset store to initial state before each test
|
||||
useProjectStore.setState({
|
||||
projects: [],
|
||||
selectedProjectId: null,
|
||||
isLoading: false,
|
||||
error: null,
|
||||
openProjectIds: [],
|
||||
activeProjectId: null,
|
||||
tabOrder: []
|
||||
});
|
||||
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
describe('openProjectTab', () => {
|
||||
it('should open a new project tab', () => {
|
||||
const project = createTestProject({ id: 'project-1' });
|
||||
useProjectStore.setState({ projects: [project] });
|
||||
|
||||
useProjectStore.getState().openProjectTab('project-1');
|
||||
|
||||
expect(useProjectStore.getState().openProjectIds).toContain('project-1');
|
||||
expect(useProjectStore.getState().activeProjectId).toBe('project-1');
|
||||
expect(useProjectStore.getState().tabOrder).toContain('project-1');
|
||||
});
|
||||
|
||||
it('should add to existing open tabs', () => {
|
||||
const project1 = createTestProject({ id: 'project-1' });
|
||||
const project2 = createTestProject({ id: 'project-2' });
|
||||
useProjectStore.setState({
|
||||
projects: [project1, project2],
|
||||
openProjectIds: ['project-1'],
|
||||
activeProjectId: 'project-1',
|
||||
tabOrder: ['project-1']
|
||||
});
|
||||
|
||||
useProjectStore.getState().openProjectTab('project-2');
|
||||
|
||||
expect(useProjectStore.getState().openProjectIds).toEqual(['project-1', 'project-2']);
|
||||
expect(useProjectStore.getState().activeProjectId).toBe('project-2');
|
||||
expect(useProjectStore.getState().tabOrder).toEqual(['project-1', 'project-2']);
|
||||
});
|
||||
|
||||
it('should not duplicate existing tab', () => {
|
||||
const project = createTestProject({ id: 'project-1' });
|
||||
useProjectStore.setState({
|
||||
projects: [project],
|
||||
openProjectIds: ['project-1'],
|
||||
activeProjectId: 'project-1',
|
||||
tabOrder: ['project-1']
|
||||
});
|
||||
|
||||
useProjectStore.getState().openProjectTab('project-1');
|
||||
|
||||
// Should only have one entry
|
||||
expect(useProjectStore.getState().openProjectIds).toEqual(['project-1']);
|
||||
expect(useProjectStore.getState().tabOrder).toEqual(['project-1']);
|
||||
// Should still make it active
|
||||
expect(useProjectStore.getState().activeProjectId).toBe('project-1');
|
||||
});
|
||||
|
||||
it('should preserve existing tab order when adding new tab', () => {
|
||||
const project1 = createTestProject({ id: 'project-1' });
|
||||
const project2 = createTestProject({ id: 'project-2' });
|
||||
const project3 = createTestProject({ id: 'project-3' });
|
||||
useProjectStore.setState({
|
||||
projects: [project1, project2, project3],
|
||||
openProjectIds: ['project-1', 'project-3'],
|
||||
activeProjectId: 'project-3',
|
||||
tabOrder: ['project-1', 'project-3']
|
||||
});
|
||||
|
||||
useProjectStore.getState().openProjectTab('project-2');
|
||||
|
||||
expect(useProjectStore.getState().tabOrder).toEqual(['project-1', 'project-3', 'project-2']);
|
||||
});
|
||||
});
|
||||
|
||||
describe('closeProjectTab', () => {
|
||||
it('should close a project tab', () => {
|
||||
useProjectStore.setState({
|
||||
openProjectIds: ['project-1', 'project-2'],
|
||||
activeProjectId: 'project-2',
|
||||
tabOrder: ['project-1', 'project-2']
|
||||
});
|
||||
|
||||
useProjectStore.getState().closeProjectTab('project-1');
|
||||
|
||||
expect(useProjectStore.getState().openProjectIds).toEqual(['project-2']);
|
||||
expect(useProjectStore.getState().tabOrder).toEqual(['project-2']);
|
||||
});
|
||||
|
||||
it('should activate first remaining tab when closing active tab', () => {
|
||||
useProjectStore.setState({
|
||||
openProjectIds: ['project-1', 'project-2', 'project-3'],
|
||||
activeProjectId: 'project-2',
|
||||
tabOrder: ['project-1', 'project-2', 'project-3']
|
||||
});
|
||||
|
||||
useProjectStore.getState().closeProjectTab('project-2');
|
||||
|
||||
// After removing project-2 from tabOrder, we get ['project-1', 'project-3']
|
||||
// The first tab in the remaining order is 'project-1'
|
||||
expect(useProjectStore.getState().activeProjectId).toBe('project-1');
|
||||
});
|
||||
|
||||
it('should activate previous tab when closing active tab and no next tab', () => {
|
||||
useProjectStore.setState({
|
||||
openProjectIds: ['project-1', 'project-2'],
|
||||
activeProjectId: 'project-2',
|
||||
tabOrder: ['project-1', 'project-2']
|
||||
});
|
||||
|
||||
useProjectStore.getState().closeProjectTab('project-2');
|
||||
|
||||
expect(useProjectStore.getState().activeProjectId).toBe('project-1');
|
||||
});
|
||||
|
||||
it('should set activeProjectId to null when closing last tab', () => {
|
||||
useProjectStore.setState({
|
||||
openProjectIds: ['project-1'],
|
||||
activeProjectId: 'project-1',
|
||||
tabOrder: ['project-1']
|
||||
});
|
||||
|
||||
useProjectStore.getState().closeProjectTab('project-1');
|
||||
|
||||
expect(useProjectStore.getState().activeProjectId).toBeNull();
|
||||
});
|
||||
|
||||
it('should not affect activeProjectId when closing non-active tab', () => {
|
||||
useProjectStore.setState({
|
||||
openProjectIds: ['project-1', 'project-2'],
|
||||
activeProjectId: 'project-2',
|
||||
tabOrder: ['project-1', 'project-2']
|
||||
});
|
||||
|
||||
useProjectStore.getState().closeProjectTab('project-1');
|
||||
|
||||
expect(useProjectStore.getState().activeProjectId).toBe('project-2');
|
||||
});
|
||||
});
|
||||
|
||||
describe('setActiveProject', () => {
|
||||
it('should set active project', () => {
|
||||
useProjectStore.setState({ activeProjectId: null });
|
||||
|
||||
useProjectStore.getState().setActiveProject('project-1');
|
||||
|
||||
expect(useProjectStore.getState().activeProjectId).toBe('project-1');
|
||||
});
|
||||
|
||||
it('should clear active project with null', () => {
|
||||
useProjectStore.setState({ activeProjectId: 'project-1' });
|
||||
|
||||
useProjectStore.getState().setActiveProject(null);
|
||||
|
||||
expect(useProjectStore.getState().activeProjectId).toBeNull();
|
||||
});
|
||||
|
||||
it('should also update selectedProjectId for backward compatibility', () => {
|
||||
useProjectStore.setState({ selectedProjectId: null });
|
||||
|
||||
useProjectStore.getState().setActiveProject('project-1');
|
||||
|
||||
expect(useProjectStore.getState().selectedProjectId).toBe('project-1');
|
||||
});
|
||||
});
|
||||
|
||||
describe('reorderTabs', () => {
|
||||
it('should reorder tabs by moving from index to index', () => {
|
||||
useProjectStore.setState({
|
||||
tabOrder: ['project-1', 'project-2', 'project-3', 'project-4']
|
||||
});
|
||||
|
||||
// Move project-3 from index 2 to index 1
|
||||
useProjectStore.getState().reorderTabs(2, 1);
|
||||
|
||||
expect(useProjectStore.getState().tabOrder).toEqual(['project-1', 'project-3', 'project-2', 'project-4']);
|
||||
});
|
||||
|
||||
it('should handle moving tab to the end', () => {
|
||||
useProjectStore.setState({
|
||||
tabOrder: ['project-1', 'project-2', 'project-3']
|
||||
});
|
||||
|
||||
// Move project-1 from index 0 to index 2
|
||||
useProjectStore.getState().reorderTabs(0, 2);
|
||||
|
||||
expect(useProjectStore.getState().tabOrder).toEqual(['project-2', 'project-3', 'project-1']);
|
||||
});
|
||||
|
||||
it('should handle moving tab to the beginning', () => {
|
||||
useProjectStore.setState({
|
||||
tabOrder: ['project-1', 'project-2', 'project-3']
|
||||
});
|
||||
|
||||
// Move project-3 from index 2 to index 0
|
||||
useProjectStore.getState().reorderTabs(2, 0);
|
||||
|
||||
expect(useProjectStore.getState().tabOrder).toEqual(['project-3', 'project-1', 'project-2']);
|
||||
});
|
||||
|
||||
it('should handle no-op reordering (same index)', () => {
|
||||
useProjectStore.setState({
|
||||
tabOrder: ['project-1', 'project-2', 'project-3']
|
||||
});
|
||||
|
||||
useProjectStore.getState().reorderTabs(1, 1);
|
||||
|
||||
expect(useProjectStore.getState().tabOrder).toEqual(['project-1', 'project-2', 'project-3']);
|
||||
});
|
||||
});
|
||||
|
||||
describe('restoreTabState', () => {
|
||||
it('should be a no-op (tab state is now loaded via IPC in loadProjects)', () => {
|
||||
// Set up some initial state
|
||||
useProjectStore.setState({
|
||||
openProjectIds: ['existing'],
|
||||
activeProjectId: 'existing',
|
||||
tabOrder: ['existing']
|
||||
});
|
||||
|
||||
// restoreTabState is now a no-op - it just logs
|
||||
useProjectStore.getState().restoreTabState();
|
||||
|
||||
// State should remain unchanged (not modified by restoreTabState)
|
||||
expect(useProjectStore.getState().openProjectIds).toEqual(['existing']);
|
||||
expect(useProjectStore.getState().activeProjectId).toBe('existing');
|
||||
expect(useProjectStore.getState().tabOrder).toEqual(['existing']);
|
||||
});
|
||||
});
|
||||
|
||||
describe('getOpenProjects', () => {
|
||||
it('should return projects that are open', () => {
|
||||
const project1 = createTestProject({ id: 'project-1' });
|
||||
const project2 = createTestProject({ id: 'project-2' });
|
||||
const project3 = createTestProject({ id: 'project-3' });
|
||||
|
||||
useProjectStore.setState({
|
||||
projects: [project1, project2, project3],
|
||||
openProjectIds: ['project-1', 'project-3']
|
||||
});
|
||||
|
||||
const openProjects = useProjectStore.getState().getOpenProjects();
|
||||
|
||||
expect(openProjects).toHaveLength(2);
|
||||
expect(openProjects.map(p => p.id)).toEqual(['project-1', 'project-3']);
|
||||
});
|
||||
|
||||
it('should return empty array when no projects are open', () => {
|
||||
const project1 = createTestProject({ id: 'project-1' });
|
||||
|
||||
useProjectStore.setState({
|
||||
projects: [project1],
|
||||
openProjectIds: []
|
||||
});
|
||||
|
||||
const openProjects = useProjectStore.getState().getOpenProjects();
|
||||
|
||||
expect(openProjects).toHaveLength(0);
|
||||
});
|
||||
|
||||
it('should handle open project IDs that dont exist in projects', () => {
|
||||
const project1 = createTestProject({ id: 'project-1' });
|
||||
|
||||
useProjectStore.setState({
|
||||
projects: [project1],
|
||||
openProjectIds: ['project-1', 'non-existent']
|
||||
});
|
||||
|
||||
const openProjects = useProjectStore.getState().getOpenProjects();
|
||||
|
||||
expect(openProjects).toHaveLength(1);
|
||||
expect(openProjects[0].id).toBe('project-1');
|
||||
});
|
||||
});
|
||||
|
||||
describe('getActiveProject', () => {
|
||||
it('should return the active project', () => {
|
||||
const project1 = createTestProject({ id: 'project-1' });
|
||||
const project2 = createTestProject({ id: 'project-2' });
|
||||
|
||||
useProjectStore.setState({
|
||||
projects: [project1, project2],
|
||||
activeProjectId: 'project-2'
|
||||
});
|
||||
|
||||
const activeProject = useProjectStore.getState().getActiveProject();
|
||||
|
||||
expect(activeProject).toBeDefined();
|
||||
expect(activeProject?.id).toBe('project-2');
|
||||
});
|
||||
|
||||
it('should return undefined when no active project', () => {
|
||||
const project1 = createTestProject({ id: 'project-1' });
|
||||
|
||||
useProjectStore.setState({
|
||||
projects: [project1],
|
||||
activeProjectId: null
|
||||
});
|
||||
|
||||
const activeProject = useProjectStore.getState().getActiveProject();
|
||||
|
||||
expect(activeProject).toBeUndefined();
|
||||
});
|
||||
|
||||
it('should return undefined when active project ID does not exist', () => {
|
||||
const project1 = createTestProject({ id: 'project-1' });
|
||||
|
||||
useProjectStore.setState({
|
||||
projects: [project1],
|
||||
activeProjectId: 'non-existent'
|
||||
});
|
||||
|
||||
const activeProject = useProjectStore.getState().getActiveProject();
|
||||
|
||||
expect(activeProject).toBeUndefined();
|
||||
});
|
||||
});
|
||||
|
||||
describe('getProjectTabs', () => {
|
||||
it('should return projects in tab order', () => {
|
||||
const project1 = createTestProject({ id: 'project-1', name: 'Project 1' });
|
||||
const project2 = createTestProject({ id: 'project-2', name: 'Project 2' });
|
||||
const project3 = createTestProject({ id: 'project-3', name: 'Project 3' });
|
||||
|
||||
useProjectStore.setState({
|
||||
projects: [project1, project2, project3],
|
||||
openProjectIds: ['project-1', 'project-2', 'project-3'],
|
||||
tabOrder: ['project-3', 'project-1', 'project-2']
|
||||
});
|
||||
|
||||
const tabs = useProjectStore.getState().getProjectTabs();
|
||||
|
||||
expect(tabs).toHaveLength(3);
|
||||
expect(tabs.map(p => p.id)).toEqual(['project-3', 'project-1', 'project-2']);
|
||||
});
|
||||
|
||||
it('should append open projects not in tab order', () => {
|
||||
const project1 = createTestProject({ id: 'project-1', name: 'Project 1' });
|
||||
const project2 = createTestProject({ id: 'project-2', name: 'Project 2' });
|
||||
const project3 = createTestProject({ id: 'project-3', name: 'Project 3' });
|
||||
|
||||
useProjectStore.setState({
|
||||
projects: [project1, project2, project3],
|
||||
openProjectIds: ['project-1', 'project-2', 'project-3'],
|
||||
tabOrder: ['project-2'] // Only project-2 is in tabOrder
|
||||
});
|
||||
|
||||
const tabs = useProjectStore.getState().getProjectTabs();
|
||||
|
||||
expect(tabs).toHaveLength(3);
|
||||
// project-2 should be first (from tabOrder), others appended
|
||||
expect(tabs[0].id).toBe('project-2');
|
||||
expect(tabs.slice(1).map(p => p.id)).toContain('project-1');
|
||||
expect(tabs.slice(1).map(p => p.id)).toContain('project-3');
|
||||
});
|
||||
|
||||
it('should return empty array when no tabs are open', () => {
|
||||
const project1 = createTestProject({ id: 'project-1' });
|
||||
|
||||
useProjectStore.setState({
|
||||
projects: [project1],
|
||||
openProjectIds: [],
|
||||
tabOrder: []
|
||||
});
|
||||
|
||||
const tabs = useProjectStore.getState().getProjectTabs();
|
||||
|
||||
expect(tabs).toHaveLength(0);
|
||||
});
|
||||
|
||||
it('should handle tab order entries for projects that are not open', () => {
|
||||
const project1 = createTestProject({ id: 'project-1' });
|
||||
const project2 = createTestProject({ id: 'project-2' });
|
||||
|
||||
useProjectStore.setState({
|
||||
projects: [project1, project2],
|
||||
openProjectIds: ['project-1'], // Only project-1 is actually open
|
||||
tabOrder: ['project-2', 'project-1'] // tabOrder has project-2
|
||||
});
|
||||
|
||||
const tabs = useProjectStore.getState().getProjectTabs();
|
||||
|
||||
// getProjectTabs returns all projects in tabOrder, then adds open projects not in tabOrder
|
||||
// So it returns project-2 (from tabOrder) and project-1 (from tabOrder)
|
||||
// Even though project-2 is not in openProjectIds
|
||||
expect(tabs).toHaveLength(2);
|
||||
expect(tabs[0].id).toBe('project-2'); // First in tabOrder
|
||||
expect(tabs[1].id).toBe('project-1'); // Second in tabOrder
|
||||
});
|
||||
});
|
||||
|
||||
describe('Integration with existing project operations', () => {
|
||||
it('should open tab when adding project', () => {
|
||||
const project = createTestProject({ id: 'project-1' });
|
||||
|
||||
useProjectStore.setState({ projects: [] });
|
||||
useProjectStore.getState().addProject(project);
|
||||
useProjectStore.getState().selectProject(project.id);
|
||||
useProjectStore.getState().openProjectTab(project.id);
|
||||
|
||||
expect(useProjectStore.getState().projects).toContain(project);
|
||||
expect(useProjectStore.getState().selectedProjectId).toBe(project.id);
|
||||
expect(useProjectStore.getState().openProjectIds).toContain(project.id);
|
||||
expect(useProjectStore.getState().activeProjectId).toBe(project.id);
|
||||
});
|
||||
|
||||
it('should update selectedProjectId when removing project', () => {
|
||||
const project1 = createTestProject({ id: 'project-1' });
|
||||
const project2 = createTestProject({ id: 'project-2' });
|
||||
|
||||
useProjectStore.setState({
|
||||
projects: [project1, project2],
|
||||
openProjectIds: ['project-1', 'project-2'],
|
||||
activeProjectId: 'project-2',
|
||||
selectedProjectId: 'project-1'
|
||||
});
|
||||
|
||||
useProjectStore.getState().removeProject('project-1');
|
||||
|
||||
expect(useProjectStore.getState().projects).not.toContain(
|
||||
expect.objectContaining({ id: 'project-1' })
|
||||
);
|
||||
// removeProject clears selectedProjectId if it matches the removed project
|
||||
expect(useProjectStore.getState().selectedProjectId).toBeNull();
|
||||
// Note: openProjectIds is not automatically cleared by removeProject
|
||||
// This would be handled by the UI layer when it detects the project was removed
|
||||
});
|
||||
});
|
||||
});
|
||||
@@ -149,11 +149,11 @@ export function AgentProfileSelector({
|
||||
description: profile.description
|
||||
};
|
||||
}
|
||||
// Default to balanced
|
||||
// Default to auto profile (the actual default)
|
||||
return {
|
||||
icon: Scale,
|
||||
label: 'Balanced',
|
||||
description: 'Good balance of speed and quality'
|
||||
icon: Sparkles,
|
||||
label: 'Auto (Optimized)',
|
||||
description: 'Uses Opus across all phases with optimized thinking levels'
|
||||
};
|
||||
};
|
||||
|
||||
|
||||
@@ -0,0 +1,236 @@
|
||||
import { useState, useEffect, useCallback, useRef, useMemo } from 'react';
|
||||
import { File, Folder, ChevronRight } from 'lucide-react';
|
||||
import { cn } from '../lib/utils';
|
||||
import { useFileExplorerStore } from '../stores/file-explorer-store';
|
||||
import type { FileNode } from '../../shared/types';
|
||||
|
||||
interface FileAutocompleteProps {
|
||||
query: string;
|
||||
projectPath: string;
|
||||
position: { top: number; left: number };
|
||||
onSelect: (filename: string, fullPath: string) => void;
|
||||
onClose: () => void;
|
||||
maxResults?: number;
|
||||
}
|
||||
|
||||
/**
|
||||
* Autocomplete popup for @ file mentions in the task description.
|
||||
* Shows filtered list of files based on the query after @.
|
||||
*/
|
||||
export function FileAutocomplete({
|
||||
query,
|
||||
projectPath,
|
||||
position,
|
||||
onSelect,
|
||||
onClose,
|
||||
maxResults = 10
|
||||
}: FileAutocompleteProps) {
|
||||
const [selectedIndex, setSelectedIndex] = useState(0);
|
||||
const listRef = useRef<HTMLDivElement>(null);
|
||||
const { files, loadDirectory } = useFileExplorerStore();
|
||||
|
||||
// Load root directory if not cached
|
||||
useEffect(() => {
|
||||
if (projectPath && !files.has(projectPath)) {
|
||||
loadDirectory(projectPath);
|
||||
}
|
||||
}, [projectPath, files, loadDirectory]);
|
||||
|
||||
// Collect all files from cache (flatten the tree)
|
||||
const allFiles = useMemo(() => {
|
||||
const result: FileNode[] = [];
|
||||
|
||||
// Recursive function to collect all cached files
|
||||
const collectFiles = (dirPath: string, visited = new Set<string>()) => {
|
||||
if (visited.has(dirPath)) return;
|
||||
visited.add(dirPath);
|
||||
|
||||
const dirFiles = files.get(dirPath);
|
||||
if (!dirFiles) return;
|
||||
|
||||
for (const file of dirFiles) {
|
||||
result.push(file);
|
||||
// For directories, also load and collect their children if cached
|
||||
if (file.isDirectory && files.has(file.path)) {
|
||||
collectFiles(file.path, visited);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
collectFiles(projectPath);
|
||||
return result;
|
||||
}, [files, projectPath]);
|
||||
|
||||
// Filter files based on query
|
||||
const filteredFiles = useMemo(() => {
|
||||
if (!query) {
|
||||
// Show most recently accessed or common files when no query
|
||||
return allFiles.filter(f => !f.isDirectory).slice(0, maxResults);
|
||||
}
|
||||
|
||||
const lowerQuery = query.toLowerCase();
|
||||
|
||||
// Score files by match quality
|
||||
const scored = allFiles
|
||||
.filter(f => !f.isDirectory) // Only files, not directories
|
||||
.map(file => {
|
||||
const name = file.name.toLowerCase();
|
||||
const path = file.path.toLowerCase();
|
||||
|
||||
let score = 0;
|
||||
|
||||
// Exact name match (highest priority)
|
||||
if (name === lowerQuery) {
|
||||
score = 1000;
|
||||
}
|
||||
// Name starts with query
|
||||
else if (name.startsWith(lowerQuery)) {
|
||||
score = 100;
|
||||
}
|
||||
// Name contains query
|
||||
else if (name.includes(lowerQuery)) {
|
||||
score = 50;
|
||||
}
|
||||
// Path contains query
|
||||
else if (path.includes(lowerQuery)) {
|
||||
score = 10;
|
||||
}
|
||||
|
||||
return { file, score };
|
||||
})
|
||||
.filter(item => item.score > 0)
|
||||
.sort((a, b) => b.score - a.score)
|
||||
.slice(0, maxResults)
|
||||
.map(item => item.file);
|
||||
|
||||
return scored;
|
||||
}, [allFiles, query, maxResults]);
|
||||
|
||||
// Reset selection when results change
|
||||
useEffect(() => {
|
||||
setSelectedIndex(0);
|
||||
}, [filteredFiles]);
|
||||
|
||||
// Scroll selected item into view
|
||||
useEffect(() => {
|
||||
const list = listRef.current;
|
||||
if (!list) return;
|
||||
|
||||
const selectedElement = list.children[selectedIndex] as HTMLElement;
|
||||
if (selectedElement) {
|
||||
selectedElement.scrollIntoView({ block: 'nearest' });
|
||||
}
|
||||
}, [selectedIndex]);
|
||||
|
||||
// Handle keyboard navigation
|
||||
const handleKeyDown = useCallback((e: KeyboardEvent) => {
|
||||
switch (e.key) {
|
||||
case 'ArrowDown':
|
||||
e.preventDefault();
|
||||
setSelectedIndex(prev =>
|
||||
prev < filteredFiles.length - 1 ? prev + 1 : prev
|
||||
);
|
||||
break;
|
||||
case 'ArrowUp':
|
||||
e.preventDefault();
|
||||
setSelectedIndex(prev => prev > 0 ? prev - 1 : prev);
|
||||
break;
|
||||
case 'Enter':
|
||||
e.preventDefault();
|
||||
if (filteredFiles[selectedIndex]) {
|
||||
const file = filteredFiles[selectedIndex];
|
||||
onSelect(file.name, file.path);
|
||||
}
|
||||
break;
|
||||
case 'Escape':
|
||||
e.preventDefault();
|
||||
onClose();
|
||||
break;
|
||||
case 'Tab':
|
||||
e.preventDefault();
|
||||
if (filteredFiles[selectedIndex]) {
|
||||
const file = filteredFiles[selectedIndex];
|
||||
onSelect(file.name, file.path);
|
||||
}
|
||||
break;
|
||||
}
|
||||
}, [filteredFiles, selectedIndex, onSelect, onClose]);
|
||||
|
||||
// Attach keyboard listener
|
||||
useEffect(() => {
|
||||
document.addEventListener('keydown', handleKeyDown);
|
||||
return () => document.removeEventListener('keydown', handleKeyDown);
|
||||
}, [handleKeyDown]);
|
||||
|
||||
// Get relative path from project root
|
||||
const getRelativePath = (fullPath: string) => {
|
||||
if (fullPath.startsWith(projectPath)) {
|
||||
return fullPath.slice(projectPath.length + 1); // +1 for the slash
|
||||
}
|
||||
return fullPath;
|
||||
};
|
||||
|
||||
// Don't render if no results
|
||||
if (filteredFiles.length === 0) {
|
||||
return (
|
||||
<div
|
||||
className="absolute z-50 bg-popover border border-border rounded-md shadow-lg p-3 text-sm text-muted-foreground"
|
||||
style={{
|
||||
top: position.top,
|
||||
left: position.left,
|
||||
minWidth: '200px'
|
||||
}}
|
||||
>
|
||||
No files found
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
return (
|
||||
<div
|
||||
className="absolute z-50 bg-popover border border-border rounded-md shadow-lg overflow-hidden"
|
||||
style={{
|
||||
top: position.top,
|
||||
left: position.left,
|
||||
minWidth: '280px',
|
||||
maxWidth: '400px',
|
||||
maxHeight: '240px'
|
||||
}}
|
||||
>
|
||||
<div
|
||||
ref={listRef}
|
||||
className="overflow-y-auto max-h-[240px]"
|
||||
>
|
||||
{filteredFiles.map((file, index) => (
|
||||
<button
|
||||
key={file.path}
|
||||
className={cn(
|
||||
'w-full flex items-center gap-2 px-3 py-2 text-left text-sm',
|
||||
'hover:bg-accent hover:text-accent-foreground',
|
||||
'focus:outline-none transition-colors',
|
||||
index === selectedIndex && 'bg-accent text-accent-foreground'
|
||||
)}
|
||||
onClick={() => onSelect(file.name, file.path)}
|
||||
onMouseEnter={() => setSelectedIndex(index)}
|
||||
>
|
||||
{file.isDirectory ? (
|
||||
<Folder className="h-4 w-4 text-muted-foreground shrink-0" />
|
||||
) : (
|
||||
<File className="h-4 w-4 text-muted-foreground shrink-0" />
|
||||
)}
|
||||
<div className="flex-1 min-w-0">
|
||||
<div className="font-medium truncate">{file.name}</div>
|
||||
<div className="text-xs text-muted-foreground truncate flex items-center gap-1">
|
||||
<ChevronRight className="h-3 w-3 shrink-0" />
|
||||
{getRelativePath(file.path)}
|
||||
</div>
|
||||
</div>
|
||||
</button>
|
||||
))}
|
||||
</div>
|
||||
<div className="border-t border-border px-3 py-1.5 text-[10px] text-muted-foreground bg-muted/30">
|
||||
<span className="font-medium">↑↓</span> navigate · <span className="font-medium">Enter</span> select · <span className="font-medium">Esc</span> close
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
@@ -7,7 +7,13 @@ import {
|
||||
CheckCircle2,
|
||||
AlertCircle,
|
||||
ChevronRight,
|
||||
Sparkles
|
||||
Sparkles,
|
||||
Plus,
|
||||
Link,
|
||||
Lock,
|
||||
Globe,
|
||||
Building,
|
||||
User
|
||||
} from 'lucide-react';
|
||||
import { Button } from './ui/button';
|
||||
import {
|
||||
@@ -19,6 +25,7 @@ import {
|
||||
DialogTitle
|
||||
} from './ui/dialog';
|
||||
import { Label } from './ui/label';
|
||||
import { Input } from './ui/input';
|
||||
import {
|
||||
Select,
|
||||
SelectContent,
|
||||
@@ -38,7 +45,7 @@ interface GitHubSetupModalProps {
|
||||
onSkip?: () => void;
|
||||
}
|
||||
|
||||
type SetupStep = 'github-auth' | 'claude-auth' | 'repo' | 'branch' | 'complete';
|
||||
type SetupStep = 'github-auth' | 'claude-auth' | 'repo-confirm' | 'repo' | 'branch' | 'complete';
|
||||
|
||||
/**
|
||||
* Setup Modal - Required setup flow after Auto Claude initialization
|
||||
@@ -67,10 +74,23 @@ export function GitHubSetupModal({
|
||||
const [isLoadingRepo, setIsLoadingRepo] = useState(false);
|
||||
const [error, setError] = useState<string | null>(null);
|
||||
|
||||
// Reset state when modal opens
|
||||
// Repo setup state (for when no remote is detected)
|
||||
const [repoAction, setRepoAction] = useState<'create' | 'link' | null>(null);
|
||||
const [newRepoName, setNewRepoName] = useState('');
|
||||
const [isPrivateRepo, setIsPrivateRepo] = useState(true);
|
||||
const [existingRepoName, setExistingRepoName] = useState('');
|
||||
const [isCreatingRepo, setIsCreatingRepo] = useState(false);
|
||||
|
||||
// Organization selection state
|
||||
const [githubUsername, setGithubUsername] = useState<string | null>(null);
|
||||
const [organizations, setOrganizations] = useState<Array<{ login: string; avatarUrl?: string }>>([]);
|
||||
const [selectedOwner, setSelectedOwner] = useState<string | null>(null);
|
||||
const [isLoadingOrgs, setIsLoadingOrgs] = useState(false);
|
||||
|
||||
// Reset state and check existing auth when modal opens
|
||||
useEffect(() => {
|
||||
if (open) {
|
||||
setStep('github-auth');
|
||||
// Reset all state first
|
||||
setGithubToken(null);
|
||||
setGithubRepo(null);
|
||||
setDetectedRepo(null);
|
||||
@@ -78,9 +98,84 @@ export function GitHubSetupModal({
|
||||
setSelectedBranch(null);
|
||||
setRecommendedBranch(null);
|
||||
setError(null);
|
||||
// Reset repo setup state
|
||||
setRepoAction(null);
|
||||
setNewRepoName(project.name.replace(/[^A-Za-z0-9_.-]/g, '-'));
|
||||
setIsPrivateRepo(true);
|
||||
setExistingRepoName('');
|
||||
setIsCreatingRepo(false);
|
||||
// Reset organization state
|
||||
setGithubUsername(null);
|
||||
setOrganizations([]);
|
||||
setSelectedOwner(null);
|
||||
setIsLoadingOrgs(false);
|
||||
|
||||
// Check for existing authentication and skip to appropriate step
|
||||
const checkExistingAuth = async () => {
|
||||
try {
|
||||
// Check for existing GitHub token
|
||||
const ghTokenResult = await window.electronAPI.getGitHubToken();
|
||||
const hasGitHubAuth = ghTokenResult.success && ghTokenResult.data?.token;
|
||||
|
||||
// Check for existing Claude authentication
|
||||
const profilesResult = await window.electronAPI.getClaudeProfiles();
|
||||
let hasClaudeAuth = false;
|
||||
if (profilesResult.success && profilesResult.data) {
|
||||
const activeProfile = profilesResult.data.profiles.find(
|
||||
(p) => p.id === profilesResult.data!.activeProfileId
|
||||
);
|
||||
hasClaudeAuth = !!(activeProfile?.oauthToken || (activeProfile?.isDefault && activeProfile?.configDir));
|
||||
}
|
||||
|
||||
// Determine starting step based on existing auth
|
||||
if (hasGitHubAuth && hasClaudeAuth) {
|
||||
// Both authenticated, go directly to repo detection
|
||||
setGithubToken(ghTokenResult.data!.token);
|
||||
// detectRepository will be called and set the step
|
||||
setStep('repo'); // Temporary, detectRepository will update
|
||||
await detectRepository();
|
||||
} else if (hasGitHubAuth) {
|
||||
// Only GitHub authenticated, go to Claude auth
|
||||
setGithubToken(ghTokenResult.data!.token);
|
||||
setStep('claude-auth');
|
||||
} else {
|
||||
// No auth, start from beginning
|
||||
setStep('github-auth');
|
||||
}
|
||||
} catch (err) {
|
||||
console.error('Failed to check existing auth:', err);
|
||||
// On error, start from beginning
|
||||
setStep('github-auth');
|
||||
}
|
||||
};
|
||||
|
||||
checkExistingAuth();
|
||||
}
|
||||
}, [open]);
|
||||
|
||||
// Load user info and organizations
|
||||
const loadUserAndOrgs = async () => {
|
||||
setIsLoadingOrgs(true);
|
||||
try {
|
||||
// Get current user
|
||||
const userResult = await window.electronAPI.getGitHubUser();
|
||||
if (userResult.success && userResult.data) {
|
||||
setGithubUsername(userResult.data.username);
|
||||
setSelectedOwner(userResult.data.username); // Default to personal account
|
||||
}
|
||||
|
||||
// Get organizations
|
||||
const orgsResult = await window.electronAPI.listGitHubOrgs();
|
||||
if (orgsResult.success && orgsResult.data) {
|
||||
setOrganizations(orgsResult.data.orgs);
|
||||
}
|
||||
} catch (err) {
|
||||
console.error('Failed to load user/orgs:', err);
|
||||
} finally {
|
||||
setIsLoadingOrgs(false);
|
||||
}
|
||||
};
|
||||
|
||||
// Detect repository from git remote when auth succeeds
|
||||
const detectRepository = async () => {
|
||||
setIsLoadingRepo(true);
|
||||
@@ -92,15 +187,16 @@ export function GitHubSetupModal({
|
||||
if (result.success && result.data) {
|
||||
setDetectedRepo(result.data);
|
||||
setGithubRepo(result.data);
|
||||
setStep('branch');
|
||||
// Immediately load branches
|
||||
await loadBranches(result.data);
|
||||
// Go to confirmation step instead of directly to branch
|
||||
setStep('repo-confirm');
|
||||
} else {
|
||||
// No remote detected, show repo input step
|
||||
// No remote detected, load orgs and show repo setup step
|
||||
await loadUserAndOrgs();
|
||||
setStep('repo');
|
||||
}
|
||||
} catch (err) {
|
||||
setError(err instanceof Error ? err.message : 'Failed to detect repository');
|
||||
await loadUserAndOrgs();
|
||||
setStep('repo');
|
||||
} finally {
|
||||
setIsLoadingRepo(false);
|
||||
@@ -146,7 +242,27 @@ export function GitHubSetupModal({
|
||||
// Handle GitHub OAuth success
|
||||
const handleGitHubAuthSuccess = async (token: string) => {
|
||||
setGithubToken(token);
|
||||
// Move to Claude auth step
|
||||
|
||||
// Check if Claude is already authenticated before showing auth step
|
||||
try {
|
||||
const profilesResult = await window.electronAPI.getClaudeProfiles();
|
||||
if (profilesResult.success && profilesResult.data) {
|
||||
const activeProfile = profilesResult.data.profiles.find(
|
||||
(p) => p.id === profilesResult.data!.activeProfileId
|
||||
);
|
||||
// Check if active profile has authentication (oauthToken or default with configDir)
|
||||
if (activeProfile?.oauthToken || (activeProfile?.isDefault && activeProfile?.configDir)) {
|
||||
// Already authenticated, skip Claude auth and go directly to repo detection
|
||||
await detectRepository();
|
||||
return;
|
||||
}
|
||||
}
|
||||
} catch (err) {
|
||||
console.error('Failed to check Claude profiles:', err);
|
||||
// On error, fall through to show Claude auth step
|
||||
}
|
||||
|
||||
// Not authenticated, show Claude auth step
|
||||
setStep('claude-auth');
|
||||
};
|
||||
|
||||
@@ -157,6 +273,92 @@ export function GitHubSetupModal({
|
||||
await detectRepository();
|
||||
};
|
||||
|
||||
// Handle creating a new GitHub repository
|
||||
const handleCreateRepo = async () => {
|
||||
if (!newRepoName.trim()) {
|
||||
setError('Please enter a repository name');
|
||||
return;
|
||||
}
|
||||
|
||||
if (!selectedOwner) {
|
||||
setError('Please select an owner for the repository');
|
||||
return;
|
||||
}
|
||||
|
||||
setIsCreatingRepo(true);
|
||||
setError(null);
|
||||
|
||||
try {
|
||||
const result = await window.electronAPI.createGitHubRepo(newRepoName.trim(), {
|
||||
isPrivate: isPrivateRepo,
|
||||
projectPath: project.path,
|
||||
owner: selectedOwner !== githubUsername ? selectedOwner : undefined // Only pass owner if it's an org
|
||||
});
|
||||
|
||||
if (result.success && result.data) {
|
||||
// Repo created and remote added automatically by gh CLI
|
||||
setGithubRepo(result.data.fullName);
|
||||
setDetectedRepo(result.data.fullName);
|
||||
setStep('branch');
|
||||
await loadBranches(result.data.fullName);
|
||||
} else {
|
||||
setError(result.error || 'Failed to create repository');
|
||||
}
|
||||
} catch (err) {
|
||||
setError(err instanceof Error ? err.message : 'Failed to create repository');
|
||||
} finally {
|
||||
setIsCreatingRepo(false);
|
||||
}
|
||||
};
|
||||
|
||||
// Handle confirming the detected repository
|
||||
const handleConfirmRepo = async () => {
|
||||
if (detectedRepo) {
|
||||
setStep('branch');
|
||||
await loadBranches(detectedRepo);
|
||||
}
|
||||
};
|
||||
|
||||
// Handle changing the repository (go to repo setup)
|
||||
const handleChangeRepo = async () => {
|
||||
await loadUserAndOrgs();
|
||||
setStep('repo');
|
||||
};
|
||||
|
||||
// Handle linking to an existing GitHub repository
|
||||
const handleLinkRepo = async () => {
|
||||
if (!existingRepoName.trim()) {
|
||||
setError('Please enter a repository name (owner/repo format)');
|
||||
return;
|
||||
}
|
||||
|
||||
// Validate format
|
||||
if (!/^[A-Za-z0-9_.-]+\/[A-Za-z0-9_.-]+$/.test(existingRepoName.trim())) {
|
||||
setError('Invalid format. Use owner/repo (e.g., username/my-project)');
|
||||
return;
|
||||
}
|
||||
|
||||
setIsCreatingRepo(true);
|
||||
setError(null);
|
||||
|
||||
try {
|
||||
const result = await window.electronAPI.addGitRemote(project.path, existingRepoName.trim());
|
||||
|
||||
if (result.success) {
|
||||
setGithubRepo(existingRepoName.trim());
|
||||
setDetectedRepo(existingRepoName.trim());
|
||||
setStep('branch');
|
||||
await loadBranches(existingRepoName.trim());
|
||||
} else {
|
||||
setError(result.error || 'Failed to add remote');
|
||||
}
|
||||
} catch (err) {
|
||||
setError(err instanceof Error ? err.message : 'Failed to add remote');
|
||||
} finally {
|
||||
setIsCreatingRepo(false);
|
||||
}
|
||||
};
|
||||
|
||||
// Handle branch selection complete
|
||||
const handleComplete = () => {
|
||||
if (githubToken && githubRepo && selectedBranch) {
|
||||
@@ -215,35 +417,236 @@ export function GitHubSetupModal({
|
||||
</>
|
||||
);
|
||||
|
||||
case 'repo-confirm':
|
||||
return (
|
||||
<>
|
||||
<DialogHeader>
|
||||
<DialogTitle className="flex items-center gap-2">
|
||||
<Github className="h-5 w-5" />
|
||||
Confirm Repository
|
||||
</DialogTitle>
|
||||
<DialogDescription>
|
||||
We detected a GitHub repository for this project. Please confirm or change it.
|
||||
</DialogDescription>
|
||||
</DialogHeader>
|
||||
|
||||
<div className="py-4 space-y-4">
|
||||
<div className="rounded-lg border bg-muted/50 p-4">
|
||||
<div className="flex items-center gap-3">
|
||||
<CheckCircle2 className="h-6 w-6 text-green-500" />
|
||||
<div>
|
||||
<p className="font-medium">Repository Detected</p>
|
||||
<p className="text-sm text-muted-foreground font-mono">
|
||||
{detectedRepo}
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<p className="text-sm text-muted-foreground">
|
||||
Auto Claude will use this repository for managing task branches and keeping your code up to date.
|
||||
</p>
|
||||
|
||||
{error && (
|
||||
<div className="rounded-lg bg-destructive/10 border border-destructive/30 p-3 text-sm text-destructive">
|
||||
{error}
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
|
||||
<DialogFooter>
|
||||
<Button variant="outline" onClick={handleChangeRepo}>
|
||||
Use Different Repository
|
||||
</Button>
|
||||
<Button onClick={handleConfirmRepo}>
|
||||
<CheckCircle2 className="mr-2 h-4 w-4" />
|
||||
Confirm & Continue
|
||||
</Button>
|
||||
</DialogFooter>
|
||||
</>
|
||||
);
|
||||
|
||||
case 'repo':
|
||||
return (
|
||||
<>
|
||||
<DialogHeader>
|
||||
<DialogTitle className="flex items-center gap-2">
|
||||
<Github className="h-5 w-5" />
|
||||
Repository Not Detected
|
||||
Connect to GitHub
|
||||
</DialogTitle>
|
||||
<DialogDescription>
|
||||
We couldn't detect a GitHub repository for this project. Please ensure your project has a GitHub remote configured.
|
||||
Your project needs a GitHub repository. Create a new one or link to an existing repository.
|
||||
</DialogDescription>
|
||||
</DialogHeader>
|
||||
|
||||
<div className="py-4 space-y-4">
|
||||
<div className="rounded-lg border border-warning/30 bg-warning/10 p-4">
|
||||
<div className="flex items-start gap-3">
|
||||
<AlertCircle className="h-5 w-5 text-warning mt-0.5" />
|
||||
{/* Action selection */}
|
||||
{!repoAction && (
|
||||
<div className="grid grid-cols-2 gap-3">
|
||||
<button
|
||||
onClick={() => setRepoAction('create')}
|
||||
className="flex flex-col items-center gap-2 p-4 rounded-lg border-2 border-dashed hover:border-primary hover:bg-primary/5 transition-colors"
|
||||
>
|
||||
<Plus className="h-8 w-8 text-muted-foreground" />
|
||||
<span className="text-sm font-medium">Create New Repo</span>
|
||||
<span className="text-xs text-muted-foreground text-center">
|
||||
Create a new repository on GitHub
|
||||
</span>
|
||||
</button>
|
||||
<button
|
||||
onClick={() => setRepoAction('link')}
|
||||
className="flex flex-col items-center gap-2 p-4 rounded-lg border-2 border-dashed hover:border-primary hover:bg-primary/5 transition-colors"
|
||||
>
|
||||
<Link className="h-8 w-8 text-muted-foreground" />
|
||||
<span className="text-sm font-medium">Link Existing</span>
|
||||
<span className="text-xs text-muted-foreground text-center">
|
||||
Connect to an existing repository
|
||||
</span>
|
||||
</button>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Create new repo form */}
|
||||
{repoAction === 'create' && (
|
||||
<div className="space-y-4">
|
||||
<div className="flex items-center gap-2 text-sm text-muted-foreground">
|
||||
<button
|
||||
onClick={() => setRepoAction(null)}
|
||||
className="text-primary hover:underline"
|
||||
>
|
||||
← Back
|
||||
</button>
|
||||
<span>Create a new repository</span>
|
||||
</div>
|
||||
|
||||
{/* Owner selection */}
|
||||
<div className="space-y-2">
|
||||
<p className="text-sm font-medium">No GitHub remote found</p>
|
||||
<Label>Owner</Label>
|
||||
{isLoadingOrgs ? (
|
||||
<div className="flex items-center gap-2 text-sm text-muted-foreground">
|
||||
<Loader2 className="h-4 w-4 animate-spin" />
|
||||
Loading accounts...
|
||||
</div>
|
||||
) : (
|
||||
<div className="flex flex-wrap gap-2">
|
||||
{/* Personal account */}
|
||||
{githubUsername && (
|
||||
<button
|
||||
onClick={() => setSelectedOwner(githubUsername)}
|
||||
className={`flex items-center gap-2 px-3 py-2 rounded-md border ${
|
||||
selectedOwner === githubUsername
|
||||
? 'border-primary bg-primary/10 text-primary'
|
||||
: 'border-muted hover:border-primary/50'
|
||||
}`}
|
||||
disabled={isCreatingRepo}
|
||||
>
|
||||
<User className="h-4 w-4" />
|
||||
<span className="text-sm">{githubUsername}</span>
|
||||
</button>
|
||||
)}
|
||||
{/* Organizations */}
|
||||
{organizations.map((org) => (
|
||||
<button
|
||||
key={org.login}
|
||||
onClick={() => setSelectedOwner(org.login)}
|
||||
className={`flex items-center gap-2 px-3 py-2 rounded-md border ${
|
||||
selectedOwner === org.login
|
||||
? 'border-primary bg-primary/10 text-primary'
|
||||
: 'border-muted hover:border-primary/50'
|
||||
}`}
|
||||
disabled={isCreatingRepo}
|
||||
>
|
||||
<Building className="h-4 w-4" />
|
||||
<span className="text-sm">{org.login}</span>
|
||||
</button>
|
||||
))}
|
||||
</div>
|
||||
)}
|
||||
{organizations.length > 0 && (
|
||||
<p className="text-xs text-muted-foreground">
|
||||
To use Auto Claude, your project needs to be connected to a GitHub repository.
|
||||
Select your personal account or an organization
|
||||
</p>
|
||||
<div className="text-xs font-mono bg-muted p-2 rounded mt-2">
|
||||
git remote add origin https://github.com/owner/repo.git
|
||||
)}
|
||||
</div>
|
||||
|
||||
<div className="space-y-2">
|
||||
<Label htmlFor="repo-name">Repository Name</Label>
|
||||
<div className="flex items-center gap-2">
|
||||
<span className="text-sm text-muted-foreground">
|
||||
{selectedOwner || '...'} /
|
||||
</span>
|
||||
<Input
|
||||
id="repo-name"
|
||||
value={newRepoName}
|
||||
onChange={(e) => setNewRepoName(e.target.value)}
|
||||
placeholder="my-project"
|
||||
disabled={isCreatingRepo}
|
||||
className="flex-1"
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="space-y-2">
|
||||
<Label>Visibility</Label>
|
||||
<div className="flex gap-2">
|
||||
<button
|
||||
onClick={() => setIsPrivateRepo(true)}
|
||||
className={`flex items-center gap-2 px-3 py-2 rounded-md border ${
|
||||
isPrivateRepo
|
||||
? 'border-primary bg-primary/10 text-primary'
|
||||
: 'border-muted hover:border-primary/50'
|
||||
}`}
|
||||
disabled={isCreatingRepo}
|
||||
>
|
||||
<Lock className="h-4 w-4" />
|
||||
<span className="text-sm">Private</span>
|
||||
</button>
|
||||
<button
|
||||
onClick={() => setIsPrivateRepo(false)}
|
||||
className={`flex items-center gap-2 px-3 py-2 rounded-md border ${
|
||||
!isPrivateRepo
|
||||
? 'border-primary bg-primary/10 text-primary'
|
||||
: 'border-muted hover:border-primary/50'
|
||||
}`}
|
||||
disabled={isCreatingRepo}
|
||||
>
|
||||
<Globe className="h-4 w-4" />
|
||||
<span className="text-sm">Public</span>
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Link existing repo form */}
|
||||
{repoAction === 'link' && (
|
||||
<div className="space-y-4">
|
||||
<div className="flex items-center gap-2 text-sm text-muted-foreground">
|
||||
<button
|
||||
onClick={() => setRepoAction(null)}
|
||||
className="text-primary hover:underline"
|
||||
>
|
||||
← Back
|
||||
</button>
|
||||
<span>Link to existing repository</span>
|
||||
</div>
|
||||
|
||||
<div className="space-y-2">
|
||||
<Label htmlFor="existing-repo">Repository</Label>
|
||||
<Input
|
||||
id="existing-repo"
|
||||
value={existingRepoName}
|
||||
onChange={(e) => setExistingRepoName(e.target.value)}
|
||||
placeholder="username/repository"
|
||||
disabled={isCreatingRepo}
|
||||
/>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Enter the full repository path (e.g., octocat/hello-world)
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{error && (
|
||||
<div className="rounded-lg bg-destructive/10 border border-destructive/30 p-3 text-sm text-destructive">
|
||||
{error}
|
||||
@@ -253,11 +656,42 @@ export function GitHubSetupModal({
|
||||
|
||||
<DialogFooter>
|
||||
{onSkip && (
|
||||
<Button variant="outline" onClick={onSkip}>
|
||||
<Button variant="outline" onClick={onSkip} disabled={isCreatingRepo}>
|
||||
Skip for now
|
||||
</Button>
|
||||
)}
|
||||
<Button onClick={detectRepository} disabled={isLoadingRepo}>
|
||||
{repoAction === 'create' && (
|
||||
<Button onClick={handleCreateRepo} disabled={isCreatingRepo || !newRepoName.trim()}>
|
||||
{isCreatingRepo ? (
|
||||
<>
|
||||
<Loader2 className="mr-2 h-4 w-4 animate-spin" />
|
||||
Creating...
|
||||
</>
|
||||
) : (
|
||||
<>
|
||||
<Plus className="mr-2 h-4 w-4" />
|
||||
Create Repository
|
||||
</>
|
||||
)}
|
||||
</Button>
|
||||
)}
|
||||
{repoAction === 'link' && (
|
||||
<Button onClick={handleLinkRepo} disabled={isCreatingRepo || !existingRepoName.trim()}>
|
||||
{isCreatingRepo ? (
|
||||
<>
|
||||
<Loader2 className="mr-2 h-4 w-4 animate-spin" />
|
||||
Linking...
|
||||
</>
|
||||
) : (
|
||||
<>
|
||||
<Link className="mr-2 h-4 w-4" />
|
||||
Link Repository
|
||||
</>
|
||||
)}
|
||||
</Button>
|
||||
)}
|
||||
{!repoAction && (
|
||||
<Button variant="outline" onClick={detectRepository} disabled={isLoadingRepo}>
|
||||
{isLoadingRepo ? (
|
||||
<>
|
||||
<Loader2 className="mr-2 h-4 w-4 animate-spin" />
|
||||
@@ -267,6 +701,7 @@ export function GitHubSetupModal({
|
||||
'Retry Detection'
|
||||
)}
|
||||
</Button>
|
||||
)}
|
||||
</DialogFooter>
|
||||
</>
|
||||
);
|
||||
|
||||
@@ -79,14 +79,9 @@ export function ProjectSettings({ project, open, onOpenChange }: ProjectSettings
|
||||
const {
|
||||
infrastructureStatus,
|
||||
isCheckingInfrastructure,
|
||||
isStartingFalkorDB,
|
||||
isOpeningDocker,
|
||||
handleStartFalkorDB,
|
||||
handleOpenDockerDesktop,
|
||||
handleDownloadDocker,
|
||||
} = useInfrastructureStatus(
|
||||
envConfig?.graphitiEnabled,
|
||||
envConfig?.graphitiFalkorDbPort,
|
||||
envConfig?.graphitiDbPath,
|
||||
open
|
||||
);
|
||||
|
||||
@@ -253,11 +248,6 @@ export function ProjectSettings({ project, open, onOpenChange }: ProjectSettings
|
||||
onUpdateSettings={(updates) => setSettings({ ...settings, ...updates })}
|
||||
infrastructureStatus={infrastructureStatus}
|
||||
isCheckingInfrastructure={isCheckingInfrastructure}
|
||||
isStartingFalkorDB={isStartingFalkorDB}
|
||||
isOpeningDocker={isOpeningDocker}
|
||||
onStartFalkorDB={handleStartFalkorDB}
|
||||
onOpenDockerDesktop={handleOpenDockerDesktop}
|
||||
onDownloadDocker={handleDownloadDocker}
|
||||
/>
|
||||
|
||||
<Separator />
|
||||
|
||||
@@ -0,0 +1,115 @@
|
||||
import { useEffect } from 'react';
|
||||
import { Plus } from 'lucide-react';
|
||||
import { cn } from '../lib/utils';
|
||||
import { Button } from './ui/button';
|
||||
import { SortableProjectTab } from './SortableProjectTab';
|
||||
import type { Project } from '../../shared/types';
|
||||
|
||||
interface ProjectTabBarProps {
|
||||
projects: Project[];
|
||||
activeProjectId: string | null;
|
||||
onProjectSelect: (projectId: string) => void;
|
||||
onProjectClose: (projectId: string) => void;
|
||||
onAddProject: () => void;
|
||||
className?: string;
|
||||
}
|
||||
|
||||
export function ProjectTabBar({
|
||||
projects,
|
||||
activeProjectId,
|
||||
onProjectSelect,
|
||||
onProjectClose,
|
||||
onAddProject,
|
||||
className
|
||||
}: ProjectTabBarProps) {
|
||||
// Keyboard shortcuts for tab navigation
|
||||
useEffect(() => {
|
||||
const handleKeyDown = (e: KeyboardEvent) => {
|
||||
// Skip if in input fields
|
||||
if (
|
||||
e.target instanceof HTMLInputElement ||
|
||||
e.target instanceof HTMLTextAreaElement ||
|
||||
(e.target as HTMLElement)?.isContentEditable
|
||||
) {
|
||||
return;
|
||||
}
|
||||
|
||||
const isMod = e.metaKey || e.ctrlKey;
|
||||
if (!isMod) return;
|
||||
|
||||
// Cmd/Ctrl + 1-9: Switch to tab N
|
||||
if (e.key >= '1' && e.key <= '9') {
|
||||
e.preventDefault();
|
||||
const index = parseInt(e.key) - 1;
|
||||
if (index < projects.length) {
|
||||
onProjectSelect(projects[index].id);
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
// Cmd/Ctrl + Tab: Next tab
|
||||
// Cmd/Ctrl + Shift + Tab: Previous tab
|
||||
if (e.key === 'Tab') {
|
||||
e.preventDefault();
|
||||
const currentIndex = projects.findIndex((p) => p.id === activeProjectId);
|
||||
if (currentIndex === -1 || projects.length === 0) return;
|
||||
|
||||
const nextIndex = e.shiftKey
|
||||
? (currentIndex - 1 + projects.length) % projects.length
|
||||
: (currentIndex + 1) % projects.length;
|
||||
onProjectSelect(projects[nextIndex].id);
|
||||
return;
|
||||
}
|
||||
|
||||
// Cmd/Ctrl + W: Close current tab (only if more than one tab)
|
||||
if (e.key === 'w' && activeProjectId && projects.length > 1) {
|
||||
e.preventDefault();
|
||||
onProjectClose(activeProjectId);
|
||||
}
|
||||
};
|
||||
|
||||
window.addEventListener('keydown', handleKeyDown);
|
||||
return () => window.removeEventListener('keydown', handleKeyDown);
|
||||
}, [projects, activeProjectId, onProjectSelect, onProjectClose]);
|
||||
|
||||
if (projects.length === 0) {
|
||||
return null;
|
||||
}
|
||||
|
||||
return (
|
||||
<div className={cn(
|
||||
'flex items-center border-b border-border bg-background',
|
||||
'overflow-x-auto scrollbar-thin scrollbar-thumb-border scrollbar-track-transparent',
|
||||
className
|
||||
)}>
|
||||
<div className="flex items-center flex-1 min-w-0">
|
||||
{projects.map((project, index) => (
|
||||
<SortableProjectTab
|
||||
key={project.id}
|
||||
project={project}
|
||||
isActive={activeProjectId === project.id}
|
||||
canClose={projects.length > 1}
|
||||
tabIndex={index}
|
||||
onSelect={() => onProjectSelect(project.id)}
|
||||
onClose={(e) => {
|
||||
e.stopPropagation();
|
||||
onProjectClose(project.id);
|
||||
}}
|
||||
/>
|
||||
))}
|
||||
</div>
|
||||
|
||||
<div className="flex items-center px-2 py-1">
|
||||
<Button
|
||||
variant="ghost"
|
||||
size="icon"
|
||||
className="h-8 w-8"
|
||||
onClick={onAddProject}
|
||||
title="Add Project"
|
||||
>
|
||||
<Plus className="h-4 w-4" />
|
||||
</Button>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
@@ -1,6 +1,5 @@
|
||||
import { useState, useEffect } from 'react';
|
||||
import {
|
||||
FolderOpen,
|
||||
Plus,
|
||||
Settings,
|
||||
Trash2,
|
||||
@@ -21,13 +20,6 @@ import {
|
||||
import { Button } from './ui/button';
|
||||
import { ScrollArea } from './ui/scroll-area';
|
||||
import { Separator } from './ui/separator';
|
||||
import {
|
||||
Select,
|
||||
SelectContent,
|
||||
SelectItem,
|
||||
SelectTrigger,
|
||||
SelectValue
|
||||
} from './ui/select';
|
||||
import {
|
||||
Tooltip,
|
||||
TooltipContent,
|
||||
@@ -255,13 +247,6 @@ export function Sidebar({
|
||||
await removeProject(projectId);
|
||||
};
|
||||
|
||||
const handleProjectChange = (projectId: string) => {
|
||||
if (projectId === '__add_new__') {
|
||||
handleAddProject();
|
||||
} else {
|
||||
selectProject(projectId);
|
||||
}
|
||||
};
|
||||
|
||||
const handleNavClick = (view: SidebarView) => {
|
||||
onViewChange?.(view);
|
||||
@@ -304,67 +289,6 @@ export function Sidebar({
|
||||
|
||||
<Separator className="mt-2" />
|
||||
|
||||
{/* Project Selector Dropdown */}
|
||||
<div className="px-4 py-4">
|
||||
<Select
|
||||
value={selectedProjectId || ''}
|
||||
onValueChange={handleProjectChange}
|
||||
>
|
||||
<SelectTrigger className="w-full [&_span]:truncate">
|
||||
<div className="flex items-center gap-2 min-w-0 flex-1 overflow-hidden">
|
||||
<FolderOpen className="h-4 w-4 shrink-0 text-muted-foreground" />
|
||||
<SelectValue placeholder="Select a project..." className="truncate min-w-0 flex-1" />
|
||||
</div>
|
||||
</SelectTrigger>
|
||||
<SelectContent className="min-w-(--radix-select-trigger-width) max-w-(--radix-select-trigger-width)">
|
||||
{projects.length === 0 ? (
|
||||
<div className="px-2 py-4 text-center text-sm text-muted-foreground">
|
||||
<p>No projects yet</p>
|
||||
</div>
|
||||
) : (
|
||||
projects.map((project) => (
|
||||
<div key={project.id} className="relative flex items-center">
|
||||
<SelectItem value={project.id} className="flex-1 pr-10">
|
||||
<span className="truncate" title={`${project.name} - ${project.path}`}>
|
||||
{project.name}
|
||||
</span>
|
||||
</SelectItem>
|
||||
<button
|
||||
type="button"
|
||||
className="absolute right-2 flex h-6 w-6 items-center justify-center rounded-md hover:bg-destructive/10 transition-colors"
|
||||
onPointerDown={(e) => {
|
||||
e.stopPropagation();
|
||||
}}
|
||||
onClick={(e) => {
|
||||
e.stopPropagation();
|
||||
e.preventDefault();
|
||||
removeProject(project.id);
|
||||
}}
|
||||
>
|
||||
<Trash2 className="h-3.5 w-3.5 text-destructive" />
|
||||
</button>
|
||||
</div>
|
||||
))
|
||||
)}
|
||||
<Separator className="my-1" />
|
||||
<SelectItem value="__add_new__">
|
||||
<div className="flex items-center gap-2">
|
||||
<Plus className="h-4 w-4 shrink-0" />
|
||||
<span>Add Project...</span>
|
||||
</div>
|
||||
</SelectItem>
|
||||
</SelectContent>
|
||||
</Select>
|
||||
|
||||
{/* Project path - shown when project is selected */}
|
||||
{selectedProject && (
|
||||
<div className="mt-2">
|
||||
<span className="truncate block text-xs text-muted-foreground" title={selectedProject.path}>
|
||||
{selectedProject.path}
|
||||
</span>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
|
||||
<Separator />
|
||||
|
||||
|
||||
@@ -0,0 +1,129 @@
|
||||
import { useSortable } from '@dnd-kit/sortable';
|
||||
import { CSS } from '@dnd-kit/utilities';
|
||||
import { cn } from '../lib/utils';
|
||||
import { Tooltip, TooltipContent, TooltipTrigger } from './ui/tooltip';
|
||||
import type { Project } from '../../shared/types';
|
||||
|
||||
interface SortableProjectTabProps {
|
||||
project: Project;
|
||||
isActive: boolean;
|
||||
canClose: boolean;
|
||||
tabIndex: number;
|
||||
onSelect: () => void;
|
||||
onClose: (e: React.MouseEvent) => void;
|
||||
}
|
||||
|
||||
// Detect if running on macOS for keyboard shortcut display
|
||||
const isMac = typeof navigator !== 'undefined' && navigator.platform.toUpperCase().indexOf('MAC') >= 0;
|
||||
const modKey = isMac ? '⌘' : 'Ctrl+';
|
||||
|
||||
export function SortableProjectTab({
|
||||
project,
|
||||
isActive,
|
||||
canClose,
|
||||
tabIndex,
|
||||
onSelect,
|
||||
onClose
|
||||
}: SortableProjectTabProps) {
|
||||
// Build tooltip with keyboard shortcut hint (only for tabs 1-9)
|
||||
const shortcutHint = tabIndex < 9 ? `${modKey}${tabIndex + 1}` : '';
|
||||
const closeShortcut = `${modKey}W`;
|
||||
const {
|
||||
attributes,
|
||||
listeners,
|
||||
setNodeRef,
|
||||
transform,
|
||||
transition,
|
||||
isDragging
|
||||
} = useSortable({ id: project.id });
|
||||
|
||||
const style = {
|
||||
transform: CSS.Transform.toString(transform),
|
||||
transition,
|
||||
// Prevent z-index stacking issues during drag
|
||||
zIndex: isDragging ? 50 : undefined
|
||||
};
|
||||
|
||||
return (
|
||||
<div
|
||||
ref={setNodeRef}
|
||||
style={style}
|
||||
className={cn(
|
||||
'group relative flex items-center min-w-0 max-w-[200px]',
|
||||
'border-r border-border last:border-r-0',
|
||||
'touch-none transition-all duration-200',
|
||||
isDragging && 'opacity-60 scale-[0.98] shadow-lg'
|
||||
)}
|
||||
{...attributes}
|
||||
>
|
||||
<Tooltip delayDuration={200}>
|
||||
<TooltipTrigger asChild>
|
||||
<div
|
||||
className={cn(
|
||||
'flex-1 flex items-center gap-2 px-4 py-2.5 text-sm',
|
||||
'min-w-0 truncate hover:bg-muted/50 transition-colors',
|
||||
'border-b-2 border-transparent cursor-pointer',
|
||||
isActive && [
|
||||
'bg-background border-b-primary text-foreground',
|
||||
'hover:bg-background'
|
||||
],
|
||||
!isActive && [
|
||||
'text-muted-foreground',
|
||||
'hover:text-foreground'
|
||||
]
|
||||
)}
|
||||
onClick={onSelect}
|
||||
>
|
||||
{/* Drag handle - visible on hover */}
|
||||
<div
|
||||
{...listeners}
|
||||
className={cn(
|
||||
'opacity-0 group-hover:opacity-60 transition-opacity',
|
||||
'cursor-grab active:cursor-grabbing',
|
||||
'w-1 h-4 bg-muted-foreground rounded-full'
|
||||
)}
|
||||
/>
|
||||
<span className="truncate font-medium">
|
||||
{project.name}
|
||||
</span>
|
||||
</div>
|
||||
</TooltipTrigger>
|
||||
<TooltipContent side="bottom" className="flex items-center gap-2">
|
||||
<span>{project.name}</span>
|
||||
{shortcutHint && (
|
||||
<kbd className="px-1.5 py-0.5 text-xs bg-muted rounded border border-border font-mono">
|
||||
{shortcutHint}
|
||||
</kbd>
|
||||
)}
|
||||
</TooltipContent>
|
||||
</Tooltip>
|
||||
|
||||
{canClose && (
|
||||
<Tooltip delayDuration={200}>
|
||||
<TooltipTrigger asChild>
|
||||
<button
|
||||
className={cn(
|
||||
'h-6 w-6 p-0 mr-1 opacity-0 group-hover:opacity-100',
|
||||
'transition-opacity duration-200 rounded',
|
||||
'hover:bg-destructive hover:text-destructive-foreground',
|
||||
'flex items-center justify-center',
|
||||
isActive && 'opacity-100'
|
||||
)}
|
||||
onClick={onClose}
|
||||
>
|
||||
<svg className="h-3 w-3" fill="none" viewBox="0 0 24 24" stroke="currentColor">
|
||||
<path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M6 18L18 6M6 6l12 12" />
|
||||
</svg>
|
||||
</button>
|
||||
</TooltipTrigger>
|
||||
<TooltipContent side="bottom" className="flex items-center gap-2">
|
||||
<span>Close tab</span>
|
||||
<kbd className="px-1.5 py-0.5 text-xs bg-muted rounded border border-border font-mono">
|
||||
{closeShortcut}
|
||||
</kbd>
|
||||
</TooltipContent>
|
||||
</Tooltip>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
@@ -21,16 +21,15 @@ import {
|
||||
SelectValue
|
||||
} from './ui/select';
|
||||
import {
|
||||
ImageUpload,
|
||||
generateImageId,
|
||||
blobToBase64,
|
||||
createThumbnail,
|
||||
isValidImageMimeType,
|
||||
resolveFilename
|
||||
} from './ImageUpload';
|
||||
import { ReferencedFilesSection } from './ReferencedFilesSection';
|
||||
import { TaskFileExplorerDrawer } from './TaskFileExplorerDrawer';
|
||||
import { AgentProfileSelector } from './AgentProfileSelector';
|
||||
import { FileAutocomplete } from './FileAutocomplete';
|
||||
import { createTask, saveDraft, loadDraft, clearDraft, isDraftEmpty } from '../stores/task-store';
|
||||
import { useProjectStore } from '../stores/project-store';
|
||||
import { cn } from '../lib/utils';
|
||||
@@ -42,7 +41,6 @@ import {
|
||||
TASK_COMPLEXITY_LABELS,
|
||||
TASK_IMPACT_LABELS,
|
||||
MAX_IMAGES_PER_TASK,
|
||||
MAX_REFERENCED_FILES,
|
||||
ALLOWED_IMAGE_TYPES_DISPLAY,
|
||||
DEFAULT_AGENT_PROFILES,
|
||||
DEFAULT_PHASE_MODELS,
|
||||
@@ -72,7 +70,6 @@ export function TaskCreationWizard({
|
||||
const [isCreating, setIsCreating] = useState(false);
|
||||
const [error, setError] = useState<string | null>(null);
|
||||
const [showAdvanced, setShowAdvanced] = useState(false);
|
||||
const [showImages, setShowImages] = useState(false);
|
||||
const [showFileExplorer, setShowFileExplorer] = useState(false);
|
||||
const [showGitOptions, setShowGitOptions] = useState(false);
|
||||
|
||||
@@ -126,9 +123,20 @@ export function TaskCreationWizard({
|
||||
// Ref for the textarea to handle paste events
|
||||
const descriptionRef = useRef<HTMLTextAreaElement>(null);
|
||||
|
||||
// Ref for the form scroll container (for drag auto-scroll)
|
||||
const formContainerRef = useRef<HTMLDivElement>(null);
|
||||
|
||||
// Drag-and-drop state for images over textarea
|
||||
const [isDragOverTextarea, setIsDragOverTextarea] = useState(false);
|
||||
|
||||
// @ autocomplete state
|
||||
const [autocomplete, setAutocomplete] = useState<{
|
||||
show: boolean;
|
||||
query: string;
|
||||
startPos: number;
|
||||
position: { top: number; left: number };
|
||||
} | null>(null);
|
||||
|
||||
// Load draft when dialog opens, or initialize from selected profile
|
||||
useEffect(() => {
|
||||
if (open && projectId) {
|
||||
@@ -155,10 +163,6 @@ export function TaskCreationWizard({
|
||||
if (draft.category || draft.priority || draft.complexity || draft.impact) {
|
||||
setShowAdvanced(true);
|
||||
}
|
||||
if (draft.images.length > 0) {
|
||||
setShowImages(true);
|
||||
}
|
||||
// Note: Referenced Files section is always visible, no need to expand
|
||||
} else {
|
||||
// No draft - initialize from selected profile and custom settings
|
||||
setProfileId(settings.selectedAgentProfile || 'auto');
|
||||
@@ -308,14 +312,128 @@ export function TaskCreationWizard({
|
||||
|
||||
if (newImages.length > 0) {
|
||||
setImages(prev => [...prev, ...newImages]);
|
||||
// Auto-expand images section
|
||||
setShowImages(true);
|
||||
// Show success feedback
|
||||
setPasteSuccess(true);
|
||||
setTimeout(() => setPasteSuccess(false), 2000);
|
||||
}
|
||||
}, [images]);
|
||||
|
||||
/**
|
||||
* Detect @ mention being typed and show autocomplete
|
||||
*/
|
||||
const detectAtMention = useCallback((text: string, cursorPos: number) => {
|
||||
const beforeCursor = text.slice(0, cursorPos);
|
||||
// Match @ followed by optional path characters (letters, numbers, dots, dashes, slashes)
|
||||
const match = beforeCursor.match(/@([\w\-./\\]*)$/);
|
||||
|
||||
if (match) {
|
||||
return {
|
||||
query: match[1],
|
||||
startPos: cursorPos - match[0].length
|
||||
};
|
||||
}
|
||||
return null;
|
||||
}, []);
|
||||
|
||||
/**
|
||||
* Handle description change and check for @ mentions
|
||||
*/
|
||||
const handleDescriptionChange = useCallback((e: React.ChangeEvent<HTMLTextAreaElement>) => {
|
||||
const newValue = e.target.value;
|
||||
const cursorPos = e.target.selectionStart || 0;
|
||||
|
||||
setDescription(newValue);
|
||||
|
||||
// Check for @ mention at cursor
|
||||
const mention = detectAtMention(newValue, cursorPos);
|
||||
|
||||
if (mention) {
|
||||
// Calculate popup position based on cursor
|
||||
const textarea = descriptionRef.current;
|
||||
if (textarea) {
|
||||
const rect = textarea.getBoundingClientRect();
|
||||
const textareaStyle = window.getComputedStyle(textarea);
|
||||
const lineHeight = parseFloat(textareaStyle.lineHeight) || 20;
|
||||
const paddingTop = parseFloat(textareaStyle.paddingTop) || 8;
|
||||
const paddingLeft = parseFloat(textareaStyle.paddingLeft) || 12;
|
||||
|
||||
// Estimate cursor position (simplified - assumes fixed-width font)
|
||||
const textBeforeCursor = newValue.slice(0, cursorPos);
|
||||
const lines = textBeforeCursor.split('\n');
|
||||
const currentLineIndex = lines.length - 1;
|
||||
const currentLineLength = lines[currentLineIndex].length;
|
||||
|
||||
// Calculate position relative to textarea
|
||||
const charWidth = 8; // Approximate character width
|
||||
const top = paddingTop + (currentLineIndex + 1) * lineHeight + 4;
|
||||
const left = paddingLeft + Math.min(currentLineLength * charWidth, rect.width - 300);
|
||||
|
||||
setAutocomplete({
|
||||
show: true,
|
||||
query: mention.query,
|
||||
startPos: mention.startPos,
|
||||
position: { top, left: Math.max(0, left) }
|
||||
});
|
||||
}
|
||||
} else {
|
||||
// No @ mention at cursor, close autocomplete
|
||||
if (autocomplete?.show) {
|
||||
setAutocomplete(null);
|
||||
}
|
||||
}
|
||||
}, [detectAtMention, autocomplete?.show]);
|
||||
|
||||
/**
|
||||
* Handle autocomplete selection
|
||||
*/
|
||||
const handleAutocompleteSelect = useCallback((filename: string) => {
|
||||
if (!autocomplete) return;
|
||||
|
||||
const textarea = descriptionRef.current;
|
||||
if (!textarea) return;
|
||||
|
||||
// Replace the @query with @filename
|
||||
const beforeMention = description.slice(0, autocomplete.startPos);
|
||||
const afterMention = description.slice(autocomplete.startPos + 1 + autocomplete.query.length);
|
||||
const newDescription = beforeMention + '@' + filename + afterMention;
|
||||
|
||||
setDescription(newDescription);
|
||||
setAutocomplete(null);
|
||||
|
||||
// Set cursor after the inserted mention
|
||||
setTimeout(() => {
|
||||
const newCursorPos = autocomplete.startPos + 1 + filename.length;
|
||||
textarea.focus();
|
||||
textarea.setSelectionRange(newCursorPos, newCursorPos);
|
||||
}, 0);
|
||||
}, [autocomplete, description]);
|
||||
|
||||
/**
|
||||
* Close autocomplete
|
||||
*/
|
||||
const handleAutocompleteClose = useCallback(() => {
|
||||
setAutocomplete(null);
|
||||
}, []);
|
||||
|
||||
/**
|
||||
* Handle drag over the form container to auto-scroll when dragging near edges
|
||||
*/
|
||||
const handleContainerDragOver = useCallback((e: DragEvent<HTMLDivElement>) => {
|
||||
const container = formContainerRef.current;
|
||||
if (!container) return;
|
||||
|
||||
const rect = container.getBoundingClientRect();
|
||||
const edgeThreshold = 60; // px from edge to trigger scroll
|
||||
const scrollSpeed = 8;
|
||||
|
||||
// Auto-scroll when dragging near top or bottom edges
|
||||
if (e.clientY < rect.top + edgeThreshold) {
|
||||
container.scrollTop -= scrollSpeed;
|
||||
} else if (e.clientY > rect.bottom - edgeThreshold) {
|
||||
container.scrollTop += scrollSpeed;
|
||||
}
|
||||
}, []);
|
||||
|
||||
/**
|
||||
* Handle drag over textarea for image drops
|
||||
*/
|
||||
@@ -439,8 +557,6 @@ export function TaskCreationWizard({
|
||||
|
||||
if (newImages.length > 0) {
|
||||
setImages(prev => [...prev, ...newImages]);
|
||||
// Auto-expand images section
|
||||
setShowImages(true);
|
||||
// Show success feedback
|
||||
setPasteSuccess(true);
|
||||
setTimeout(() => setPasteSuccess(false), 2000);
|
||||
@@ -555,7 +671,6 @@ export function TaskCreationWizard({
|
||||
setBaseBranch(PROJECT_DEFAULT_BRANCH);
|
||||
setError(null);
|
||||
setShowAdvanced(false);
|
||||
setShowImages(false);
|
||||
setShowFileExplorer(false);
|
||||
setShowGitOptions(false);
|
||||
setIsDraftRestored(false);
|
||||
@@ -602,7 +717,11 @@ export function TaskCreationWizard({
|
||||
>
|
||||
<div className="flex h-full min-h-0 overflow-hidden">
|
||||
{/* Form content */}
|
||||
<div className="flex-1 flex flex-col p-6 min-w-0 min-h-0 overflow-y-auto relative">
|
||||
<div
|
||||
ref={formContainerRef}
|
||||
onDragOver={handleContainerDragOver}
|
||||
className="flex-1 flex flex-col p-6 min-w-0 min-h-0 overflow-y-auto relative"
|
||||
>
|
||||
<DialogHeader>
|
||||
<div className="flex items-center justify-between">
|
||||
<DialogTitle className="text-foreground">Create New Task</DialogTitle>
|
||||
@@ -668,9 +787,9 @@ export function TaskCreationWizard({
|
||||
<Textarea
|
||||
ref={descriptionRef}
|
||||
id="description"
|
||||
placeholder="Describe the feature, bug fix, or improvement you want to implement. Be as specific as possible about requirements, constraints, and expected behavior."
|
||||
placeholder="Describe the feature, bug fix, or improvement you want to implement. Be as specific as possible about requirements, constraints, and expected behavior. Type @ to reference files."
|
||||
value={description}
|
||||
onChange={(e) => setDescription(e.target.value)}
|
||||
onChange={handleDescriptionChange}
|
||||
onPaste={handlePaste}
|
||||
onDragOver={handleTextareaDragOver}
|
||||
onDragLeave={handleTextareaDragLeave}
|
||||
@@ -684,10 +803,62 @@ export function TaskCreationWizard({
|
||||
)}
|
||||
style={{ caretColor: 'auto' }}
|
||||
/>
|
||||
{/* File autocomplete popup */}
|
||||
{autocomplete?.show && projectPath && (
|
||||
<FileAutocomplete
|
||||
query={autocomplete.query}
|
||||
projectPath={projectPath}
|
||||
position={autocomplete.position}
|
||||
onSelect={handleAutocompleteSelect}
|
||||
onClose={handleAutocompleteClose}
|
||||
/>
|
||||
)}
|
||||
</div>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Tip: Drag files from the explorer to insert @references, or paste screenshots with {navigator.platform.includes('Mac') ? '⌘V' : 'Ctrl+V'}.
|
||||
Files and images can be copy/pasted or dragged & dropped into the description.
|
||||
</p>
|
||||
|
||||
{/* Image Thumbnails - displayed inline below description */}
|
||||
{images.length > 0 && (
|
||||
<div className="flex flex-wrap gap-2 mt-2">
|
||||
{images.map((image) => (
|
||||
<div
|
||||
key={image.id}
|
||||
className="relative group rounded-md border border-border overflow-hidden cursor-pointer hover:ring-2 hover:ring-primary/50 transition-all"
|
||||
style={{ width: '64px', height: '64px' }}
|
||||
onClick={() => {
|
||||
// Open full-size image in a new window/modal could be added here
|
||||
}}
|
||||
title={image.filename}
|
||||
>
|
||||
{image.thumbnail ? (
|
||||
<img
|
||||
src={image.thumbnail}
|
||||
alt={image.filename}
|
||||
className="w-full h-full object-cover"
|
||||
/>
|
||||
) : (
|
||||
<div className="w-full h-full flex items-center justify-center bg-muted">
|
||||
<ImageIcon className="h-6 w-6 text-muted-foreground" />
|
||||
</div>
|
||||
)}
|
||||
{/* Remove button */}
|
||||
{!isCreating && (
|
||||
<button
|
||||
type="button"
|
||||
className="absolute top-0.5 right-0.5 h-4 w-4 flex items-center justify-center rounded-full bg-destructive text-destructive-foreground opacity-0 group-hover:opacity-100 transition-opacity"
|
||||
onClick={(e) => {
|
||||
e.stopPropagation();
|
||||
setImages(prev => prev.filter(img => img.id !== image.id));
|
||||
}}
|
||||
>
|
||||
<X className="h-3 w-3" />
|
||||
</button>
|
||||
)}
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
|
||||
{/* Title (Optional - Auto-generated if empty) */}
|
||||
@@ -855,79 +1026,6 @@ export function TaskCreationWizard({
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Images Toggle */}
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => setShowImages(!showImages)}
|
||||
className={cn(
|
||||
'flex items-center gap-2 text-sm text-muted-foreground hover:text-foreground transition-colors',
|
||||
'w-full justify-between py-2 px-3 rounded-md hover:bg-muted/50'
|
||||
)}
|
||||
disabled={isCreating}
|
||||
>
|
||||
<span className="flex items-center gap-2">
|
||||
<ImageIcon className="h-4 w-4" />
|
||||
Reference Images (optional)
|
||||
{images.length > 0 && (
|
||||
<span className="text-xs bg-primary/10 text-primary px-1.5 py-0.5 rounded">
|
||||
{images.length}
|
||||
</span>
|
||||
)}
|
||||
</span>
|
||||
{showImages ? (
|
||||
<ChevronUp className="h-4 w-4" />
|
||||
) : (
|
||||
<ChevronDown className="h-4 w-4" />
|
||||
)}
|
||||
</button>
|
||||
|
||||
{/* Image Upload Section */}
|
||||
{showImages && (
|
||||
<div className="space-y-3 p-4 rounded-lg border border-border bg-muted/30">
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Attach screenshots, mockups, or diagrams to provide visual context for the AI.
|
||||
</p>
|
||||
<ImageUpload
|
||||
images={images}
|
||||
onImagesChange={setImages}
|
||||
disabled={isCreating}
|
||||
/>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Referenced Files Section - Always visible, clean list */}
|
||||
<div className="space-y-3 p-4 rounded-lg border border-border bg-muted/30">
|
||||
{/* Header */}
|
||||
<div className="flex items-center gap-2">
|
||||
<FolderTree className="h-4 w-4 text-muted-foreground" />
|
||||
<span className="text-sm font-medium text-foreground">Referenced Files</span>
|
||||
{referencedFiles.length > 0 && (
|
||||
<span className="text-xs bg-primary/10 text-primary px-1.5 py-0.5 rounded">
|
||||
{referencedFiles.length}/{MAX_REFERENCED_FILES}
|
||||
</span>
|
||||
)}
|
||||
</div>
|
||||
|
||||
{/* Empty state hint */}
|
||||
{referencedFiles.length === 0 ? (
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Drag files from the file explorer anywhere onto this form to add references, or use the "Browse Files" button below.
|
||||
</p>
|
||||
) : (
|
||||
<>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
These files will provide context for the AI when working on your task.
|
||||
</p>
|
||||
<ReferencedFilesSection
|
||||
files={referencedFiles}
|
||||
onRemove={(id) => setReferencedFiles(prev => prev.filter(f => f.id !== id))}
|
||||
maxFiles={MAX_REFERENCED_FILES}
|
||||
disabled={isCreating}
|
||||
/>
|
||||
</>
|
||||
)}
|
||||
</div>
|
||||
|
||||
{/* Review Requirement Toggle */}
|
||||
<div className="flex items-start gap-3 p-4 rounded-lg border border-border bg-muted/30">
|
||||
<Checkbox
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
import { motion, AnimatePresence } from 'motion/react';
|
||||
import { X, FolderTree, RefreshCw } from 'lucide-react';
|
||||
import { Button } from './ui/button';
|
||||
import { ScrollArea } from './ui/scroll-area';
|
||||
import { FileTree } from './FileTree';
|
||||
import { useFileExplorerStore } from '../stores/file-explorer-store';
|
||||
|
||||
@@ -105,10 +104,10 @@ export function TaskFileExplorerDrawer({ isOpen, onClose, projectPath }: TaskFil
|
||||
</p>
|
||||
</div>
|
||||
|
||||
{/* File tree */}
|
||||
<ScrollArea className="flex-1">
|
||||
{/* File tree - no ScrollArea wrapper as FileTree uses virtualization with its own scroll container */}
|
||||
<div className="flex-1 overflow-hidden min-h-0">
|
||||
<FileTree rootPath={projectPath} />
|
||||
</ScrollArea>
|
||||
</div>
|
||||
</motion.div>
|
||||
</motion.div>
|
||||
)}
|
||||
|
||||
@@ -33,9 +33,10 @@ import type { SessionDateInfo } from '../../shared/types';
|
||||
interface TerminalGridProps {
|
||||
projectPath?: string;
|
||||
onNewTaskClick?: () => void;
|
||||
isActive?: boolean;
|
||||
}
|
||||
|
||||
export function TerminalGrid({ projectPath, onNewTaskClick }: TerminalGridProps) {
|
||||
export function TerminalGrid({ projectPath, onNewTaskClick, isActive = false }: TerminalGridProps) {
|
||||
const terminals = useTerminalStore((state) => state.terminals);
|
||||
const activeTerminalId = useTerminalStore((state) => state.activeTerminalId);
|
||||
const addTerminal = useTerminalStore((state) => state.addTerminal);
|
||||
@@ -167,8 +168,10 @@ export function TerminalGrid({ projectPath, onNewTaskClick }: TerminalGridProps)
|
||||
removeTerminal(id);
|
||||
}, [removeTerminal]);
|
||||
|
||||
// Handle keyboard shortcut for new terminal
|
||||
// Handle keyboard shortcut for new terminal (only when this view is active)
|
||||
useEffect(() => {
|
||||
if (!isActive) return;
|
||||
|
||||
const handleKeyDown = (e: KeyboardEvent) => {
|
||||
// Ctrl+T or Cmd+T for new terminal
|
||||
if ((e.ctrlKey || e.metaKey) && e.key === 't') {
|
||||
@@ -186,7 +189,7 @@ export function TerminalGrid({ projectPath, onNewTaskClick }: TerminalGridProps)
|
||||
|
||||
window.addEventListener('keydown', handleKeyDown);
|
||||
return () => window.removeEventListener('keydown', handleKeyDown);
|
||||
}, [addTerminal, canAddTerminal, projectPath, activeTerminalId, handleCloseTerminal]);
|
||||
}, [isActive, addTerminal, canAddTerminal, projectPath, activeTerminalId, handleCloseTerminal]);
|
||||
|
||||
const handleAddTerminal = useCallback(() => {
|
||||
if (canAddTerminal()) {
|
||||
|
||||
@@ -0,0 +1,483 @@
|
||||
/**
|
||||
* Unit tests for ProjectTabBar component
|
||||
* Tests project tab rendering, interaction handling, and state display
|
||||
*
|
||||
* @vitest-environment jsdom
|
||||
*/
|
||||
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
||||
import type { Project } from '../../../shared/types';
|
||||
|
||||
// Helper to create test projects
|
||||
function createTestProject(overrides: Partial<Project> = {}): Project {
|
||||
return {
|
||||
id: `project-${Date.now()}-${Math.random().toString(36).substring(7)}`,
|
||||
name: 'Test Project',
|
||||
path: '/path/to/test-project',
|
||||
autoBuildPath: '/path/to/test-project/.auto-claude',
|
||||
settings: {
|
||||
model: 'claude-3-haiku-20240307',
|
||||
memoryBackend: 'file',
|
||||
linearSync: false,
|
||||
notifications: {
|
||||
onTaskComplete: true,
|
||||
onTaskFailed: true,
|
||||
onReviewNeeded: true,
|
||||
sound: false
|
||||
},
|
||||
graphitiMcpEnabled: false
|
||||
},
|
||||
createdAt: new Date(),
|
||||
updatedAt: new Date(),
|
||||
...overrides
|
||||
};
|
||||
}
|
||||
|
||||
describe('ProjectTabBar', () => {
|
||||
// Mock callbacks
|
||||
const mockOnProjectSelect = vi.fn();
|
||||
const mockOnProjectClose = vi.fn();
|
||||
const mockOnAddProject = vi.fn();
|
||||
|
||||
beforeEach(() => {
|
||||
// Reset all mocks
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
describe('Rendering Logic', () => {
|
||||
it('should return null when projects array is empty', () => {
|
||||
const projects: Project[] = [];
|
||||
const activeProjectId = null;
|
||||
|
||||
// Component returns null when projects.length === 0
|
||||
expect(projects.length).toBe(0);
|
||||
expect(activeProjectId).toBeNull();
|
||||
});
|
||||
|
||||
it('should render when projects array has at least one project', () => {
|
||||
const projects = [createTestProject()];
|
||||
const activeProjectId = projects[0].id;
|
||||
|
||||
// Component renders when projects.length > 0
|
||||
expect(projects.length).toBeGreaterThan(0);
|
||||
expect(activeProjectId).toBe(projects[0].id);
|
||||
});
|
||||
|
||||
it('should render all projects in the array', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-1', name: 'Project 1' }),
|
||||
createTestProject({ id: 'proj-2', name: 'Project 2' }),
|
||||
createTestProject({ id: 'proj-3', name: 'Project 3' })
|
||||
];
|
||||
|
||||
expect(projects).toHaveLength(3);
|
||||
expect(projects.map(p => p.name)).toEqual(['Project 1', 'Project 2', 'Project 3']);
|
||||
});
|
||||
|
||||
it('should render tabs in the order they appear in the projects array', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-1', name: 'Alpha' }),
|
||||
createTestProject({ id: 'proj-2', name: 'Beta' }),
|
||||
createTestProject({ id: 'proj-3', name: 'Gamma' })
|
||||
];
|
||||
|
||||
const expectedOrder = ['Alpha', 'Beta', 'Gamma'];
|
||||
const actualOrder = projects.map(p => p.name);
|
||||
|
||||
expect(actualOrder).toEqual(expectedOrder);
|
||||
});
|
||||
});
|
||||
|
||||
describe('Active Project State', () => {
|
||||
it('should identify active project correctly', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-1', name: 'Work' }),
|
||||
createTestProject({ id: 'proj-2', name: 'Personal' })
|
||||
];
|
||||
const activeProjectId = 'proj-2';
|
||||
|
||||
// Check which project is active
|
||||
const activeProject = projects.find(p => p.id === activeProjectId);
|
||||
expect(activeProject?.name).toBe('Personal');
|
||||
|
||||
// Check isActive logic for each project
|
||||
projects.forEach(project => {
|
||||
const isActive = project.id === activeProjectId;
|
||||
if (project.id === 'proj-2') {
|
||||
expect(isActive).toBe(true);
|
||||
} else {
|
||||
expect(isActive).toBe(false);
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
it('should handle when no project is active', () => {
|
||||
const projects = [createTestProject({ id: 'proj-1', name: 'Solo' })];
|
||||
const activeProjectId = null;
|
||||
|
||||
const activeProject = projects.find(p => p.id === activeProjectId);
|
||||
expect(activeProject).toBeUndefined();
|
||||
|
||||
// Check isActive logic for the project
|
||||
projects.forEach(project => {
|
||||
const isActive = project.id === activeProjectId;
|
||||
expect(isActive).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
it('should handle active project that is not in the projects array', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-1', name: 'Current' })
|
||||
];
|
||||
const activeProjectId = 'proj-not-in-array';
|
||||
|
||||
// No project should be active
|
||||
projects.forEach(project => {
|
||||
const isActive = project.id === activeProjectId;
|
||||
expect(isActive).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
it('should handle multiple projects with the same name but different IDs', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-1', name: 'My Project' }),
|
||||
createTestProject({ id: 'proj-2', name: 'My Project' })
|
||||
];
|
||||
const activeProjectId = 'proj-2';
|
||||
|
||||
// Both projects have same name but different IDs
|
||||
expect(projects[0].name).toBe(projects[1].name);
|
||||
expect(projects[0].id).not.toBe(projects[1].id);
|
||||
|
||||
// Only proj-2 should be active
|
||||
projects.forEach(project => {
|
||||
const isActive = project.id === activeProjectId;
|
||||
if (project.id === 'proj-2') {
|
||||
expect(isActive).toBe(true);
|
||||
} else {
|
||||
expect(isActive).toBe(false);
|
||||
}
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('Project Selection', () => {
|
||||
it('should call onProjectSelect with correct project ID when tab is clicked', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-1', name: 'Project 1' }),
|
||||
createTestProject({ id: 'proj-2', name: 'Project 2' })
|
||||
];
|
||||
|
||||
// Simulate clicking on project 2
|
||||
const selectedProjectId = 'proj-2';
|
||||
mockOnProjectSelect(selectedProjectId);
|
||||
|
||||
expect(mockOnProjectSelect).toHaveBeenCalledWith('proj-2');
|
||||
expect(mockOnProjectSelect).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it('should handle project selection for the first project', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-first', name: 'First Project' }),
|
||||
createTestProject({ id: 'proj-second', name: 'Second Project' })
|
||||
];
|
||||
|
||||
const selectedProjectId = 'proj-first';
|
||||
mockOnProjectSelect(selectedProjectId);
|
||||
|
||||
expect(mockOnProjectSelect).toHaveBeenCalledWith('proj-first');
|
||||
});
|
||||
|
||||
it('should handle project selection for the last project', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-a', name: 'Project A' }),
|
||||
createTestProject({ id: 'proj-b', name: 'Project B' }),
|
||||
createTestProject({ id: 'proj-c', name: 'Project C' })
|
||||
];
|
||||
|
||||
const selectedProjectId = 'proj-c';
|
||||
mockOnProjectSelect(selectedProjectId);
|
||||
|
||||
expect(mockOnProjectSelect).toHaveBeenCalledWith('proj-c');
|
||||
});
|
||||
});
|
||||
|
||||
describe('Project Closing', () => {
|
||||
it('should call onProjectClose with correct project ID when close button is clicked', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-1', name: 'Project 1' }),
|
||||
createTestProject({ id: 'proj-2', name: 'Project 2' })
|
||||
];
|
||||
|
||||
// Simulate clicking close button for project 1
|
||||
const closedProjectId = 'proj-1';
|
||||
|
||||
// Create mock event
|
||||
const mockEvent = {
|
||||
stopPropagation: vi.fn()
|
||||
} as unknown as React.MouseEvent;
|
||||
|
||||
mockOnProjectClose(closedProjectId);
|
||||
|
||||
expect(mockOnProjectClose).toHaveBeenCalledWith('proj-1');
|
||||
expect(mockOnProjectClose).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it('should prevent event propagation when close button is clicked', () => {
|
||||
const mockEvent = {
|
||||
stopPropagation: vi.fn()
|
||||
} as unknown as React.MouseEvent;
|
||||
|
||||
// Simulate the event handling logic
|
||||
const onClose = (e: React.MouseEvent) => {
|
||||
e.stopPropagation();
|
||||
mockOnProjectClose('proj-1');
|
||||
};
|
||||
|
||||
onClose(mockEvent);
|
||||
|
||||
expect(mockEvent.stopPropagation).toHaveBeenCalled();
|
||||
expect(mockOnProjectClose).toHaveBeenCalledWith('proj-1');
|
||||
});
|
||||
|
||||
it('should allow closing when there are multiple projects', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-1', name: 'Project 1' }),
|
||||
createTestProject({ id: 'proj-2', name: 'Project 2' })
|
||||
];
|
||||
|
||||
// canClose = projects.length > 1
|
||||
const canClose = projects.length > 1;
|
||||
expect(canClose).toBe(true);
|
||||
});
|
||||
|
||||
it('should not allow closing when there is only one project', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-only', name: 'Only Project' })
|
||||
];
|
||||
|
||||
// canClose = projects.length > 1
|
||||
const canClose = projects.length > 1;
|
||||
expect(canClose).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
describe('Add Project Button', () => {
|
||||
it('should call onAddProject when add button is clicked', () => {
|
||||
mockOnAddProject();
|
||||
|
||||
expect(mockOnAddProject).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it('should render add button with correct attributes', () => {
|
||||
// Check button attributes from component
|
||||
const buttonVariant = 'ghost';
|
||||
const buttonSize = 'icon';
|
||||
const buttonTitle = 'Add Project';
|
||||
const buttonClasses = 'h-8 w-8';
|
||||
|
||||
expect(buttonVariant).toBe('ghost');
|
||||
expect(buttonSize).toBe('icon');
|
||||
expect(buttonTitle).toBe('Add Project');
|
||||
expect(buttonClasses).toBe('h-8 w-8');
|
||||
});
|
||||
|
||||
it('should render Plus icon in add button', () => {
|
||||
// Component uses Plus from lucide-react
|
||||
const iconClass = 'h-4 w-4';
|
||||
expect(iconClass).toBe('h-4 w-4');
|
||||
});
|
||||
});
|
||||
|
||||
describe('Container Layout and Styling', () => {
|
||||
it('should apply correct container classes', () => {
|
||||
// From component: className={cn(
|
||||
// 'flex items-center border-b border-border bg-background',
|
||||
// 'overflow-x-auto scrollbar-thin scrollbar-thumb-border scrollbar-track-transparent',
|
||||
// className
|
||||
// )}
|
||||
const expectedClasses = [
|
||||
'flex',
|
||||
'items-center',
|
||||
'border-b',
|
||||
'border-border',
|
||||
'bg-background',
|
||||
'overflow-x-auto',
|
||||
'scrollbar-thin',
|
||||
'scrollbar-thumb-border',
|
||||
'scrollbar-track-transparent'
|
||||
];
|
||||
|
||||
expectedClasses.forEach(cls => {
|
||||
expect(cls).toBeTruthy();
|
||||
});
|
||||
});
|
||||
|
||||
it('should apply correct flex container for tabs', () => {
|
||||
// From component: <div className="flex items-center flex-1 min-w-0">
|
||||
const tabContainerClasses = [
|
||||
'flex',
|
||||
'items-center',
|
||||
'flex-1',
|
||||
'min-w-0'
|
||||
];
|
||||
|
||||
tabContainerClasses.forEach(cls => {
|
||||
expect(cls).toBeTruthy();
|
||||
});
|
||||
});
|
||||
|
||||
it('should apply correct add button container classes', () => {
|
||||
// From component: <div className="flex items-center px-2 py-1">
|
||||
const addButtonContainerClasses = [
|
||||
'flex',
|
||||
'items-center',
|
||||
'px-2',
|
||||
'py-1'
|
||||
];
|
||||
|
||||
addButtonContainerClasses.forEach(cls => {
|
||||
expect(cls).toBeTruthy();
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('Props Handling', () => {
|
||||
it('should accept and use custom className', () => {
|
||||
const customClassName = 'custom-test-class';
|
||||
const baseClasses = [
|
||||
'flex items-center border-b border-border bg-background',
|
||||
'overflow-x-auto scrollbar-thin scrollbar-thumb-border scrollbar-track-transparent'
|
||||
];
|
||||
|
||||
// The cn function combines base classes with custom className
|
||||
expect(customClassName).toBe('custom-test-class');
|
||||
baseClasses.forEach(cls => {
|
||||
expect(cls).toBeTruthy();
|
||||
});
|
||||
});
|
||||
|
||||
it('should handle all required props correctly', () => {
|
||||
const projects = [createTestProject()];
|
||||
const activeProjectId = projects[0].id;
|
||||
const className = undefined;
|
||||
|
||||
// All required props should be available
|
||||
expect(projects).toBeDefined();
|
||||
expect(activeProjectId).toBeDefined();
|
||||
expect(mockOnProjectSelect).toBeDefined();
|
||||
expect(mockOnProjectClose).toBeDefined();
|
||||
expect(mockOnAddProject).toBeDefined();
|
||||
expect(className).toBeUndefined(); // Optional prop
|
||||
});
|
||||
|
||||
it('should handle optional className prop', () => {
|
||||
const projects = [createTestProject()];
|
||||
const activeProjectId = projects[0].id;
|
||||
const customClassName = 'my-custom-class';
|
||||
|
||||
// Optional prop should be handled correctly
|
||||
expect(customClassName).toBe('my-custom-class');
|
||||
});
|
||||
});
|
||||
|
||||
describe('Tab Key Generation', () => {
|
||||
it('should use project.id as key for tabs', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'unique-id-1', name: 'Project 1' }),
|
||||
createTestProject({ id: 'unique-id-2', name: 'Project 2' })
|
||||
];
|
||||
|
||||
// Each tab should use project.id as its key
|
||||
projects.forEach(project => {
|
||||
const key = project.id;
|
||||
expect(key).toBeDefined();
|
||||
expect(typeof key).toBe('string');
|
||||
expect(key.length).toBeGreaterThan(0);
|
||||
});
|
||||
|
||||
// Keys should be unique
|
||||
const keys = projects.map(p => p.id);
|
||||
const uniqueKeys = new Set(keys);
|
||||
expect(uniqueKeys.size).toBe(keys.length);
|
||||
});
|
||||
|
||||
it('should handle projects with special characters in ID', () => {
|
||||
const specialIds = ['proj-with-123', 'proj_with_underscore', 'proj.with.dots'];
|
||||
|
||||
specialIds.forEach(id => {
|
||||
const project = createTestProject({ id });
|
||||
expect(project.id).toBe(id);
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('Integration with SortableProjectTab', () => {
|
||||
it('should pass correct props to SortableProjectTab', () => {
|
||||
const projects = [
|
||||
createTestProject({ id: 'proj-1', name: 'Test Project' })
|
||||
];
|
||||
const activeProjectId = 'proj-1';
|
||||
|
||||
// Props that should be passed to SortableProjectTab
|
||||
const tabProps = {
|
||||
project: projects[0],
|
||||
isActive: activeProjectId === projects[0].id,
|
||||
canClose: projects.length > 1,
|
||||
tabIndex: 0,
|
||||
onSelect: expect.any(Function),
|
||||
onClose: expect.any(Function)
|
||||
};
|
||||
|
||||
expect(tabProps.project.id).toBe('proj-1');
|
||||
expect(tabProps.isActive).toBe(true);
|
||||
expect(tabProps.canClose).toBe(false); // Only one project
|
||||
});
|
||||
|
||||
it('should pass canClose correctly based on project count', () => {
|
||||
const singleProject = [createTestProject({ id: 'proj-single' })];
|
||||
const multipleProjects = [
|
||||
createTestProject({ id: 'proj-a' }),
|
||||
createTestProject({ id: 'proj-b' })
|
||||
];
|
||||
|
||||
// For single project
|
||||
const canCloseSingle = singleProject.length > 1;
|
||||
expect(canCloseSingle).toBe(false);
|
||||
|
||||
// For multiple projects
|
||||
const canCloseMultiple = multipleProjects.length > 1;
|
||||
expect(canCloseMultiple).toBe(true);
|
||||
});
|
||||
|
||||
it('should pass correct onSelect function that calls onProjectSelect with project ID', () => {
|
||||
const projects = [createTestProject({ id: 'proj-callback' })];
|
||||
|
||||
// Create the onSelect function that would be passed to SortableProjectTab
|
||||
const projectId = 'proj-callback';
|
||||
const onSelect = () => mockOnProjectSelect(projectId);
|
||||
|
||||
onSelect();
|
||||
|
||||
expect(mockOnProjectSelect).toHaveBeenCalledWith('proj-callback');
|
||||
});
|
||||
|
||||
it('should pass correct onClose function that stops propagation and calls onProjectClose', () => {
|
||||
const projects = [createTestProject({ id: 'proj-close' })];
|
||||
|
||||
const mockEvent = {
|
||||
stopPropagation: vi.fn()
|
||||
} as unknown as React.MouseEvent;
|
||||
|
||||
const projectId = 'proj-close';
|
||||
const onClose = (e: React.MouseEvent) => {
|
||||
e.stopPropagation();
|
||||
mockOnProjectClose(projectId);
|
||||
};
|
||||
|
||||
onClose(mockEvent);
|
||||
|
||||
expect(mockEvent.stopPropagation).toHaveBeenCalled();
|
||||
expect(mockOnProjectClose).toHaveBeenCalledWith('proj-close');
|
||||
});
|
||||
});
|
||||
});
|
||||
@@ -60,6 +60,8 @@ export function Step2ConfigureGenerate(props: Step2ConfigureGenerateProps) {
|
||||
} = props;
|
||||
|
||||
const selectedProjectId = useProjectStore((state) => state.selectedProjectId);
|
||||
const projects = useProjectStore((state) => state.projects);
|
||||
const selectedProject = projects.find((p) => p.id === selectedProjectId);
|
||||
const selectedTasks = doneTasks.filter((t) => selectedTaskIds.includes(t.id));
|
||||
|
||||
const summaryInfo = getSummaryInfo(
|
||||
@@ -112,6 +114,7 @@ export function Step2ConfigureGenerate(props: Step2ConfigureGenerateProps) {
|
||||
isDragOver={imageUpload.isDragOver}
|
||||
imageError={imageUpload.imageError}
|
||||
textareaRef={imageUpload.textareaRef}
|
||||
projectPath={selectedProject?.path}
|
||||
onSave={props.onSave}
|
||||
onCopy={props.onCopy}
|
||||
onChangelogEdit={onChangelogEdit}
|
||||
|
||||
@@ -1,11 +1,82 @@
|
||||
import { useState } from 'react';
|
||||
import { FileText, Copy, Save, CheckCircle, Image as ImageIcon } from 'lucide-react';
|
||||
import { useState, useMemo, useEffect } from 'react';
|
||||
import { FileText, Copy, Save, CheckCircle, Image as ImageIcon, Loader2 } from 'lucide-react';
|
||||
import { Button } from '../ui/button';
|
||||
import { Textarea } from '../ui/textarea';
|
||||
import { Tooltip, TooltipContent, TooltipTrigger } from '../ui/tooltip';
|
||||
import ReactMarkdown from 'react-markdown';
|
||||
import ReactMarkdown, { Components } from 'react-markdown';
|
||||
import remarkGfm from 'remark-gfm';
|
||||
|
||||
// Component for loading local images via IPC
|
||||
interface LocalImageProps {
|
||||
src: string;
|
||||
alt: string;
|
||||
projectPath?: string;
|
||||
}
|
||||
|
||||
function LocalImage({ src, alt, projectPath }: LocalImageProps) {
|
||||
const [imageSrc, setImageSrc] = useState<string | null>(null);
|
||||
const [loading, setLoading] = useState(true);
|
||||
const [error, setError] = useState<string | null>(null);
|
||||
|
||||
useEffect(() => {
|
||||
// If it's already an absolute URL or data URL, use it directly
|
||||
if (src.startsWith('http://') || src.startsWith('https://') || src.startsWith('data:')) {
|
||||
setImageSrc(src);
|
||||
setLoading(false);
|
||||
return;
|
||||
}
|
||||
|
||||
// If no project path, we can't load local images
|
||||
if (!projectPath) {
|
||||
setError('Cannot load local image: no project path');
|
||||
setLoading(false);
|
||||
return;
|
||||
}
|
||||
|
||||
// Load local image via IPC
|
||||
const loadImage = async () => {
|
||||
try {
|
||||
setLoading(true);
|
||||
setError(null);
|
||||
// Handle relative paths like .github/assets/... or ./path/to/image
|
||||
const relativePath = src.startsWith('./') ? src.slice(2) : src;
|
||||
const result = await window.electronAPI.readLocalImage(projectPath, relativePath);
|
||||
if (result.success && result.data) {
|
||||
setImageSrc(result.data);
|
||||
} else {
|
||||
setError(result.error || 'Failed to load image');
|
||||
}
|
||||
} catch (err) {
|
||||
setError(err instanceof Error ? err.message : 'Failed to load image');
|
||||
} finally {
|
||||
setLoading(false);
|
||||
}
|
||||
};
|
||||
|
||||
loadImage();
|
||||
}, [src, projectPath]);
|
||||
|
||||
if (loading) {
|
||||
return (
|
||||
<span className="inline-flex items-center gap-2 rounded border border-border bg-muted/50 px-3 py-2 text-sm text-muted-foreground">
|
||||
<Loader2 className="h-4 w-4 animate-spin" />
|
||||
<span>Loading image...</span>
|
||||
</span>
|
||||
);
|
||||
}
|
||||
|
||||
if (error || !imageSrc) {
|
||||
return (
|
||||
<span className="inline-flex items-center gap-2 rounded border border-destructive/50 bg-destructive/10 px-3 py-2 text-sm text-destructive">
|
||||
<ImageIcon className="h-4 w-4" />
|
||||
<span>{error || 'Image not found'}</span>
|
||||
</span>
|
||||
);
|
||||
}
|
||||
|
||||
return <img src={imageSrc} alt={alt} className="max-w-full h-auto" />;
|
||||
}
|
||||
|
||||
interface PreviewPanelProps {
|
||||
generatedChangelog: string;
|
||||
saveSuccess: boolean;
|
||||
@@ -14,6 +85,7 @@ interface PreviewPanelProps {
|
||||
isDragOver: boolean;
|
||||
imageError: string | null;
|
||||
textareaRef: React.RefObject<HTMLTextAreaElement | null>;
|
||||
projectPath?: string;
|
||||
onSave: () => void;
|
||||
onCopy: () => void;
|
||||
onChangelogEdit: (content: string) => void;
|
||||
@@ -31,6 +103,7 @@ export function PreviewPanel({
|
||||
isDragOver,
|
||||
imageError,
|
||||
textareaRef,
|
||||
projectPath,
|
||||
onSave,
|
||||
onCopy,
|
||||
onChangelogEdit,
|
||||
@@ -41,6 +114,13 @@ export function PreviewPanel({
|
||||
}: PreviewPanelProps) {
|
||||
const [viewMode, setViewMode] = useState<'markdown' | 'preview'>('markdown');
|
||||
|
||||
// Custom components for ReactMarkdown to handle local image paths
|
||||
const markdownComponents: Components = useMemo(() => ({
|
||||
img: ({ src, alt }) => {
|
||||
return <LocalImage src={src || ''} alt={alt || ''} projectPath={projectPath} />;
|
||||
}
|
||||
}), [projectPath]);
|
||||
|
||||
return (
|
||||
<div className="flex-1 flex flex-col overflow-hidden">
|
||||
{/* Preview Header */}
|
||||
@@ -146,7 +226,10 @@ export function PreviewPanel({
|
||||
) : (
|
||||
<div className="h-full overflow-auto">
|
||||
<div className="prose prose-sm dark:prose-invert max-w-none">
|
||||
<ReactMarkdown remarkPlugins={[remarkGfm]}>
|
||||
<ReactMarkdown
|
||||
remarkPlugins={[remarkGfm]}
|
||||
components={markdownComponents}
|
||||
>
|
||||
{generatedChangelog}
|
||||
</ReactMarkdown>
|
||||
</div>
|
||||
|
||||
@@ -79,7 +79,7 @@ export function MemoriesTab({
|
||||
<>
|
||||
<div className="grid gap-3 sm:grid-cols-3 text-sm">
|
||||
<InfoItem label="Database" value={memoryStatus.database || 'auto_claude_memory'} />
|
||||
<InfoItem label="Host" value={`${memoryStatus.host}:${memoryStatus.port}`} />
|
||||
<InfoItem label="Path" value={memoryStatus.dbPath || '~/.auto-claude/graphs'} />
|
||||
{memoryState && (
|
||||
<InfoItem label="Episodes" value={memoryState.episode_count.toString()} />
|
||||
)}
|
||||
@@ -94,7 +94,7 @@ export function MemoriesTab({
|
||||
<div className="text-sm text-muted-foreground">
|
||||
<p>{memoryStatus?.reason || 'Graphiti memory is not configured'}</p>
|
||||
<p className="mt-2 text-xs">
|
||||
To enable graph memory, set <code className="bg-muted px-1 py-0.5 rounded">GRAPHITI_ENABLED=true</code> and configure FalkorDB.
|
||||
To enable graph memory, set <code className="bg-muted px-1 py-0.5 rounded">GRAPHITI_ENABLED=true</code> in project settings.
|
||||
</p>
|
||||
</div>
|
||||
)}
|
||||
|
||||
@@ -47,7 +47,7 @@ function parseMemoryContent(content: string): ParsedSessionInsight | null {
|
||||
try {
|
||||
return JSON.parse(content);
|
||||
} catch {
|
||||
// Try to parse nested JSON (from our FalkorDB query)
|
||||
// Try to parse nested JSON (from our LadybugDB query)
|
||||
try {
|
||||
const outer = JSON.parse(content);
|
||||
if (typeof outer === 'object') {
|
||||
|
||||
@@ -9,7 +9,6 @@ import {
|
||||
ExternalLink,
|
||||
Eye,
|
||||
EyeOff,
|
||||
Server,
|
||||
Zap,
|
||||
XCircle
|
||||
} from 'lucide-react';
|
||||
@@ -55,17 +54,17 @@ const EMBEDDING_PROVIDERS: Array<{
|
||||
description: string;
|
||||
requiresApiKey: boolean;
|
||||
}> = [
|
||||
{ id: 'ollama', name: 'Ollama', description: 'Local embeddings (free)', requiresApiKey: false },
|
||||
{ id: 'openai', name: 'OpenAI', description: 'text-embedding-3-small (recommended)', requiresApiKey: true },
|
||||
{ id: 'voyage', name: 'Voyage AI', description: 'voyage-3 (great with Anthropic)', requiresApiKey: true },
|
||||
{ id: 'google', name: 'Google AI', description: 'Gemini text-embedding-004', requiresApiKey: true },
|
||||
{ id: 'huggingface', name: 'HuggingFace', description: 'Open source models', requiresApiKey: true },
|
||||
{ id: 'azure_openai', name: 'Azure OpenAI', description: 'Enterprise Azure embeddings', requiresApiKey: true },
|
||||
{ id: 'ollama', name: 'Ollama', description: 'Local embeddings (free)', requiresApiKey: false }
|
||||
{ id: 'azure_openai', name: 'Azure OpenAI', description: 'Enterprise Azure embeddings', requiresApiKey: true }
|
||||
];
|
||||
|
||||
interface GraphitiConfig {
|
||||
enabled: boolean;
|
||||
falkorDbUri: string;
|
||||
database: string;
|
||||
dbPath: string;
|
||||
llmProvider: GraphitiLLMProvider;
|
||||
embeddingProvider: GraphitiEmbeddingProvider;
|
||||
// OpenAI
|
||||
@@ -93,12 +92,13 @@ interface GraphitiConfig {
|
||||
}
|
||||
|
||||
interface ValidationStatus {
|
||||
falkordb: { tested: boolean; success: boolean; message: string } | null;
|
||||
database: { tested: boolean; success: boolean; message: string } | null;
|
||||
provider: { tested: boolean; success: boolean; message: string } | null;
|
||||
}
|
||||
|
||||
/**
|
||||
* Graphiti/FalkorDB configuration step for the onboarding wizard.
|
||||
* Graphiti memory configuration step for the onboarding wizard.
|
||||
* Uses LadybugDB (embedded database) - no Docker required.
|
||||
* Allows users to optionally configure Graphiti memory backend with multiple provider options.
|
||||
* This step is entirely optional and can be skipped.
|
||||
*/
|
||||
@@ -106,7 +106,8 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
const { settings, updateSettings } = useSettingsStore();
|
||||
const [config, setConfig] = useState<GraphitiConfig>({
|
||||
enabled: false,
|
||||
falkorDbUri: 'bolt://localhost:6380',
|
||||
database: 'auto_claude_memory',
|
||||
dbPath: '',
|
||||
llmProvider: 'openai',
|
||||
embeddingProvider: 'openai',
|
||||
openaiApiKey: settings.globalOpenAIApiKey || '',
|
||||
@@ -128,33 +129,25 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
const [isSaving, setIsSaving] = useState(false);
|
||||
const [error, setError] = useState<string | null>(null);
|
||||
const [success, setSuccess] = useState(false);
|
||||
const [isCheckingDocker, setIsCheckingDocker] = useState(true);
|
||||
const [dockerAvailable, setDockerAvailable] = useState<boolean | null>(null);
|
||||
const [isCheckingInfra, setIsCheckingInfra] = useState(true);
|
||||
const [kuzuAvailable, setKuzuAvailable] = useState<boolean | null>(null);
|
||||
const [isValidating, setIsValidating] = useState(false);
|
||||
const [validationStatus, setValidationStatus] = useState<ValidationStatus>({
|
||||
falkordb: null,
|
||||
database: null,
|
||||
provider: null
|
||||
});
|
||||
|
||||
// Check Docker/Infrastructure availability on mount
|
||||
// Check LadybugDB/Kuzu availability on mount
|
||||
useEffect(() => {
|
||||
const checkInfrastructure = async () => {
|
||||
setIsCheckingDocker(true);
|
||||
setIsCheckingInfra(true);
|
||||
try {
|
||||
const result = await window.electronAPI.getInfrastructureStatus();
|
||||
setDockerAvailable(result?.success && result?.data?.docker?.running ? true : false);
|
||||
|
||||
if (result?.success && result?.data?.falkordb?.containerRunning) {
|
||||
const detectedPort = result.data.falkordb.port;
|
||||
setConfig(prev => ({
|
||||
...prev,
|
||||
falkorDbUri: `bolt://localhost:${detectedPort}`
|
||||
}));
|
||||
}
|
||||
const result = await window.electronAPI.getMemoryInfrastructureStatus();
|
||||
setKuzuAvailable(result?.success && result?.data?.memory?.kuzuInstalled ? true : false);
|
||||
} catch {
|
||||
setDockerAvailable(false);
|
||||
setKuzuAvailable(false);
|
||||
} finally {
|
||||
setIsCheckingDocker(false);
|
||||
setIsCheckingInfra(false);
|
||||
}
|
||||
};
|
||||
|
||||
@@ -165,7 +158,7 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
setConfig(prev => ({ ...prev, enabled: checked }));
|
||||
setError(null);
|
||||
setSuccess(false);
|
||||
setValidationStatus({ falkordb: null, provider: null });
|
||||
setValidationStatus({ database: null, provider: null });
|
||||
};
|
||||
|
||||
const toggleShowApiKey = (key: string) => {
|
||||
@@ -202,9 +195,6 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
if (llmProvider === 'groq') {
|
||||
if (!config.groqApiKey.trim()) return 'Groq API key';
|
||||
}
|
||||
if (embeddingProvider === 'huggingface') {
|
||||
if (!config.huggingfaceApiKey.trim()) return 'HuggingFace API key';
|
||||
}
|
||||
if (llmProvider === 'ollama') {
|
||||
if (!config.ollamaLlmModel.trim()) return 'Ollama LLM model name';
|
||||
}
|
||||
@@ -224,41 +214,48 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
|
||||
setIsValidating(true);
|
||||
setError(null);
|
||||
setValidationStatus({ falkordb: null, provider: null });
|
||||
setValidationStatus({ database: null, provider: null });
|
||||
|
||||
try {
|
||||
// For now, use the existing OpenAI validation - this will be expanded
|
||||
// Get the API key for the current LLM provider
|
||||
const apiKey = config.llmProvider === 'openai' ? config.openaiApiKey :
|
||||
config.llmProvider === 'anthropic' ? config.anthropicApiKey :
|
||||
config.llmProvider === 'google' ? config.googleApiKey :
|
||||
config.llmProvider === 'groq' ? config.groqApiKey :
|
||||
config.llmProvider === 'azure_openai' ? config.azureOpenaiApiKey :
|
||||
config.llmProvider === 'ollama' ? '' : // Ollama doesn't need API key
|
||||
config.embeddingProvider === 'openai' ? config.openaiApiKey : '';
|
||||
|
||||
const result = await window.electronAPI.testGraphitiConnection(
|
||||
config.falkorDbUri,
|
||||
apiKey.trim()
|
||||
);
|
||||
const result = await window.electronAPI.testGraphitiConnection({
|
||||
dbPath: config.dbPath || undefined,
|
||||
database: config.database || 'auto_claude_memory',
|
||||
llmProvider: config.llmProvider,
|
||||
apiKey: apiKey.trim()
|
||||
});
|
||||
|
||||
if (result?.success && result?.data) {
|
||||
setValidationStatus({
|
||||
falkordb: {
|
||||
database: {
|
||||
tested: true,
|
||||
success: result.data.falkordb.success,
|
||||
message: result.data.falkordb.message
|
||||
success: result.data.database.success,
|
||||
message: result.data.database.message
|
||||
},
|
||||
provider: {
|
||||
tested: true,
|
||||
success: result.data.openai.success,
|
||||
message: result.data.openai.success
|
||||
success: result.data.llmProvider.success,
|
||||
message: result.data.llmProvider.success
|
||||
? `${config.llmProvider} / ${config.embeddingProvider} providers configured`
|
||||
: result.data.openai.message
|
||||
: result.data.llmProvider.message
|
||||
}
|
||||
});
|
||||
|
||||
if (!result.data.ready) {
|
||||
const errors: string[] = [];
|
||||
if (!result.data.falkordb.success) {
|
||||
errors.push(`FalkorDB: ${result.data.falkordb.message}`);
|
||||
if (!result.data.database.success) {
|
||||
errors.push(`Database: ${result.data.database.message}`);
|
||||
}
|
||||
if (!result.data.openai.success) {
|
||||
errors.push(`Provider: ${result.data.openai.message}`);
|
||||
if (!result.data.llmProvider.success) {
|
||||
errors.push(`Provider: ${result.data.llmProvider.message}`);
|
||||
}
|
||||
if (errors.length > 0) {
|
||||
setError(errors.join('\n'));
|
||||
@@ -359,7 +356,6 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
const needsVoyage = embeddingProvider === 'voyage';
|
||||
const needsGoogle = llmProvider === 'google' || embeddingProvider === 'google';
|
||||
const needsGroq = llmProvider === 'groq';
|
||||
const needsHuggingFace = embeddingProvider === 'huggingface';
|
||||
const needsOllama = llmProvider === 'ollama' || embeddingProvider === 'ollama';
|
||||
|
||||
return (
|
||||
@@ -611,39 +607,6 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* HuggingFace API Key */}
|
||||
{needsHuggingFace && (
|
||||
<div className="space-y-2">
|
||||
<Label htmlFor="huggingface-key" className="text-sm font-medium text-foreground">
|
||||
HuggingFace API Key
|
||||
</Label>
|
||||
<div className="relative">
|
||||
<Input
|
||||
id="huggingface-key"
|
||||
type={showApiKey['huggingface'] ? 'text' : 'password'}
|
||||
value={config.huggingfaceApiKey}
|
||||
onChange={(e) => setConfig(prev => ({ ...prev, huggingfaceApiKey: e.target.value }))}
|
||||
placeholder="hf_..."
|
||||
className="pr-10 font-mono text-sm"
|
||||
disabled={isSaving || isValidating}
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => toggleShowApiKey('huggingface')}
|
||||
className="absolute right-3 top-1/2 -translate-y-1/2 text-muted-foreground hover:text-foreground"
|
||||
>
|
||||
{showApiKey['huggingface'] ? <EyeOff className="h-4 w-4" /> : <Eye className="h-4 w-4" />}
|
||||
</button>
|
||||
</div>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Get your key from{' '}
|
||||
<a href="https://huggingface.co/settings/tokens" target="_blank" rel="noopener noreferrer" className="text-primary hover:text-primary/80">
|
||||
HuggingFace
|
||||
</a>
|
||||
</p>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Ollama Settings */}
|
||||
{needsOllama && (
|
||||
<div className="space-y-3 p-3 rounded-md bg-muted/50">
|
||||
@@ -732,15 +695,15 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
</p>
|
||||
</div>
|
||||
|
||||
{/* Loading state for Docker check */}
|
||||
{isCheckingDocker && (
|
||||
{/* Loading state for infrastructure check */}
|
||||
{isCheckingInfra && (
|
||||
<div className="flex items-center justify-center py-12">
|
||||
<Loader2 className="h-8 w-8 animate-spin text-muted-foreground" />
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Main content */}
|
||||
{!isCheckingDocker && (
|
||||
{!isCheckingInfra && (
|
||||
<div className="space-y-6">
|
||||
{/* Success state */}
|
||||
{success && (
|
||||
@@ -789,19 +752,19 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
</Card>
|
||||
)}
|
||||
|
||||
{/* Docker warning */}
|
||||
{dockerAvailable === false && (
|
||||
<Card className="border border-warning/30 bg-warning/10">
|
||||
{/* Kuzu status notice */}
|
||||
{kuzuAvailable === false && (
|
||||
<Card className="border border-info/30 bg-info/10">
|
||||
<CardContent className="p-4">
|
||||
<div className="flex items-start gap-3">
|
||||
<AlertCircle className="h-5 w-5 text-warning shrink-0 mt-0.5" />
|
||||
<Info className="h-5 w-5 text-info shrink-0 mt-0.5" />
|
||||
<div className="flex-1">
|
||||
<p className="text-sm font-medium text-warning">
|
||||
Docker not detected
|
||||
<p className="text-sm font-medium text-info">
|
||||
Database will be created automatically
|
||||
</p>
|
||||
<p className="text-sm text-warning/80 mt-1">
|
||||
FalkorDB requires Docker to run. You can still configure Graphiti now
|
||||
and set up Docker later.
|
||||
<p className="text-sm text-info/80 mt-1">
|
||||
LadybugDB uses an embedded database - no Docker required.
|
||||
The database will be created when you first use memory features.
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
@@ -827,6 +790,7 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
<li>Persistent memory across coding sessions</li>
|
||||
<li>Better understanding of your codebase over time</li>
|
||||
<li>Reduces repetitive explanations</li>
|
||||
<li>No Docker required - uses embedded database</li>
|
||||
</ul>
|
||||
<button
|
||||
onClick={handleOpenDocs}
|
||||
@@ -851,7 +815,7 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
Enable Graphiti Memory
|
||||
</Label>
|
||||
<p className="text-xs text-muted-foreground mt-0.5">
|
||||
Requires FalkorDB (Docker) and an LLM/embedding provider
|
||||
Uses LadybugDB (embedded) and an LLM/embedding provider
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
@@ -867,42 +831,42 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
{/* Configuration fields (shown when enabled) */}
|
||||
{config.enabled && (
|
||||
<div className="space-y-4 animate-in slide-in-from-top-2 duration-200">
|
||||
{/* FalkorDB URI */}
|
||||
{/* Database Settings */}
|
||||
<div className="space-y-2">
|
||||
<div className="flex items-center justify-between">
|
||||
<div className="flex items-center gap-2">
|
||||
<Server className="h-4 w-4 text-muted-foreground" />
|
||||
<Label htmlFor="falkordb-uri" className="text-sm font-medium text-foreground">
|
||||
FalkorDB URI
|
||||
<Database className="h-4 w-4 text-muted-foreground" />
|
||||
<Label htmlFor="database-name" className="text-sm font-medium text-foreground">
|
||||
Database Name
|
||||
</Label>
|
||||
</div>
|
||||
{validationStatus.falkordb && (
|
||||
{validationStatus.database && (
|
||||
<div className="flex items-center gap-1.5">
|
||||
{validationStatus.falkordb.success ? (
|
||||
{validationStatus.database.success ? (
|
||||
<CheckCircle2 className="h-4 w-4 text-success" />
|
||||
) : (
|
||||
<XCircle className="h-4 w-4 text-destructive" />
|
||||
)}
|
||||
<span className={`text-xs ${validationStatus.falkordb.success ? 'text-success' : 'text-destructive'}`}>
|
||||
{validationStatus.falkordb.success ? 'Connected' : 'Failed'}
|
||||
<span className={`text-xs ${validationStatus.database.success ? 'text-success' : 'text-destructive'}`}>
|
||||
{validationStatus.database.success ? 'Ready' : 'Issue'}
|
||||
</span>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
<Input
|
||||
id="falkordb-uri"
|
||||
id="database-name"
|
||||
type="text"
|
||||
value={config.falkorDbUri}
|
||||
value={config.database}
|
||||
onChange={(e) => {
|
||||
setConfig(prev => ({ ...prev, falkorDbUri: e.target.value }));
|
||||
setValidationStatus(prev => ({ ...prev, falkordb: null }));
|
||||
setConfig(prev => ({ ...prev, database: e.target.value }));
|
||||
setValidationStatus(prev => ({ ...prev, database: null }));
|
||||
}}
|
||||
placeholder="bolt://localhost:6379"
|
||||
placeholder="auto_claude_memory"
|
||||
className="font-mono text-sm"
|
||||
disabled={isSaving || isValidating}
|
||||
/>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Auto-detected from Docker if FalkorDB is running
|
||||
Stored in ~/.auto-claude/graphs/
|
||||
</p>
|
||||
</div>
|
||||
|
||||
@@ -990,7 +954,7 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
</>
|
||||
)}
|
||||
</Button>
|
||||
{validationStatus.falkordb?.success && validationStatus.provider?.success && (
|
||||
{validationStatus.database?.success && validationStatus.provider?.success && (
|
||||
<p className="text-xs text-success text-center mt-2">
|
||||
All connections validated successfully!
|
||||
</p>
|
||||
@@ -1032,7 +996,7 @@ export function GraphitiStep({ onNext, onBack, onSkip }: GraphitiStepProps) {
|
||||
</Button>
|
||||
<Button
|
||||
onClick={handleContinue}
|
||||
disabled={isCheckingDocker || (config.enabled && !!getRequiredApiKey() && !success) || isSaving || isValidating}
|
||||
disabled={isCheckingInfra || (config.enabled && !!getRequiredApiKey() && !success) || isSaving || isValidating}
|
||||
>
|
||||
{isSaving ? (
|
||||
<>
|
||||
|
||||
@@ -0,0 +1,532 @@
|
||||
import { useState, useEffect } from 'react';
|
||||
import {
|
||||
Brain,
|
||||
Database,
|
||||
Info,
|
||||
Loader2,
|
||||
CheckCircle2,
|
||||
AlertCircle,
|
||||
Eye,
|
||||
EyeOff,
|
||||
} from 'lucide-react';
|
||||
import { Button } from '../ui/button';
|
||||
import { Input } from '../ui/input';
|
||||
import { Label } from '../ui/label';
|
||||
import { Card, CardContent } from '../ui/card';
|
||||
import {
|
||||
Select,
|
||||
SelectContent,
|
||||
SelectItem,
|
||||
SelectTrigger,
|
||||
SelectValue
|
||||
} from '../ui/select';
|
||||
import { OllamaModelSelector } from './OllamaModelSelector';
|
||||
import { useSettingsStore } from '../../stores/settings-store';
|
||||
import type { GraphitiEmbeddingProvider, AppSettings } from '../../../shared/types';
|
||||
|
||||
interface MemoryStepProps {
|
||||
onNext: () => void;
|
||||
onBack: () => void;
|
||||
}
|
||||
|
||||
// Embedding provider configurations (LLM provider removed - Claude SDK handles RAG)
|
||||
const EMBEDDING_PROVIDERS: Array<{
|
||||
id: GraphitiEmbeddingProvider;
|
||||
name: string;
|
||||
description: string;
|
||||
requiresApiKey: boolean;
|
||||
}> = [
|
||||
{ id: 'ollama', name: 'Ollama (Local)', description: 'Free, local embeddings', requiresApiKey: false },
|
||||
{ id: 'openai', name: 'OpenAI', description: 'text-embedding-3-small', requiresApiKey: true },
|
||||
{ id: 'voyage', name: 'Voyage AI', description: 'voyage-3 (high quality)', requiresApiKey: true },
|
||||
{ id: 'google', name: 'Google AI', description: 'text-embedding-004', requiresApiKey: true },
|
||||
{ id: 'azure_openai', name: 'Azure OpenAI', description: 'Enterprise deployment', requiresApiKey: true },
|
||||
];
|
||||
|
||||
interface MemoryConfig {
|
||||
database: string;
|
||||
embeddingProvider: GraphitiEmbeddingProvider;
|
||||
// OpenAI
|
||||
openaiApiKey: string;
|
||||
// Azure OpenAI
|
||||
azureOpenaiApiKey: string;
|
||||
azureOpenaiBaseUrl: string;
|
||||
azureOpenaiEmbeddingDeployment: string;
|
||||
// Voyage
|
||||
voyageApiKey: string;
|
||||
// Google
|
||||
googleApiKey: string;
|
||||
// Ollama
|
||||
ollamaBaseUrl: string;
|
||||
ollamaEmbeddingModel: string;
|
||||
ollamaEmbeddingDim: number;
|
||||
}
|
||||
|
||||
/**
|
||||
* Memory configuration step for the onboarding wizard.
|
||||
*
|
||||
* Key simplifications from the previous GraphitiStep:
|
||||
* - Memory is always enabled (no toggle)
|
||||
* - LLM provider removed (Claude SDK handles RAG queries)
|
||||
* - Ollama is the default with model discovery + download
|
||||
* - Keyword search works as fallback without embeddings
|
||||
*/
|
||||
export function MemoryStep({ onNext, onBack }: MemoryStepProps) {
|
||||
const { settings, updateSettings } = useSettingsStore();
|
||||
const [config, setConfig] = useState<MemoryConfig>({
|
||||
database: 'auto_claude_memory',
|
||||
embeddingProvider: 'ollama',
|
||||
openaiApiKey: settings.globalOpenAIApiKey || '',
|
||||
azureOpenaiApiKey: '',
|
||||
azureOpenaiBaseUrl: '',
|
||||
azureOpenaiEmbeddingDeployment: '',
|
||||
voyageApiKey: '',
|
||||
googleApiKey: settings.globalGoogleApiKey || '',
|
||||
ollamaBaseUrl: settings.ollamaBaseUrl || 'http://localhost:11434',
|
||||
ollamaEmbeddingModel: 'embeddinggemma',
|
||||
ollamaEmbeddingDim: 768,
|
||||
});
|
||||
const [showApiKey, setShowApiKey] = useState<Record<string, boolean>>({});
|
||||
const [isSaving, setIsSaving] = useState(false);
|
||||
const [error, setError] = useState<string | null>(null);
|
||||
const [isCheckingInfra, setIsCheckingInfra] = useState(true);
|
||||
const [kuzuAvailable, setKuzuAvailable] = useState<boolean | null>(null);
|
||||
|
||||
// Check LadybugDB/Kuzu availability on mount
|
||||
useEffect(() => {
|
||||
const checkInfrastructure = async () => {
|
||||
setIsCheckingInfra(true);
|
||||
try {
|
||||
const result = await window.electronAPI.getMemoryInfrastructureStatus();
|
||||
setKuzuAvailable(result?.success && result?.data?.memory?.kuzuInstalled ? true : false);
|
||||
} catch {
|
||||
setKuzuAvailable(false);
|
||||
} finally {
|
||||
setIsCheckingInfra(false);
|
||||
}
|
||||
};
|
||||
|
||||
checkInfrastructure();
|
||||
}, []);
|
||||
|
||||
const toggleShowApiKey = (key: string) => {
|
||||
setShowApiKey(prev => ({ ...prev, [key]: !prev[key] }));
|
||||
};
|
||||
|
||||
// Check if we have valid configuration
|
||||
const isConfigValid = (): boolean => {
|
||||
const { embeddingProvider } = config;
|
||||
|
||||
// Ollama just needs a model selected
|
||||
if (embeddingProvider === 'ollama') {
|
||||
return !!config.ollamaEmbeddingModel.trim();
|
||||
}
|
||||
|
||||
// Other providers need API keys
|
||||
if (embeddingProvider === 'openai' && !config.openaiApiKey.trim()) return false;
|
||||
if (embeddingProvider === 'voyage' && !config.voyageApiKey.trim()) return false;
|
||||
if (embeddingProvider === 'google' && !config.googleApiKey.trim()) return false;
|
||||
if (embeddingProvider === 'azure_openai') {
|
||||
if (!config.azureOpenaiApiKey.trim()) return false;
|
||||
if (!config.azureOpenaiBaseUrl.trim()) return false;
|
||||
if (!config.azureOpenaiEmbeddingDeployment.trim()) return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
};
|
||||
|
||||
const handleSave = async () => {
|
||||
setIsSaving(true);
|
||||
setError(null);
|
||||
|
||||
try {
|
||||
// Save the API keys to global settings
|
||||
const settingsToSave: Record<string, string | undefined> = {};
|
||||
|
||||
if (config.openaiApiKey.trim()) {
|
||||
settingsToSave.globalOpenAIApiKey = config.openaiApiKey.trim();
|
||||
}
|
||||
if (config.googleApiKey.trim()) {
|
||||
settingsToSave.globalGoogleApiKey = config.googleApiKey.trim();
|
||||
}
|
||||
if (config.ollamaBaseUrl.trim()) {
|
||||
settingsToSave.ollamaBaseUrl = config.ollamaBaseUrl.trim();
|
||||
}
|
||||
|
||||
const result = await window.electronAPI.saveSettings(settingsToSave);
|
||||
|
||||
if (result?.success) {
|
||||
// Update local settings store
|
||||
const storeUpdate: Partial<Pick<AppSettings, 'globalOpenAIApiKey' | 'globalGoogleApiKey' | 'ollamaBaseUrl'>> = {};
|
||||
if (config.openaiApiKey.trim()) storeUpdate.globalOpenAIApiKey = config.openaiApiKey.trim();
|
||||
if (config.googleApiKey.trim()) storeUpdate.globalGoogleApiKey = config.googleApiKey.trim();
|
||||
if (config.ollamaBaseUrl.trim()) storeUpdate.ollamaBaseUrl = config.ollamaBaseUrl.trim();
|
||||
updateSettings(storeUpdate);
|
||||
onNext();
|
||||
} else {
|
||||
setError(result?.error || 'Failed to save memory configuration');
|
||||
}
|
||||
} catch (err) {
|
||||
setError(err instanceof Error ? err.message : 'Unknown error occurred');
|
||||
} finally {
|
||||
setIsSaving(false);
|
||||
}
|
||||
};
|
||||
|
||||
const handleContinue = () => {
|
||||
handleSave();
|
||||
};
|
||||
|
||||
const handleOllamaModelSelect = (modelName: string, dim: number) => {
|
||||
setConfig(prev => ({
|
||||
...prev,
|
||||
ollamaEmbeddingModel: modelName,
|
||||
ollamaEmbeddingDim: dim,
|
||||
}));
|
||||
};
|
||||
|
||||
// Render provider-specific configuration fields
|
||||
const renderProviderFields = () => {
|
||||
const { embeddingProvider } = config;
|
||||
|
||||
if (embeddingProvider === 'ollama') {
|
||||
return (
|
||||
<div className="space-y-4">
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">
|
||||
Select Embedding Model
|
||||
</Label>
|
||||
<OllamaModelSelector
|
||||
selectedModel={config.ollamaEmbeddingModel}
|
||||
onModelSelect={handleOllamaModelSelect}
|
||||
disabled={isSaving}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
if (embeddingProvider === 'openai') {
|
||||
return (
|
||||
<div className="space-y-2">
|
||||
<Label htmlFor="openai-key" className="text-sm font-medium text-foreground">
|
||||
OpenAI API Key
|
||||
</Label>
|
||||
<div className="relative">
|
||||
<Input
|
||||
id="openai-key"
|
||||
type={showApiKey['openai'] ? 'text' : 'password'}
|
||||
value={config.openaiApiKey}
|
||||
onChange={(e) => setConfig(prev => ({ ...prev, openaiApiKey: e.target.value }))}
|
||||
placeholder="sk-..."
|
||||
className="pr-10 font-mono text-sm"
|
||||
disabled={isSaving}
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => toggleShowApiKey('openai')}
|
||||
className="absolute right-3 top-1/2 -translate-y-1/2 text-muted-foreground hover:text-foreground"
|
||||
>
|
||||
{showApiKey['openai'] ? <EyeOff className="h-4 w-4" /> : <Eye className="h-4 w-4" />}
|
||||
</button>
|
||||
</div>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Get your key from{' '}
|
||||
<a href="https://platform.openai.com/api-keys" target="_blank" rel="noopener noreferrer" className="text-primary hover:text-primary/80">
|
||||
OpenAI
|
||||
</a>
|
||||
</p>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
if (embeddingProvider === 'voyage') {
|
||||
return (
|
||||
<div className="space-y-2">
|
||||
<Label htmlFor="voyage-key" className="text-sm font-medium text-foreground">
|
||||
Voyage API Key
|
||||
</Label>
|
||||
<div className="relative">
|
||||
<Input
|
||||
id="voyage-key"
|
||||
type={showApiKey['voyage'] ? 'text' : 'password'}
|
||||
value={config.voyageApiKey}
|
||||
onChange={(e) => setConfig(prev => ({ ...prev, voyageApiKey: e.target.value }))}
|
||||
placeholder="pa-..."
|
||||
className="pr-10 font-mono text-sm"
|
||||
disabled={isSaving}
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => toggleShowApiKey('voyage')}
|
||||
className="absolute right-3 top-1/2 -translate-y-1/2 text-muted-foreground hover:text-foreground"
|
||||
>
|
||||
{showApiKey['voyage'] ? <EyeOff className="h-4 w-4" /> : <Eye className="h-4 w-4" />}
|
||||
</button>
|
||||
</div>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Get your key from{' '}
|
||||
<a href="https://dash.voyageai.com/api-keys" target="_blank" rel="noopener noreferrer" className="text-primary hover:text-primary/80">
|
||||
Voyage AI
|
||||
</a>
|
||||
</p>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
if (embeddingProvider === 'google') {
|
||||
return (
|
||||
<div className="space-y-2">
|
||||
<Label htmlFor="google-key" className="text-sm font-medium text-foreground">
|
||||
Google API Key
|
||||
</Label>
|
||||
<div className="relative">
|
||||
<Input
|
||||
id="google-key"
|
||||
type={showApiKey['google'] ? 'text' : 'password'}
|
||||
value={config.googleApiKey}
|
||||
onChange={(e) => setConfig(prev => ({ ...prev, googleApiKey: e.target.value }))}
|
||||
placeholder="AIza..."
|
||||
className="pr-10 font-mono text-sm"
|
||||
disabled={isSaving}
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => toggleShowApiKey('google')}
|
||||
className="absolute right-3 top-1/2 -translate-y-1/2 text-muted-foreground hover:text-foreground"
|
||||
>
|
||||
{showApiKey['google'] ? <EyeOff className="h-4 w-4" /> : <Eye className="h-4 w-4" />}
|
||||
</button>
|
||||
</div>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Get your key from{' '}
|
||||
<a href="https://aistudio.google.com/apikey" target="_blank" rel="noopener noreferrer" className="text-primary hover:text-primary/80">
|
||||
Google AI Studio
|
||||
</a>
|
||||
</p>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
if (embeddingProvider === 'azure_openai') {
|
||||
return (
|
||||
<div className="space-y-3 p-3 rounded-md bg-muted/50">
|
||||
<p className="text-sm font-medium text-foreground">Azure OpenAI Settings</p>
|
||||
<div className="space-y-2">
|
||||
<Label htmlFor="azure-key" className="text-xs text-muted-foreground">API Key</Label>
|
||||
<div className="relative">
|
||||
<Input
|
||||
id="azure-key"
|
||||
type={showApiKey['azure'] ? 'text' : 'password'}
|
||||
value={config.azureOpenaiApiKey}
|
||||
onChange={(e) => setConfig(prev => ({ ...prev, azureOpenaiApiKey: e.target.value }))}
|
||||
placeholder="Azure API key"
|
||||
className="pr-10 font-mono text-sm"
|
||||
disabled={isSaving}
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => toggleShowApiKey('azure')}
|
||||
className="absolute right-3 top-1/2 -translate-y-1/2 text-muted-foreground hover:text-foreground"
|
||||
>
|
||||
{showApiKey['azure'] ? <EyeOff className="h-4 w-4" /> : <Eye className="h-4 w-4" />}
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
<div className="space-y-2">
|
||||
<Label htmlFor="azure-url" className="text-xs text-muted-foreground">Base URL</Label>
|
||||
<Input
|
||||
id="azure-url"
|
||||
type="text"
|
||||
value={config.azureOpenaiBaseUrl}
|
||||
onChange={(e) => setConfig(prev => ({ ...prev, azureOpenaiBaseUrl: e.target.value }))}
|
||||
placeholder="https://your-resource.openai.azure.com"
|
||||
className="font-mono text-sm"
|
||||
disabled={isSaving}
|
||||
/>
|
||||
</div>
|
||||
<div className="space-y-2">
|
||||
<Label htmlFor="azure-embedding-deployment" className="text-xs text-muted-foreground">Embedding Deployment Name</Label>
|
||||
<Input
|
||||
id="azure-embedding-deployment"
|
||||
type="text"
|
||||
value={config.azureOpenaiEmbeddingDeployment}
|
||||
onChange={(e) => setConfig(prev => ({ ...prev, azureOpenaiEmbeddingDeployment: e.target.value }))}
|
||||
placeholder="text-embedding-ada-002"
|
||||
className="font-mono text-sm"
|
||||
disabled={isSaving}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
return null;
|
||||
};
|
||||
|
||||
return (
|
||||
<div className="flex h-full flex-col items-center justify-center px-8 py-6">
|
||||
<div className="w-full max-w-2xl">
|
||||
{/* Header */}
|
||||
<div className="text-center mb-8">
|
||||
<div className="flex justify-center mb-4">
|
||||
<div className="flex h-14 w-14 items-center justify-center rounded-full bg-primary/10 text-primary">
|
||||
<Brain className="h-7 w-7" />
|
||||
</div>
|
||||
</div>
|
||||
<h1 className="text-2xl font-bold text-foreground tracking-tight">
|
||||
Memory
|
||||
</h1>
|
||||
<p className="mt-2 text-muted-foreground">
|
||||
Auto Claude Memory helps remember context across your coding sessions
|
||||
</p>
|
||||
</div>
|
||||
|
||||
{/* Loading state for infrastructure check */}
|
||||
{isCheckingInfra && (
|
||||
<div className="flex items-center justify-center py-12">
|
||||
<Loader2 className="h-8 w-8 animate-spin text-muted-foreground" />
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Main content */}
|
||||
{!isCheckingInfra && (
|
||||
<div className="space-y-6">
|
||||
{/* Error banner */}
|
||||
{error && (
|
||||
<Card className="border border-destructive/30 bg-destructive/10">
|
||||
<CardContent className="p-4">
|
||||
<div className="flex items-start gap-3">
|
||||
<AlertCircle className="h-5 w-5 text-destructive shrink-0 mt-0.5" />
|
||||
<p className="text-sm text-destructive whitespace-pre-line">{error}</p>
|
||||
</div>
|
||||
</CardContent>
|
||||
</Card>
|
||||
)}
|
||||
|
||||
{/* Kuzu status notice */}
|
||||
{kuzuAvailable === false && (
|
||||
<Card className="border border-info/30 bg-info/10">
|
||||
<CardContent className="p-4">
|
||||
<div className="flex items-start gap-3">
|
||||
<Info className="h-5 w-5 text-info shrink-0 mt-0.5" />
|
||||
<div className="flex-1">
|
||||
<p className="text-sm font-medium text-info">
|
||||
Database will be created automatically
|
||||
</p>
|
||||
<p className="text-sm text-info/80 mt-1">
|
||||
Memory uses an embedded database - no Docker required.
|
||||
It will be created when you first use memory features.
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
</CardContent>
|
||||
</Card>
|
||||
)}
|
||||
|
||||
{/* Info card about Memory */}
|
||||
<Card className="border border-info/30 bg-info/10">
|
||||
<CardContent className="p-5">
|
||||
<div className="flex items-start gap-4">
|
||||
<Info className="h-5 w-5 text-info shrink-0 mt-0.5" />
|
||||
<div className="flex-1 space-y-3">
|
||||
<p className="text-sm font-medium text-foreground">
|
||||
What does Memory do?
|
||||
</p>
|
||||
<p className="text-sm text-muted-foreground">
|
||||
Memory stores discoveries, patterns, and insights about your codebase
|
||||
so future sessions start with context already loaded.
|
||||
</p>
|
||||
<ul className="text-sm text-muted-foreground space-y-1.5 list-disc list-inside">
|
||||
<li>Remembers patterns across sessions</li>
|
||||
<li>Understands your codebase over time</li>
|
||||
<li>Works offline - no cloud required</li>
|
||||
</ul>
|
||||
</div>
|
||||
</div>
|
||||
</CardContent>
|
||||
</Card>
|
||||
|
||||
{/* Database info */}
|
||||
<div className="flex items-center gap-3 p-3 rounded-md bg-muted/50">
|
||||
<Database className="h-5 w-5 text-muted-foreground" />
|
||||
<div>
|
||||
<p className="text-sm font-medium text-foreground">
|
||||
Memory Database
|
||||
</p>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Stored in ~/.auto-claude/memories/
|
||||
</p>
|
||||
</div>
|
||||
{kuzuAvailable && (
|
||||
<CheckCircle2 className="h-4 w-4 text-success ml-auto" />
|
||||
)}
|
||||
</div>
|
||||
|
||||
{/* Embedding Provider Selection */}
|
||||
<div className="space-y-4">
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">
|
||||
Embedding Provider (for semantic search)
|
||||
</Label>
|
||||
<Select
|
||||
value={config.embeddingProvider}
|
||||
onValueChange={(value: GraphitiEmbeddingProvider) => {
|
||||
setConfig(prev => ({ ...prev, embeddingProvider: value }));
|
||||
}}
|
||||
disabled={isSaving}
|
||||
>
|
||||
<SelectTrigger>
|
||||
<SelectValue />
|
||||
</SelectTrigger>
|
||||
<SelectContent>
|
||||
{EMBEDDING_PROVIDERS.map(p => (
|
||||
<SelectItem key={p.id} value={p.id}>
|
||||
<div className="flex flex-col">
|
||||
<span>{p.name}</span>
|
||||
<span className="text-xs text-muted-foreground">{p.description}</span>
|
||||
</div>
|
||||
</SelectItem>
|
||||
))}
|
||||
</SelectContent>
|
||||
</Select>
|
||||
</div>
|
||||
|
||||
{/* Provider-specific fields */}
|
||||
{renderProviderFields()}
|
||||
</div>
|
||||
|
||||
{/* Fallback info */}
|
||||
<p className="text-xs text-muted-foreground text-center">
|
||||
No embedding provider? Memory still works with keyword search. Semantic search is an upgrade.
|
||||
</p>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Action Buttons */}
|
||||
<div className="flex justify-between items-center mt-10 pt-6 border-t border-border">
|
||||
<Button
|
||||
variant="ghost"
|
||||
onClick={onBack}
|
||||
className="text-muted-foreground hover:text-foreground"
|
||||
>
|
||||
Back
|
||||
</Button>
|
||||
<Button
|
||||
onClick={handleContinue}
|
||||
disabled={isCheckingInfra || !isConfigValid() || isSaving}
|
||||
>
|
||||
{isSaving ? (
|
||||
<>
|
||||
<Loader2 className="h-4 w-4 animate-spin mr-2" />
|
||||
Saving...
|
||||
</>
|
||||
) : (
|
||||
'Save & Continue'
|
||||
)}
|
||||
</Button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,287 @@
|
||||
import { useState, useEffect } from 'react';
|
||||
import {
|
||||
Check,
|
||||
Download,
|
||||
Loader2,
|
||||
AlertCircle,
|
||||
RefreshCw
|
||||
} from 'lucide-react';
|
||||
import { Button } from '../ui/button';
|
||||
import { cn } from '../../lib/utils';
|
||||
|
||||
interface OllamaModel {
|
||||
name: string;
|
||||
description: string;
|
||||
size_estimate?: string;
|
||||
dim: number;
|
||||
installed: boolean;
|
||||
}
|
||||
|
||||
interface OllamaModelSelectorProps {
|
||||
selectedModel: string;
|
||||
onModelSelect: (model: string, dim: number) => void;
|
||||
disabled?: boolean;
|
||||
className?: string;
|
||||
}
|
||||
|
||||
// Recommended embedding models for Auto Claude Memory
|
||||
// embeddinggemma is first as the recommended default
|
||||
const RECOMMENDED_MODELS: OllamaModel[] = [
|
||||
{
|
||||
name: 'embeddinggemma',
|
||||
description: "Google's lightweight embedding model (Recommended)",
|
||||
size_estimate: '621 MB',
|
||||
dim: 768,
|
||||
installed: false,
|
||||
},
|
||||
{
|
||||
name: 'nomic-embed-text',
|
||||
description: 'Popular general-purpose embeddings',
|
||||
size_estimate: '274 MB',
|
||||
dim: 768,
|
||||
installed: false,
|
||||
},
|
||||
{
|
||||
name: 'mxbai-embed-large',
|
||||
description: 'MixedBread AI large embeddings',
|
||||
size_estimate: '670 MB',
|
||||
dim: 1024,
|
||||
installed: false,
|
||||
},
|
||||
];
|
||||
|
||||
/**
|
||||
* OllamaModelSelector - Select or download Ollama embedding models
|
||||
*
|
||||
* Shows installed models with checkmarks and recommended models with download buttons.
|
||||
* Automatically refreshes the list after successful downloads.
|
||||
*/
|
||||
export function OllamaModelSelector({
|
||||
selectedModel,
|
||||
onModelSelect,
|
||||
disabled = false,
|
||||
className,
|
||||
}: OllamaModelSelectorProps) {
|
||||
const [models, setModels] = useState<OllamaModel[]>(RECOMMENDED_MODELS);
|
||||
const [isLoading, setIsLoading] = useState(true);
|
||||
const [isDownloading, setIsDownloading] = useState<string | null>(null);
|
||||
const [error, setError] = useState<string | null>(null);
|
||||
const [ollamaAvailable, setOllamaAvailable] = useState(true);
|
||||
|
||||
// Check installed models - used by both mount effect and refresh after download
|
||||
const checkInstalledModels = async (abortSignal?: AbortSignal) => {
|
||||
setIsLoading(true);
|
||||
setError(null);
|
||||
|
||||
try {
|
||||
// Check Ollama status first
|
||||
const statusResult = await window.electronAPI.checkOllamaStatus();
|
||||
if (abortSignal?.aborted) return;
|
||||
|
||||
if (!statusResult?.success || !statusResult?.data?.running) {
|
||||
setOllamaAvailable(false);
|
||||
setIsLoading(false);
|
||||
return;
|
||||
}
|
||||
|
||||
setOllamaAvailable(true);
|
||||
|
||||
// Get list of installed embedding models
|
||||
const result = await window.electronAPI.listOllamaEmbeddingModels();
|
||||
if (abortSignal?.aborted) return;
|
||||
|
||||
if (result?.success && result?.data?.embedding_models) {
|
||||
const installedNames = new Set(
|
||||
result.data.embedding_models.map((m: { name: string }) => {
|
||||
// Normalize: "embeddinggemma:latest" -> "embeddinggemma"
|
||||
const name = m.name;
|
||||
return name.includes(':') ? name.split(':')[0] : name;
|
||||
})
|
||||
);
|
||||
|
||||
// Update models with installation status
|
||||
setModels(
|
||||
RECOMMENDED_MODELS.map(model => {
|
||||
const baseName = model.name.includes(':') ? model.name.split(':')[0] : model.name;
|
||||
return {
|
||||
...model,
|
||||
installed: installedNames.has(baseName) || installedNames.has(model.name),
|
||||
};
|
||||
})
|
||||
);
|
||||
}
|
||||
} catch (err) {
|
||||
if (!abortSignal?.aborted) {
|
||||
console.error('Failed to check Ollama models:', err);
|
||||
setError('Failed to check Ollama models');
|
||||
}
|
||||
} finally {
|
||||
if (!abortSignal?.aborted) {
|
||||
setIsLoading(false);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
// Fetch installed models on mount with cleanup
|
||||
useEffect(() => {
|
||||
const controller = new AbortController();
|
||||
checkInstalledModels(controller.signal);
|
||||
return () => controller.abort();
|
||||
}, []);
|
||||
|
||||
const handleDownload = async (modelName: string) => {
|
||||
setIsDownloading(modelName);
|
||||
setError(null);
|
||||
|
||||
try {
|
||||
const result = await window.electronAPI.pullOllamaModel(modelName);
|
||||
if (result?.success) {
|
||||
// Refresh the model list
|
||||
await checkInstalledModels();
|
||||
} else {
|
||||
setError(result?.error || `Failed to download ${modelName}`);
|
||||
}
|
||||
} catch (err) {
|
||||
setError(err instanceof Error ? err.message : 'Download failed');
|
||||
} finally {
|
||||
setIsDownloading(null);
|
||||
}
|
||||
};
|
||||
|
||||
const handleSelect = (model: OllamaModel) => {
|
||||
if (!model.installed || disabled) return;
|
||||
onModelSelect(model.name, model.dim);
|
||||
};
|
||||
|
||||
if (isLoading) {
|
||||
return (
|
||||
<div className={cn('flex items-center justify-center py-8', className)}>
|
||||
<Loader2 className="h-5 w-5 animate-spin text-muted-foreground" />
|
||||
<span className="ml-2 text-sm text-muted-foreground">Checking Ollama models...</span>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
if (!ollamaAvailable) {
|
||||
return (
|
||||
<div className={cn('rounded-lg border border-warning/30 bg-warning/10 p-4', className)}>
|
||||
<div className="flex items-start gap-3">
|
||||
<AlertCircle className="h-5 w-5 text-warning shrink-0 mt-0.5" />
|
||||
<div className="flex-1">
|
||||
<p className="text-sm font-medium text-warning">Ollama not running</p>
|
||||
<p className="text-sm text-warning/80 mt-1">
|
||||
Start Ollama to use local embedding models. Memory will still work with keyword search.
|
||||
</p>
|
||||
<Button
|
||||
variant="outline"
|
||||
size="sm"
|
||||
onClick={() => checkInstalledModels()}
|
||||
className="mt-3"
|
||||
>
|
||||
<RefreshCw className="h-3.5 w-3.5 mr-1.5" />
|
||||
Retry
|
||||
</Button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
return (
|
||||
<div className={cn('space-y-3', className)}>
|
||||
{error && (
|
||||
<div className="rounded-md border border-destructive/30 bg-destructive/10 p-3">
|
||||
<p className="text-sm text-destructive">{error}</p>
|
||||
</div>
|
||||
)}
|
||||
|
||||
<div className="space-y-2">
|
||||
{models.map(model => {
|
||||
const isSelected = selectedModel === model.name;
|
||||
const isCurrentlyDownloading = isDownloading === model.name;
|
||||
|
||||
return (
|
||||
<div
|
||||
key={model.name}
|
||||
className={cn(
|
||||
'flex items-center justify-between rounded-lg border p-3 transition-colors',
|
||||
model.installed && !disabled
|
||||
? 'cursor-pointer hover:bg-accent/50'
|
||||
: 'cursor-default',
|
||||
isSelected && 'border-primary bg-primary/5',
|
||||
!model.installed && 'bg-muted/30'
|
||||
)}
|
||||
onClick={() => handleSelect(model)}
|
||||
>
|
||||
<div className="flex items-center gap-3">
|
||||
{/* Selection/Status indicator */}
|
||||
<div
|
||||
className={cn(
|
||||
'flex h-5 w-5 items-center justify-center rounded-full border-2',
|
||||
isSelected
|
||||
? 'border-primary bg-primary text-primary-foreground'
|
||||
: model.installed
|
||||
? 'border-muted-foreground/30'
|
||||
: 'border-muted-foreground/20 bg-muted/50'
|
||||
)}
|
||||
>
|
||||
{isSelected && <Check className="h-3 w-3" />}
|
||||
</div>
|
||||
|
||||
<div>
|
||||
<div className="flex items-center gap-2">
|
||||
<span className="text-sm font-medium">{model.name}</span>
|
||||
<span className="text-xs text-muted-foreground">
|
||||
({model.dim} dim)
|
||||
</span>
|
||||
{model.installed && (
|
||||
<span className="inline-flex items-center rounded-full bg-success/10 px-2 py-0.5 text-xs text-success">
|
||||
Installed
|
||||
</span>
|
||||
)}
|
||||
</div>
|
||||
<p className="text-xs text-muted-foreground">{model.description}</p>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{/* Download button for non-installed models */}
|
||||
{!model.installed && (
|
||||
<Button
|
||||
variant="outline"
|
||||
size="sm"
|
||||
onClick={(e) => {
|
||||
e.stopPropagation();
|
||||
handleDownload(model.name);
|
||||
}}
|
||||
disabled={isCurrentlyDownloading || disabled}
|
||||
className="shrink-0"
|
||||
>
|
||||
{isCurrentlyDownloading ? (
|
||||
<>
|
||||
<Loader2 className="h-3.5 w-3.5 animate-spin mr-1.5" />
|
||||
Downloading...
|
||||
</>
|
||||
) : (
|
||||
<>
|
||||
<Download className="h-3.5 w-3.5 mr-1.5" />
|
||||
Download
|
||||
{model.size_estimate && (
|
||||
<span className="ml-1 text-muted-foreground">
|
||||
({model.size_estimate})
|
||||
</span>
|
||||
)}
|
||||
</>
|
||||
)}
|
||||
</Button>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
})}
|
||||
</div>
|
||||
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Select an installed model for semantic search. Memory works with keyword search even without embeddings.
|
||||
</p>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
@@ -12,7 +12,7 @@ import { ScrollArea } from '../ui/scroll-area';
|
||||
import { WizardProgress, WizardStep } from './WizardProgress';
|
||||
import { WelcomeStep } from './WelcomeStep';
|
||||
import { OAuthStep } from './OAuthStep';
|
||||
import { GraphitiStep } from './GraphitiStep';
|
||||
import { MemoryStep } from './MemoryStep';
|
||||
import { CompletionStep } from './CompletionStep';
|
||||
import { useSettingsStore } from '../../stores/settings-store';
|
||||
|
||||
@@ -24,13 +24,13 @@ interface OnboardingWizardProps {
|
||||
}
|
||||
|
||||
// Wizard step identifiers
|
||||
type WizardStepId = 'welcome' | 'oauth' | 'graphiti' | 'completion';
|
||||
type WizardStepId = 'welcome' | 'oauth' | 'memory' | 'completion';
|
||||
|
||||
// Step configuration
|
||||
const WIZARD_STEPS: { id: WizardStepId; label: string }[] = [
|
||||
{ id: 'welcome', label: 'Welcome' },
|
||||
{ id: 'oauth', label: 'Auth' },
|
||||
{ id: 'graphiti', label: 'Memory' },
|
||||
{ id: 'memory', label: 'Memory' },
|
||||
{ id: 'completion', label: 'Done' }
|
||||
];
|
||||
|
||||
@@ -153,12 +153,11 @@ export function OnboardingWizard({
|
||||
onSkip={skipWizard}
|
||||
/>
|
||||
);
|
||||
case 'graphiti':
|
||||
case 'memory':
|
||||
return (
|
||||
<GraphitiStep
|
||||
<MemoryStep
|
||||
onNext={goToNextStep}
|
||||
onBack={goToPreviousStep}
|
||||
onSkip={skipWizard}
|
||||
/>
|
||||
);
|
||||
case 'completion':
|
||||
|
||||
@@ -6,7 +6,11 @@
|
||||
export { OnboardingWizard } from './OnboardingWizard';
|
||||
export { WelcomeStep } from './WelcomeStep';
|
||||
export { OAuthStep } from './OAuthStep';
|
||||
export { GraphitiStep } from './GraphitiStep';
|
||||
export { MemoryStep } from './MemoryStep';
|
||||
export { OllamaModelSelector } from './OllamaModelSelector';
|
||||
export { FirstSpecStep } from './FirstSpecStep';
|
||||
export { CompletionStep } from './CompletionStep';
|
||||
export { WizardProgress, type WizardStep } from './WizardProgress';
|
||||
|
||||
// Legacy export for backward compatibility
|
||||
export { GraphitiStep } from './GraphitiStep';
|
||||
|
||||
@@ -1,133 +1,91 @@
|
||||
import { Loader2, CheckCircle2, AlertCircle, Download, Zap } from 'lucide-react';
|
||||
import { Button } from '../ui/button';
|
||||
import { Loader2, CheckCircle2, AlertCircle, Database } from 'lucide-react';
|
||||
import type { InfrastructureStatus as InfrastructureStatusType } from '../../../shared/types';
|
||||
|
||||
interface InfrastructureStatusProps {
|
||||
infrastructureStatus: InfrastructureStatusType | null;
|
||||
isCheckingInfrastructure: boolean;
|
||||
isStartingFalkorDB: boolean;
|
||||
isOpeningDocker: boolean;
|
||||
onStartFalkorDB: () => void;
|
||||
onOpenDockerDesktop: () => void;
|
||||
onDownloadDocker: () => void;
|
||||
}
|
||||
|
||||
/**
|
||||
* Memory Infrastructure Status Component
|
||||
* Shows status of LadybugDB (embedded database) - no Docker required
|
||||
*/
|
||||
export function InfrastructureStatus({
|
||||
infrastructureStatus,
|
||||
isCheckingInfrastructure,
|
||||
isStartingFalkorDB,
|
||||
isOpeningDocker,
|
||||
onStartFalkorDB,
|
||||
onOpenDockerDesktop,
|
||||
onDownloadDocker,
|
||||
}: InfrastructureStatusProps) {
|
||||
return (
|
||||
<div className="rounded-lg border border-border bg-muted/30 p-3 space-y-3">
|
||||
<div className="flex items-center justify-between">
|
||||
<span className="text-sm font-medium text-foreground">Infrastructure Status</span>
|
||||
<span className="text-sm font-medium text-foreground">Memory Infrastructure</span>
|
||||
{isCheckingInfrastructure && (
|
||||
<Loader2 className="h-4 w-4 animate-spin text-muted-foreground" />
|
||||
)}
|
||||
</div>
|
||||
|
||||
{/* Docker Status */}
|
||||
{/* Kuzu Installation Status */}
|
||||
<div className="flex items-center justify-between">
|
||||
<div className="flex items-center gap-2">
|
||||
{infrastructureStatus?.docker.running ? (
|
||||
{infrastructureStatus?.memory.kuzuInstalled ? (
|
||||
<CheckCircle2 className="h-4 w-4 text-success" />
|
||||
) : infrastructureStatus?.docker.installed ? (
|
||||
<AlertCircle className="h-4 w-4 text-warning" />
|
||||
) : (
|
||||
<AlertCircle className="h-4 w-4 text-destructive" />
|
||||
<AlertCircle className="h-4 w-4 text-warning" />
|
||||
)}
|
||||
<span className="text-xs text-foreground">Docker</span>
|
||||
<span className="text-xs text-foreground">Kuzu Database</span>
|
||||
</div>
|
||||
<div className="flex items-center gap-2">
|
||||
{infrastructureStatus?.docker.running ? (
|
||||
<span className="text-xs text-success">Running</span>
|
||||
) : infrastructureStatus?.docker.installed ? (
|
||||
<>
|
||||
<span className="text-xs text-warning">Not Running</span>
|
||||
<Button
|
||||
size="sm"
|
||||
variant="outline"
|
||||
onClick={onOpenDockerDesktop}
|
||||
disabled={isOpeningDocker}
|
||||
className="h-6 text-xs"
|
||||
>
|
||||
{isOpeningDocker ? (
|
||||
<Loader2 className="h-3 w-3 animate-spin mr-1" />
|
||||
) : null}
|
||||
Start Docker
|
||||
</Button>
|
||||
</>
|
||||
{infrastructureStatus?.memory.kuzuInstalled ? (
|
||||
<span className="text-xs text-success">Installed</span>
|
||||
) : (
|
||||
<>
|
||||
<span className="text-xs text-destructive">Not Installed</span>
|
||||
<Button
|
||||
size="sm"
|
||||
variant="outline"
|
||||
onClick={onDownloadDocker}
|
||||
className="h-6 text-xs"
|
||||
>
|
||||
<Download className="h-3 w-3 mr-1" />
|
||||
Install
|
||||
</Button>
|
||||
</>
|
||||
<span className="text-xs text-warning">Not Available</span>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{/* FalkorDB Status */}
|
||||
{/* Database Status */}
|
||||
<div className="flex items-center justify-between">
|
||||
<div className="flex items-center gap-2">
|
||||
{infrastructureStatus?.falkordb.healthy ? (
|
||||
<CheckCircle2 className="h-4 w-4 text-success" />
|
||||
) : infrastructureStatus?.falkordb.containerRunning ? (
|
||||
<Loader2 className="h-4 w-4 animate-spin text-warning" />
|
||||
{infrastructureStatus?.memory.databaseExists ? (
|
||||
<Database className="h-4 w-4 text-success" />
|
||||
) : (
|
||||
<AlertCircle className="h-4 w-4 text-muted-foreground" />
|
||||
<Database className="h-4 w-4 text-muted-foreground" />
|
||||
)}
|
||||
<span className="text-xs text-foreground">FalkorDB</span>
|
||||
<span className="text-xs text-foreground">Database</span>
|
||||
</div>
|
||||
<div className="flex items-center gap-2">
|
||||
{infrastructureStatus?.falkordb.healthy ? (
|
||||
{infrastructureStatus?.memory.databaseExists ? (
|
||||
<span className="text-xs text-success">Ready</span>
|
||||
) : infrastructureStatus?.falkordb.containerRunning ? (
|
||||
<span className="text-xs text-warning">Starting...</span>
|
||||
) : infrastructureStatus?.docker.running ? (
|
||||
<>
|
||||
<span className="text-xs text-muted-foreground">Not Running</span>
|
||||
<Button
|
||||
size="sm"
|
||||
variant="outline"
|
||||
onClick={onStartFalkorDB}
|
||||
disabled={isStartingFalkorDB}
|
||||
className="h-6 text-xs"
|
||||
>
|
||||
{isStartingFalkorDB ? (
|
||||
<Loader2 className="h-3 w-3 animate-spin mr-1" />
|
||||
) : infrastructureStatus?.memory.kuzuInstalled ? (
|
||||
<span className="text-xs text-muted-foreground">Will be created on first use</span>
|
||||
) : (
|
||||
<Zap className="h-3 w-3 mr-1" />
|
||||
)}
|
||||
Start
|
||||
</Button>
|
||||
</>
|
||||
) : (
|
||||
<span className="text-xs text-muted-foreground">Requires Docker</span>
|
||||
<span className="text-xs text-muted-foreground">Requires Kuzu</span>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{/* Available Databases */}
|
||||
{infrastructureStatus?.memory.databases && infrastructureStatus.memory.databases.length > 0 && (
|
||||
<div className="text-xs text-muted-foreground">
|
||||
Available databases: {infrastructureStatus.memory.databases.join(', ')}
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Overall Status Message */}
|
||||
{infrastructureStatus?.ready ? (
|
||||
<div className="text-xs text-success flex items-center gap-1">
|
||||
<CheckCircle2 className="h-3 w-3" />
|
||||
Graph memory is ready to use
|
||||
</div>
|
||||
) : infrastructureStatus && !infrastructureStatus.docker.installed && (
|
||||
) : infrastructureStatus && !infrastructureStatus.memory.kuzuInstalled && (
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Docker Desktop is required for graph-based memory.
|
||||
Graph memory requires Python 3.12+ with LadybugDB. No Docker needed.
|
||||
</p>
|
||||
)}
|
||||
|
||||
{/* Error Display */}
|
||||
{infrastructureStatus?.memory.error && (
|
||||
<p className="text-xs text-destructive">
|
||||
{infrastructureStatus.memory.error}
|
||||
</p>
|
||||
)}
|
||||
</div>
|
||||
|
||||
@@ -1,4 +1,5 @@
|
||||
import { Database, Globe } from 'lucide-react';
|
||||
import { useState, useEffect, useCallback } from 'react';
|
||||
import { Database, Globe, RefreshCw, CheckCircle2, AlertCircle, Loader2 } from 'lucide-react';
|
||||
import { CollapsibleSection } from './CollapsibleSection';
|
||||
import { InfrastructureStatus } from './InfrastructureStatus';
|
||||
import { PasswordInput } from './PasswordInput';
|
||||
@@ -7,8 +8,16 @@ import { Input } from '../ui/input';
|
||||
import { Switch } from '../ui/switch';
|
||||
import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from '../ui/select';
|
||||
import { Separator } from '../ui/separator';
|
||||
import { Button } from '../ui/button';
|
||||
import type { ProjectEnvConfig, ProjectSettings, InfrastructureStatus as InfrastructureStatusType } from '../../../shared/types';
|
||||
|
||||
interface OllamaEmbeddingModel {
|
||||
name: string;
|
||||
embedding_dim: number | null;
|
||||
description: string;
|
||||
size_gb: number;
|
||||
}
|
||||
|
||||
interface MemoryBackendSectionProps {
|
||||
isExpanded: boolean;
|
||||
onToggle: () => void;
|
||||
@@ -18,13 +27,12 @@ interface MemoryBackendSectionProps {
|
||||
onUpdateSettings: (updates: Partial<ProjectSettings>) => void;
|
||||
infrastructureStatus: InfrastructureStatusType | null;
|
||||
isCheckingInfrastructure: boolean;
|
||||
isStartingFalkorDB: boolean;
|
||||
isOpeningDocker: boolean;
|
||||
onStartFalkorDB: () => void;
|
||||
onOpenDockerDesktop: () => void;
|
||||
onDownloadDocker: () => void;
|
||||
}
|
||||
|
||||
/**
|
||||
* Memory Backend Section Component
|
||||
* Configures Graphiti memory using LadybugDB (embedded database - no Docker required)
|
||||
*/
|
||||
export function MemoryBackendSection({
|
||||
isExpanded,
|
||||
onToggle,
|
||||
@@ -34,25 +42,67 @@ export function MemoryBackendSection({
|
||||
onUpdateSettings,
|
||||
infrastructureStatus,
|
||||
isCheckingInfrastructure,
|
||||
isStartingFalkorDB,
|
||||
isOpeningDocker,
|
||||
onStartFalkorDB,
|
||||
onOpenDockerDesktop,
|
||||
onDownloadDocker,
|
||||
}: MemoryBackendSectionProps) {
|
||||
// Ollama model detection state
|
||||
const [ollamaModels, setOllamaModels] = useState<OllamaEmbeddingModel[]>([]);
|
||||
const [ollamaStatus, setOllamaStatus] = useState<'idle' | 'checking' | 'connected' | 'disconnected'>('idle');
|
||||
const [ollamaError, setOllamaError] = useState<string | null>(null);
|
||||
|
||||
const embeddingProvider = envConfig.graphitiProviderConfig?.embeddingProvider || 'openai';
|
||||
const ollamaBaseUrl = envConfig.graphitiProviderConfig?.ollamaBaseUrl || 'http://localhost:11434';
|
||||
|
||||
// Detect Ollama embedding models
|
||||
const detectOllamaModels = useCallback(async () => {
|
||||
if (!envConfig.graphitiEnabled || embeddingProvider !== 'ollama') return;
|
||||
|
||||
setOllamaStatus('checking');
|
||||
setOllamaError(null);
|
||||
|
||||
try {
|
||||
// Check Ollama status first
|
||||
const statusResult = await window.electronAPI.checkOllamaStatus(ollamaBaseUrl);
|
||||
if (!statusResult.success || !statusResult.data?.running) {
|
||||
setOllamaStatus('disconnected');
|
||||
setOllamaError(statusResult.data?.message || 'Ollama is not running');
|
||||
return;
|
||||
}
|
||||
|
||||
// Get embedding models
|
||||
const modelsResult = await window.electronAPI.listOllamaEmbeddingModels(ollamaBaseUrl);
|
||||
if (!modelsResult.success) {
|
||||
setOllamaStatus('connected');
|
||||
setOllamaError(modelsResult.error || 'Failed to list models');
|
||||
return;
|
||||
}
|
||||
|
||||
setOllamaModels(modelsResult.data?.embedding_models || []);
|
||||
setOllamaStatus('connected');
|
||||
} catch (err) {
|
||||
setOllamaStatus('disconnected');
|
||||
setOllamaError(err instanceof Error ? err.message : 'Failed to detect Ollama models');
|
||||
}
|
||||
}, [envConfig.graphitiEnabled, embeddingProvider, ollamaBaseUrl]);
|
||||
|
||||
// Auto-detect when Ollama is selected
|
||||
useEffect(() => {
|
||||
if (embeddingProvider === 'ollama' && envConfig.graphitiEnabled) {
|
||||
detectOllamaModels();
|
||||
}
|
||||
}, [embeddingProvider, envConfig.graphitiEnabled, detectOllamaModels]);
|
||||
|
||||
const badge = (
|
||||
<span className={`px-2 py-0.5 text-xs rounded-full ${
|
||||
envConfig.graphitiEnabled
|
||||
? 'bg-success/10 text-success'
|
||||
: 'bg-muted text-muted-foreground'
|
||||
}`}>
|
||||
{envConfig.graphitiEnabled ? 'Graphiti' : 'File-based'}
|
||||
{envConfig.graphitiEnabled ? 'Enabled' : 'Disabled'}
|
||||
</span>
|
||||
);
|
||||
|
||||
return (
|
||||
<CollapsibleSection
|
||||
title="Memory Backend"
|
||||
title="Memory"
|
||||
icon={<Database className="h-4 w-4" />}
|
||||
isExpanded={isExpanded}
|
||||
onToggle={onToggle}
|
||||
@@ -60,9 +110,9 @@ export function MemoryBackendSection({
|
||||
>
|
||||
<div className="flex items-center justify-between">
|
||||
<div className="space-y-0.5">
|
||||
<Label className="font-normal text-foreground">Use Graphiti (Recommended)</Label>
|
||||
<Label className="font-normal text-foreground">Enable Memory</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Persistent cross-session memory using FalkorDB graph database
|
||||
Persistent cross-session memory using embedded graph database
|
||||
</p>
|
||||
</div>
|
||||
<Switch
|
||||
@@ -79,22 +129,17 @@ export function MemoryBackendSection({
|
||||
<div className="rounded-lg border border-border bg-muted/30 p-3">
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Using file-based memory. Session insights are stored locally in JSON files.
|
||||
Enable Graphiti for persistent cross-session memory with semantic search.
|
||||
Enable Memory for persistent cross-session context with semantic search.
|
||||
</p>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{envConfig.graphitiEnabled && (
|
||||
<>
|
||||
{/* Infrastructure Status - Dynamic Docker/FalkorDB check */}
|
||||
{/* Infrastructure Status - LadybugDB check */}
|
||||
<InfrastructureStatus
|
||||
infrastructureStatus={infrastructureStatus}
|
||||
isCheckingInfrastructure={isCheckingInfrastructure}
|
||||
isStartingFalkorDB={isStartingFalkorDB}
|
||||
isOpeningDocker={isOpeningDocker}
|
||||
onStartFalkorDB={onStartFalkorDB}
|
||||
onOpenDockerDesktop={onOpenDockerDesktop}
|
||||
onDownloadDocker={onDownloadDocker}
|
||||
/>
|
||||
|
||||
{/* Graphiti MCP Server Toggle */}
|
||||
@@ -117,66 +162,30 @@ export function MemoryBackendSection({
|
||||
<div className="space-y-2 ml-6">
|
||||
<Label className="text-sm font-medium text-foreground">Graphiti MCP Server URL</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
URL of the Graphiti MCP server (requires Docker container)
|
||||
URL of the Graphiti MCP server
|
||||
</p>
|
||||
<Input
|
||||
placeholder="http://localhost:8000/mcp/"
|
||||
value={settings.graphitiMcpUrl || ''}
|
||||
onChange={(e) => onUpdateSettings({ graphitiMcpUrl: e.target.value || undefined })}
|
||||
/>
|
||||
<div className="rounded-lg border border-info/30 bg-info/5 p-3">
|
||||
<p className="text-xs text-info">
|
||||
Start the MCP server with:{' '}
|
||||
<code className="px-1 bg-info/10 rounded">docker run -d -p 8000:8000 falkordb/graphiti-knowledge-graph-mcp</code>
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
<Separator />
|
||||
|
||||
{/* LLM Provider Selection - V2 Multi-provider support */}
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">LLM Provider</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Provider for graph operations (extraction, search, reasoning)
|
||||
</p>
|
||||
<Select
|
||||
value={envConfig.graphitiProviderConfig?.llmProvider || 'openai'}
|
||||
onValueChange={(value) => onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
llmProvider: value as 'openai' | 'anthropic' | 'azure_openai' | 'ollama' | 'google' | 'groq',
|
||||
embeddingProvider: envConfig.graphitiProviderConfig?.embeddingProvider || 'openai',
|
||||
}
|
||||
})}
|
||||
>
|
||||
<SelectTrigger>
|
||||
<SelectValue placeholder="Select LLM provider" />
|
||||
</SelectTrigger>
|
||||
<SelectContent>
|
||||
<SelectItem value="openai">OpenAI (GPT-4o-mini)</SelectItem>
|
||||
<SelectItem value="anthropic">Anthropic (Claude)</SelectItem>
|
||||
<SelectItem value="google">Google AI (Gemini)</SelectItem>
|
||||
<SelectItem value="azure_openai">Azure OpenAI</SelectItem>
|
||||
<SelectItem value="ollama">Ollama (Local)</SelectItem>
|
||||
</SelectContent>
|
||||
</Select>
|
||||
</div>
|
||||
|
||||
{/* Embedding Provider Selection */}
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">Embedding Provider</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Provider for semantic search embeddings
|
||||
Provider for semantic search (optional - keyword search works without)
|
||||
</p>
|
||||
<Select
|
||||
value={envConfig.graphitiProviderConfig?.embeddingProvider || 'openai'}
|
||||
value={embeddingProvider}
|
||||
onValueChange={(value) => onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
llmProvider: envConfig.graphitiProviderConfig?.llmProvider || 'openai',
|
||||
embeddingProvider: value as 'openai' | 'voyage' | 'azure_openai' | 'ollama' | 'google' | 'huggingface',
|
||||
embeddingProvider: value as 'openai' | 'voyage' | 'azure_openai' | 'ollama' | 'google',
|
||||
}
|
||||
})}
|
||||
>
|
||||
@@ -184,17 +193,20 @@ export function MemoryBackendSection({
|
||||
<SelectValue placeholder="Select embedding provider" />
|
||||
</SelectTrigger>
|
||||
<SelectContent>
|
||||
<SelectItem value="ollama">Ollama (Local - Free)</SelectItem>
|
||||
<SelectItem value="openai">OpenAI</SelectItem>
|
||||
<SelectItem value="voyage">Voyage AI</SelectItem>
|
||||
<SelectItem value="google">Google AI</SelectItem>
|
||||
<SelectItem value="azure_openai">Azure OpenAI</SelectItem>
|
||||
<SelectItem value="ollama">Ollama (Local)</SelectItem>
|
||||
</SelectContent>
|
||||
</Select>
|
||||
</div>
|
||||
|
||||
<Separator />
|
||||
|
||||
{/* Provider-specific credential fields */}
|
||||
{/* OpenAI */}
|
||||
{embeddingProvider === 'openai' && (
|
||||
<div className="space-y-2">
|
||||
<div className="flex items-center justify-between">
|
||||
<Label className="text-sm font-medium text-foreground">
|
||||
@@ -213,7 +225,7 @@ export function MemoryBackendSection({
|
||||
</p>
|
||||
) : (
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Required when using OpenAI as LLM or embedding provider
|
||||
Required for OpenAI embeddings
|
||||
</p>
|
||||
)}
|
||||
<PasswordInput
|
||||
@@ -222,44 +234,270 @@ export function MemoryBackendSection({
|
||||
placeholder={envConfig.openaiKeyIsGlobal ? 'Enter to override global key...' : 'sk-xxxxxxxx'}
|
||||
/>
|
||||
</div>
|
||||
)}
|
||||
|
||||
<div className="grid grid-cols-2 gap-4">
|
||||
{/* Voyage AI */}
|
||||
{embeddingProvider === 'voyage' && (
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">FalkorDB Host</Label>
|
||||
<Label className="text-sm font-medium text-foreground">Voyage AI API Key</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Required for Voyage AI embeddings
|
||||
</p>
|
||||
<PasswordInput
|
||||
value={envConfig.graphitiProviderConfig?.voyageApiKey || ''}
|
||||
onChange={(value) => onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'voyage',
|
||||
voyageApiKey: value || undefined,
|
||||
}
|
||||
})}
|
||||
placeholder="pa-xxxxxxxx"
|
||||
/>
|
||||
<div className="space-y-1">
|
||||
<Label className="text-xs text-muted-foreground">Embedding Model</Label>
|
||||
<Input
|
||||
placeholder="localhost"
|
||||
value={envConfig.graphitiFalkorDbHost || ''}
|
||||
onChange={(e) => onUpdateConfig({ graphitiFalkorDbHost: e.target.value })}
|
||||
placeholder="voyage-3"
|
||||
value={envConfig.graphitiProviderConfig?.voyageEmbeddingModel || ''}
|
||||
onChange={(e) => onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'voyage',
|
||||
voyageEmbeddingModel: e.target.value || undefined,
|
||||
}
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Google AI */}
|
||||
{embeddingProvider === 'google' && (
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">FalkorDB Port</Label>
|
||||
<Label className="text-sm font-medium text-foreground">Google AI API Key</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Required for Google AI embeddings
|
||||
</p>
|
||||
<PasswordInput
|
||||
value={envConfig.graphitiProviderConfig?.googleApiKey || ''}
|
||||
onChange={(value) => onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'google',
|
||||
googleApiKey: value || undefined,
|
||||
}
|
||||
})}
|
||||
placeholder="AIzaSy..."
|
||||
/>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Azure OpenAI */}
|
||||
{embeddingProvider === 'azure_openai' && (
|
||||
<div className="space-y-3">
|
||||
<Label className="text-sm font-medium text-foreground">Azure OpenAI Configuration</Label>
|
||||
<div className="space-y-2">
|
||||
<Label className="text-xs text-muted-foreground">API Key</Label>
|
||||
<PasswordInput
|
||||
value={envConfig.graphitiProviderConfig?.azureOpenaiApiKey || ''}
|
||||
onChange={(value) => onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'azure_openai',
|
||||
azureOpenaiApiKey: value || undefined,
|
||||
}
|
||||
})}
|
||||
placeholder="Azure API Key"
|
||||
/>
|
||||
</div>
|
||||
<div className="space-y-1">
|
||||
<Label className="text-xs text-muted-foreground">Base URL</Label>
|
||||
<Input
|
||||
placeholder="https://your-resource.openai.azure.com"
|
||||
value={envConfig.graphitiProviderConfig?.azureOpenaiBaseUrl || ''}
|
||||
onChange={(e) => onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'azure_openai',
|
||||
azureOpenaiBaseUrl: e.target.value || undefined,
|
||||
}
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
<div className="space-y-1">
|
||||
<Label className="text-xs text-muted-foreground">Embedding Deployment Name</Label>
|
||||
<Input
|
||||
placeholder="text-embedding-ada-002"
|
||||
value={envConfig.graphitiProviderConfig?.azureOpenaiEmbeddingDeployment || ''}
|
||||
onChange={(e) => onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'azure_openai',
|
||||
azureOpenaiEmbeddingDeployment: e.target.value || undefined,
|
||||
}
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Ollama (Local) */}
|
||||
{embeddingProvider === 'ollama' && (
|
||||
<div className="space-y-3">
|
||||
<div className="flex items-center justify-between">
|
||||
<Label className="text-sm font-medium text-foreground">Ollama Configuration</Label>
|
||||
<div className="flex items-center gap-2">
|
||||
{ollamaStatus === 'checking' && (
|
||||
<span className="flex items-center gap-1 text-xs text-muted-foreground">
|
||||
<Loader2 className="h-3 w-3 animate-spin" />
|
||||
Checking...
|
||||
</span>
|
||||
)}
|
||||
{ollamaStatus === 'connected' && (
|
||||
<span className="flex items-center gap-1 text-xs text-success">
|
||||
<CheckCircle2 className="h-3 w-3" />
|
||||
Connected
|
||||
</span>
|
||||
)}
|
||||
{ollamaStatus === 'disconnected' && (
|
||||
<span className="flex items-center gap-1 text-xs text-destructive">
|
||||
<AlertCircle className="h-3 w-3" />
|
||||
Not running
|
||||
</span>
|
||||
)}
|
||||
<Button
|
||||
variant="ghost"
|
||||
size="sm"
|
||||
onClick={detectOllamaModels}
|
||||
disabled={ollamaStatus === 'checking'}
|
||||
className="h-6 px-2"
|
||||
>
|
||||
<RefreshCw className={`h-3 w-3 ${ollamaStatus === 'checking' ? 'animate-spin' : ''}`} />
|
||||
</Button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="space-y-2">
|
||||
<Label className="text-xs text-muted-foreground">Base URL</Label>
|
||||
<Input
|
||||
placeholder="http://localhost:11434"
|
||||
value={envConfig.graphitiProviderConfig?.ollamaBaseUrl || ''}
|
||||
onChange={(e) => onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'ollama',
|
||||
ollamaBaseUrl: e.target.value || undefined,
|
||||
}
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
|
||||
{ollamaError && (
|
||||
<div className="rounded-md bg-destructive/10 p-2 text-xs text-destructive">
|
||||
{ollamaError}
|
||||
</div>
|
||||
)}
|
||||
|
||||
<div className="space-y-2">
|
||||
<Label className="text-xs text-muted-foreground">Embedding Model</Label>
|
||||
{ollamaModels.length > 0 ? (
|
||||
<Select
|
||||
value={envConfig.graphitiProviderConfig?.ollamaEmbeddingModel || ''}
|
||||
onValueChange={(value) => {
|
||||
const model = ollamaModels.find(m => m.name === value);
|
||||
onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'ollama',
|
||||
ollamaEmbeddingModel: value,
|
||||
ollamaEmbeddingDim: model?.embedding_dim || undefined,
|
||||
}
|
||||
});
|
||||
}}
|
||||
>
|
||||
<SelectTrigger>
|
||||
<SelectValue placeholder="Select embedding model" />
|
||||
</SelectTrigger>
|
||||
<SelectContent>
|
||||
{ollamaModels.map((model) => (
|
||||
<SelectItem key={model.name} value={model.name}>
|
||||
<div className="flex items-center gap-2">
|
||||
<span>{model.name}</span>
|
||||
{model.embedding_dim && (
|
||||
<span className="text-xs text-muted-foreground">
|
||||
({model.embedding_dim}d)
|
||||
</span>
|
||||
)}
|
||||
</div>
|
||||
</SelectItem>
|
||||
))}
|
||||
</SelectContent>
|
||||
</Select>
|
||||
) : (
|
||||
<Input
|
||||
placeholder="nomic-embed-text"
|
||||
value={envConfig.graphitiProviderConfig?.ollamaEmbeddingModel || ''}
|
||||
onChange={(e) => onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'ollama',
|
||||
ollamaEmbeddingModel: e.target.value || undefined,
|
||||
}
|
||||
})}
|
||||
/>
|
||||
)}
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Common models: nomic-embed-text, bge-base-en, mxbai-embed-large
|
||||
</p>
|
||||
</div>
|
||||
|
||||
<div className="space-y-1">
|
||||
<Label className="text-xs text-muted-foreground">Embedding Dimension</Label>
|
||||
<Input
|
||||
type="number"
|
||||
placeholder="6380"
|
||||
value={envConfig.graphitiFalkorDbPort || ''}
|
||||
onChange={(e) => onUpdateConfig({ graphitiFalkorDbPort: parseInt(e.target.value) || undefined })}
|
||||
placeholder="768"
|
||||
value={envConfig.graphitiProviderConfig?.ollamaEmbeddingDim || ''}
|
||||
onChange={(e) => onUpdateConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'ollama',
|
||||
ollamaEmbeddingDim: parseInt(e.target.value) || undefined,
|
||||
}
|
||||
})}
|
||||
/>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Required for Ollama embeddings (e.g., 768 for nomic-embed-text)
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">FalkorDB Password (Optional)</Label>
|
||||
<PasswordInput
|
||||
value={envConfig.graphitiFalkorDbPassword || ''}
|
||||
onChange={(value) => onUpdateConfig({ graphitiFalkorDbPassword: value })}
|
||||
placeholder="Leave empty if none"
|
||||
/>
|
||||
</div>
|
||||
{/* Database Settings */}
|
||||
<Separator />
|
||||
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">Database Name</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Name for the memory database (stored in ~/.auto-claude/memories/)
|
||||
</p>
|
||||
<Input
|
||||
placeholder="auto_claude_memory"
|
||||
value={envConfig.graphitiDatabase || ''}
|
||||
onChange={(e) => onUpdateConfig({ graphitiDatabase: e.target.value })}
|
||||
/>
|
||||
</div>
|
||||
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">Database Path (Optional)</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Custom storage location. Default: ~/.auto-claude/memories/
|
||||
</p>
|
||||
<Input
|
||||
placeholder="~/.auto-claude/memories"
|
||||
value={envConfig.graphitiDbPath || ''}
|
||||
onChange={(e) => onUpdateConfig({ graphitiDbPath: e.target.value || undefined })}
|
||||
/>
|
||||
</div>
|
||||
</>
|
||||
)}
|
||||
</CollapsibleSection>
|
||||
|
||||
@@ -45,8 +45,6 @@ export function ProjectSettings({ project, open, onOpenChange }: ProjectSettings
|
||||
setShowLinearKey,
|
||||
showOpenAIKey,
|
||||
setShowOpenAIKey,
|
||||
showFalkorPassword,
|
||||
setShowFalkorPassword,
|
||||
showGitHubToken,
|
||||
setShowGitHubToken,
|
||||
expandedSections,
|
||||
@@ -146,8 +144,6 @@ export function ProjectSettings({ project, open, onOpenChange }: ProjectSettings
|
||||
updateEnvConfig={updateEnvConfig}
|
||||
showOpenAIKey={showOpenAIKey}
|
||||
setShowOpenAIKey={setShowOpenAIKey}
|
||||
showFalkorPassword={showFalkorPassword}
|
||||
setShowFalkorPassword={setShowFalkorPassword}
|
||||
expanded={expandedSections.graphiti}
|
||||
onToggle={() => toggleSection('graphiti')}
|
||||
/>
|
||||
|
||||
@@ -1,3 +1,4 @@
|
||||
import { useState, useEffect } from 'react';
|
||||
import {
|
||||
Database,
|
||||
Eye,
|
||||
@@ -17,7 +18,8 @@ import {
|
||||
SelectValue
|
||||
} from '../ui/select';
|
||||
import { Separator } from '../ui/separator';
|
||||
import type { ProjectEnvConfig, ProjectSettings as ProjectSettingsType } from '../../../shared/types';
|
||||
import { OllamaModelSelector } from '../onboarding/OllamaModelSelector';
|
||||
import type { ProjectEnvConfig, ProjectSettings as ProjectSettingsType, GraphitiEmbeddingProvider } from '../../../shared/types';
|
||||
|
||||
interface SecuritySettingsProps {
|
||||
envConfig: ProjectEnvConfig | null;
|
||||
@@ -28,8 +30,6 @@ interface SecuritySettingsProps {
|
||||
// Password visibility
|
||||
showOpenAIKey: boolean;
|
||||
setShowOpenAIKey: React.Dispatch<React.SetStateAction<boolean>>;
|
||||
showFalkorPassword: boolean;
|
||||
setShowFalkorPassword: React.Dispatch<React.SetStateAction<boolean>>;
|
||||
|
||||
// Collapsible section
|
||||
expanded: boolean;
|
||||
@@ -43,13 +43,276 @@ export function SecuritySettings({
|
||||
updateEnvConfig,
|
||||
showOpenAIKey,
|
||||
setShowOpenAIKey,
|
||||
showFalkorPassword,
|
||||
setShowFalkorPassword,
|
||||
expanded,
|
||||
onToggle
|
||||
}: SecuritySettingsProps) {
|
||||
// Password visibility for multiple providers
|
||||
const [showApiKey, setShowApiKey] = useState<Record<string, boolean>>({
|
||||
openai: showOpenAIKey,
|
||||
voyage: false,
|
||||
google: false,
|
||||
azure: false
|
||||
});
|
||||
|
||||
// Sync parent's showOpenAIKey prop to local state
|
||||
useEffect(() => {
|
||||
setShowApiKey(prev => ({ ...prev, openai: showOpenAIKey }));
|
||||
}, [showOpenAIKey]);
|
||||
|
||||
const embeddingProvider = envConfig?.graphitiProviderConfig?.embeddingProvider || 'ollama';
|
||||
|
||||
// Toggle API key visibility
|
||||
const toggleShowApiKey = (key: string) => {
|
||||
const newValue = !showApiKey[key];
|
||||
setShowApiKey(prev => ({ ...prev, [key]: newValue }));
|
||||
// Sync with parent for OpenAI
|
||||
if (key === 'openai') {
|
||||
setShowOpenAIKey(newValue);
|
||||
}
|
||||
};
|
||||
|
||||
// Handle Ollama model selection
|
||||
const handleOllamaModelSelect = (modelName: string, dim: number) => {
|
||||
updateEnvConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig?.graphitiProviderConfig,
|
||||
embeddingProvider: 'ollama',
|
||||
ollamaEmbeddingModel: modelName,
|
||||
ollamaEmbeddingDim: dim,
|
||||
}
|
||||
});
|
||||
};
|
||||
|
||||
if (!envConfig) return null;
|
||||
|
||||
// Render provider-specific configuration fields
|
||||
const renderProviderFields = () => {
|
||||
// OpenAI
|
||||
if (embeddingProvider === 'openai') {
|
||||
return (
|
||||
<div className="space-y-2">
|
||||
<div className="flex items-center justify-between">
|
||||
<Label className="text-sm font-medium text-foreground">
|
||||
OpenAI API Key {envConfig.openaiKeyIsGlobal ? '(Override)' : ''}
|
||||
</Label>
|
||||
{envConfig.openaiKeyIsGlobal && (
|
||||
<span className="flex items-center gap-1 text-xs text-info">
|
||||
<Globe className="h-3 w-3" />
|
||||
Using global key
|
||||
</span>
|
||||
)}
|
||||
</div>
|
||||
{envConfig.openaiKeyIsGlobal ? (
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Using key from App Settings. Enter a project-specific key below to override.
|
||||
</p>
|
||||
) : (
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Required for OpenAI embeddings
|
||||
</p>
|
||||
)}
|
||||
<div className="relative">
|
||||
<Input
|
||||
type={showApiKey['openai'] ? 'text' : 'password'}
|
||||
placeholder={envConfig.openaiKeyIsGlobal ? 'Enter to override global key...' : 'sk-xxxxxxxx'}
|
||||
value={envConfig.openaiKeyIsGlobal ? '' : (envConfig.openaiApiKey || '')}
|
||||
onChange={(e) => updateEnvConfig({ openaiApiKey: e.target.value || undefined })}
|
||||
className="pr-10"
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => toggleShowApiKey('openai')}
|
||||
className="absolute right-3 top-1/2 -translate-y-1/2 text-muted-foreground hover:text-foreground"
|
||||
aria-label={showApiKey['openai'] ? 'Hide OpenAI API key' : 'Show OpenAI API key'}
|
||||
>
|
||||
{showApiKey['openai'] ? <EyeOff className="h-4 w-4" /> : <Eye className="h-4 w-4" />}
|
||||
</button>
|
||||
</div>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Get your key from{' '}
|
||||
<a href="https://platform.openai.com/api-keys" target="_blank" rel="noopener noreferrer" className="text-primary hover:text-primary/80">
|
||||
OpenAI
|
||||
</a>
|
||||
</p>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
// Voyage AI
|
||||
if (embeddingProvider === 'voyage') {
|
||||
return (
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">Voyage AI API Key</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Required for Voyage AI embeddings
|
||||
</p>
|
||||
<div className="relative">
|
||||
<Input
|
||||
type={showApiKey['voyage'] ? 'text' : 'password'}
|
||||
value={envConfig.graphitiProviderConfig?.voyageApiKey || ''}
|
||||
onChange={(e) => updateEnvConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'voyage',
|
||||
voyageApiKey: e.target.value || undefined,
|
||||
}
|
||||
})}
|
||||
placeholder="pa-xxxxxxxx"
|
||||
className="pr-10"
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => toggleShowApiKey('voyage')}
|
||||
className="absolute right-3 top-1/2 -translate-y-1/2 text-muted-foreground hover:text-foreground"
|
||||
aria-label={showApiKey['voyage'] ? 'Hide Voyage AI API key' : 'Show Voyage AI API key'}
|
||||
>
|
||||
{showApiKey['voyage'] ? <EyeOff className="h-4 w-4" /> : <Eye className="h-4 w-4" />}
|
||||
</button>
|
||||
</div>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Get your key from{' '}
|
||||
<a href="https://dash.voyageai.com/api-keys" target="_blank" rel="noopener noreferrer" className="text-primary hover:text-primary/80">
|
||||
Voyage AI
|
||||
</a>
|
||||
</p>
|
||||
<div className="space-y-1 mt-3">
|
||||
<Label className="text-xs text-muted-foreground">Embedding Model (optional)</Label>
|
||||
<Input
|
||||
placeholder="voyage-3"
|
||||
value={envConfig.graphitiProviderConfig?.voyageEmbeddingModel || ''}
|
||||
onChange={(e) => updateEnvConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'voyage',
|
||||
voyageEmbeddingModel: e.target.value || undefined,
|
||||
}
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
// Google AI
|
||||
if (embeddingProvider === 'google') {
|
||||
return (
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">Google AI API Key</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Required for Google AI embeddings
|
||||
</p>
|
||||
<div className="relative">
|
||||
<Input
|
||||
type={showApiKey['google'] ? 'text' : 'password'}
|
||||
value={envConfig.graphitiProviderConfig?.googleApiKey || ''}
|
||||
onChange={(e) => updateEnvConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'google',
|
||||
googleApiKey: e.target.value || undefined,
|
||||
}
|
||||
})}
|
||||
placeholder="AIzaSy..."
|
||||
className="pr-10"
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => toggleShowApiKey('google')}
|
||||
className="absolute right-3 top-1/2 -translate-y-1/2 text-muted-foreground hover:text-foreground"
|
||||
aria-label={showApiKey['google'] ? 'Hide Google API key' : 'Show Google API key'}
|
||||
>
|
||||
{showApiKey['google'] ? <EyeOff className="h-4 w-4" /> : <Eye className="h-4 w-4" />}
|
||||
</button>
|
||||
</div>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Get your key from{' '}
|
||||
<a href="https://aistudio.google.com/apikey" target="_blank" rel="noopener noreferrer" className="text-primary hover:text-primary/80">
|
||||
Google AI Studio
|
||||
</a>
|
||||
</p>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
// Azure OpenAI
|
||||
if (embeddingProvider === 'azure_openai') {
|
||||
return (
|
||||
<div className="space-y-3 p-3 rounded-md bg-muted/50">
|
||||
<Label className="text-sm font-medium text-foreground">Azure OpenAI Configuration</Label>
|
||||
<div className="space-y-2">
|
||||
<Label className="text-xs text-muted-foreground">API Key</Label>
|
||||
<div className="relative">
|
||||
<Input
|
||||
type={showApiKey['azure'] ? 'text' : 'password'}
|
||||
value={envConfig.graphitiProviderConfig?.azureOpenaiApiKey || ''}
|
||||
onChange={(e) => updateEnvConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'azure_openai',
|
||||
azureOpenaiApiKey: e.target.value || undefined,
|
||||
}
|
||||
})}
|
||||
placeholder="Azure API Key"
|
||||
className="pr-10"
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => toggleShowApiKey('azure')}
|
||||
className="absolute right-3 top-1/2 -translate-y-1/2 text-muted-foreground hover:text-foreground"
|
||||
aria-label={showApiKey['azure'] ? 'Hide Azure OpenAI API key' : 'Show Azure OpenAI API key'}
|
||||
>
|
||||
{showApiKey['azure'] ? <EyeOff className="h-4 w-4" /> : <Eye className="h-4 w-4" />}
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
<div className="space-y-1">
|
||||
<Label className="text-xs text-muted-foreground">Base URL</Label>
|
||||
<Input
|
||||
placeholder="https://your-resource.openai.azure.com"
|
||||
value={envConfig.graphitiProviderConfig?.azureOpenaiBaseUrl || ''}
|
||||
onChange={(e) => updateEnvConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'azure_openai',
|
||||
azureOpenaiBaseUrl: e.target.value || undefined,
|
||||
}
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
<div className="space-y-1">
|
||||
<Label className="text-xs text-muted-foreground">Embedding Deployment Name</Label>
|
||||
<Input
|
||||
placeholder="text-embedding-ada-002"
|
||||
value={envConfig.graphitiProviderConfig?.azureOpenaiEmbeddingDeployment || ''}
|
||||
onChange={(e) => updateEnvConfig({
|
||||
graphitiProviderConfig: {
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: 'azure_openai',
|
||||
azureOpenaiEmbeddingDeployment: e.target.value || undefined,
|
||||
}
|
||||
})}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
// Ollama (Local) - uses OllamaModelSelector component
|
||||
if (embeddingProvider === 'ollama') {
|
||||
return (
|
||||
<div className="space-y-3">
|
||||
<Label className="text-sm font-medium text-foreground">Select Embedding Model</Label>
|
||||
<OllamaModelSelector
|
||||
selectedModel={envConfig.graphitiProviderConfig?.ollamaEmbeddingModel || ''}
|
||||
onModelSelect={handleOllamaModelSelect}
|
||||
/>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
return null;
|
||||
};
|
||||
|
||||
return (
|
||||
<section className="space-y-3">
|
||||
<button
|
||||
@@ -58,13 +321,13 @@ export function SecuritySettings({
|
||||
>
|
||||
<div className="flex items-center gap-2">
|
||||
<Database className="h-4 w-4" />
|
||||
Memory Backend
|
||||
Memory
|
||||
<span className={`px-2 py-0.5 text-xs rounded-full ${
|
||||
envConfig.graphitiEnabled
|
||||
? 'bg-success/10 text-success'
|
||||
: 'bg-muted text-muted-foreground'
|
||||
}`}>
|
||||
{envConfig.graphitiEnabled ? 'Graphiti' : 'File-based'}
|
||||
{envConfig.graphitiEnabled ? 'Enabled' : 'Disabled'}
|
||||
</span>
|
||||
</div>
|
||||
{expanded ? (
|
||||
@@ -78,9 +341,9 @@ export function SecuritySettings({
|
||||
<div className="space-y-4 pl-6 pt-2">
|
||||
<div className="flex items-center justify-between">
|
||||
<div className="space-y-0.5">
|
||||
<Label className="font-normal text-foreground">Use Graphiti (Recommended)</Label>
|
||||
<Label className="font-normal text-foreground">Enable Memory</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Persistent cross-session memory using FalkorDB graph database
|
||||
Persistent cross-session memory using LadybugDB (embedded database)
|
||||
</p>
|
||||
</div>
|
||||
<Switch
|
||||
@@ -96,20 +359,13 @@ export function SecuritySettings({
|
||||
<div className="rounded-lg border border-border bg-muted/30 p-3">
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Using file-based memory. Session insights are stored locally in JSON files.
|
||||
Enable Graphiti for persistent cross-session memory with semantic search.
|
||||
Enable Memory for persistent cross-session context with semantic search.
|
||||
</p>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{envConfig.graphitiEnabled && (
|
||||
<>
|
||||
<div className="rounded-lg border border-warning/30 bg-warning/5 p-3">
|
||||
<p className="text-xs text-warning">
|
||||
Requires FalkorDB running. Start with:{' '}
|
||||
<code className="px-1 bg-warning/10 rounded">docker-compose up -d falkordb</code>
|
||||
</p>
|
||||
</div>
|
||||
|
||||
{/* Graphiti MCP Server Toggle */}
|
||||
<div className="flex items-center justify-between">
|
||||
<div className="space-y-0.5">
|
||||
@@ -130,75 +386,31 @@ export function SecuritySettings({
|
||||
<div className="space-y-2 ml-6">
|
||||
<Label className="text-sm font-medium text-foreground">Graphiti MCP Server URL</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
URL of the Graphiti MCP server (requires Docker container)
|
||||
URL of the Graphiti MCP server for agent memory access
|
||||
</p>
|
||||
<Input
|
||||
placeholder="http://localhost:8000/mcp/"
|
||||
value={settings.graphitiMcpUrl || ''}
|
||||
onChange={(e) => setSettings({ ...settings, graphitiMcpUrl: e.target.value || undefined })}
|
||||
/>
|
||||
<div className="rounded-lg border border-info/30 bg-info/5 p-3">
|
||||
<p className="text-xs text-info">
|
||||
Start the MCP server with:{' '}
|
||||
<code className="px-1 bg-info/10 rounded">docker run -d -p 8000:8000 falkordb/graphiti-knowledge-graph-mcp</code>
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
<Separator />
|
||||
|
||||
{/* LLM Provider Selection */}
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">LLM Provider</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Provider for graph operations (extraction, search, reasoning)
|
||||
</p>
|
||||
<Select
|
||||
value={envConfig.graphitiProviderConfig?.llmProvider || 'openai'}
|
||||
onValueChange={(value) => {
|
||||
const currentConfig = envConfig.graphitiProviderConfig || {
|
||||
llmProvider: 'openai' as const,
|
||||
embeddingProvider: 'openai' as const
|
||||
};
|
||||
updateEnvConfig({
|
||||
graphitiProviderConfig: {
|
||||
...currentConfig,
|
||||
llmProvider: value as 'openai' | 'anthropic' | 'azure_openai' | 'ollama' | 'google' | 'groq',
|
||||
}
|
||||
});
|
||||
}}
|
||||
>
|
||||
<SelectTrigger>
|
||||
<SelectValue placeholder="Select LLM provider" />
|
||||
</SelectTrigger>
|
||||
<SelectContent>
|
||||
<SelectItem value="openai">OpenAI (GPT-4o-mini)</SelectItem>
|
||||
<SelectItem value="anthropic">Anthropic (Claude)</SelectItem>
|
||||
<SelectItem value="google">Google AI (Gemini)</SelectItem>
|
||||
<SelectItem value="azure_openai">Azure OpenAI</SelectItem>
|
||||
<SelectItem value="ollama">Ollama (Local)</SelectItem>
|
||||
</SelectContent>
|
||||
</Select>
|
||||
</div>
|
||||
|
||||
{/* Embedding Provider Selection */}
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">Embedding Provider</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Provider for semantic search embeddings
|
||||
Provider for semantic search (optional - keyword search works without)
|
||||
</p>
|
||||
<Select
|
||||
value={envConfig.graphitiProviderConfig?.embeddingProvider || 'openai'}
|
||||
onValueChange={(value) => {
|
||||
const currentConfig = envConfig.graphitiProviderConfig || {
|
||||
llmProvider: 'openai' as const,
|
||||
embeddingProvider: 'openai' as const
|
||||
};
|
||||
value={embeddingProvider}
|
||||
onValueChange={(value: GraphitiEmbeddingProvider) => {
|
||||
updateEnvConfig({
|
||||
graphitiProviderConfig: {
|
||||
...currentConfig,
|
||||
embeddingProvider: value as 'openai' | 'voyage' | 'azure_openai' | 'ollama' | 'google' | 'huggingface',
|
||||
...envConfig.graphitiProviderConfig,
|
||||
embeddingProvider: value,
|
||||
}
|
||||
});
|
||||
}}
|
||||
@@ -207,104 +419,44 @@ export function SecuritySettings({
|
||||
<SelectValue placeholder="Select embedding provider" />
|
||||
</SelectTrigger>
|
||||
<SelectContent>
|
||||
<SelectItem value="ollama">Ollama (Local - Free)</SelectItem>
|
||||
<SelectItem value="openai">OpenAI</SelectItem>
|
||||
<SelectItem value="voyage">Voyage AI</SelectItem>
|
||||
<SelectItem value="google">Google AI</SelectItem>
|
||||
<SelectItem value="azure_openai">Azure OpenAI</SelectItem>
|
||||
<SelectItem value="ollama">Ollama (Local)</SelectItem>
|
||||
</SelectContent>
|
||||
</Select>
|
||||
</div>
|
||||
|
||||
{/* Provider-specific fields */}
|
||||
{renderProviderFields()}
|
||||
|
||||
<Separator />
|
||||
|
||||
<div className="space-y-2">
|
||||
<div className="flex items-center justify-between">
|
||||
<Label className="text-sm font-medium text-foreground">
|
||||
OpenAI API Key {envConfig.openaiKeyIsGlobal ? '(Override)' : ''}
|
||||
</Label>
|
||||
{envConfig.openaiKeyIsGlobal && (
|
||||
<span className="flex items-center gap-1 text-xs text-info">
|
||||
<Globe className="h-3 w-3" />
|
||||
Using global key
|
||||
</span>
|
||||
)}
|
||||
</div>
|
||||
{envConfig.openaiKeyIsGlobal ? (
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Using key from App Settings. Enter a project-specific key below to override.
|
||||
</p>
|
||||
) : (
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Required when using OpenAI as LLM or embedding provider
|
||||
</p>
|
||||
)}
|
||||
<div className="relative">
|
||||
<Input
|
||||
type={showOpenAIKey ? 'text' : 'password'}
|
||||
placeholder={envConfig.openaiKeyIsGlobal ? 'Enter to override global key...' : 'sk-xxxxxxxx'}
|
||||
value={envConfig.openaiKeyIsGlobal ? '' : (envConfig.openaiApiKey || '')}
|
||||
onChange={(e) => updateEnvConfig({ openaiApiKey: e.target.value || undefined })}
|
||||
className="pr-10"
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => setShowOpenAIKey(!showOpenAIKey)}
|
||||
className="absolute right-3 top-1/2 -translate-y-1/2 text-muted-foreground hover:text-foreground"
|
||||
>
|
||||
{showOpenAIKey ? <EyeOff className="h-4 w-4" /> : <Eye className="h-4 w-4" />}
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="grid grid-cols-2 gap-4">
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">FalkorDB Host</Label>
|
||||
<Input
|
||||
placeholder="localhost"
|
||||
value={envConfig.graphitiFalkorDbHost || ''}
|
||||
onChange={(e) => updateEnvConfig({ graphitiFalkorDbHost: e.target.value })}
|
||||
/>
|
||||
</div>
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">FalkorDB Port</Label>
|
||||
<Input
|
||||
type="number"
|
||||
placeholder="6380"
|
||||
value={envConfig.graphitiFalkorDbPort || ''}
|
||||
onChange={(e) => updateEnvConfig({ graphitiFalkorDbPort: parseInt(e.target.value) || undefined })}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">FalkorDB Password (Optional)</Label>
|
||||
<div className="relative">
|
||||
<Input
|
||||
type={showFalkorPassword ? 'text' : 'password'}
|
||||
placeholder="Leave empty if none"
|
||||
value={envConfig.graphitiFalkorDbPassword || ''}
|
||||
onChange={(e) => updateEnvConfig({ graphitiFalkorDbPassword: e.target.value })}
|
||||
className="pr-10"
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
onClick={() => setShowFalkorPassword(!showFalkorPassword)}
|
||||
className="absolute right-3 top-1/2 -translate-y-1/2 text-muted-foreground hover:text-foreground"
|
||||
>
|
||||
{showFalkorPassword ? <EyeOff className="h-4 w-4" /> : <Eye className="h-4 w-4" />}
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{/* Database Settings */}
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">Database Name</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Stored in ~/.auto-claude/memories/
|
||||
</p>
|
||||
<Input
|
||||
placeholder="auto_claude_memory"
|
||||
value={envConfig.graphitiDatabase || ''}
|
||||
onChange={(e) => updateEnvConfig({ graphitiDatabase: e.target.value })}
|
||||
/>
|
||||
</div>
|
||||
|
||||
<div className="space-y-2">
|
||||
<Label className="text-sm font-medium text-foreground">Database Path (Optional)</Label>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Custom storage location. Default: ~/.auto-claude/memories/
|
||||
</p>
|
||||
<Input
|
||||
placeholder="~/.auto-claude/memories"
|
||||
value={envConfig.graphitiDbPath || ''}
|
||||
onChange={(e) => updateEnvConfig({ graphitiDbPath: e.target.value || undefined })}
|
||||
/>
|
||||
</div>
|
||||
</>
|
||||
)}
|
||||
</div>
|
||||
|
||||
@@ -44,8 +44,6 @@ export interface UseProjectSettingsReturn {
|
||||
setShowLinearKey: React.Dispatch<React.SetStateAction<boolean>>;
|
||||
showOpenAIKey: boolean;
|
||||
setShowOpenAIKey: React.Dispatch<React.SetStateAction<boolean>>;
|
||||
showFalkorPassword: boolean;
|
||||
setShowFalkorPassword: React.Dispatch<React.SetStateAction<boolean>>;
|
||||
showGitHubToken: boolean;
|
||||
setShowGitHubToken: React.Dispatch<React.SetStateAction<boolean>>;
|
||||
|
||||
@@ -97,7 +95,6 @@ export function useProjectSettings(
|
||||
const [showClaudeToken, setShowClaudeToken] = useState(false);
|
||||
const [showLinearKey, setShowLinearKey] = useState(false);
|
||||
const [showOpenAIKey, setShowOpenAIKey] = useState(false);
|
||||
const [showFalkorPassword, setShowFalkorPassword] = useState(false);
|
||||
|
||||
// Collapsible sections
|
||||
const [expandedSections, setExpandedSections] = useState<Record<string, boolean>>({
|
||||
@@ -386,8 +383,6 @@ export function useProjectSettings(
|
||||
setShowLinearKey,
|
||||
showOpenAIKey,
|
||||
setShowOpenAIKey,
|
||||
showFalkorPassword,
|
||||
setShowFalkorPassword,
|
||||
showGitHubToken,
|
||||
setShowGitHubToken,
|
||||
expandedSections,
|
||||
|
||||
@@ -86,8 +86,6 @@ function ProjectSettingsContentInner({
|
||||
setShowLinearKey,
|
||||
showOpenAIKey,
|
||||
setShowOpenAIKey,
|
||||
showFalkorPassword,
|
||||
setShowFalkorPassword,
|
||||
showGitHubToken,
|
||||
setShowGitHubToken,
|
||||
expandedSections: _expandedSections,
|
||||
@@ -138,8 +136,6 @@ function ProjectSettingsContentInner({
|
||||
setShowLinearKey={setShowLinearKey}
|
||||
showOpenAIKey={showOpenAIKey}
|
||||
setShowOpenAIKey={setShowOpenAIKey}
|
||||
showFalkorPassword={showFalkorPassword}
|
||||
setShowFalkorPassword={setShowFalkorPassword}
|
||||
showGitHubToken={showGitHubToken}
|
||||
setShowGitHubToken={setShowGitHubToken}
|
||||
gitHubConnectionStatus={gitHubConnectionStatus}
|
||||
|
||||
@@ -26,8 +26,6 @@ interface SectionRouterProps {
|
||||
setShowLinearKey: React.Dispatch<React.SetStateAction<boolean>>;
|
||||
showOpenAIKey: boolean;
|
||||
setShowOpenAIKey: React.Dispatch<React.SetStateAction<boolean>>;
|
||||
showFalkorPassword: boolean;
|
||||
setShowFalkorPassword: React.Dispatch<React.SetStateAction<boolean>>;
|
||||
showGitHubToken: boolean;
|
||||
setShowGitHubToken: React.Dispatch<React.SetStateAction<boolean>>;
|
||||
gitHubConnectionStatus: GitHubSyncStatus | null;
|
||||
@@ -64,8 +62,6 @@ export function SectionRouter({
|
||||
setShowLinearKey,
|
||||
showOpenAIKey,
|
||||
setShowOpenAIKey,
|
||||
showFalkorPassword,
|
||||
setShowFalkorPassword,
|
||||
showGitHubToken,
|
||||
setShowGitHubToken,
|
||||
gitHubConnectionStatus,
|
||||
@@ -178,13 +174,13 @@ export function SectionRouter({
|
||||
case 'memory':
|
||||
return (
|
||||
<SettingsSection
|
||||
title="Memory Backend"
|
||||
description="Configure how agents store and retrieve memory"
|
||||
title="Memory"
|
||||
description="Configure persistent cross-session memory for agents"
|
||||
>
|
||||
<InitializationGuard
|
||||
initialized={!!project.autoBuildPath}
|
||||
title="Memory Backend"
|
||||
description="Configure agent memory storage"
|
||||
title="Memory"
|
||||
description="Configure persistent memory"
|
||||
>
|
||||
<SecuritySettings
|
||||
envConfig={envConfig}
|
||||
@@ -193,8 +189,6 @@ export function SectionRouter({
|
||||
updateEnvConfig={updateEnvConfig}
|
||||
showOpenAIKey={showOpenAIKey}
|
||||
setShowOpenAIKey={setShowOpenAIKey}
|
||||
showFalkorPassword={showFalkorPassword}
|
||||
setShowFalkorPassword={setShowFalkorPassword}
|
||||
expanded={true}
|
||||
onToggle={() => {}}
|
||||
/>
|
||||
|
||||
@@ -33,8 +33,6 @@ export function createHookProxy(
|
||||
get setShowLinearKey() { return hookRef.current.setShowLinearKey; },
|
||||
get showOpenAIKey() { return hookRef.current.showOpenAIKey; },
|
||||
get setShowOpenAIKey() { return hookRef.current.setShowOpenAIKey; },
|
||||
get showFalkorPassword() { return hookRef.current.showFalkorPassword; },
|
||||
get setShowFalkorPassword() { return hookRef.current.setShowFalkorPassword; },
|
||||
get showGitHubToken() { return hookRef.current.showGitHubToken; },
|
||||
get setShowGitHubToken() { return hookRef.current.setShowGitHubToken; },
|
||||
get expandedSections() { return hookRef.current.expandedSections; },
|
||||
|
||||
@@ -248,8 +248,8 @@ function TaskDetailModalContent({ open, task, onOpenChange }: { open: boolean; t
|
||||
{/* Header */}
|
||||
<div className="p-5 pb-4 border-b border-border shrink-0">
|
||||
<div className="flex items-start justify-between gap-4">
|
||||
<div className="flex-1 min-w-0">
|
||||
<DialogPrimitive.Title className="text-xl font-semibold leading-tight text-foreground pr-4">
|
||||
<div className="flex-1 min-w-0 overflow-hidden">
|
||||
<DialogPrimitive.Title className="text-xl font-semibold leading-tight text-foreground truncate">
|
||||
{task.title}
|
||||
</DialogPrimitive.Title>
|
||||
<DialogPrimitive.Description asChild>
|
||||
@@ -298,7 +298,7 @@ function TaskDetailModalContent({ open, task, onOpenChange }: { open: boolean; t
|
||||
</div>
|
||||
</DialogPrimitive.Description>
|
||||
</div>
|
||||
<div className="flex items-center gap-1 shrink-0">
|
||||
<div className="flex items-center gap-1 shrink-0 electron-no-drag">
|
||||
<Button
|
||||
variant="ghost"
|
||||
size="icon"
|
||||
|
||||
@@ -1,15 +1,17 @@
|
||||
import { useState, useEffect } from 'react';
|
||||
import type { InfrastructureStatus } from '../../shared/types';
|
||||
|
||||
/**
|
||||
* Hook for checking memory infrastructure status (LadybugDB)
|
||||
* No Docker required - uses embedded database
|
||||
*/
|
||||
export function useInfrastructureStatus(
|
||||
graphitiEnabled: boolean | undefined,
|
||||
falkorDbPort: number | undefined,
|
||||
dbPath: string | undefined,
|
||||
open: boolean
|
||||
) {
|
||||
const [infrastructureStatus, setInfrastructureStatus] = useState<InfrastructureStatus | null>(null);
|
||||
const [isCheckingInfrastructure, setIsCheckingInfrastructure] = useState(false);
|
||||
const [isStartingFalkorDB, setIsStartingFalkorDB] = useState(false);
|
||||
const [isOpeningDocker, setIsOpeningDocker] = useState(false);
|
||||
|
||||
useEffect(() => {
|
||||
const checkInfrastructure = async () => {
|
||||
@@ -20,8 +22,7 @@ export function useInfrastructureStatus(
|
||||
|
||||
setIsCheckingInfrastructure(true);
|
||||
try {
|
||||
const port = falkorDbPort || 6380;
|
||||
const result = await window.electronAPI.getInfrastructureStatus(port);
|
||||
const result = await window.electronAPI.getMemoryInfrastructureStatus(dbPath);
|
||||
if (result.success && result.data) {
|
||||
setInfrastructureStatus(result.data);
|
||||
}
|
||||
@@ -41,57 +42,10 @@ export function useInfrastructureStatus(
|
||||
return () => {
|
||||
if (interval) clearInterval(interval);
|
||||
};
|
||||
}, [graphitiEnabled, falkorDbPort, open]);
|
||||
|
||||
const handleStartFalkorDB = async () => {
|
||||
setIsStartingFalkorDB(true);
|
||||
try {
|
||||
const port = falkorDbPort || 6380;
|
||||
const result = await window.electronAPI.startFalkorDB(port);
|
||||
if (result.success && result.data?.success) {
|
||||
// Refresh status after starting
|
||||
const statusResult = await window.electronAPI.getInfrastructureStatus(port);
|
||||
if (statusResult.success && statusResult.data) {
|
||||
setInfrastructureStatus(statusResult.data);
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
// Error handling is implicit in the status check
|
||||
} finally {
|
||||
setIsStartingFalkorDB(false);
|
||||
}
|
||||
};
|
||||
|
||||
const handleOpenDockerDesktop = async () => {
|
||||
setIsOpeningDocker(true);
|
||||
try {
|
||||
await window.electronAPI.openDockerDesktop();
|
||||
// Wait a bit then refresh status
|
||||
setTimeout(async () => {
|
||||
const port = falkorDbPort || 6380;
|
||||
const result = await window.electronAPI.getInfrastructureStatus(port);
|
||||
if (result.success && result.data) {
|
||||
setInfrastructureStatus(result.data);
|
||||
}
|
||||
setIsOpeningDocker(false);
|
||||
}, 3000);
|
||||
} catch {
|
||||
setIsOpeningDocker(false);
|
||||
}
|
||||
};
|
||||
|
||||
const handleDownloadDocker = async () => {
|
||||
const url = await window.electronAPI.getDockerDownloadUrl();
|
||||
window.electronAPI.openExternal(url);
|
||||
};
|
||||
}, [graphitiEnabled, dbPath, open]);
|
||||
|
||||
return {
|
||||
infrastructureStatus,
|
||||
isCheckingInfrastructure,
|
||||
isStartingFalkorDB,
|
||||
isOpeningDocker,
|
||||
handleStartFalkorDB,
|
||||
handleOpenDockerDesktop,
|
||||
handleDownloadDocker,
|
||||
};
|
||||
}
|
||||
|
||||
@@ -0,0 +1,149 @@
|
||||
/**
|
||||
* Centralized Icon Exports
|
||||
*
|
||||
* This file serves as the single source of truth for all lucide-react icons used
|
||||
* throughout the application. By consolidating imports here, we enable:
|
||||
*
|
||||
* 1. Better tracking of which icons are actually used
|
||||
* 2. Potential code-splitting opportunities
|
||||
* 3. Easier future migration to alternative icon solutions
|
||||
* 4. Reduced bundle size through optimized tree-shaking
|
||||
*
|
||||
* Usage:
|
||||
* import { AlertCircle, Check, X } from '@/lib/icons';
|
||||
*
|
||||
* When adding new icons:
|
||||
* 1. Import the icon from 'lucide-react'
|
||||
* 2. Add it to the export statement in alphabetical order
|
||||
*/
|
||||
|
||||
export {
|
||||
Activity,
|
||||
AlertCircle,
|
||||
AlertTriangle,
|
||||
Archive,
|
||||
ArrowLeft,
|
||||
ArrowRight,
|
||||
BarChart3,
|
||||
Bell,
|
||||
BookOpen,
|
||||
Bot,
|
||||
Box,
|
||||
Brain,
|
||||
Bug,
|
||||
Calendar,
|
||||
Check,
|
||||
CheckCircle,
|
||||
CheckCircle2,
|
||||
CheckSquare,
|
||||
ChevronDown,
|
||||
ChevronRight,
|
||||
ChevronUp,
|
||||
Circle,
|
||||
Clock,
|
||||
CloudDownload,
|
||||
Code,
|
||||
Code2,
|
||||
Cog,
|
||||
Copy,
|
||||
Cpu,
|
||||
CreditCard,
|
||||
Database,
|
||||
Download,
|
||||
ExternalLink,
|
||||
Eye,
|
||||
EyeOff,
|
||||
File,
|
||||
FileCode,
|
||||
FileDown,
|
||||
FileImage,
|
||||
FileJson,
|
||||
FileText,
|
||||
Filter,
|
||||
FlaskConical,
|
||||
Folder,
|
||||
FolderGit2,
|
||||
FolderOpen,
|
||||
FolderPlus,
|
||||
FolderSearch,
|
||||
FolderTree,
|
||||
FolderX,
|
||||
Gauge,
|
||||
GitBranch,
|
||||
GitCommit,
|
||||
Github,
|
||||
GitMerge,
|
||||
Globe,
|
||||
Grid2X2,
|
||||
HardDrive,
|
||||
HelpCircle,
|
||||
History,
|
||||
Image,
|
||||
Import,
|
||||
Inbox,
|
||||
Info,
|
||||
Key,
|
||||
KeyRound,
|
||||
Layers,
|
||||
LayoutGrid,
|
||||
Lightbulb,
|
||||
ListChecks,
|
||||
ListTodo,
|
||||
Loader2,
|
||||
Lock,
|
||||
LogIn,
|
||||
Mail,
|
||||
Map,
|
||||
MessageCircle,
|
||||
MessageSquare,
|
||||
Minus,
|
||||
Monitor,
|
||||
Moon,
|
||||
MoreVertical,
|
||||
Package,
|
||||
Palette,
|
||||
PanelLeft,
|
||||
PanelLeftClose,
|
||||
PartyPopper,
|
||||
Pencil,
|
||||
PenLine,
|
||||
Play,
|
||||
Plus,
|
||||
Radio,
|
||||
RefreshCw,
|
||||
Rocket,
|
||||
RotateCcw,
|
||||
Route,
|
||||
Save,
|
||||
Scale,
|
||||
Search,
|
||||
Send,
|
||||
Server,
|
||||
Settings,
|
||||
Settings2,
|
||||
Shield,
|
||||
Sliders,
|
||||
Sparkles,
|
||||
Square,
|
||||
Star,
|
||||
Sun,
|
||||
Tag,
|
||||
Target,
|
||||
Terminal,
|
||||
TerminalSquare,
|
||||
ThumbsUp,
|
||||
Trash2,
|
||||
TrendingUp,
|
||||
Upload,
|
||||
User,
|
||||
Users,
|
||||
Wand2,
|
||||
Wifi,
|
||||
Wrench,
|
||||
X,
|
||||
XCircle,
|
||||
Zap,
|
||||
} from 'lucide-react';
|
||||
|
||||
// Re-export Image as ImageIcon for components that use this alias
|
||||
export { Image as ImageIcon } from 'lucide-react';
|
||||
@@ -46,6 +46,11 @@ export const changelogMock = {
|
||||
}
|
||||
}),
|
||||
|
||||
readLocalImage: async () => ({
|
||||
success: false,
|
||||
error: 'Mock: Cannot read local images in browser mode'
|
||||
}),
|
||||
|
||||
readExistingChangelog: async () => ({
|
||||
success: true,
|
||||
data: {
|
||||
|
||||
@@ -1,60 +1,43 @@
|
||||
/**
|
||||
* Mock implementation for infrastructure, Docker, and system operations
|
||||
* Mock implementation for infrastructure and system operations
|
||||
* Updated for LadybugDB (embedded database, no Docker required)
|
||||
*/
|
||||
|
||||
export const infrastructureMock = {
|
||||
// Docker & Infrastructure Operations
|
||||
getInfrastructureStatus: async () => ({
|
||||
// Memory Infrastructure Operations (LadybugDB)
|
||||
getMemoryInfrastructureStatus: async () => ({
|
||||
success: true,
|
||||
data: {
|
||||
docker: {
|
||||
installed: true,
|
||||
running: true,
|
||||
version: 'Docker version 24.0.0 (mock)'
|
||||
},
|
||||
falkordb: {
|
||||
containerExists: true,
|
||||
containerRunning: true,
|
||||
containerName: 'auto-claude-falkordb',
|
||||
port: 6380,
|
||||
healthy: true
|
||||
memory: {
|
||||
kuzuInstalled: true,
|
||||
databasePath: '~/.auto-claude/graphs',
|
||||
databaseExists: true,
|
||||
databases: ['auto_claude_memory']
|
||||
},
|
||||
ready: true
|
||||
}
|
||||
}),
|
||||
|
||||
startFalkorDB: async () => ({
|
||||
listMemoryDatabases: async () => ({
|
||||
success: true,
|
||||
data: { success: true }
|
||||
data: ['auto_claude_memory', 'project_memory']
|
||||
}),
|
||||
|
||||
stopFalkorDB: async () => ({
|
||||
success: true,
|
||||
data: { success: true }
|
||||
}),
|
||||
|
||||
openDockerDesktop: async () => ({
|
||||
success: true,
|
||||
data: { success: true }
|
||||
}),
|
||||
|
||||
getDockerDownloadUrl: async () => 'https://www.docker.com/products/docker-desktop/',
|
||||
|
||||
// Graphiti Validation Operations
|
||||
validateFalkorDBConnection: async () => ({
|
||||
testMemoryConnection: async () => ({
|
||||
success: true,
|
||||
data: {
|
||||
success: true,
|
||||
message: 'Connected to FalkorDB at localhost:6380 (mock)',
|
||||
details: { latencyMs: 15 }
|
||||
message: 'Connected to LadybugDB database (mock)',
|
||||
details: { latencyMs: 5 }
|
||||
}
|
||||
}),
|
||||
|
||||
validateOpenAIApiKey: async () => ({
|
||||
// LLM API Validation Operations
|
||||
validateLLMApiKey: async () => ({
|
||||
success: true,
|
||||
data: {
|
||||
success: true,
|
||||
message: 'OpenAI API key is valid (mock)',
|
||||
message: 'API key is valid (mock)',
|
||||
details: { provider: 'openai', latencyMs: 100 }
|
||||
}
|
||||
}),
|
||||
@@ -62,20 +45,60 @@ export const infrastructureMock = {
|
||||
testGraphitiConnection: async () => ({
|
||||
success: true,
|
||||
data: {
|
||||
falkordb: {
|
||||
database: {
|
||||
success: true,
|
||||
message: 'Connected to FalkorDB at localhost:6380 (mock)',
|
||||
details: { latencyMs: 15 }
|
||||
message: 'Connected to LadybugDB database (mock)',
|
||||
details: { latencyMs: 5 }
|
||||
},
|
||||
openai: {
|
||||
llmProvider: {
|
||||
success: true,
|
||||
message: 'OpenAI API key is valid (mock)',
|
||||
message: 'LLM API key is valid (mock)',
|
||||
details: { provider: 'openai', latencyMs: 100 }
|
||||
},
|
||||
ready: true
|
||||
}
|
||||
}),
|
||||
|
||||
// Ollama Model Detection Operations
|
||||
checkOllamaStatus: async () => ({
|
||||
success: true,
|
||||
data: {
|
||||
running: true,
|
||||
url: 'http://localhost:11434',
|
||||
version: '0.1.0',
|
||||
}
|
||||
}),
|
||||
|
||||
listOllamaModels: async () => ({
|
||||
success: true,
|
||||
data: {
|
||||
models: [
|
||||
{ name: 'llama2', size_bytes: 4000000000, size_gb: 3.73, modified_at: '2024-01-01', is_embedding: false },
|
||||
{ name: 'nomic-embed-text', size_bytes: 500000000, size_gb: 0.47, modified_at: '2024-01-01', is_embedding: true, embedding_dim: 768, description: 'Nomic AI text embeddings' },
|
||||
],
|
||||
count: 2
|
||||
}
|
||||
}),
|
||||
|
||||
listOllamaEmbeddingModels: async () => ({
|
||||
success: true,
|
||||
data: {
|
||||
embedding_models: [
|
||||
{ name: 'nomic-embed-text', embedding_dim: 768, description: 'Nomic AI text embeddings', size_bytes: 500000000, size_gb: 0.47 },
|
||||
],
|
||||
count: 1
|
||||
}
|
||||
}),
|
||||
|
||||
pullOllamaModel: async (modelName: string) => ({
|
||||
success: true,
|
||||
data: {
|
||||
model: modelName,
|
||||
status: 'completed' as const,
|
||||
output: [`Pulling ${modelName}...`, 'Pull complete']
|
||||
}
|
||||
}),
|
||||
|
||||
// Ideation Operations
|
||||
getIdeation: async () => ({
|
||||
success: true,
|
||||
|
||||
@@ -188,5 +188,25 @@ export const integrationMock = {
|
||||
getGitHubBranches: async () => ({
|
||||
success: true,
|
||||
data: ['main', 'develop', 'feature/example']
|
||||
}),
|
||||
|
||||
createGitHubRepo: async (_repoName: string, _options: { description?: string; isPrivate?: boolean; projectPath: string; owner?: string }) => ({
|
||||
success: false,
|
||||
error: 'Not available in browser mock'
|
||||
}),
|
||||
|
||||
addGitRemote: async (_projectPath: string, _repoFullName: string) => ({
|
||||
success: false,
|
||||
error: 'Not available in browser mock'
|
||||
}),
|
||||
|
||||
listGitHubOrgs: async () => ({
|
||||
success: true,
|
||||
data: {
|
||||
orgs: [
|
||||
{ login: 'example-org', avatarUrl: 'https://avatars.githubusercontent.com/u/1?v=4' },
|
||||
{ login: 'another-org', avatarUrl: 'https://avatars.githubusercontent.com/u/2?v=4' }
|
||||
]
|
||||
}
|
||||
})
|
||||
};
|
||||
|
||||
@@ -48,6 +48,18 @@ export const projectMock = {
|
||||
}
|
||||
}),
|
||||
|
||||
// Tab state operations (persisted in main process)
|
||||
getTabState: async () => ({
|
||||
success: true,
|
||||
data: {
|
||||
openProjectIds: [],
|
||||
activeProjectId: null,
|
||||
tabOrder: []
|
||||
}
|
||||
}),
|
||||
|
||||
saveTabState: async () => ({ success: true }),
|
||||
|
||||
// Dialog operations
|
||||
selectDirectory: async () => {
|
||||
return prompt('Enter project path (browser mock):', '/Users/demo/projects/new-project');
|
||||
|
||||
@@ -1,15 +1,23 @@
|
||||
import { create } from 'zustand';
|
||||
import type { Project, ProjectSettings, AutoBuildVersionInfo, InitializationResult } from '../../shared/types';
|
||||
|
||||
// localStorage key for persisting the last selected project
|
||||
// localStorage keys for persisting project state (legacy - now using IPC)
|
||||
const LAST_SELECTED_PROJECT_KEY = 'lastSelectedProjectId';
|
||||
|
||||
// Debounce timer for saving tab state
|
||||
let saveTabStateTimeout: ReturnType<typeof setTimeout> | null = null;
|
||||
|
||||
interface ProjectState {
|
||||
projects: Project[];
|
||||
selectedProjectId: string | null;
|
||||
isLoading: boolean;
|
||||
error: string | null;
|
||||
|
||||
// Tab state
|
||||
openProjectIds: string[]; // Array of open project IDs
|
||||
activeProjectId: string | null; // Currently active tab
|
||||
tabOrder: string[]; // Order of tabs for drag and drop
|
||||
|
||||
// Actions
|
||||
setProjects: (projects: Project[]) => void;
|
||||
addProject: (project: Project) => void;
|
||||
@@ -19,8 +27,18 @@ interface ProjectState {
|
||||
setLoading: (loading: boolean) => void;
|
||||
setError: (error: string | null) => void;
|
||||
|
||||
// Tab management actions
|
||||
openProjectTab: (projectId: string) => void;
|
||||
closeProjectTab: (projectId: string) => void;
|
||||
setActiveProject: (projectId: string | null) => void;
|
||||
reorderTabs: (fromIndex: number, toIndex: number) => void;
|
||||
restoreTabState: () => void;
|
||||
|
||||
// Selectors
|
||||
getSelectedProject: () => Project | undefined;
|
||||
getOpenProjects: () => Project[];
|
||||
getActiveProject: () => Project | undefined;
|
||||
getProjectTabs: () => Project[];
|
||||
}
|
||||
|
||||
export const useProjectStore = create<ProjectState>((set, get) => ({
|
||||
@@ -29,6 +47,11 @@ export const useProjectStore = create<ProjectState>((set, get) => ({
|
||||
isLoading: false,
|
||||
error: null,
|
||||
|
||||
// Tab state - initialized empty, loaded via IPC from main process for reliability
|
||||
openProjectIds: [],
|
||||
activeProjectId: null,
|
||||
tabOrder: [],
|
||||
|
||||
setProjects: (projects) => set({ projects }),
|
||||
|
||||
addProject: (project) =>
|
||||
@@ -70,12 +93,144 @@ export const useProjectStore = create<ProjectState>((set, get) => ({
|
||||
|
||||
setError: (error) => set({ error }),
|
||||
|
||||
// Tab management actions
|
||||
openProjectTab: (projectId) => {
|
||||
const state = get();
|
||||
console.log('[ProjectStore] openProjectTab called:', {
|
||||
projectId,
|
||||
currentOpenProjectIds: state.openProjectIds,
|
||||
currentTabOrder: state.tabOrder
|
||||
});
|
||||
if (!state.openProjectIds.includes(projectId)) {
|
||||
const newOpenProjectIds = [...state.openProjectIds, projectId];
|
||||
const newTabOrder = state.tabOrder.includes(projectId)
|
||||
? state.tabOrder
|
||||
: [...state.tabOrder, projectId];
|
||||
|
||||
console.log('[ProjectStore] Adding new tab:', {
|
||||
newOpenProjectIds,
|
||||
newTabOrder
|
||||
});
|
||||
|
||||
set({
|
||||
openProjectIds: newOpenProjectIds,
|
||||
tabOrder: newTabOrder,
|
||||
activeProjectId: projectId
|
||||
});
|
||||
|
||||
// Save to main process (debounced)
|
||||
saveTabStateToMain();
|
||||
} else {
|
||||
console.log('[ProjectStore] Project already open, just activating');
|
||||
// Project already open, just make it active
|
||||
get().setActiveProject(projectId);
|
||||
}
|
||||
},
|
||||
|
||||
closeProjectTab: (projectId) => {
|
||||
const state = get();
|
||||
const newOpenProjectIds = state.openProjectIds.filter(id => id !== projectId);
|
||||
const newTabOrder = state.tabOrder.filter(id => id !== projectId);
|
||||
|
||||
// If closing the active project, select another one or null
|
||||
let newActiveProjectId = state.activeProjectId;
|
||||
if (state.activeProjectId === projectId) {
|
||||
const remainingTabs = newTabOrder.length > 0 ? newTabOrder : [];
|
||||
newActiveProjectId = remainingTabs.length > 0 ? remainingTabs[0] : null;
|
||||
}
|
||||
|
||||
set({
|
||||
openProjectIds: newOpenProjectIds,
|
||||
tabOrder: newTabOrder,
|
||||
activeProjectId: newActiveProjectId
|
||||
});
|
||||
|
||||
// Save to main process (debounced)
|
||||
saveTabStateToMain();
|
||||
},
|
||||
|
||||
setActiveProject: (projectId) => {
|
||||
set({ activeProjectId: projectId });
|
||||
// Also update selectedProjectId for backward compatibility
|
||||
get().selectProject(projectId);
|
||||
// Save to main process (debounced)
|
||||
saveTabStateToMain();
|
||||
},
|
||||
|
||||
reorderTabs: (fromIndex, toIndex) => {
|
||||
const state = get();
|
||||
const newTabOrder = [...state.tabOrder];
|
||||
const [movedTab] = newTabOrder.splice(fromIndex, 1);
|
||||
newTabOrder.splice(toIndex, 0, movedTab);
|
||||
|
||||
set({ tabOrder: newTabOrder });
|
||||
// Save to main process (debounced)
|
||||
saveTabStateToMain();
|
||||
},
|
||||
|
||||
restoreTabState: () => {
|
||||
// This is now handled by loadTabStateFromMain() called during loadProjects()
|
||||
console.log('[ProjectStore] restoreTabState called - now handled by IPC');
|
||||
},
|
||||
|
||||
|
||||
// Original selectors
|
||||
getSelectedProject: () => {
|
||||
const state = get();
|
||||
return state.projects.find((p) => p.id === state.selectedProjectId);
|
||||
},
|
||||
|
||||
// New selectors for tab functionality
|
||||
getOpenProjects: () => {
|
||||
const state = get();
|
||||
return state.projects.filter((p) => state.openProjectIds.includes(p.id));
|
||||
},
|
||||
|
||||
getActiveProject: () => {
|
||||
const state = get();
|
||||
return state.projects.find((p) => p.id === state.activeProjectId);
|
||||
},
|
||||
|
||||
getProjectTabs: () => {
|
||||
const state = get();
|
||||
const orderedProjects = state.tabOrder
|
||||
.map(id => state.projects.find(p => p.id === id))
|
||||
.filter(Boolean) as Project[];
|
||||
|
||||
// Add any open projects not in tabOrder to the end
|
||||
const remainingProjects = state.projects
|
||||
.filter(p => state.openProjectIds.includes(p.id) && !state.tabOrder.includes(p.id));
|
||||
|
||||
return [...orderedProjects, ...remainingProjects];
|
||||
}
|
||||
}));
|
||||
|
||||
/**
|
||||
* Save tab state to main process (debounced to avoid excessive IPC calls)
|
||||
*/
|
||||
function saveTabStateToMain(): void {
|
||||
// Clear any pending save
|
||||
if (saveTabStateTimeout) {
|
||||
clearTimeout(saveTabStateTimeout);
|
||||
}
|
||||
|
||||
// Debounce saves to avoid excessive IPC calls
|
||||
saveTabStateTimeout = setTimeout(async () => {
|
||||
const store = useProjectStore.getState();
|
||||
const tabState = {
|
||||
openProjectIds: store.openProjectIds,
|
||||
activeProjectId: store.activeProjectId,
|
||||
tabOrder: store.tabOrder
|
||||
};
|
||||
console.log('[ProjectStore] Saving tab state to main process:', tabState);
|
||||
try {
|
||||
await window.electronAPI.saveTabState(tabState);
|
||||
} catch (err) {
|
||||
console.error('[ProjectStore] Failed to save tab state:', err);
|
||||
}
|
||||
}, 100);
|
||||
}
|
||||
|
||||
/**
|
||||
* Load projects from main process
|
||||
*/
|
||||
@@ -85,17 +240,80 @@ export async function loadProjects(): Promise<void> {
|
||||
store.setError(null);
|
||||
|
||||
try {
|
||||
// First, load tab state from main process (reliable persistence)
|
||||
const tabStateResult = await window.electronAPI.getTabState();
|
||||
console.log('[ProjectStore] Loaded tab state from main process:', tabStateResult.data);
|
||||
|
||||
if (tabStateResult.success && tabStateResult.data) {
|
||||
useProjectStore.setState({
|
||||
openProjectIds: tabStateResult.data.openProjectIds || [],
|
||||
activeProjectId: tabStateResult.data.activeProjectId || null,
|
||||
tabOrder: tabStateResult.data.tabOrder || []
|
||||
});
|
||||
}
|
||||
|
||||
// Then load projects
|
||||
const result = await window.electronAPI.getProjects();
|
||||
console.log('[ProjectStore] getProjects result:', {
|
||||
success: result.success,
|
||||
projectCount: result.data?.length,
|
||||
projectIds: result.data?.map(p => p.id)
|
||||
});
|
||||
|
||||
if (result.success && result.data) {
|
||||
store.setProjects(result.data);
|
||||
|
||||
// Restore last selected project from localStorage, or fall back to first project
|
||||
if (!store.selectedProjectId && result.data.length > 0) {
|
||||
// Get current tab state (may have been loaded from IPC)
|
||||
const currentState = useProjectStore.getState();
|
||||
|
||||
// Clean up tab state - remove any project IDs that no longer exist
|
||||
const validOpenProjectIds = currentState.openProjectIds.filter(id =>
|
||||
result.data?.some((p) => p.id === id) ?? false
|
||||
);
|
||||
const validTabOrder = currentState.tabOrder.filter(id =>
|
||||
result.data?.some((p) => p.id === id) ?? false
|
||||
);
|
||||
const validActiveProjectId = currentState.activeProjectId &&
|
||||
result.data?.some((p) => p.id === currentState.activeProjectId)
|
||||
? currentState.activeProjectId
|
||||
: null;
|
||||
|
||||
console.log('[ProjectStore] Tab state cleanup:', {
|
||||
originalOpenProjectIds: currentState.openProjectIds,
|
||||
validOpenProjectIds,
|
||||
originalTabOrder: currentState.tabOrder,
|
||||
validTabOrder,
|
||||
originalActiveProjectId: currentState.activeProjectId,
|
||||
validActiveProjectId
|
||||
});
|
||||
|
||||
// Update store with cleaned tab state if needed
|
||||
if (validOpenProjectIds.length !== currentState.openProjectIds.length ||
|
||||
validTabOrder.length !== currentState.tabOrder.length ||
|
||||
validActiveProjectId !== currentState.activeProjectId) {
|
||||
console.log('[ProjectStore] Updating cleaned tab state');
|
||||
useProjectStore.setState({
|
||||
openProjectIds: validOpenProjectIds,
|
||||
tabOrder: validTabOrder,
|
||||
activeProjectId: validActiveProjectId
|
||||
});
|
||||
// Save cleaned state back to main process
|
||||
saveTabStateToMain();
|
||||
} else {
|
||||
console.log('[ProjectStore] Tab state is valid, no cleanup needed');
|
||||
}
|
||||
|
||||
// Restore last selected project from localStorage for backward compatibility,
|
||||
// or fall back to active project, or first project
|
||||
const updatedState = useProjectStore.getState();
|
||||
if (!updatedState.selectedProjectId && result.data.length > 0) {
|
||||
const lastSelectedId = localStorage.getItem(LAST_SELECTED_PROJECT_KEY);
|
||||
const projectExists = lastSelectedId && result.data.some((p) => p.id === lastSelectedId);
|
||||
|
||||
if (projectExists) {
|
||||
store.selectProject(lastSelectedId);
|
||||
} else if (updatedState.activeProjectId) {
|
||||
store.selectProject(updatedState.activeProjectId);
|
||||
} else {
|
||||
store.selectProject(result.data[0].id);
|
||||
}
|
||||
@@ -121,6 +339,8 @@ export async function addProject(projectPath: string): Promise<Project | null> {
|
||||
if (result.success && result.data) {
|
||||
store.addProject(result.data);
|
||||
store.selectProject(result.data.id);
|
||||
// Also open a tab for the new project
|
||||
store.openProjectTab(result.data.id);
|
||||
return result.data;
|
||||
} else {
|
||||
store.setError(result.error || 'Failed to add project');
|
||||
@@ -142,6 +362,10 @@ export async function removeProject(projectId: string): Promise<boolean> {
|
||||
const result = await window.electronAPI.removeProject(projectId);
|
||||
if (result.success) {
|
||||
store.removeProject(projectId);
|
||||
// Also close the tab if it's open
|
||||
if (store.openProjectIds.includes(projectId)) {
|
||||
store.closeProjectTab(projectId);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
|
||||
@@ -211,5 +211,3 @@ export function canCreateRelease(): boolean {
|
||||
!store.isCreatingRelease
|
||||
);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -13,6 +13,10 @@ export const IPC_CHANNELS = {
|
||||
PROJECT_UPDATE_AUTOBUILD: 'project:updateAutoBuild',
|
||||
PROJECT_CHECK_VERSION: 'project:checkVersion',
|
||||
|
||||
// Tab state operations (persisted in main process)
|
||||
TAB_STATE_GET: 'tabState:get',
|
||||
TAB_STATE_SAVE: 'tabState:save',
|
||||
|
||||
// Task operations
|
||||
TASK_LIST: 'task:list',
|
||||
TASK_CREATE: 'task:create',
|
||||
@@ -192,24 +196,31 @@ export const IPC_CHANNELS = {
|
||||
GITHUB_LIST_USER_REPOS: 'github:listUserRepos',
|
||||
GITHUB_DETECT_REPO: 'github:detectRepo',
|
||||
GITHUB_GET_BRANCHES: 'github:getBranches',
|
||||
GITHUB_CREATE_REPO: 'github:createRepo',
|
||||
GITHUB_ADD_REMOTE: 'github:addRemote',
|
||||
GITHUB_LIST_ORGS: 'github:listOrgs',
|
||||
|
||||
// GitHub events (main -> renderer)
|
||||
GITHUB_INVESTIGATION_PROGRESS: 'github:investigationProgress',
|
||||
GITHUB_INVESTIGATION_COMPLETE: 'github:investigationComplete',
|
||||
GITHUB_INVESTIGATION_ERROR: 'github:investigationError',
|
||||
|
||||
// Docker & Infrastructure status
|
||||
DOCKER_STATUS: 'docker:status',
|
||||
DOCKER_START_FALKORDB: 'docker:startFalkordb',
|
||||
DOCKER_STOP_FALKORDB: 'docker:stopFalkordb',
|
||||
DOCKER_OPEN_DESKTOP: 'docker:openDesktop',
|
||||
DOCKER_GET_DOWNLOAD_URL: 'docker:getDownloadUrl',
|
||||
// Memory Infrastructure status (LadybugDB - no Docker required)
|
||||
MEMORY_STATUS: 'memory:status',
|
||||
MEMORY_LIST_DATABASES: 'memory:listDatabases',
|
||||
MEMORY_TEST_CONNECTION: 'memory:testConnection',
|
||||
|
||||
// Graphiti validation
|
||||
GRAPHITI_VALIDATE_FALKORDB: 'graphiti:validateFalkordb',
|
||||
GRAPHITI_VALIDATE_OPENAI: 'graphiti:validateOpenai',
|
||||
GRAPHITI_VALIDATE_LLM: 'graphiti:validateLlm',
|
||||
GRAPHITI_TEST_CONNECTION: 'graphiti:testConnection',
|
||||
|
||||
// Ollama model detection and management
|
||||
OLLAMA_CHECK_STATUS: 'ollama:checkStatus',
|
||||
OLLAMA_LIST_MODELS: 'ollama:listModels',
|
||||
OLLAMA_LIST_EMBEDDING_MODELS: 'ollama:listEmbeddingModels',
|
||||
OLLAMA_PULL_MODEL: 'ollama:pullModel',
|
||||
OLLAMA_PULL_PROGRESS: 'ollama:pullProgress',
|
||||
|
||||
// Auto Claude source updates
|
||||
AUTOBUILD_SOURCE_CHECK: 'autobuild:source:check',
|
||||
AUTOBUILD_SOURCE_DOWNLOAD: 'autobuild:source:download',
|
||||
@@ -235,6 +246,7 @@ export const IPC_CHANNELS = {
|
||||
CHANGELOG_GET_TAGS: 'changelog:getTags',
|
||||
CHANGELOG_GET_COMMITS_PREVIEW: 'changelog:getCommitsPreview',
|
||||
CHANGELOG_SAVE_IMAGE: 'changelog:saveImage',
|
||||
CHANGELOG_READ_LOCAL_IMAGE: 'changelog:readLocalImage',
|
||||
|
||||
// Changelog events (main -> renderer)
|
||||
CHANGELOG_GENERATION_PROGRESS: 'changelog:generationProgress',
|
||||
|
||||
@@ -135,5 +135,5 @@ export const DEFAULT_AGENT_PROFILES: AgentProfile[] = [
|
||||
|
||||
export const MEMORY_BACKENDS = [
|
||||
{ value: 'file', label: 'File-based (default)' },
|
||||
{ value: 'graphiti', label: 'Graphiti (FalkorDB)' }
|
||||
{ value: 'graphiti', label: 'Graphiti (LadybugDB)' }
|
||||
] as const;
|
||||
|
||||
@@ -109,6 +109,13 @@ import type {
|
||||
} from './integrations';
|
||||
|
||||
// Electron API exposed via contextBridge
|
||||
// Tab state interface (persisted in main process)
|
||||
export interface TabState {
|
||||
openProjectIds: string[];
|
||||
activeProjectId: string | null;
|
||||
tabOrder: string[];
|
||||
}
|
||||
|
||||
export interface ElectronAPI {
|
||||
// Project operations
|
||||
addProject: (projectPath: string) => Promise<IPCResult<Project>>;
|
||||
@@ -119,6 +126,10 @@ export interface ElectronAPI {
|
||||
updateProjectAutoBuild: (projectId: string) => Promise<IPCResult<InitializationResult>>;
|
||||
checkProjectVersion: (projectId: string) => Promise<IPCResult<AutoBuildVersionInfo>>;
|
||||
|
||||
// Tab State (persisted in main process for reliability)
|
||||
getTabState: () => Promise<IPCResult<TabState>>;
|
||||
saveTabState: (tabState: TabState) => Promise<IPCResult>;
|
||||
|
||||
// Task operations
|
||||
getTasks: (projectId: string) => Promise<IPCResult<Task[]>>;
|
||||
createTask: (projectId: string, title: string, description: string, metadata?: TaskMetadata) => Promise<IPCResult<Task>>;
|
||||
@@ -279,20 +290,19 @@ export interface ElectronAPI {
|
||||
checkClaudeAuth: (projectId: string) => Promise<IPCResult<ClaudeAuthResult>>;
|
||||
invokeClaudeSetup: (projectId: string) => Promise<IPCResult<ClaudeAuthResult>>;
|
||||
|
||||
// Docker & Infrastructure operations (for Graphiti/FalkorDB)
|
||||
getInfrastructureStatus: (port?: number) => Promise<IPCResult<InfrastructureStatus>>;
|
||||
startFalkorDB: (port?: number) => Promise<IPCResult<{ success: boolean; error?: string }>>;
|
||||
stopFalkorDB: () => Promise<IPCResult<{ success: boolean; error?: string }>>;
|
||||
openDockerDesktop: () => Promise<IPCResult<{ success: boolean; error?: string }>>;
|
||||
getDockerDownloadUrl: () => Promise<string>;
|
||||
// Memory Infrastructure operations (LadybugDB - no Docker required)
|
||||
getMemoryInfrastructureStatus: (dbPath?: string) => Promise<IPCResult<InfrastructureStatus>>;
|
||||
listMemoryDatabases: (dbPath?: string) => Promise<IPCResult<string[]>>;
|
||||
testMemoryConnection: (dbPath?: string, database?: string) => Promise<IPCResult<GraphitiValidationResult>>;
|
||||
|
||||
// Graphiti validation operations
|
||||
validateFalkorDBConnection: (uri: string) => Promise<IPCResult<GraphitiValidationResult>>;
|
||||
validateOpenAIApiKey: (apiKey: string) => Promise<IPCResult<GraphitiValidationResult>>;
|
||||
testGraphitiConnection: (
|
||||
falkorDbUri: string,
|
||||
openAiApiKey: string
|
||||
) => Promise<IPCResult<GraphitiConnectionTestResult>>;
|
||||
validateLLMApiKey: (provider: string, apiKey: string) => Promise<IPCResult<GraphitiValidationResult>>;
|
||||
testGraphitiConnection: (config: {
|
||||
dbPath?: string;
|
||||
database?: string;
|
||||
llmProvider: string;
|
||||
apiKey: string;
|
||||
}) => Promise<IPCResult<GraphitiConnectionTestResult>>;
|
||||
|
||||
// Linear integration operations
|
||||
getLinearTeams: (projectId: string) => Promise<IPCResult<LinearTeam[]>>;
|
||||
@@ -332,6 +342,15 @@ export interface ElectronAPI {
|
||||
listGitHubUserRepos: () => Promise<IPCResult<{ repos: Array<{ fullName: string; description: string | null; isPrivate: boolean }> }>>;
|
||||
detectGitHubRepo: (projectPath: string) => Promise<IPCResult<string>>;
|
||||
getGitHubBranches: (repo: string, token: string) => Promise<IPCResult<string[]>>;
|
||||
createGitHubRepo: (
|
||||
repoName: string,
|
||||
options: { description?: string; isPrivate?: boolean; projectPath: string; owner?: string }
|
||||
) => Promise<IPCResult<{ fullName: string; url: string }>>;
|
||||
addGitRemote: (
|
||||
projectPath: string,
|
||||
repoFullName: string
|
||||
) => Promise<IPCResult<{ remoteUrl: string }>>;
|
||||
listGitHubOrgs: () => Promise<IPCResult<{ orgs: Array<{ login: string; avatarUrl?: string }> }>>;
|
||||
|
||||
// GitHub event listeners
|
||||
onGitHubInvestigationProgress: (
|
||||
@@ -458,6 +477,10 @@ export interface ElectronAPI {
|
||||
imageData: string,
|
||||
filename: string
|
||||
) => Promise<IPCResult<{ relativePath: string; url: string }>>;
|
||||
readLocalImage: (
|
||||
projectPath: string,
|
||||
relativePath: string
|
||||
) => Promise<IPCResult<string>>;
|
||||
|
||||
// Changelog event listeners
|
||||
onChangelogGenerationProgress: (
|
||||
@@ -520,6 +543,41 @@ export interface ElectronAPI {
|
||||
detectMainBranch: (projectPath: string) => Promise<IPCResult<string | null>>;
|
||||
checkGitStatus: (projectPath: string) => Promise<IPCResult<GitStatus>>;
|
||||
initializeGit: (projectPath: string) => Promise<IPCResult<InitializationResult>>;
|
||||
|
||||
// Ollama model detection operations
|
||||
checkOllamaStatus: (baseUrl?: string) => Promise<IPCResult<{
|
||||
running: boolean;
|
||||
url: string;
|
||||
version?: string;
|
||||
message?: string;
|
||||
}>>;
|
||||
listOllamaModels: (baseUrl?: string) => Promise<IPCResult<{
|
||||
models: Array<{
|
||||
name: string;
|
||||
size_bytes: number;
|
||||
size_gb: number;
|
||||
modified_at: string;
|
||||
is_embedding: boolean;
|
||||
embedding_dim?: number | null;
|
||||
description?: string;
|
||||
}>;
|
||||
count: number;
|
||||
}>>;
|
||||
listOllamaEmbeddingModels: (baseUrl?: string) => Promise<IPCResult<{
|
||||
embedding_models: Array<{
|
||||
name: string;
|
||||
embedding_dim: number | null;
|
||||
description: string;
|
||||
size_bytes: number;
|
||||
size_gb: number;
|
||||
}>;
|
||||
count: number;
|
||||
}>>;
|
||||
pullOllamaModel: (modelName: string, baseUrl?: string) => Promise<IPCResult<{
|
||||
model: string;
|
||||
status: 'completed' | 'failed';
|
||||
output: string[];
|
||||
}>>;
|
||||
}
|
||||
|
||||
declare global {
|
||||
|
||||
@@ -139,33 +139,23 @@ export interface ConventionsInfo {
|
||||
export interface GraphitiMemoryStatus {
|
||||
enabled: boolean;
|
||||
available: boolean;
|
||||
host?: string;
|
||||
port?: number;
|
||||
database?: string;
|
||||
dbPath?: string;
|
||||
reason?: string;
|
||||
}
|
||||
|
||||
// Docker & Infrastructure Types
|
||||
export interface DockerStatus {
|
||||
installed: boolean;
|
||||
running: boolean;
|
||||
version?: string;
|
||||
error?: string;
|
||||
}
|
||||
|
||||
export interface FalkorDBStatus {
|
||||
containerExists: boolean;
|
||||
containerRunning: boolean;
|
||||
containerName: string;
|
||||
port: number;
|
||||
healthy: boolean;
|
||||
// Memory Infrastructure Types
|
||||
export interface MemoryDatabaseStatus {
|
||||
kuzuInstalled: boolean;
|
||||
databasePath: string;
|
||||
databaseExists: boolean;
|
||||
databases: string[];
|
||||
error?: string;
|
||||
}
|
||||
|
||||
export interface InfrastructureStatus {
|
||||
docker: DockerStatus;
|
||||
falkordb: FalkorDBStatus;
|
||||
ready: boolean; // True if both Docker is running and FalkorDB is healthy
|
||||
memory: MemoryDatabaseStatus;
|
||||
ready: boolean; // True if memory database is available
|
||||
}
|
||||
|
||||
// Graphiti Validation Types
|
||||
@@ -180,71 +170,50 @@ export interface GraphitiValidationResult {
|
||||
}
|
||||
|
||||
export interface GraphitiConnectionTestResult {
|
||||
falkordb: GraphitiValidationResult;
|
||||
openai: GraphitiValidationResult;
|
||||
database: GraphitiValidationResult;
|
||||
llmProvider: GraphitiValidationResult;
|
||||
ready: boolean;
|
||||
}
|
||||
|
||||
// Graphiti Provider Types (Memory System V2)
|
||||
// LLM Providers: OpenAI, Anthropic, Azure OpenAI, Ollama (local), Google, Groq
|
||||
export type GraphitiLLMProvider = 'openai' | 'anthropic' | 'azure_openai' | 'ollama' | 'google' | 'groq';
|
||||
// Embedding Providers: OpenAI, Voyage AI, Azure OpenAI, Ollama (local), Google, HuggingFace
|
||||
export type GraphitiEmbeddingProvider = 'openai' | 'voyage' | 'azure_openai' | 'ollama' | 'google' | 'huggingface';
|
||||
// Memory Provider Types
|
||||
// Embedding Providers: OpenAI, Voyage AI, Azure OpenAI, Ollama (local), Google
|
||||
// Note: LLM provider removed - Claude SDK handles RAG queries
|
||||
export type GraphitiEmbeddingProvider = 'openai' | 'voyage' | 'azure_openai' | 'ollama' | 'google';
|
||||
|
||||
// Legacy type alias for backward compatibility
|
||||
// Legacy type aliases for backward compatibility
|
||||
export type GraphitiLLMProvider = 'openai' | 'anthropic' | 'azure_openai' | 'ollama' | 'google' | 'groq';
|
||||
export type GraphitiProviderType = GraphitiLLMProvider;
|
||||
|
||||
export interface GraphitiProviderConfig {
|
||||
// LLM Provider
|
||||
llmProvider: GraphitiLLMProvider;
|
||||
llmModel?: string; // Model name, uses provider default if not specified
|
||||
|
||||
// Embedding Provider
|
||||
// Embedding Provider (LLM provider removed - Claude SDK handles RAG)
|
||||
embeddingProvider: GraphitiEmbeddingProvider;
|
||||
embeddingModel?: string; // Embedding model, uses provider default if not specified
|
||||
|
||||
// OpenAI settings
|
||||
// OpenAI Embeddings
|
||||
openaiApiKey?: string;
|
||||
openaiModel?: string;
|
||||
openaiEmbeddingModel?: string;
|
||||
|
||||
// Anthropic settings (LLM only - needs separate embedder)
|
||||
anthropicApiKey?: string;
|
||||
anthropicModel?: string;
|
||||
|
||||
// Azure OpenAI settings
|
||||
// Azure OpenAI Embeddings
|
||||
azureOpenaiApiKey?: string;
|
||||
azureOpenaiBaseUrl?: string;
|
||||
azureOpenaiLlmDeployment?: string;
|
||||
azureOpenaiEmbeddingDeployment?: string;
|
||||
|
||||
// Voyage AI settings (embeddings only - commonly used with Anthropic)
|
||||
// Voyage AI Embeddings
|
||||
voyageApiKey?: string;
|
||||
voyageEmbeddingModel?: string;
|
||||
|
||||
// Google AI settings (LLM and embeddings)
|
||||
// Google AI Embeddings
|
||||
googleApiKey?: string;
|
||||
googleLlmModel?: string;
|
||||
googleEmbeddingModel?: string;
|
||||
|
||||
// Ollama settings (local LLM, no API key required)
|
||||
// Ollama Embeddings (local, no API key required)
|
||||
ollamaBaseUrl?: string; // Default: http://localhost:11434
|
||||
ollamaLlmModel?: string;
|
||||
ollamaEmbeddingModel?: string;
|
||||
ollamaEmbeddingDim?: number;
|
||||
|
||||
// Groq settings
|
||||
groqApiKey?: string;
|
||||
groqModel?: string;
|
||||
|
||||
// HuggingFace settings (embeddings only)
|
||||
huggingfaceApiKey?: string;
|
||||
huggingfaceEmbeddingModel?: string;
|
||||
|
||||
// FalkorDB connection (required for all providers)
|
||||
falkorDbHost?: string;
|
||||
falkorDbPort?: number;
|
||||
falkorDbPassword?: string;
|
||||
// LadybugDB settings (embedded database - no Docker required)
|
||||
database?: string; // Database name (default: auto_claude_memory)
|
||||
dbPath?: string; // Database storage path (default: ~/.auto-claude/memories)
|
||||
}
|
||||
|
||||
export interface GraphitiProviderInfo {
|
||||
@@ -268,7 +237,7 @@ export interface GraphitiMemoryState {
|
||||
|
||||
export interface MemoryEpisode {
|
||||
id: string;
|
||||
type: 'session_insight' | 'codebase_discovery' | 'codebase_map' | 'pattern' | 'gotcha';
|
||||
type: 'session_insight' | 'codebase_discovery' | 'codebase_map' | 'pattern' | 'gotcha' | 'task_outcome';
|
||||
timestamp: string;
|
||||
content: string;
|
||||
session_number?: number;
|
||||
@@ -318,16 +287,15 @@ export interface ProjectEnvConfig {
|
||||
defaultBranch?: string; // Base branch for worktree creation (e.g., 'main', 'develop')
|
||||
|
||||
// Graphiti Memory Integration (V2 - Multi-provider support)
|
||||
// Uses LadybugDB embedded database (no Docker required, Python 3.12+)
|
||||
graphitiEnabled: boolean;
|
||||
graphitiProviderConfig?: GraphitiProviderConfig; // New V2 provider configuration
|
||||
graphitiProviderConfig?: GraphitiProviderConfig; // Provider configuration
|
||||
// Legacy fields (still supported for backward compatibility)
|
||||
openaiApiKey?: string;
|
||||
// Indicates if the OpenAI key is from global settings (not project-specific)
|
||||
openaiKeyIsGlobal?: boolean;
|
||||
graphitiFalkorDbHost?: string;
|
||||
graphitiFalkorDbPort?: number;
|
||||
graphitiFalkorDbPassword?: string;
|
||||
graphitiDatabase?: string;
|
||||
graphitiDbPath?: string;
|
||||
|
||||
// UI Settings
|
||||
enableFancyUi: boolean;
|
||||
|
||||
+23
-72
@@ -117,25 +117,36 @@
|
||||
# ELECTRON_DEBUG_PORT=9222
|
||||
|
||||
# =============================================================================
|
||||
# GRAPHITI MEMORY INTEGRATION V2 (OPTIONAL)
|
||||
# GRAPHITI MEMORY INTEGRATION (OPTIONAL)
|
||||
# =============================================================================
|
||||
# Enable Graphiti-based persistent memory layer for cross-session context
|
||||
# retention. Uses FalkorDB as the graph database backend.
|
||||
# retention. Uses LadybugDB as the embedded graph database.
|
||||
#
|
||||
# V2 supports multiple LLM and embedder providers:
|
||||
# REQUIREMENTS:
|
||||
# - Python 3.12 or higher
|
||||
# - Install: pip install real_ladybug graphiti-core
|
||||
#
|
||||
# Supports multiple LLM and embedder providers:
|
||||
# - OpenAI (default)
|
||||
# - Anthropic (LLM only, use with Voyage for embeddings)
|
||||
# - Azure OpenAI
|
||||
# - Ollama (local, fully offline)
|
||||
#
|
||||
# Prerequisites:
|
||||
# 1. Start FalkorDB: docker-compose up -d falkordb
|
||||
# 2. Install Graphiti: pip install graphiti-core[falkordb]
|
||||
# 3. Configure your chosen provider (see below)
|
||||
# - Google AI (Gemini)
|
||||
|
||||
# Enable Graphiti integration (default: false)
|
||||
# GRAPHITI_ENABLED=true
|
||||
|
||||
# =============================================================================
|
||||
# GRAPHITI: Database Settings
|
||||
# =============================================================================
|
||||
# LadybugDB stores data in a local directory (no Docker required).
|
||||
|
||||
# Database name (default: auto_claude_memory)
|
||||
# GRAPHITI_DATABASE=auto_claude_memory
|
||||
|
||||
# Database storage path (default: ~/.auto-claude/memories)
|
||||
# GRAPHITI_DB_PATH=~/.auto-claude/memories
|
||||
|
||||
# =============================================================================
|
||||
# GRAPHITI: Provider Selection
|
||||
# =============================================================================
|
||||
@@ -157,8 +168,8 @@
|
||||
# OpenAI API Key
|
||||
# OPENAI_API_KEY=sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
|
||||
|
||||
# OpenAI Model for LLM (default: gpt-5-mini)
|
||||
# OPENAI_MODEL=gpt-5-mini
|
||||
# OpenAI Model for LLM (default: gpt-4o-mini)
|
||||
# OPENAI_MODEL=gpt-4o-mini
|
||||
|
||||
# OpenAI Model for embeddings (default: text-embedding-3-small)
|
||||
# Available: text-embedding-3-small (1536 dim), text-embedding-3-large (3072 dim)
|
||||
@@ -225,7 +236,7 @@
|
||||
# AZURE_OPENAI_BASE_URL=https://your-resource.openai.azure.com/openai/deployments/your-deployment
|
||||
|
||||
# Azure OpenAI Deployment Names
|
||||
# AZURE_OPENAI_LLM_DEPLOYMENT=gpt-5
|
||||
# AZURE_OPENAI_LLM_DEPLOYMENT=gpt-4
|
||||
# AZURE_OPENAI_EMBEDDING_DEPLOYMENT=text-embedding-3-small
|
||||
|
||||
# =============================================================================
|
||||
@@ -257,66 +268,6 @@
|
||||
# Common values: nomic-embed-text=768, mxbai-embed-large=1024, all-minilm=384
|
||||
# OLLAMA_EMBEDDING_DIM=768
|
||||
|
||||
# =============================================================================
|
||||
# GRAPHITI MCP SERVER (OPTIONAL - Agent-accessible knowledge graph)
|
||||
# =============================================================================
|
||||
# Enable Graphiti MCP server for direct agent access to knowledge graph.
|
||||
# This is SEPARATE from GRAPHITI_ENABLED (Python library integration).
|
||||
#
|
||||
# With MCP server enabled, Claude agents can directly:
|
||||
# - Search for relevant context (search_nodes, search_facts)
|
||||
# - Add discoveries to the graph (add_episode)
|
||||
# - Retrieve recent sessions (get_episodes)
|
||||
#
|
||||
# Quick Start (uses docker-compose.yml in project root):
|
||||
# 1. Set OPENAI_API_KEY in your .env file (or export it)
|
||||
# 2. Run: docker-compose up -d
|
||||
# 3. Set GRAPHITI_MCP_URL below
|
||||
#
|
||||
# Manual Docker (alternative):
|
||||
# docker run -d -p 8000:8000 \
|
||||
# -e DATABASE_TYPE=falkordb \
|
||||
# -e OPENAI_API_KEY=$OPENAI_API_KEY \
|
||||
# -e FALKORDB_URI=redis://host.docker.internal:6379 \
|
||||
# falkordb/graphiti-knowledge-graph-mcp
|
||||
#
|
||||
# See: https://docs.falkordb.com/agentic-memory/graphiti-mcp-server.html
|
||||
|
||||
# Graphiti MCP Server URL (setting this enables MCP integration)
|
||||
# GRAPHITI_MCP_URL=http://localhost:8000/mcp/
|
||||
|
||||
# Graphiti MCP Server Port (for docker-compose, default: 8000)
|
||||
# GRAPHITI_MCP_PORT=8000
|
||||
|
||||
# Optional: Override model settings for MCP server
|
||||
# GRAPHITI_MODEL_NAME=gpt-4o-mini
|
||||
# GRAPHITI_EMBEDDING_MODEL=text-embedding-3-small
|
||||
|
||||
# =============================================================================
|
||||
# GRAPHITI: FalkorDB Connection Settings
|
||||
# =============================================================================
|
||||
# Configure the FalkorDB graph database connection.
|
||||
# Start FalkorDB: docker run -p 6379:6379 falkordb/falkordb:latest
|
||||
|
||||
# FalkorDB Host (default: localhost)
|
||||
# GRAPHITI_FALKORDB_HOST=localhost
|
||||
|
||||
# FalkorDB Port (default: 6379)
|
||||
# GRAPHITI_FALKORDB_PORT=6379
|
||||
|
||||
# FalkorDB Password (default: empty)
|
||||
# GRAPHITI_FALKORDB_PASSWORD=
|
||||
|
||||
# Graph Database Name (default: auto_claude_memory)
|
||||
# GRAPHITI_DATABASE=auto_claude_memory
|
||||
|
||||
# =============================================================================
|
||||
# GRAPHITI: Miscellaneous Settings
|
||||
# =============================================================================
|
||||
|
||||
# Disable Graphiti telemetry (default: true)
|
||||
# GRAPHITI_TELEMETRY_ENABLED=false
|
||||
|
||||
# =============================================================================
|
||||
# GRAPHITI: Example Configurations
|
||||
# =============================================================================
|
||||
@@ -348,7 +299,7 @@
|
||||
# GRAPHITI_EMBEDDER_PROVIDER=azure_openai
|
||||
# AZURE_OPENAI_API_KEY=xxxxxxxx
|
||||
# AZURE_OPENAI_BASE_URL=https://your-resource.openai.azure.com/...
|
||||
# AZURE_OPENAI_LLM_DEPLOYMENT=gpt-5
|
||||
# AZURE_OPENAI_LLM_DEPLOYMENT=gpt-4
|
||||
# AZURE_OPENAI_EMBEDDING_DEPLOYMENT=text-embedding-3-small
|
||||
#
|
||||
# --- Example 5: Google AI (Gemini) ---
|
||||
|
||||
@@ -5,6 +5,7 @@
|
||||
|
||||
# Virtual environment
|
||||
.venv/
|
||||
.venv*/
|
||||
venv/
|
||||
env/
|
||||
|
||||
|
||||
@@ -311,7 +311,9 @@ def main() -> None:
|
||||
if args.merge_preview:
|
||||
from cli.workspace_commands import handle_merge_preview_command
|
||||
|
||||
result = handle_merge_preview_command(project_dir, spec_dir.name)
|
||||
result = handle_merge_preview_command(
|
||||
project_dir, spec_dir.name, base_branch=args.base_branch
|
||||
)
|
||||
# Output as JSON for the UI to parse
|
||||
import json
|
||||
|
||||
@@ -320,7 +322,10 @@ def main() -> None:
|
||||
|
||||
if args.merge:
|
||||
success = handle_merge_command(
|
||||
project_dir, spec_dir.name, no_commit=args.no_commit
|
||||
project_dir,
|
||||
spec_dir.name,
|
||||
no_commit=args.no_commit,
|
||||
base_branch=args.base_branch,
|
||||
)
|
||||
if not success:
|
||||
sys.exit(1)
|
||||
|
||||
@@ -145,7 +145,8 @@ def validate_environment(spec_dir: Path) -> bool:
|
||||
if graphiti_status["available"]:
|
||||
print("Graphiti memory: ENABLED")
|
||||
print(f" Database: {graphiti_status['database']}")
|
||||
print(f" Host: {graphiti_status['host']}:{graphiti_status['port']}")
|
||||
if graphiti_status.get("db_path"):
|
||||
print(f" Path: {graphiti_status['db_path']}")
|
||||
elif graphiti_status["enabled"]:
|
||||
print(
|
||||
f"Graphiti memory: CONFIGURED but unavailable ({graphiti_status['reason']})"
|
||||
|
||||
@@ -176,7 +176,10 @@ MODULE = "cli.workspace_commands"
|
||||
|
||||
|
||||
def handle_merge_command(
|
||||
project_dir: Path, spec_name: str, no_commit: bool = False
|
||||
project_dir: Path,
|
||||
spec_name: str,
|
||||
no_commit: bool = False,
|
||||
base_branch: str | None = None,
|
||||
) -> bool:
|
||||
"""
|
||||
Handle the --merge command.
|
||||
@@ -185,11 +188,14 @@ def handle_merge_command(
|
||||
project_dir: Project root directory
|
||||
spec_name: Name of the spec
|
||||
no_commit: If True, stage changes but don't commit
|
||||
base_branch: Branch to compare against (default: auto-detect)
|
||||
|
||||
Returns:
|
||||
True if merge succeeded, False otherwise
|
||||
"""
|
||||
success = merge_existing_build(project_dir, spec_name, no_commit=no_commit)
|
||||
success = merge_existing_build(
|
||||
project_dir, spec_name, no_commit=no_commit, base_branch=base_branch
|
||||
)
|
||||
|
||||
# Generate commit message suggestion if staging succeeded (no_commit mode)
|
||||
if success and no_commit:
|
||||
@@ -508,7 +514,11 @@ def _check_git_merge_conflicts(project_dir: Path, spec_name: str) -> dict:
|
||||
return result
|
||||
|
||||
|
||||
def handle_merge_preview_command(project_dir: Path, spec_name: str) -> dict:
|
||||
def handle_merge_preview_command(
|
||||
project_dir: Path,
|
||||
spec_name: str,
|
||||
base_branch: str | None = None,
|
||||
) -> dict:
|
||||
"""
|
||||
Handle the --merge-preview command.
|
||||
|
||||
@@ -523,6 +533,7 @@ def handle_merge_preview_command(project_dir: Path, spec_name: str) -> dict:
|
||||
Args:
|
||||
project_dir: Project root directory
|
||||
spec_name: Name of the spec
|
||||
base_branch: Branch the task was created from (for comparison). If None, auto-detect.
|
||||
|
||||
Returns:
|
||||
Dictionary with preview information
|
||||
@@ -565,15 +576,20 @@ def handle_merge_preview_command(project_dir: Path, spec_name: str) -> dict:
|
||||
# First, check for git-level conflicts (diverged branches)
|
||||
git_conflicts = _check_git_merge_conflicts(project_dir, spec_name)
|
||||
|
||||
# Determine the task's source branch (where the task was created from)
|
||||
# Use provided base_branch (from task metadata), or fall back to detected default
|
||||
task_source_branch = base_branch
|
||||
if not task_source_branch:
|
||||
# Auto-detect the default branch (main/master) that worktrees are typically created from
|
||||
task_source_branch = _detect_default_branch(project_dir)
|
||||
|
||||
# Get actual changed files from git diff (this is the authoritative count)
|
||||
# Detect the default branch (main/master) that worktrees are created from
|
||||
# Note: git_conflicts["base_branch"] is the current project branch, not the
|
||||
# worktree base, so we detect the default branch separately
|
||||
default_branch = _detect_default_branch(project_dir)
|
||||
all_changed_files = _get_changed_files_from_git(worktree_path, default_branch)
|
||||
all_changed_files = _get_changed_files_from_git(
|
||||
worktree_path, task_source_branch
|
||||
)
|
||||
debug(
|
||||
MODULE,
|
||||
f"Git diff against '{default_branch}' shows {len(all_changed_files)} changed files",
|
||||
f"Git diff against '{task_source_branch}' shows {len(all_changed_files)} changed files",
|
||||
changed_files=all_changed_files[:10], # Log first 10
|
||||
)
|
||||
|
||||
@@ -587,8 +603,15 @@ def handle_merge_preview_command(project_dir: Path, spec_name: str) -> dict:
|
||||
)
|
||||
|
||||
# Refresh evolution data from the worktree
|
||||
debug(MODULE, f"Refreshing evolution data from worktree: {worktree_path}")
|
||||
orchestrator.evolution_tracker.refresh_from_git(spec_name, worktree_path)
|
||||
# Compare against the task's source branch (where the task was created from)
|
||||
debug(
|
||||
MODULE,
|
||||
f"Refreshing evolution data from worktree: {worktree_path}",
|
||||
task_source_branch=task_source_branch,
|
||||
)
|
||||
orchestrator.evolution_tracker.refresh_from_git(
|
||||
spec_name, worktree_path, target_branch=task_source_branch
|
||||
)
|
||||
|
||||
# Get merge preview (semantic conflicts between parallel tasks)
|
||||
debug(MODULE, "Generating merge preview...")
|
||||
|
||||
@@ -134,6 +134,7 @@ def merge_existing_build(
|
||||
spec_name: str,
|
||||
no_commit: bool = False,
|
||||
use_smart_merge: bool = True,
|
||||
base_branch: str | None = None,
|
||||
) -> bool:
|
||||
"""
|
||||
Merge an existing build into the project using intent-aware merge.
|
||||
@@ -152,6 +153,7 @@ def merge_existing_build(
|
||||
spec_name: Name of the spec
|
||||
no_commit: If True, merge changes but don't commit (stage only for review in IDE)
|
||||
use_smart_merge: If True, use intent-aware merge (default True)
|
||||
base_branch: The branch the task was created from (for comparison). If None, auto-detect.
|
||||
|
||||
Returns:
|
||||
True if merge succeeded
|
||||
@@ -217,7 +219,12 @@ def merge_existing_build(
|
||||
# Try smart merge first if enabled
|
||||
if use_smart_merge:
|
||||
smart_result = _try_smart_merge(
|
||||
project_dir, spec_name, worktree_path, manager, no_commit=no_commit
|
||||
project_dir,
|
||||
spec_name,
|
||||
worktree_path,
|
||||
manager,
|
||||
no_commit=no_commit,
|
||||
task_source_branch=base_branch,
|
||||
)
|
||||
|
||||
if smart_result is not None:
|
||||
@@ -313,6 +320,7 @@ def _try_smart_merge(
|
||||
worktree_path: Path,
|
||||
manager: WorktreeManager,
|
||||
no_commit: bool = False,
|
||||
task_source_branch: str | None = None,
|
||||
) -> dict | None:
|
||||
"""
|
||||
Try to use the intent-aware merge system.
|
||||
@@ -322,6 +330,10 @@ def _try_smart_merge(
|
||||
|
||||
Uses a lock file to prevent concurrent merges for the same spec.
|
||||
|
||||
Args:
|
||||
task_source_branch: The branch the task was created from (for comparison).
|
||||
If None, auto-detect.
|
||||
|
||||
Returns:
|
||||
Dict with results, or None if smart merge not applicable
|
||||
"""
|
||||
@@ -329,7 +341,12 @@ def _try_smart_merge(
|
||||
try:
|
||||
with MergeLock(project_dir, spec_name):
|
||||
return _try_smart_merge_inner(
|
||||
project_dir, spec_name, worktree_path, manager, no_commit
|
||||
project_dir,
|
||||
spec_name,
|
||||
worktree_path,
|
||||
manager,
|
||||
no_commit,
|
||||
task_source_branch=task_source_branch,
|
||||
)
|
||||
except MergeLockError as e:
|
||||
print(warning(f" {e}"))
|
||||
@@ -346,6 +363,7 @@ def _try_smart_merge_inner(
|
||||
worktree_path: Path,
|
||||
manager: WorktreeManager,
|
||||
no_commit: bool = False,
|
||||
task_source_branch: str | None = None,
|
||||
) -> dict | None:
|
||||
"""Inner implementation of smart merge (called with lock held)."""
|
||||
debug(
|
||||
@@ -381,8 +399,17 @@ def _try_smart_merge_inner(
|
||||
)
|
||||
|
||||
# Refresh evolution data from the worktree
|
||||
debug(MODULE, "Refreshing evolution data from git", spec_name=spec_name)
|
||||
orchestrator.evolution_tracker.refresh_from_git(spec_name, worktree_path)
|
||||
# Use task_source_branch (where task branched from) for comparing what files changed
|
||||
# If not provided, auto-detection will find main/master
|
||||
debug(
|
||||
MODULE,
|
||||
"Refreshing evolution data from git",
|
||||
spec_name=spec_name,
|
||||
task_source_branch=task_source_branch,
|
||||
)
|
||||
orchestrator.evolution_tracker.refresh_from_git(
|
||||
spec_name, worktree_path, target_branch=task_source_branch
|
||||
)
|
||||
|
||||
# Check for git-level conflicts first (branch divergence)
|
||||
debug(MODULE, "Checking for git-level conflicts")
|
||||
|
||||
@@ -5,15 +5,21 @@ Graphiti Integration Configuration
|
||||
Constants, status mappings, and configuration helpers for Graphiti memory integration.
|
||||
Follows the same patterns as linear_config.py for consistency.
|
||||
|
||||
Uses LadybugDB as the embedded graph database (no Docker required, requires Python 3.12+).
|
||||
|
||||
Multi-Provider Support (V2):
|
||||
- LLM Providers: OpenAI, Anthropic, Azure OpenAI, Ollama
|
||||
- Embedder Providers: OpenAI, Voyage AI, Azure OpenAI, Ollama
|
||||
- LLM Providers: OpenAI, Anthropic, Azure OpenAI, Ollama, Google AI
|
||||
- Embedder Providers: OpenAI, Voyage AI, Azure OpenAI, Ollama, Google AI
|
||||
|
||||
Environment Variables:
|
||||
# Core
|
||||
GRAPHITI_ENABLED: Set to "true" to enable Graphiti integration
|
||||
GRAPHITI_LLM_PROVIDER: openai|anthropic|azure_openai|ollama (default: openai)
|
||||
GRAPHITI_EMBEDDER_PROVIDER: openai|voyage|azure_openai|ollama (default: openai)
|
||||
GRAPHITI_LLM_PROVIDER: openai|anthropic|azure_openai|ollama|google (default: openai)
|
||||
GRAPHITI_EMBEDDER_PROVIDER: openai|voyage|azure_openai|ollama|google (default: openai)
|
||||
|
||||
# Database
|
||||
GRAPHITI_DATABASE: Graph database name (default: auto_claude_memory)
|
||||
GRAPHITI_DB_PATH: Database storage path (default: ~/.auto-claude/memories)
|
||||
|
||||
# OpenAI
|
||||
OPENAI_API_KEY: Required for OpenAI provider
|
||||
@@ -22,7 +28,7 @@ Environment Variables:
|
||||
|
||||
# Anthropic (LLM only - needs separate embedder)
|
||||
ANTHROPIC_API_KEY: Required for Anthropic provider
|
||||
GRAPHITI_ANTHROPIC_MODEL: Model for LLM (default: claude-sonnet-4-5-latest)
|
||||
GRAPHITI_ANTHROPIC_MODEL: Model for LLM (default: claude-sonnet-4-5)
|
||||
|
||||
# Azure OpenAI
|
||||
AZURE_OPENAI_API_KEY: Required for Azure provider
|
||||
@@ -34,18 +40,19 @@ Environment Variables:
|
||||
VOYAGE_API_KEY: Required for Voyage embedder
|
||||
VOYAGE_EMBEDDING_MODEL: Model (default: voyage-3)
|
||||
|
||||
# Google AI
|
||||
GOOGLE_API_KEY: Required for Google provider
|
||||
GOOGLE_LLM_MODEL: Model for LLM (default: gemini-2.0-flash)
|
||||
GOOGLE_EMBEDDING_MODEL: Model for embeddings (default: text-embedding-004)
|
||||
|
||||
# Ollama (local)
|
||||
OLLAMA_BASE_URL: Ollama server URL (default: http://localhost:11434)
|
||||
OLLAMA_LLM_MODEL: Model for LLM (e.g., deepseek-r1:7b)
|
||||
OLLAMA_EMBEDDING_MODEL: Model for embeddings (e.g., nomic-embed-text)
|
||||
OLLAMA_EMBEDDING_DIM: Embedding dimension (required for Ollama, e.g., 768)
|
||||
|
||||
# FalkorDB
|
||||
GRAPHITI_FALKORDB_HOST: FalkorDB host (default: localhost)
|
||||
GRAPHITI_FALKORDB_PORT: FalkorDB port (default: 6380)
|
||||
GRAPHITI_FALKORDB_PASSWORD: FalkorDB password (default: empty)
|
||||
GRAPHITI_DATABASE: Graph database name (default: auto_claude_memory)
|
||||
GRAPHITI_TELEMETRY_ENABLED: Set to "false" to disable telemetry (default: true)
|
||||
OLLAMA_EMBEDDING_MODEL: Model for embeddings. Supported models with auto-detected dimensions:
|
||||
- embeddinggemma (768) - Google's lightweight embedding model
|
||||
- qwen3-embedding:0.6b (1024), :4b (2560), :8b (4096) - Qwen3 series
|
||||
- nomic-embed-text (768), mxbai-embed-large (1024), bge-large (1024)
|
||||
OLLAMA_EMBEDDING_DIM: Override dimension (optional if using known model)
|
||||
"""
|
||||
|
||||
import json
|
||||
@@ -57,9 +64,8 @@ from pathlib import Path
|
||||
from typing import Optional
|
||||
|
||||
# Default configuration values
|
||||
DEFAULT_FALKORDB_HOST = "localhost"
|
||||
DEFAULT_FALKORDB_PORT = 6380
|
||||
DEFAULT_DATABASE = "auto_claude_memory"
|
||||
DEFAULT_DB_PATH = "~/.auto-claude/memories"
|
||||
DEFAULT_OLLAMA_BASE_URL = "http://localhost:11434"
|
||||
|
||||
# Graphiti state marker file (stores connection info and status)
|
||||
@@ -97,19 +103,19 @@ class EmbedderProvider(str, Enum):
|
||||
|
||||
@dataclass
|
||||
class GraphitiConfig:
|
||||
"""Configuration for Graphiti memory integration with multi-provider support."""
|
||||
"""Configuration for Graphiti memory integration with multi-provider support.
|
||||
|
||||
Uses LadybugDB as the embedded graph database (no Docker required, requires Python 3.12+).
|
||||
"""
|
||||
|
||||
# Core settings
|
||||
enabled: bool = False
|
||||
llm_provider: str = "openai"
|
||||
embedder_provider: str = "openai"
|
||||
|
||||
# FalkorDB connection
|
||||
falkordb_host: str = DEFAULT_FALKORDB_HOST
|
||||
falkordb_port: int = DEFAULT_FALKORDB_PORT
|
||||
falkordb_password: str = ""
|
||||
# Database settings (LadybugDB - embedded, no Docker required)
|
||||
database: str = DEFAULT_DATABASE
|
||||
telemetry_enabled: bool = True
|
||||
db_path: str = DEFAULT_DB_PATH
|
||||
|
||||
# OpenAI settings
|
||||
openai_api_key: str = ""
|
||||
@@ -118,7 +124,7 @@ class GraphitiConfig:
|
||||
|
||||
# Anthropic settings (LLM only)
|
||||
anthropic_api_key: str = ""
|
||||
anthropic_model: str = "claude-sonnet-4-5-latest"
|
||||
anthropic_model: str = "claude-sonnet-4-5"
|
||||
|
||||
# Azure OpenAI settings
|
||||
azure_openai_api_key: str = ""
|
||||
@@ -154,22 +160,9 @@ class GraphitiConfig:
|
||||
"GRAPHITI_EMBEDDER_PROVIDER", "openai"
|
||||
).lower()
|
||||
|
||||
# FalkorDB connection settings
|
||||
falkordb_host = os.environ.get("GRAPHITI_FALKORDB_HOST", DEFAULT_FALKORDB_HOST)
|
||||
|
||||
try:
|
||||
falkordb_port = int(
|
||||
os.environ.get("GRAPHITI_FALKORDB_PORT", str(DEFAULT_FALKORDB_PORT))
|
||||
)
|
||||
except ValueError:
|
||||
falkordb_port = DEFAULT_FALKORDB_PORT
|
||||
|
||||
falkordb_password = os.environ.get("GRAPHITI_FALKORDB_PASSWORD", "")
|
||||
# Database settings (LadybugDB - embedded)
|
||||
database = os.environ.get("GRAPHITI_DATABASE", DEFAULT_DATABASE)
|
||||
|
||||
# Telemetry setting
|
||||
telemetry_str = os.environ.get("GRAPHITI_TELEMETRY_ENABLED", "true").lower()
|
||||
telemetry_enabled = telemetry_str not in ("false", "0", "no")
|
||||
db_path = os.environ.get("GRAPHITI_DB_PATH", DEFAULT_DB_PATH)
|
||||
|
||||
# OpenAI settings
|
||||
openai_api_key = os.environ.get("OPENAI_API_KEY", "")
|
||||
@@ -181,7 +174,7 @@ class GraphitiConfig:
|
||||
# Anthropic settings
|
||||
anthropic_api_key = os.environ.get("ANTHROPIC_API_KEY", "")
|
||||
anthropic_model = os.environ.get(
|
||||
"GRAPHITI_ANTHROPIC_MODEL", "claude-sonnet-4-5-latest"
|
||||
"GRAPHITI_ANTHROPIC_MODEL", "claude-sonnet-4-5"
|
||||
)
|
||||
|
||||
# Azure OpenAI settings
|
||||
@@ -218,11 +211,8 @@ class GraphitiConfig:
|
||||
enabled=enabled,
|
||||
llm_provider=llm_provider,
|
||||
embedder_provider=embedder_provider,
|
||||
falkordb_host=falkordb_host,
|
||||
falkordb_port=falkordb_port,
|
||||
falkordb_password=falkordb_password,
|
||||
database=database,
|
||||
telemetry_enabled=telemetry_enabled,
|
||||
db_path=db_path,
|
||||
openai_api_key=openai_api_key,
|
||||
openai_model=openai_model,
|
||||
openai_embedding_model=openai_embedding_model,
|
||||
@@ -249,40 +239,17 @@ class GraphitiConfig:
|
||||
|
||||
Returns True if:
|
||||
- GRAPHITI_ENABLED is true
|
||||
- LLM provider is configured correctly
|
||||
- Embedder provider is configured correctly
|
||||
- Embedder provider is configured (optional - keyword search works without)
|
||||
|
||||
Note: LLM provider is no longer required - Claude Agent SDK handles RAG queries.
|
||||
"""
|
||||
if not self.enabled:
|
||||
return False
|
||||
|
||||
# Validate LLM provider
|
||||
if not self._validate_llm_provider():
|
||||
return False
|
||||
|
||||
# Validate embedder provider
|
||||
if not self._validate_embedder_provider():
|
||||
return False
|
||||
|
||||
# Embedder validation is optional - memory works with keyword search fallback
|
||||
# Return True if enabled, embedder config is a bonus for semantic search
|
||||
return True
|
||||
|
||||
def _validate_llm_provider(self) -> bool:
|
||||
"""Validate LLM provider configuration."""
|
||||
if self.llm_provider == "openai":
|
||||
return bool(self.openai_api_key)
|
||||
elif self.llm_provider == "anthropic":
|
||||
return bool(self.anthropic_api_key)
|
||||
elif self.llm_provider == "azure_openai":
|
||||
return bool(
|
||||
self.azure_openai_api_key
|
||||
and self.azure_openai_base_url
|
||||
and self.azure_openai_llm_deployment
|
||||
)
|
||||
elif self.llm_provider == "ollama":
|
||||
return bool(self.ollama_llm_model)
|
||||
elif self.llm_provider == "google":
|
||||
return bool(self.google_api_key)
|
||||
return False
|
||||
|
||||
def _validate_embedder_provider(self) -> bool:
|
||||
"""Validate embedder provider configuration."""
|
||||
if self.embedder_provider == "openai":
|
||||
@@ -296,7 +263,8 @@ class GraphitiConfig:
|
||||
and self.azure_openai_embedding_deployment
|
||||
)
|
||||
elif self.embedder_provider == "ollama":
|
||||
return bool(self.ollama_embedding_model and self.ollama_embedding_dim)
|
||||
# Only require model - dimension is auto-detected for known models
|
||||
return bool(self.ollama_embedding_model)
|
||||
elif self.embedder_provider == "google":
|
||||
return bool(self.google_api_key)
|
||||
return False
|
||||
@@ -309,34 +277,10 @@ class GraphitiConfig:
|
||||
errors.append("GRAPHITI_ENABLED must be set to true")
|
||||
return errors
|
||||
|
||||
# LLM provider validation
|
||||
if self.llm_provider == "openai":
|
||||
if not self.openai_api_key:
|
||||
errors.append("OpenAI LLM provider requires OPENAI_API_KEY")
|
||||
elif self.llm_provider == "anthropic":
|
||||
if not self.anthropic_api_key:
|
||||
errors.append("Anthropic LLM provider requires ANTHROPIC_API_KEY")
|
||||
elif self.llm_provider == "azure_openai":
|
||||
if not self.azure_openai_api_key:
|
||||
errors.append("Azure OpenAI LLM provider requires AZURE_OPENAI_API_KEY")
|
||||
if not self.azure_openai_base_url:
|
||||
errors.append(
|
||||
"Azure OpenAI LLM provider requires AZURE_OPENAI_BASE_URL"
|
||||
)
|
||||
if not self.azure_openai_llm_deployment:
|
||||
errors.append(
|
||||
"Azure OpenAI LLM provider requires AZURE_OPENAI_LLM_DEPLOYMENT"
|
||||
)
|
||||
elif self.llm_provider == "ollama":
|
||||
if not self.ollama_llm_model:
|
||||
errors.append("Ollama LLM provider requires OLLAMA_LLM_MODEL")
|
||||
elif self.llm_provider == "google":
|
||||
if not self.google_api_key:
|
||||
errors.append("Google LLM provider requires GOOGLE_API_KEY")
|
||||
else:
|
||||
errors.append(f"Unknown LLM provider: {self.llm_provider}")
|
||||
# Note: LLM provider validation removed - Claude Agent SDK handles RAG queries
|
||||
# Memory works with keyword search even without embedder, so embedder errors are warnings
|
||||
|
||||
# Embedder provider validation
|
||||
# Embedder provider validation (optional - keyword search works without)
|
||||
if self.embedder_provider == "openai":
|
||||
if not self.openai_api_key:
|
||||
errors.append("OpenAI embedder provider requires OPENAI_API_KEY")
|
||||
@@ -361,8 +305,7 @@ class GraphitiConfig:
|
||||
errors.append(
|
||||
"Ollama embedder provider requires OLLAMA_EMBEDDING_MODEL"
|
||||
)
|
||||
if not self.ollama_embedding_dim:
|
||||
errors.append("Ollama embedder provider requires OLLAMA_EMBEDDING_DIM")
|
||||
# Note: OLLAMA_EMBEDDING_DIM is optional - auto-detected for known models
|
||||
elif self.embedder_provider == "google":
|
||||
if not self.google_api_key:
|
||||
errors.append("Google embedder provider requires GOOGLE_API_KEY")
|
||||
@@ -371,16 +314,103 @@ class GraphitiConfig:
|
||||
|
||||
return errors
|
||||
|
||||
def get_connection_uri(self) -> str:
|
||||
"""Get the FalkorDB connection URI."""
|
||||
if self.falkordb_password:
|
||||
return f"redis://:{self.falkordb_password}@{self.falkordb_host}:{self.falkordb_port}"
|
||||
return f"redis://{self.falkordb_host}:{self.falkordb_port}"
|
||||
def get_db_path(self) -> Path:
|
||||
"""
|
||||
Get the resolved database path.
|
||||
|
||||
Expands ~ to home directory and appends the database name.
|
||||
Creates the parent directory if it doesn't exist (not the final
|
||||
database file/directory itself, which is created by the driver).
|
||||
"""
|
||||
base_path = Path(self.db_path).expanduser()
|
||||
full_path = base_path / self.database
|
||||
full_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
return full_path
|
||||
|
||||
def get_provider_summary(self) -> str:
|
||||
"""Get a summary of configured providers."""
|
||||
return f"LLM: {self.llm_provider}, Embedder: {self.embedder_provider}"
|
||||
|
||||
def get_embedding_dimension(self) -> int:
|
||||
"""
|
||||
Get the embedding dimension for the current embedder provider.
|
||||
|
||||
Returns:
|
||||
Embedding dimension (e.g., 768, 1024, 1536)
|
||||
"""
|
||||
if self.embedder_provider == "ollama":
|
||||
if self.ollama_embedding_dim > 0:
|
||||
return self.ollama_embedding_dim
|
||||
# Auto-detect for known models
|
||||
model = self.ollama_embedding_model.lower()
|
||||
if "embeddinggemma" in model or "nomic-embed-text" in model:
|
||||
return 768
|
||||
elif "mxbai" in model or "bge-large" in model:
|
||||
return 1024
|
||||
elif "qwen3" in model:
|
||||
if "0.6b" in model:
|
||||
return 1024
|
||||
elif "4b" in model:
|
||||
return 2560
|
||||
elif "8b" in model:
|
||||
return 4096
|
||||
return 768 # Default fallback
|
||||
elif self.embedder_provider == "openai":
|
||||
# OpenAI text-embedding-3-small default is 1536
|
||||
return 1536
|
||||
elif self.embedder_provider == "voyage":
|
||||
# Voyage-3 uses 1024 dimensions
|
||||
return 1024
|
||||
elif self.embedder_provider == "google":
|
||||
# Google text-embedding-004 uses 768 dimensions
|
||||
return 768
|
||||
elif self.embedder_provider == "azure_openai":
|
||||
# Depends on the deployment, default to 1536
|
||||
return 1536
|
||||
return 768 # Safe default
|
||||
|
||||
def get_provider_signature(self) -> str:
|
||||
"""
|
||||
Get a unique signature for the current embedding provider configuration.
|
||||
|
||||
Used to generate provider-specific database names to prevent mixing
|
||||
incompatible embeddings.
|
||||
|
||||
Returns:
|
||||
Provider signature string (e.g., "openai_1536", "ollama_768")
|
||||
"""
|
||||
provider = self.embedder_provider
|
||||
dim = self.get_embedding_dimension()
|
||||
|
||||
if provider == "ollama":
|
||||
# Include model name for Ollama
|
||||
model = self.ollama_embedding_model.replace(":", "_").replace(".", "_")
|
||||
return f"ollama_{model}_{dim}"
|
||||
else:
|
||||
return f"{provider}_{dim}"
|
||||
|
||||
def get_provider_specific_database_name(self, base_name: str = None) -> str:
|
||||
"""
|
||||
Get a provider-specific database name to prevent embedding dimension mismatches.
|
||||
|
||||
Args:
|
||||
base_name: Base database name (default: from config)
|
||||
|
||||
Returns:
|
||||
Database name with provider signature (e.g., "auto_claude_memory_ollama_768")
|
||||
"""
|
||||
if base_name is None:
|
||||
base_name = self.database
|
||||
|
||||
# Remove existing provider suffix if present
|
||||
for provider in ["openai", "ollama", "voyage", "google", "azure_openai"]:
|
||||
if f"_{provider}_" in base_name:
|
||||
base_name = base_name.split(f"_{provider}_")[0]
|
||||
break
|
||||
|
||||
signature = self.get_provider_signature()
|
||||
return f"{base_name}_{signature}"
|
||||
|
||||
|
||||
@dataclass
|
||||
class GraphitiState:
|
||||
@@ -454,6 +484,43 @@ class GraphitiState:
|
||||
# Keep only last 10 errors
|
||||
self.error_log = self.error_log[-10:]
|
||||
|
||||
def has_provider_changed(self, config: GraphitiConfig) -> bool:
|
||||
"""
|
||||
Check if the embedding provider has changed since initialization.
|
||||
|
||||
Args:
|
||||
config: Current GraphitiConfig
|
||||
|
||||
Returns:
|
||||
True if provider has changed (requiring migration)
|
||||
"""
|
||||
if not self.initialized or not self.embedder_provider:
|
||||
return False
|
||||
|
||||
return self.embedder_provider != config.embedder_provider
|
||||
|
||||
def get_migration_info(self, config: GraphitiConfig) -> dict:
|
||||
"""
|
||||
Get information about provider migration needs.
|
||||
|
||||
Args:
|
||||
config: Current GraphitiConfig
|
||||
|
||||
Returns:
|
||||
Dict with migration details or None if no migration needed
|
||||
"""
|
||||
if not self.has_provider_changed(config):
|
||||
return None
|
||||
|
||||
return {
|
||||
"old_provider": self.embedder_provider,
|
||||
"new_provider": config.embedder_provider,
|
||||
"old_database": self.database,
|
||||
"new_database": config.get_provider_specific_database_name(),
|
||||
"episode_count": self.episode_count,
|
||||
"requires_migration": True,
|
||||
}
|
||||
|
||||
|
||||
def is_graphiti_enabled() -> bool:
|
||||
"""
|
||||
@@ -475,9 +542,8 @@ def get_graphiti_status() -> dict:
|
||||
Dict with status information:
|
||||
- enabled: bool
|
||||
- available: bool (has required dependencies)
|
||||
- host: str
|
||||
- port: int
|
||||
- database: str
|
||||
- db_path: str
|
||||
- llm_provider: str
|
||||
- embedder_provider: str
|
||||
- reason: str (why unavailable if not available)
|
||||
@@ -488,9 +554,8 @@ def get_graphiti_status() -> dict:
|
||||
status = {
|
||||
"enabled": config.enabled,
|
||||
"available": False,
|
||||
"host": config.falkordb_host,
|
||||
"port": config.falkordb_port,
|
||||
"database": config.database,
|
||||
"db_path": config.db_path,
|
||||
"llm_provider": config.llm_provider,
|
||||
"embedder_provider": config.embedder_provider,
|
||||
"reason": "",
|
||||
@@ -501,14 +566,17 @@ def get_graphiti_status() -> dict:
|
||||
status["reason"] = "GRAPHITI_ENABLED not set to true"
|
||||
return status
|
||||
|
||||
# Get validation errors
|
||||
# Get validation errors (these are warnings, not blockers)
|
||||
errors = config.get_validation_errors()
|
||||
if errors:
|
||||
status["errors"] = errors
|
||||
status["reason"] = errors[0] # First error as primary reason
|
||||
return status
|
||||
# Errors are informational - embedder is optional (keyword search fallback)
|
||||
|
||||
# Available if is_valid() returns True (just needs enabled flag)
|
||||
status["available"] = config.is_valid()
|
||||
if not status["available"]:
|
||||
status["reason"] = errors[0] if errors else "Configuration invalid"
|
||||
|
||||
status["available"] = True
|
||||
return status
|
||||
|
||||
|
||||
|
||||
@@ -25,8 +25,15 @@ see graphiti/graphiti.py.
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
# Re-export from modular system
|
||||
from graphiti import (
|
||||
# Import config utilities
|
||||
from graphiti_config import (
|
||||
GraphitiConfig,
|
||||
is_graphiti_enabled,
|
||||
)
|
||||
|
||||
# Re-export from modular system (queries_pkg)
|
||||
from .queries_pkg.graphiti import GraphitiMemory
|
||||
from .queries_pkg.schema import (
|
||||
EPISODE_TYPE_CODEBASE_DISCOVERY,
|
||||
EPISODE_TYPE_GOTCHA,
|
||||
EPISODE_TYPE_HISTORICAL_CONTEXT,
|
||||
@@ -35,16 +42,9 @@ from graphiti import (
|
||||
EPISODE_TYPE_SESSION_INSIGHT,
|
||||
EPISODE_TYPE_TASK_OUTCOME,
|
||||
MAX_CONTEXT_RESULTS,
|
||||
GraphitiMemory,
|
||||
GroupIdMode,
|
||||
)
|
||||
|
||||
# Import config utilities
|
||||
from graphiti_config import (
|
||||
GraphitiConfig,
|
||||
is_graphiti_enabled,
|
||||
)
|
||||
|
||||
|
||||
# Convenience function for getting a memory manager
|
||||
def get_graphiti_memory(
|
||||
|
||||
@@ -0,0 +1,409 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Embedding Provider Migration Utility
|
||||
=====================================
|
||||
|
||||
Migrates Graphiti memory data from one embedding provider to another by:
|
||||
1. Reading all episodes from the source database
|
||||
2. Re-embedding content with the new provider
|
||||
3. Storing in a provider-specific target database
|
||||
|
||||
This handles the dimension mismatch issue when switching between providers
|
||||
(e.g., OpenAI 1536D → Ollama embeddinggemma 768D).
|
||||
|
||||
Usage:
|
||||
# Interactive mode (recommended)
|
||||
python integrations/graphiti/migrate_embeddings.py
|
||||
|
||||
# Automatic mode
|
||||
python integrations/graphiti/migrate_embeddings.py \
|
||||
--from-provider openai \
|
||||
--to-provider ollama \
|
||||
--auto-confirm
|
||||
|
||||
# Dry run to see what would be migrated
|
||||
python integrations/graphiti/migrate_embeddings.py --dry-run
|
||||
"""
|
||||
|
||||
import argparse
|
||||
import asyncio
|
||||
import logging
|
||||
import sys
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
|
||||
# Add auto-claude to path
|
||||
sys.path.insert(0, str(Path(__file__).parent.parent.parent))
|
||||
|
||||
from integrations.graphiti.config import GraphitiConfig
|
||||
|
||||
logging.basicConfig(
|
||||
level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
|
||||
)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class EmbeddingMigrator:
|
||||
"""Handles migration of embeddings between providers."""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
source_config: GraphitiConfig,
|
||||
target_config: GraphitiConfig,
|
||||
dry_run: bool = False,
|
||||
):
|
||||
"""
|
||||
Initialize the migrator.
|
||||
|
||||
Args:
|
||||
source_config: Config for source database
|
||||
target_config: Config for target database
|
||||
dry_run: If True, don't actually perform migration
|
||||
"""
|
||||
self.source_config = source_config
|
||||
self.target_config = target_config
|
||||
self.dry_run = dry_run
|
||||
self.source_client = None
|
||||
self.target_client = None
|
||||
|
||||
async def initialize(self) -> bool:
|
||||
"""Initialize source and target clients."""
|
||||
from integrations.graphiti.queries_pkg.client import GraphitiClient
|
||||
|
||||
logger.info("Initializing source client...")
|
||||
self.source_client = GraphitiClient(self.source_config)
|
||||
try:
|
||||
if not await self.source_client.initialize():
|
||||
logger.error("Failed to initialize source client")
|
||||
return False
|
||||
except Exception as e:
|
||||
logger.error(f"Exception initializing source client: {e}")
|
||||
return False
|
||||
|
||||
if not self.dry_run:
|
||||
logger.info("Initializing target client...")
|
||||
self.target_client = GraphitiClient(self.target_config)
|
||||
try:
|
||||
if not await self.target_client.initialize():
|
||||
logger.error("Failed to initialize target client")
|
||||
# Clean up source client on partial failure
|
||||
await self.source_client.close()
|
||||
self.source_client = None
|
||||
return False
|
||||
except Exception as e:
|
||||
logger.error(f"Exception initializing target client: {e}")
|
||||
# Clean up source client on partial failure
|
||||
await self.source_client.close()
|
||||
self.source_client = None
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
async def get_source_episodes(self) -> list[dict]:
|
||||
"""
|
||||
Retrieve all episodes from source database.
|
||||
|
||||
Returns:
|
||||
List of episode data dictionaries
|
||||
"""
|
||||
logger.info("Fetching episodes from source database...")
|
||||
|
||||
try:
|
||||
# Query all episodic nodes
|
||||
query = """
|
||||
MATCH (e:Episodic)
|
||||
RETURN
|
||||
e.uuid AS uuid,
|
||||
e.name AS name,
|
||||
e.content AS content,
|
||||
e.created_at AS created_at,
|
||||
e.valid_at AS valid_at,
|
||||
e.group_id AS group_id,
|
||||
e.source AS source,
|
||||
e.source_description AS source_description
|
||||
ORDER BY e.created_at
|
||||
"""
|
||||
|
||||
records, _, _ = await self.source_client._driver.execute_query(query)
|
||||
|
||||
episodes = []
|
||||
for record in records:
|
||||
episodes.append(
|
||||
{
|
||||
"uuid": record.get("uuid"),
|
||||
"name": record.get("name"),
|
||||
"content": record.get("content"),
|
||||
"created_at": record.get("created_at"),
|
||||
"valid_at": record.get("valid_at"),
|
||||
"group_id": record.get("group_id"),
|
||||
"source": record.get("source"),
|
||||
"source_description": record.get("source_description"),
|
||||
}
|
||||
)
|
||||
|
||||
logger.info(f"Found {len(episodes)} episodes to migrate")
|
||||
return episodes
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to fetch episodes: {e}")
|
||||
return []
|
||||
|
||||
async def migrate_episode(self, episode: dict) -> bool:
|
||||
"""
|
||||
Migrate a single episode to the target database.
|
||||
|
||||
Args:
|
||||
episode: Episode data dictionary
|
||||
|
||||
Returns:
|
||||
True if migration succeeded
|
||||
"""
|
||||
if self.dry_run:
|
||||
logger.info(f"[DRY RUN] Would migrate: {episode['name']}")
|
||||
return True
|
||||
|
||||
try:
|
||||
from graphiti_core.nodes import EpisodeType
|
||||
|
||||
# Determine episode type
|
||||
source = episode.get("source", "text")
|
||||
if source == "message":
|
||||
episode_type = EpisodeType.message
|
||||
elif source == "json":
|
||||
episode_type = EpisodeType.json
|
||||
else:
|
||||
episode_type = EpisodeType.text
|
||||
|
||||
# Parse timestamps
|
||||
valid_at = episode.get("valid_at")
|
||||
if isinstance(valid_at, str):
|
||||
valid_at = datetime.fromisoformat(valid_at.replace("Z", "+00:00"))
|
||||
|
||||
# Re-embed and save with new provider
|
||||
await self.target_client.graphiti.add_episode(
|
||||
name=episode["name"],
|
||||
episode_body=episode["content"] or "",
|
||||
source=episode_type,
|
||||
source_description=episode.get(
|
||||
"source_description", "Migrated episode"
|
||||
),
|
||||
reference_time=valid_at,
|
||||
group_id=episode.get("group_id", "default"),
|
||||
)
|
||||
|
||||
logger.info(f"Migrated: {episode['name']}")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to migrate episode {episode['name']}: {e}")
|
||||
return False
|
||||
|
||||
async def migrate_all(self) -> dict:
|
||||
"""
|
||||
Migrate all episodes from source to target.
|
||||
|
||||
Returns:
|
||||
Migration statistics dictionary
|
||||
"""
|
||||
episodes = await self.get_source_episodes()
|
||||
|
||||
stats = {
|
||||
"total": len(episodes),
|
||||
"succeeded": 0,
|
||||
"failed": 0,
|
||||
"dry_run": self.dry_run,
|
||||
}
|
||||
|
||||
for i, episode in enumerate(episodes, 1):
|
||||
logger.info(f"Processing episode {i}/{len(episodes)}")
|
||||
if await self.migrate_episode(episode):
|
||||
stats["succeeded"] += 1
|
||||
else:
|
||||
stats["failed"] += 1
|
||||
|
||||
return stats
|
||||
|
||||
async def close(self):
|
||||
"""Close client connections."""
|
||||
if self.source_client:
|
||||
await self.source_client.close()
|
||||
if self.target_client:
|
||||
await self.target_client.close()
|
||||
|
||||
|
||||
async def interactive_migration():
|
||||
"""Run interactive migration with user prompts."""
|
||||
print("\n" + "=" * 70)
|
||||
print(" GRAPHITI EMBEDDING PROVIDER MIGRATION")
|
||||
print("=" * 70 + "\n")
|
||||
|
||||
# Load current config
|
||||
current_config = GraphitiConfig.from_env()
|
||||
|
||||
print("Current Configuration:")
|
||||
print(f" Embedder Provider: {current_config.embedder_provider}")
|
||||
print(f" Embedding Dimension: {current_config.get_embedding_dimension()}")
|
||||
print(f" Database: {current_config.database}")
|
||||
print(f" Provider Signature: {current_config.get_provider_signature()}\n")
|
||||
|
||||
# Ask for source provider
|
||||
print("Which provider are you migrating FROM?")
|
||||
print(" 1. OpenAI")
|
||||
print(" 2. Ollama")
|
||||
print(" 3. Voyage AI")
|
||||
print(" 4. Google AI")
|
||||
print(" 5. Azure OpenAI")
|
||||
|
||||
source_choice = input("\nEnter choice (1-5): ").strip()
|
||||
source_map = {
|
||||
"1": "openai",
|
||||
"2": "ollama",
|
||||
"3": "voyage",
|
||||
"4": "google",
|
||||
"5": "azure_openai",
|
||||
}
|
||||
|
||||
if source_choice not in source_map:
|
||||
print("Invalid choice. Exiting.")
|
||||
return
|
||||
|
||||
source_provider = source_map[source_choice]
|
||||
|
||||
# Validate that source and target are different
|
||||
if source_provider == current_config.embedder_provider:
|
||||
print(f"\nError: Source and target providers are the same ({source_provider}).")
|
||||
print("Migration requires different providers. Exiting.")
|
||||
return
|
||||
|
||||
# Create source config with correct provider-specific database name
|
||||
source_config = GraphitiConfig.from_env()
|
||||
source_config.embedder_provider = source_provider
|
||||
# Use the source provider's signature for the database name
|
||||
source_config.database = source_config.get_provider_specific_database_name(
|
||||
"auto_claude_memory"
|
||||
)
|
||||
|
||||
print(f"\nSource: {source_provider}")
|
||||
print(f"Target: {current_config.embedder_provider}")
|
||||
print(
|
||||
f"\nThis will migrate all episodes from {source_provider} "
|
||||
f"to {current_config.embedder_provider}"
|
||||
)
|
||||
print(
|
||||
"Re-embedding may take several minutes depending on the number of episodes.\n"
|
||||
)
|
||||
|
||||
confirm = input("Continue? (yes/no): ").strip().lower()
|
||||
if confirm != "yes":
|
||||
print("Migration cancelled.")
|
||||
return
|
||||
|
||||
# Perform migration
|
||||
migrator = EmbeddingMigrator(
|
||||
source_config=source_config,
|
||||
target_config=current_config,
|
||||
dry_run=False,
|
||||
)
|
||||
|
||||
if not await migrator.initialize():
|
||||
print("Failed to initialize migration. Check configuration.")
|
||||
return
|
||||
|
||||
print("\nMigrating episodes...")
|
||||
stats = await migrator.migrate_all()
|
||||
|
||||
await migrator.close()
|
||||
|
||||
print("\n" + "=" * 70)
|
||||
print(" MIGRATION COMPLETE")
|
||||
print("=" * 70)
|
||||
print(f" Total Episodes: {stats['total']}")
|
||||
print(f" Succeeded: {stats['succeeded']}")
|
||||
print(f" Failed: {stats['failed']}")
|
||||
print("=" * 70 + "\n")
|
||||
|
||||
|
||||
async def automatic_migration(args):
|
||||
"""Run automatic migration based on command-line args."""
|
||||
current_config = GraphitiConfig.from_env()
|
||||
|
||||
if args.from_provider:
|
||||
source_config = GraphitiConfig.from_env()
|
||||
source_config.embedder_provider = args.from_provider
|
||||
# Use source provider's signature for database name
|
||||
source_config.database = source_config.get_provider_specific_database_name(
|
||||
"auto_claude_memory"
|
||||
)
|
||||
else:
|
||||
source_config = current_config
|
||||
|
||||
if args.to_provider:
|
||||
target_config = GraphitiConfig.from_env()
|
||||
target_config.embedder_provider = args.to_provider
|
||||
# Use target provider's signature for database name
|
||||
target_config.database = target_config.get_provider_specific_database_name(
|
||||
"auto_claude_memory"
|
||||
)
|
||||
else:
|
||||
target_config = current_config
|
||||
|
||||
# Validate that source and target are different
|
||||
if source_config.embedder_provider == target_config.embedder_provider:
|
||||
logger.error(
|
||||
f"Source and target providers are the same "
|
||||
f"({source_config.embedder_provider}). "
|
||||
f"Specify different --from-provider and --to-provider values."
|
||||
)
|
||||
return
|
||||
|
||||
migrator = EmbeddingMigrator(
|
||||
source_config=source_config,
|
||||
target_config=target_config,
|
||||
dry_run=args.dry_run,
|
||||
)
|
||||
|
||||
if not await migrator.initialize():
|
||||
logger.error("Failed to initialize migration")
|
||||
return
|
||||
|
||||
stats = await migrator.migrate_all()
|
||||
await migrator.close()
|
||||
|
||||
logger.info(f"Migration complete: {stats}")
|
||||
|
||||
|
||||
def main():
|
||||
"""Main entry point."""
|
||||
parser = argparse.ArgumentParser(
|
||||
description="Migrate Graphiti embeddings between providers"
|
||||
)
|
||||
parser.add_argument(
|
||||
"--from-provider",
|
||||
choices=["openai", "ollama", "voyage", "google", "azure_openai"],
|
||||
help="Source embedding provider",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--to-provider",
|
||||
choices=["openai", "ollama", "voyage", "google", "azure_openai"],
|
||||
help="Target embedding provider",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--dry-run",
|
||||
action="store_true",
|
||||
help="Show what would be migrated without actually migrating",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--auto-confirm", action="store_true", help="Skip confirmation prompts"
|
||||
)
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
# Use interactive mode if no providers specified
|
||||
if not args.from_provider and not args.to_provider:
|
||||
asyncio.run(interactive_migration())
|
||||
else:
|
||||
asyncio.run(automatic_migration(args))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -12,7 +12,11 @@ if TYPE_CHECKING:
|
||||
|
||||
from .azure_openai_embedder import create_azure_openai_embedder
|
||||
from .google_embedder import create_google_embedder
|
||||
from .ollama_embedder import create_ollama_embedder
|
||||
from .ollama_embedder import (
|
||||
KNOWN_OLLAMA_EMBEDDING_MODELS,
|
||||
create_ollama_embedder,
|
||||
get_embedding_dim_for_model,
|
||||
)
|
||||
from .openai_embedder import create_openai_embedder
|
||||
from .voyage_embedder import create_voyage_embedder
|
||||
|
||||
@@ -22,4 +26,6 @@ __all__ = [
|
||||
"create_azure_openai_embedder",
|
||||
"create_ollama_embedder",
|
||||
"create_google_embedder",
|
||||
"KNOWN_OLLAMA_EMBEDDING_MODELS",
|
||||
"get_embedding_dim_for_model",
|
||||
]
|
||||
|
||||
+74
-1
@@ -3,6 +3,15 @@ Ollama Embedder Provider
|
||||
=========================
|
||||
|
||||
Ollama embedder implementation for Graphiti (using OpenAI-compatible interface).
|
||||
|
||||
Supported models with known dimensions:
|
||||
- embeddinggemma (768) - Google's lightweight embedding model
|
||||
- qwen3-embedding:0.6b (1024) - Qwen3 small embedding model
|
||||
- qwen3-embedding:4b (2560) - Qwen3 medium embedding model
|
||||
- qwen3-embedding:8b (4096) - Qwen3 large embedding model
|
||||
- nomic-embed-text (768) - Nomic's embedding model
|
||||
- mxbai-embed-large (1024) - MixedBread AI large embedding model
|
||||
- bge-large (1024) - BAAI general embedding large
|
||||
"""
|
||||
|
||||
from typing import TYPE_CHECKING, Any
|
||||
@@ -12,6 +21,64 @@ if TYPE_CHECKING:
|
||||
|
||||
from ..exceptions import ProviderError, ProviderNotInstalled
|
||||
|
||||
# Known Ollama embedding models and their default dimensions
|
||||
# Users can override with OLLAMA_EMBEDDING_DIM env var
|
||||
KNOWN_OLLAMA_EMBEDDING_MODELS: dict[str, int] = {
|
||||
# Google EmbeddingGemma (supports 128-768 via MRL)
|
||||
"embeddinggemma": 768,
|
||||
"embeddinggemma:300m": 768,
|
||||
# Qwen3 Embedding series (support flexible dimensions)
|
||||
"qwen3-embedding": 1024, # Default tag uses 0.6b
|
||||
"qwen3-embedding:0.6b": 1024,
|
||||
"qwen3-embedding:4b": 2560,
|
||||
"qwen3-embedding:8b": 4096,
|
||||
# Other popular models
|
||||
"nomic-embed-text": 768,
|
||||
"nomic-embed-text:latest": 768,
|
||||
"mxbai-embed-large": 1024,
|
||||
"mxbai-embed-large:latest": 1024,
|
||||
"bge-large": 1024,
|
||||
"bge-large:latest": 1024,
|
||||
"bge-m3": 1024,
|
||||
"bge-m3:latest": 1024,
|
||||
"all-minilm": 384,
|
||||
"all-minilm:latest": 384,
|
||||
}
|
||||
|
||||
|
||||
def get_embedding_dim_for_model(model_name: str, configured_dim: int = 0) -> int:
|
||||
"""
|
||||
Get the embedding dimension for an Ollama model.
|
||||
|
||||
Args:
|
||||
model_name: The Ollama model name (e.g., "embeddinggemma", "qwen3-embedding:8b")
|
||||
configured_dim: User-configured dimension (takes precedence if > 0)
|
||||
|
||||
Returns:
|
||||
Embedding dimension to use
|
||||
|
||||
Raises:
|
||||
ProviderError: If model is unknown and no dimension configured
|
||||
"""
|
||||
# User override takes precedence
|
||||
if configured_dim > 0:
|
||||
return configured_dim
|
||||
|
||||
# Check known models (exact match first)
|
||||
if model_name in KNOWN_OLLAMA_EMBEDDING_MODELS:
|
||||
return KNOWN_OLLAMA_EMBEDDING_MODELS[model_name]
|
||||
|
||||
# Try without tag suffix
|
||||
base_name = model_name.split(":")[0]
|
||||
if base_name in KNOWN_OLLAMA_EMBEDDING_MODELS:
|
||||
return KNOWN_OLLAMA_EMBEDDING_MODELS[base_name]
|
||||
|
||||
raise ProviderError(
|
||||
f"Unknown Ollama embedding model: {model_name}. "
|
||||
f"Please set OLLAMA_EMBEDDING_DIM or use a known model: "
|
||||
f"{', '.join(sorted(set(k.split(':')[0] for k in KNOWN_OLLAMA_EMBEDDING_MODELS.keys())))}"
|
||||
)
|
||||
|
||||
|
||||
def create_ollama_embedder(config: "GraphitiConfig") -> Any:
|
||||
"""
|
||||
@@ -39,6 +106,12 @@ def create_ollama_embedder(config: "GraphitiConfig") -> Any:
|
||||
if not config.ollama_embedding_model:
|
||||
raise ProviderError("Ollama embedder requires OLLAMA_EMBEDDING_MODEL")
|
||||
|
||||
# Get embedding dimension (auto-detect for known models, or use configured value)
|
||||
embedding_dim = get_embedding_dim_for_model(
|
||||
config.ollama_embedding_model,
|
||||
config.ollama_embedding_dim,
|
||||
)
|
||||
|
||||
# Ensure Ollama base URL ends with /v1 for OpenAI compatibility
|
||||
base_url = config.ollama_base_url
|
||||
if not base_url.endswith("/v1"):
|
||||
@@ -47,7 +120,7 @@ def create_ollama_embedder(config: "GraphitiConfig") -> Any:
|
||||
embedder_config = OpenAIEmbedderConfig(
|
||||
api_key="ollama", # Ollama requires a dummy API key
|
||||
embedding_model=config.ollama_embedding_model,
|
||||
embedding_dim=config.ollama_embedding_dim,
|
||||
embedding_dim=embedding_dim,
|
||||
base_url=base_url,
|
||||
)
|
||||
|
||||
|
||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user