@shykaruu/jarvis-brain 0.4.0

package/src/user/profile.ts

@@ -0,0 +1,224 @@
+export type UserProfileQuestionId =
+  | 'preferred_name'
+  | 'pronouns'
+  | 'location_timezone'
+  | 'work_role'
+  | 'interests'
+  | 'current_projects'
+  | 'goals_next_90_days'
+  | 'communication_preferences'
+  | 'pet_peeves'
+  | 'routines_constraints'
+  | 'tools_stack'
+  | 'important_people'
+  | 'anything_else';
+
+export type UserProfileQuestion = {
+  id: UserProfileQuestionId;
+  step: number;
+  step_title: string;
+  label: string;
+  prompt: string;
+  description: string;
+  placeholder?: string;
+  multiline?: boolean;
+};
+
+export type UserProfileRecord = {
+  version: 1;
+  answers: Partial<Record<UserProfileQuestionId, string>>;
+  created_at: number;
+  updated_at: number;
+  completed_at: number | null;
+};
+
+export const USER_PROFILE_SETTING_KEY = 'user.profile.v1';
+
+export const USER_PROFILE_QUESTIONS: UserProfileQuestion[] = [
+  {
+    id: 'preferred_name',
+    step: 1,
+    step_title: 'Identity',
+    label: 'Preferred Name',
+    prompt: 'What should JARVIS call you?',
+    description: 'Use the name you actually want the assistant to use in conversation.',
+    placeholder: 'e.g. Alex',
+  },
+  {
+    id: 'pronouns',
+    step: 1,
+    step_title: 'Identity',
+    label: 'Pronouns',
+    prompt: 'What pronouns do you want JARVIS to use, if any?',
+    description: 'Optional, but useful for natural and respectful replies.',
+    placeholder: 'e.g. she/her, he/him, they/them',
+  },
+  {
+    id: 'location_timezone',
+    step: 1,
+    step_title: 'Identity',
+    label: 'Location / Timezone',
+    prompt: 'What location or timezone should JARVIS keep in mind?',
+    description: 'This helps with scheduling, recommendations, and time references.',
+    placeholder: 'e.g. Miami, FL / America/New_York',
+  },
+  {
+    id: 'work_role',
+    step: 1,
+    step_title: 'Identity',
+    label: 'Work / Role',
+    prompt: 'What do you do, or what roles do you usually operate in?',
+    description: 'Career, studies, side hustles, and the kinds of responsibilities you handle.',
+    placeholder: 'e.g. founder, student, engineer, creator',
+    multiline: true,
+  },
+  {
+    id: 'interests',
+    step: 2,
+    step_title: 'Interests',
+    label: 'Interests',
+    prompt: 'What are you genuinely interested in?',
+    description: 'Topics, hobbies, communities, subjects, and rabbit holes you care about.',
+    placeholder: 'e.g. AI, fitness, cars, anime, startups',
+    multiline: true,
+  },
+  {
+    id: 'current_projects',
+    step: 2,
+    step_title: 'Interests',
+    label: 'Current Projects',
+    prompt: 'What are you actively working on right now?',
+    description: 'Projects, businesses, classes, routines, or personal efforts already in motion.',
+    placeholder: 'What is already on your plate?',
+    multiline: true,
+  },
+  {
+    id: 'goals_next_90_days',
+    step: 2,
+    step_title: 'Interests',
+    label: 'Goals',
+    prompt: 'What do you want to accomplish over the next 30 to 90 days?',
+    description: 'Short-term outcomes that JARVIS should optimize around.',
+    placeholder: 'What should JARVIS help push forward?',
+    multiline: true,
+  },
+  {
+    id: 'communication_preferences',
+    step: 3,
+    step_title: 'Working Style',
+    label: 'Communication Preferences',
+    prompt: 'How do you want JARVIS to communicate with you?',
+    description: 'Tone, directness, detail level, structure, reminders, and how much pushback you want.',
+    placeholder: 'e.g. blunt, concise, actionable, no fluff',
+    multiline: true,
+  },
+  {
+    id: 'pet_peeves',
+    step: 3,
+    step_title: 'Working Style',
+    label: 'Pet Peeves',
+    prompt: 'What annoys you or wastes your time?',
+    description: 'Patterns to avoid in planning, writing, or collaboration.',
+    placeholder: 'What should JARVIS not do?',
+    multiline: true,
+  },
+  {
+    id: 'routines_constraints',
+    step: 3,
+    step_title: 'Working Style',
+    label: 'Routines & Constraints',
+    prompt: 'What routines, limits, or constraints should JARVIS know?',
+    description: 'Schedule constraints, health habits, budget limits, availability, or boundaries.',
+    placeholder: 'Anything that should shape reminders or recommendations',
+    multiline: true,
+  },
+  {
+    id: 'tools_stack',
+    step: 4,
+    step_title: 'Context',
+    label: 'Tools & Stack',
+    prompt: 'What tools, apps, or technical stack do you use most?',
+    description: 'Software, devices, languages, frameworks, and workflows JARVIS should assume.',
+    placeholder: 'e.g. GitHub, Bun, React, Notion, Telegram, Windows',
+    multiline: true,
+  },
+  {
+    id: 'important_people',
+    step: 4,
+    step_title: 'Context',
+    label: 'Important People',
+    prompt: 'Who are the important people, teams, or audiences around you?',
+    description: 'Managers, cofounders, clients, family, friends, or communities that matter in your context.',
+    placeholder: 'Who should JARVIS keep in mind?',
+    multiline: true,
+  },
+  {
+    id: 'anything_else',
+    step: 4,
+    step_title: 'Context',
+    label: 'Anything Else',
+    prompt: 'What else should JARVIS know to be useful from day one?',
+    description: 'Any extra context that does not fit the earlier prompts.',
+    placeholder: 'Anything important you want carried forward',
+    multiline: true,
+  },
+];
+
+export function createEmptyUserProfile(): UserProfileRecord {
+  const now = Date.now();
+  return {
+    version: 1,
+    answers: {},
+    created_at: now,
+    updated_at: now,
+    completed_at: null,
+  };
+}
+
+export function normalizeUserProfileAnswers(
+  input: Record<string, unknown>,
+): Partial<Record<UserProfileQuestionId, string>> {
+  const answers: Partial<Record<UserProfileQuestionId, string>> = {};
+
+  for (const question of USER_PROFILE_QUESTIONS) {
+    const raw = input[question.id];
+    if (typeof raw !== 'string') continue;
+    const value = raw.trim();
+    if (!value) continue;
+    answers[question.id] = value;
+  }
+
+  return answers;
+}
+
+export function countAnsweredUserProfileQuestions(profile: UserProfileRecord | null): number {
+  if (!profile) return 0;
+  return USER_PROFILE_QUESTIONS.filter((question) => Boolean(profile.answers[question.id]?.trim())).length;
+}
+
+export function hasUserProfile(profile: UserProfileRecord | null): boolean {
+  return countAnsweredUserProfileQuestions(profile) > 0;
+}
+
+export function formatUserProfileForPrompt(profile: UserProfileRecord | null): string | undefined {
+  if (!profile) return undefined;
+
+  const lines: string[] = [];
+  for (const question of USER_PROFILE_QUESTIONS) {
+    const answer = profile.answers[question.id]?.trim();
+    if (!answer) continue;
+    lines.push(`- ${question.label}: |`);
+    lines.push(indentPromptValue(answer));
+  }
+
+  if (lines.length === 0) return undefined;
+  return lines.join('\n');
+}
+
+function indentPromptValue(value: string): string {
+  return value
+    .replace(/\r\n/g, '\n')
+    .split('\n')
+    .map((line) => `    ${line}`)
+    .join('\n');
+}

