oh-my-githubcopilot 1.4.1 → 1.8.0-alpha.021bf87

package/dist/mcp/server.mjs

@@ -14138,11 +14138,20 @@ __export(agent_loader_exports, {
   getAgent: () => getAgent,
   loadAllAgents: () => loadAllAgents
 });
-import { readFileSync, readdirSync } from "fs";
+import { existsSync, readFileSync, readdirSync } from "fs";
 import { join } from "path";
-function loadAgentFile(filename) {
+function normalizeModelTier(modelTier, model) {
+  if (modelTier === "high" || modelTier === "standard" || modelTier === "fast") {
+    return modelTier;
+  }
+  if (!model) return "standard";
+  if (/haiku|mini|nano|fast/i.test(model)) return "fast";
+  if (/opus|pro|gpt-5\.5|sonnet-4-6/i.test(model)) return "high";
+  return "standard";
+}
+function loadAgentFile(dir, filename) {
   try {
-    const filePath = join(AGENTS_DIR, filename);
+    const filePath = join(dir, filename);
     const content = readFileSync(filePath, "utf-8");
     const parsed = parseAgentFile(content);
     if (!parsed) return null;
@@ -14150,7 +14159,7 @@ function loadAgentFile(filename) {
       id: parsed.frontmatter.name,
       name: parsed.frontmatter.name,
       description: parsed.frontmatter.description || "",
-      modelTier: parsed.frontmatter.model_tier || "standard",
+      modelTier: normalizeModelTier(parsed.frontmatter.model_tier, parsed.frontmatter.model),
       tools: parsed.frontmatter.tools || [],
       content: parsed.content
     };
@@ -14161,13 +14170,17 @@ function loadAgentFile(filename) {
 function loadAllAgents() {
   if (cache) return cache;
   cache = /* @__PURE__ */ new Map();
-  try {
-    const files = readdirSync(AGENTS_DIR).filter((f) => f.endsWith(".md"));
-    for (const file2 of files) {
-      const agent = loadAgentFile(file2);
-      if (agent) cache.set(agent.id, agent);
+  for (const dir of AGENTS_DIRS) {
+    if (!existsSync(dir)) continue;
+    try {
+      const files = readdirSync(dir).filter((f) => f.endsWith(".agent.md"));
+      for (const file2 of files) {
+        const agent = loadAgentFile(dir, file2);
+        if (agent) cache.set(agent.id, agent);
+      }
+      if (cache.size > 0) break;
+    } catch {
     }
-  } catch {
   }
   return cache;
 }
@@ -14178,12 +14191,15 @@ function getAgent(id) {
 function clearCache() {
   cache = null;
 }
-var AGENTS_DIR, cache;
+var AGENTS_DIRS, cache;
 var init_agent_loader = __esm({
   "src/utils/agent-loader.mts"() {
     "use strict";
     init_yaml_parser();
-    AGENTS_DIR = join(process.cwd(), "src", "agents");
+    AGENTS_DIRS = [
+      join(process.cwd(), "agents"),
+      join(process.cwd(), "src", "agents")
+    ];
     cache = null;
   }
 });
@@ -28224,58 +28240,71 @@ var StdioServerTransport = class {
 };
 
 // src/mcp/server.mts
-import Database from "better-sqlite3";
 import { readFileSync as readFileSync2, mkdirSync } from "fs";
 import { homedir } from "os";
 import { join as join2, dirname } from "path";
 import { randomUUID } from "crypto";
+
+// src/mcp/db-loader.mts
+import { createRequire } from "module";
+var SqliteConstructor = null;
+try {
+  SqliteConstructor = createRequire(import.meta.url)("better-sqlite3");
+} catch {
+}
+
+// src/mcp/server.mts
 function getDbPath() {
   const envPath = process.env["OMP_STATE_DB"];
   if (envPath) return envPath.replace("~", homedir());
   return join2(homedir(), ".omp", "state", "omp.db");
 }
-function ensureDbDir(dbPath2) {
-  mkdirSync(dirname(dbPath2), { recursive: true });
-}
-var dbPath = getDbPath();
-ensureDbDir(dbPath);
-var db = new Database(dbPath);
-db.exec(`
-  CREATE TABLE IF NOT EXISTS sessions (
-    id TEXT PRIMARY KEY,
-    worktree_id TEXT,
-    state_json TEXT NOT NULL,
-    created_at INTEGER NOT NULL,
-    updated_at INTEGER NOT NULL
-  );
-  CREATE INDEX IF NOT EXISTS idx_sessions_worktree ON sessions(worktree_id);
-  CREATE INDEX IF NOT EXISTS idx_sessions_updated ON sessions(updated_at);
+function ensureDbDir(dbPath) {
+  mkdirSync(dirname(dbPath), { recursive: true });
+}
+var db = null;
+if (SqliteConstructor) {
+  const dbPath = getDbPath();
+  ensureDbDir(dbPath);
+  db = new SqliteConstructor(dbPath);
+  db.pragma("journal_mode = WAL");
+  db.exec(`
+    CREATE TABLE IF NOT EXISTS sessions (
+      id TEXT PRIMARY KEY,
+      worktree_id TEXT,
+      state_json TEXT NOT NULL,
+      created_at INTEGER NOT NULL,
+      updated_at INTEGER NOT NULL
+    );
+    CREATE INDEX IF NOT EXISTS idx_sessions_worktree ON sessions(worktree_id);
+    CREATE INDEX IF NOT EXISTS idx_sessions_updated ON sessions(updated_at);
 
-  CREATE TABLE IF NOT EXISTS memory (
-    key TEXT PRIMARY KEY,
-    value TEXT NOT NULL,
-    category TEXT,
-    session_id TEXT,
-    created_at INTEGER NOT NULL,
-    updated_at INTEGER NOT NULL
-  );
-  CREATE INDEX IF NOT EXISTS idx_memory_category ON memory(category);
-  CREATE INDEX IF NOT EXISTS idx_memory_session ON memory(session_id);
+    CREATE TABLE IF NOT EXISTS memory (
+      key TEXT PRIMARY KEY,
+      value TEXT NOT NULL,
+      category TEXT,
+      session_id TEXT,
+      created_at INTEGER NOT NULL,
+      updated_at INTEGER NOT NULL
+    );
+    CREATE INDEX IF NOT EXISTS idx_memory_category ON memory(category);
+    CREATE INDEX IF NOT EXISTS idx_memory_session ON memory(session_id);
 
-  CREATE TABLE IF NOT EXISTS trace (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    session_id TEXT NOT NULL,
-    hook_id TEXT,
-    agent_id TEXT,
-    event_type TEXT NOT NULL,
-    payload TEXT,
-    duration_ms INTEGER,
-    timestamp INTEGER NOT NULL
-  );
-  CREATE INDEX IF NOT EXISTS idx_trace_session ON trace(session_id);
-  CREATE INDEX IF NOT EXISTS idx_trace_hook ON trace(hook_id);
-  CREATE INDEX IF NOT EXISTS idx_trace_agent ON trace(agent_id);
-`);
+    CREATE TABLE IF NOT EXISTS trace (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      session_id TEXT NOT NULL,
+      hook_id TEXT,
+      agent_id TEXT,
+      event_type TEXT NOT NULL,
+      payload TEXT,
+      duration_ms INTEGER,
+      timestamp INTEGER NOT NULL
+    );
+    CREATE INDEX IF NOT EXISTS idx_trace_session ON trace(session_id);
+    CREATE INDEX IF NOT EXISTS idx_trace_hook ON trace(hook_id);
+    CREATE INDEX IF NOT EXISTS idx_trace_agent ON trace(agent_id);
+  `);
+}
 var TOOLS = [
   // State tools
   {
@@ -28397,12 +28426,14 @@ function handleListTools() {
 async function handleCallTool(name, args) {
   switch (name) {
     case "omp_get_session_state": {
+      if (!db) return { content: [{ type: "text", text: "null" }] };
       const sessions = db.prepare("SELECT * FROM sessions ORDER BY updated_at DESC LIMIT 1").all();
       return { content: [{ type: "text", text: JSON.stringify(sessions[0] || null, null, 2) }] };
     }
     case "omp_save_session": {
       const sessionId = args.sessionId || randomUUID();
       const stateJson = args.stateJson || JSON.stringify({});
+      if (!db) return { content: [{ type: "text", text: JSON.stringify({ status: "ok", sessionId, note: "SQLite unavailable; state not persisted" }) }] };
       const now = Date.now();
       db.prepare(
         "INSERT OR REPLACE INTO sessions (id, worktree_id, state_json, created_at, updated_at) VALUES (?, ?, ?, ?, ?)"
@@ -28410,6 +28441,7 @@ async function handleCallTool(name, args) {
       return { content: [{ type: "text", text: JSON.stringify({ status: "ok", sessionId }) }] };
     }
     case "omp_list_sessions": {
+      if (!db) return { content: [{ type: "text", text: "[]" }] };
       const sessions = db.prepare("SELECT id, created_at, updated_at FROM sessions ORDER BY updated_at DESC").all();
       return { content: [{ type: "text", text: JSON.stringify(sessions, null, 2) }] };
     }