Overview
VectorAI is an AI-powered Visual Studio Code extension designed to improve developer productivity by bringing semantic search, embeddings generation, and intelligent code assistance directly into the development environment.
The extension provides AI capabilities for developers working with vector databases, semantic retrieval, and modern AI workflows, including PostgreSQL with pgvector and embedding-based search architectures.
The project is built with TypeScript using the Visual Studio Code Extension API, with AI integrations through Hugging Face models and vector-based technologies.
Features
🧠 Embedding Generation
Generate vector embeddings from selected text directly inside VS Code.
Capabilities:
- Transform code or text into numerical representations
- Prepare content for semantic retrieval workflows
- Support AI-powered developer tools
🔍 Semantic Search
Perform similarity-based searches using vector embeddings.
Designed to support:
- Vector databases
- Knowledge retrieval
- RAG (Retrieval Augmented Generation) workflows
💡 AI Optimization Suggestions
Receive AI-powered suggestions for improving queries and developer content.
Current scenarios:
- SQL optimization suggestions
- Intelligent recommendations
- AI-assisted analysis
⚡ Native VS Code Experience
Integrated directly into the editor:
- Command Palette
- Context actions
- VS Code notifications
- Extension commands
Architecture
VectorAI follows a layered architecture: src/ ├── extension.ts │ ├── commands/ │ ├── generateEmbedding.ts │ ├── semanticSearch.ts │ └── suggestOptimization.ts │ ├── services/ │ ├── huggingfaceService.ts │ ├── dbConnector.ts │ └── VectorAIService.ts │ ├── utils/ │ └── logger.ts │ ├── tests/ │ └── mocks/
Architectural Principles
- Separation of responsibilities
- Service-based integrations
- Testable commands
- Provider abstraction
- AI service isolation
Architecture decisions are documented using ADRs.
Commands
Available commands:
vectorAI.generateEmbedding
Generates embeddings from selected text.
vectorAI.semanticSearch
Executes semantic similarity search using vector embeddings.
vectorAI.suggestOptimization
Generates AI-powered improvement suggestions.
Technology Stack
Core
- TypeScript
- Visual Studio Code Extension API
- Node.js
AI
- Hugging Face Models
- Embedding generation
- Semantic similarity search
Database
- PostgreSQL
- pgvector support
Testing
- Jest
- ts-jest
- VS Code API mocks
Tooling
- ESLint
- TypeScript Compiler
- VSCE Packaging
Development
Install dependencies:
npm install