package/src/vault/README.md

@@ -0,0 +1,110 @@
+# Vault Module
+
+Knowledge graph storage and extraction system for J.A.R.V.I.S.
+
+## Modules
+
+- **`schema.ts`**: Database schema and initialization
+- **`entities.ts`**: Entity CRUD operations
+- **`facts.ts`**: Fact storage and queries
+- **`relationships.ts`**: Relationship management
+- **`commitments.ts`**: Promise and task tracking
+- **`observations.ts`**: Raw event storage
+- **`vectors.ts`**: Embedding storage for semantic search
+- **`extractor.ts`**: LLM-powered knowledge extraction
+- **`index.ts`**: Public API exports
+
+## Quick Start
+
+### Initialize Database
+
+```typescript
+import { initDatabase } from '@/vault';
+
+initDatabase('./data/jarvis.db');
+```
+
+### Store Knowledge Manually
+
+```typescript
+import { createEntity, createFact, createRelationship } from '@/vault';
+
+// Create entity
+const anna = createEntity('person', 'Anna', {
+  role: 'software engineer',
+});
+
+// Add fact
+createFact(anna.id, 'birthday_is', 'March 15', {
+  confidence: 1.0,
+});
+
+// Create relationship
+const google = createEntity('tool', 'Google');
+createRelationship(anna.id, google.id, 'works_at');
+```
+
+### Extract Knowledge with LLM
+
+```typescript
+import { extractAndStore } from '@/vault';
+import { AnthropicProvider } from '@/llm/anthropic';
+
+const llm = new AnthropicProvider(process.env.ANTHROPIC_API_KEY);
+
+const result = await extractAndStore(
+  "Anna's birthday is March 15th",
+  "I'll remember that!",
+  llm
+);
+```
+
+### Query Knowledge
+
+```typescript
+import { findEntities, findFacts, findCommitments } from '@/vault';
+
+// Find people
+const people = findEntities({ type: 'person' });
+
+// Find facts about Anna
+const facts = findFacts({ subject_id: anna.id });
+
+// Find pending commitments
+const pending = findCommitments({ status: 'pending' });
+```
+
+## Documentation
+
+- **Vault Extractor**: [docs/VAULT_EXTRACTOR.md](~/jarvis/docs/VAULT_EXTRACTOR.md)
+- **Entity Types**: person, project, tool, place, concept, event
+- **Commitment Statuses**: pending, active, completed, failed, escalated
+
+## Testing
+
+```bash
+# Test extractor
+bun test src/vault/extractor.test.ts
+
+# Test entire vault (if you have other tests)
+bun test src/vault/**/*.test.ts
+```
+
+## Demo
+
+```bash
+bun run examples/extractor-demo.ts
+```
+
+## Database Schema
+
+SQLite database with the following tables:
+- `entities`: People, projects, tools, places, concepts, events
+- `facts`: Atomic knowledge with confidence scores
+- `relationships`: Typed edges between entities
+- `commitments`: Promises and tasks
+- `observations`: Raw events
+- `vectors`: Embeddings for semantic search
+- `agent_messages`: Inter-agent communication
+- `personality_state`: Personality model storage
+- `conversations`: Conversation tracking

package/src/vault/awareness.ts

@@ -0,0 +1,341 @@
+/**
+ * Vault — Awareness CRUD
+ *
+ * Database operations for screen_captures, awareness_sessions, and awareness_suggestions tables.
+ * Follows the same patterns as observations.ts and commitments.ts.
+ */
+
+import { getDb, generateId } from './schema.ts';
+import type {
+  ScreenCaptureRow,
+  SessionRow,
+  SuggestionRow,
+  SuggestionType,
+  AppUsageStat,
+} from '../awareness/types.ts';
+
+// ── Screen Captures ──
+
+export function createCapture(data: {
+  timestamp: number;
+  sessionId?: string;
+  imagePath?: string;
+  thumbnailPath?: string;
+  pixelChangePct: number;
+  ocrText?: string;
+  appName?: string;
+  windowTitle?: string;
+  url?: string;
+  filePath?: string;
+  retentionTier?: 'full' | 'key_moment' | 'metadata_only';
+}): ScreenCaptureRow {
+  const db = getDb();
+  const id = generateId();
+  const now = Date.now();
+
+  db.prepare(`
+    INSERT INTO screen_captures
+      (id, timestamp, session_id, image_path, thumbnail_path, pixel_change_pct,
+       ocr_text, app_name, window_title, url, file_path, retention_tier, created_at)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+  `).run(
+    id,
+    data.timestamp,
+    data.sessionId ?? null,
+    data.imagePath ?? null,
+    data.thumbnailPath ?? null,
+    data.pixelChangePct,
+    data.ocrText ?? null,
+    data.appName ?? null,
+    data.windowTitle ?? null,
+    data.url ?? null,
+    data.filePath ?? null,
+    data.retentionTier ?? 'full',
+    now,
+  );
+
+  return {
+    id,
+    timestamp: data.timestamp,
+    session_id: data.sessionId ?? null,
+    image_path: data.imagePath ?? null,
+    thumbnail_path: data.thumbnailPath ?? null,
+    pixel_change_pct: data.pixelChangePct,
+    ocr_text: data.ocrText ?? null,
+    app_name: data.appName ?? null,
+    window_title: data.windowTitle ?? null,
+    url: data.url ?? null,
+    file_path: data.filePath ?? null,
+    retention_tier: data.retentionTier ?? 'full',
+    created_at: now,
+  };
+}
+
+export function getCapture(id: string): ScreenCaptureRow | null {
+  const db = getDb();
+  return db.prepare('SELECT * FROM screen_captures WHERE id = ?').get(id) as ScreenCaptureRow | null;
+}
+
+export function getRecentCaptures(limit: number = 50, appName?: string): ScreenCaptureRow[] {
+  const db = getDb();
+  if (appName) {
+    return db.prepare(
+      'SELECT * FROM screen_captures WHERE app_name = ? ORDER BY timestamp DESC LIMIT ?'
+    ).all(appName, limit) as ScreenCaptureRow[];
+  }
+  return db.prepare(
+    'SELECT * FROM screen_captures ORDER BY timestamp DESC LIMIT ?'
+  ).all(limit) as ScreenCaptureRow[];
+}
+
+export function getCapturesInRange(startTime: number, endTime: number): ScreenCaptureRow[] {
+  const db = getDb();
+  return db.prepare(
+    'SELECT * FROM screen_captures WHERE timestamp >= ? AND timestamp <= ? ORDER BY timestamp ASC'
+  ).all(startTime, endTime) as ScreenCaptureRow[];
+}
+
+export function getAppUsageStats(startTime: number, endTime: number): AppUsageStat[] {
+  const db = getDb();
+  const rows = db.prepare(`
+    SELECT app_name, COUNT(*) as capture_count
+    FROM screen_captures
+    WHERE timestamp >= ? AND timestamp <= ? AND app_name IS NOT NULL
+    GROUP BY app_name
+    ORDER BY capture_count DESC
+  `).all(startTime, endTime) as Array<{ app_name: string; capture_count: number }>;
+
+  const totalCaptures = rows.reduce((sum, r) => sum + r.capture_count, 0);
+
+  return rows.map(r => ({
+    app: r.app_name,
+    captureCount: r.capture_count,
+    minutes: Math.round((r.capture_count * 7) / 60),  // ~7s per capture
+    percentage: totalCaptures > 0 ? Math.round((r.capture_count / totalCaptures) * 100) : 0,
+  }));
+}
+
+export function getCaptureCountSince(timestamp: number): number {
+  const db = getDb();
+  const row = db.prepare(
+    'SELECT COUNT(*) as count FROM screen_captures WHERE timestamp >= ?'
+  ).get(timestamp) as { count: number };
+  return row.count;
+}
+
+export function updateCaptureRetention(id: string, tier: 'full' | 'key_moment' | 'metadata_only'): void {
+  const db = getDb();
+  db.prepare('UPDATE screen_captures SET retention_tier = ? WHERE id = ?').run(tier, id);
+}
+
+export function deleteCapturesBefore(timestamp: number, retentionTier: string): number {
+  const db = getDb();
+  const result = db.prepare(
+    'DELETE FROM screen_captures WHERE timestamp < ? AND retention_tier = ?'
+  ).run(timestamp, retentionTier);
+  return result.changes;
+}
+
+export function updateCaptureOcrText(id: string, ocrText: string): void {
+  const db = getDb();
+  db.prepare('UPDATE screen_captures SET ocr_text = ? WHERE id = ?').run(ocrText, id);
+}
+
+export function getCapturesForSession(sessionId: string, limit: number = 50): ScreenCaptureRow[] {
+  const db = getDb();
+  return db.prepare(
+    'SELECT * FROM screen_captures WHERE session_id = ? ORDER BY timestamp DESC LIMIT ?'
+  ).all(sessionId, limit) as ScreenCaptureRow[];
+}
+
+// ── Awareness Sessions ──
+
+export function createSession(data: {
+  startedAt: number;
+  apps?: string[];
+  projectContext?: string;
+}): SessionRow {
+  const db = getDb();
+  const id = generateId();
+  const now = Date.now();
+
+  db.prepare(`
+    INSERT INTO awareness_sessions
+      (id, started_at, ended_at, topic, apps, project_context, action_types, entity_links, summary, capture_count, created_at)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+  `).run(
+    id,
+    data.startedAt,
+    null,
+    null,
+    JSON.stringify(data.apps ?? []),
+    data.projectContext ?? null,
+    JSON.stringify([]),
+    JSON.stringify([]),
+    null,
+    0,
+    now,
+  );
+
+  return {
+    id,
+    started_at: data.startedAt,
+    ended_at: null,
+    topic: null,
+    apps: JSON.stringify(data.apps ?? []),
+    project_context: data.projectContext ?? null,
+    action_types: JSON.stringify([]),
+    entity_links: JSON.stringify([]),
+    summary: null,
+    capture_count: 0,
+    created_at: now,
+  };
+}
+
+export function getSession(id: string): SessionRow | null {
+  const db = getDb();
+  return db.prepare('SELECT * FROM awareness_sessions WHERE id = ?').get(id) as SessionRow | null;
+}
+
+export function updateSession(id: string, updates: Partial<{
+  ended_at: number | null;
+  topic: string | null;
+  apps: string[];
+  project_context: string | null;
+  action_types: string[];
+  entity_links: string[];
+  summary: string | null;
+  capture_count: number;
+}>): void {
+  const db = getDb();
+  const setClauses: string[] = [];
+  const params: unknown[] = [];
+
+  if (updates.ended_at !== undefined) { setClauses.push('ended_at = ?'); params.push(updates.ended_at); }
+  if (updates.topic !== undefined) { setClauses.push('topic = ?'); params.push(updates.topic); }
+  if (updates.apps !== undefined) { setClauses.push('apps = ?'); params.push(JSON.stringify(updates.apps)); }
+  if (updates.project_context !== undefined) { setClauses.push('project_context = ?'); params.push(updates.project_context); }
+  if (updates.action_types !== undefined) { setClauses.push('action_types = ?'); params.push(JSON.stringify(updates.action_types)); }
+  if (updates.entity_links !== undefined) { setClauses.push('entity_links = ?'); params.push(JSON.stringify(updates.entity_links)); }
+  if (updates.summary !== undefined) { setClauses.push('summary = ?'); params.push(updates.summary); }
+  if (updates.capture_count !== undefined) { setClauses.push('capture_count = ?'); params.push(updates.capture_count); }
+
+  if (setClauses.length === 0) return;
+
+  params.push(id);
+  db.prepare(`UPDATE awareness_sessions SET ${setClauses.join(', ')} WHERE id = ?`).run(...params as any[]);
+}
+
+export function endSession(id: string, summary?: string): void {
+  const db = getDb();
+  db.prepare(
+    'UPDATE awareness_sessions SET ended_at = ?, summary = ? WHERE id = ?'
+  ).run(Date.now(), summary ?? null, id);
+}
+
+export function getRecentSessions(limit: number = 20): SessionRow[] {
+  const db = getDb();
+  return db.prepare(
+    'SELECT * FROM awareness_sessions ORDER BY started_at DESC LIMIT ?'
+  ).all(limit) as SessionRow[];
+}
+
+export function incrementSessionCaptureCount(id: string): void {
+  const db = getDb();
+  db.prepare(
+    'UPDATE awareness_sessions SET capture_count = capture_count + 1 WHERE id = ?'
+  ).run(id);
+}
+
+// ── Awareness Suggestions ──
+
+export function createSuggestion(data: {
+  type: SuggestionType;
+  triggerCaptureId?: string;
+  title: string;
+  body: string;
+  context?: Record<string, unknown>;
+}): SuggestionRow {
+  const db = getDb();
+  const id = generateId();
+  const now = Date.now();
+
+  db.prepare(`
+    INSERT INTO awareness_suggestions
+      (id, type, trigger_capture_id, title, body, context, delivered, delivered_at, delivery_channel, dismissed, acted_on, created_at)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+  `).run(
+    id,
+    data.type,
+    data.triggerCaptureId ?? null,
+    data.title,
+    data.body,
+    data.context ? JSON.stringify(data.context) : null,
+    0, null, null, 0, 0,
+    now,
+  );
+
+  return {
+    id,
+    type: data.type,
+    trigger_capture_id: data.triggerCaptureId ?? null,
+    title: data.title,
+    body: data.body,
+    context: data.context ? JSON.stringify(data.context) : null,
+    delivered: 0,
+    delivered_at: null,
+    delivery_channel: null,
+    dismissed: 0,
+    acted_on: 0,
+    created_at: now,
+  };
+}
+
+export function markSuggestionDelivered(id: string, channel: string): void {
+  const db = getDb();
+  db.prepare(
+    'UPDATE awareness_suggestions SET delivered = 1, delivered_at = ?, delivery_channel = ? WHERE id = ?'
+  ).run(Date.now(), channel, id);
+}
+
+export function markSuggestionDismissed(id: string): void {
+  const db = getDb();
+  db.prepare('UPDATE awareness_suggestions SET dismissed = 1 WHERE id = ?').run(id);
+}
+
+export function markSuggestionActedOn(id: string): void {
+  const db = getDb();
+  db.prepare('UPDATE awareness_suggestions SET acted_on = 1 WHERE id = ?').run(id);
+}
+
+export function getRecentSuggestions(limit: number = 20, type?: SuggestionType): SuggestionRow[] {
+  const db = getDb();
+  if (type) {
+    return db.prepare(
+      'SELECT * FROM awareness_suggestions WHERE type = ? ORDER BY created_at DESC LIMIT ?'
+    ).all(type, limit) as SuggestionRow[];
+  }
+  return db.prepare(
+    'SELECT * FROM awareness_suggestions ORDER BY created_at DESC LIMIT ?'
+  ).all(limit) as SuggestionRow[];
+}
+
+export function getSuggestionCountSince(timestamp: number): number {
+  const db = getDb();
+  const row = db.prepare(
+    'SELECT COUNT(*) as count FROM awareness_suggestions WHERE created_at >= ?'
+  ).get(timestamp) as { count: number };
+  return row.count;
+}
+
+export function getSuggestionStats(startTime: number, endTime: number): { total: number; actedOn: number } {
+  const db = getDb();
+  const row = db.prepare(`
+    SELECT
+      COUNT(*) as total,
+      SUM(CASE WHEN acted_on = 1 THEN 1 ELSE 0 END) as acted_on
+    FROM awareness_suggestions
+    WHERE created_at >= ? AND created_at <= ?
+  `).get(startTime, endTime) as { total: number; acted_on: number };
+  return { total: row.total, actedOn: row.acted_on ?? 0 };
+}