atabey 0.0.12 → 0.0.14

package/framework-mcp/dist/src/modules/memory/embedding.js

@@ -0,0 +1,130 @@
+/**
+ * [MEMORY] Embedding Service
+ *
+ * Real embedding generation with OpenAI API integration.
+ * Falls back to TF-IDF if no API key is available.
+ *
+ * Supported methods:
+ * 1. OpenAI Embedding API (text-embedding-3-small) — 1536 dimensions
+ * 2. TF-IDF fallback (local method) — 384 dimensions
+ */
+import fs from "fs";
+import path from "path";
+import { logger } from "../../shared/logger.js";
+import { CoreMemory } from "./core.js";
+const EMBEDDING_API = "https://api.openai.com/v1/embeddings";
+const EMBEDDING_MODEL = "text-embedding-3-small";
+const EMBEDDING_DIM = 1536;
+/**
+ * Reads the API key from environment variable or .env file.
+ */
+function getApiKey() {
+    if (process.env.OPENAI_API_KEY)
+        return process.env.OPENAI_API_KEY;
+    try {
+        const envPath = path.join(process.cwd(), ".env");
+        if (fs.existsSync(envPath)) {
+            const content = fs.readFileSync(envPath, "utf8");
+            const match = content.match(/^OPENAI_API_KEY=(.+)$/m);
+            if (match)
+                return match[1].trim();
+        }
+    }
+    catch {
+        // Silently ignore .env read errors
+    }
+    return null;
+}
+/**
+ * Generates an embedding vector for the given text.
+ *
+ * Priority:
+ * 1. OpenAI API (if API key is available)
+ * 2. TF-IDF fallback (local)
+ */
+export async function generateEmbedding(text) {
+    const apiKey = getApiKey();
+    // Use OpenAI API if available
+    if (apiKey) {
+        try {
+            const response = await fetch(EMBEDDING_API, {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                    "Authorization": `Bearer ${apiKey}`,
+                },
+                body: JSON.stringify({
+                    input: text,
+                    model: EMBEDDING_MODEL,
+                }),
+            });
+            if (!response.ok) {
+                logger.warn(`OpenAI API error: ${response.status}. Falling back to TF-IDF.`);
+                return fallbackEmbedding(text);
+            }
+            const data = await response.json();
+            if (data?.data?.[0]?.embedding) {
+                return data.data[0].embedding;
+            }
+        }
+        catch (err) {
+            logger.warn(`OpenAI API call failed: ${err.message}. Falling back to TF-IDF.`);
+        }
+    }
+    // Fallback: TF-IDF
+    return fallbackEmbedding(text);
+}
+/**
+ * TF-IDF fallback embedding using CoreMemory.generateEmbedding.
+ * OpenAI embedding is 1536 dimensions, TF-IDF is 384 dimensions.
+ * Pads TF-IDF output to 1536 for compatibility.
+ */
+function fallbackEmbedding(text) {
+    const tfidfVector = CoreMemory.generateEmbedding(text); // 384 dimensions
+    // Pad 384 → 1536
+    const padded = new Array(EMBEDDING_DIM).fill(0);
+    for (let i = 0; i < tfidfVector.length; i++) {
+        padded[i] = tfidfVector[i];
+    }
+    return padded;
+}
+/**
+ * Batch embedding generation. Single API call for multiple texts.
+ * OpenAI supports batch requests natively.
+ */
+export async function generateEmbeddings(texts) {
+    const apiKey = getApiKey();
+    if (apiKey) {
+        try {
+            const response = await fetch(EMBEDDING_API, {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                    "Authorization": `Bearer ${apiKey}`,
+                },
+                body: JSON.stringify({
+                    input: texts,
+                    model: EMBEDDING_MODEL,
+                }),
+            });
+            if (response.ok) {
+                const data = await response.json();
+                if (data?.data) {
+                    return data.data.map(d => d.embedding);
+                }
+            }
+        }
+        catch {
+            // fallback
+        }
+    }
+    // Fallback for each text
+    return Promise.all(texts.map(t => fallbackEmbedding(t)));
+}
+/**
+ * Checks if a vector is an OpenAI embedding (1536 dimensions)
+ * as opposed to TF-IDF (384 dimensions).
+ */
+export function isOpenAIEmbedding(vector) {
+    return vector.length === EMBEDDING_DIM;
+}

package/framework-mcp/dist/src/modules/providers/definitions.js

@@ -0,0 +1,140 @@
+import path from "path";
+import { addMcpServerToClaude, findClaudeConfigPath } from "../../cli/utils/claude.js";
+import { writeJsonFile } from "../../shared/fs.js";
+import { registerGlobalAntigravityPlugins } from "./shared.js";
+import { registry } from "./registry.js";
+import { UI } from "../../cli/utils/ui.js";
+// ─── Register Core Adapters ──────────────────────────────────────────────────
+// [GEMINI] Gemini
+registry.register({
+    id: "gemini",
+    frameworkDir: ".gemini",
+    shimFile: "GEMINI.md",
+    shimTemplate: "src/cli/shims/gemini.md",
+    role: "commander",
+    templateDir: ".atabey",
+    nestedDirs: ["agents", "rules"],
+    agentsDir: ".gemini/agents",
+    agentsExt: ".md"
+}, (projectRoot, mcpBlock) => {
+    const frameworkDir = ".gemini";
+    writeJsonFile(path.join(projectRoot, frameworkDir, "mcp.json"), mcpBlock);
+    UI.success(`Gemini MCP registered → ${frameworkDir}/mcp.json`);
+    registerGlobalAntigravityPlugins(mcpBlock);
+});
+// [START] Claude
+registry.register({
+    id: "claude",
+    frameworkDir: ".claude",
+    shimFile: "CLAUDE.md",
+    shimTemplate: "src/cli/shims/claude.md",
+    role: "architect",
+    templateDir: ".atabey",
+    nestedDirs: ["agents", "rules"],
+    agentsDir: ".claude/agents",
+    agentsExt: ".md"
+}, (projectRoot, mcpBlock) => {
+    const configPath = findClaudeConfigPath();
+    if (configPath) {
+        const block = mcpBlock;
+        const mcpEntry = block.mcpServers["atabey"];
+        const ok = addMcpServerToClaude(configPath, "atabey", mcpEntry);
+        if (ok)
+            UI.success(`Claude MCP registered → ${configPath}`);
+    }
+    const frameworkDir = ".claude";
+    writeJsonFile(path.join(projectRoot, frameworkDir, "mcp_config.json"), mcpBlock);
+    writeJsonFile(path.join(projectRoot, ".mcp.json"), mcpBlock);
+    UI.success("Claude Code Project MCP → .mcp.json");
+});
+// [AI] Grok
+registry.register({
+    id: "grok",
+    frameworkDir: ".grok",
+    shimFile: "GROK.md",
+    shimTemplate: "src/cli/shims/grok.md",
+    role: "researcher",
+    templateDir: ".atabey",
+    nestedDirs: ["agents", "rules"],
+    agentsDir: ".grok/agents",
+    agentsExt: ".md"
+}, (projectRoot, mcpBlock) => {
+    const frameworkDir = ".grok";
+    writeJsonFile(path.join(projectRoot, frameworkDir, "mcp_config.json"), mcpBlock);
+    UI.success(`Grok MCP → ${frameworkDir}/mcp_config.json`);
+});
+// [CURSOR] Cursor
+registry.register({
+    id: "cursor",
+    frameworkDir: ".cursor",
+    shimFile: "CURSOR.md",
+    shimTemplate: "src/cli/shims/cursor.mdc",
+    role: "implementer",
+    templateDir: ".atabey",
+    nestedDirs: ["rules"],
+    agentsDir: ".cursor/rules",
+    agentsExt: ".mdc"
+}, (projectRoot, mcpBlock) => {
+    const frameworkDir = ".cursor";
+    writeJsonFile(path.join(projectRoot, frameworkDir, "mcp.json"), mcpBlock);
+    UI.success(`Cursor IDE Project MCP → ${frameworkDir}/mcp.json`);
+});
+// [TIP] Codex (Copilot)
+registry.register({
+    id: "codex",
+    frameworkDir: ".agents",
+    shimFile: "copilot-instructions.md",
+    shimTemplate: "src/cli/shims/codex.md",
+    role: "implementer",
+    templateDir: ".atabey",
+    nestedDirs: ["skills", "rules", "instructions"],
+    agentsDir: ".agents/instructions",
+    agentsExt: ".md"
+}, (projectRoot, mcpBlock) => {
+    const frameworkDir = ".agents";
+    writeJsonFile(path.join(projectRoot, frameworkDir, "mcp_config.json"), mcpBlock);
+    writeJsonFile(path.join(projectRoot, ".vscode/mcp.json"), mcpBlock);
+    writeJsonFile(path.join(projectRoot, ".mcp.json"), mcpBlock);
+    UI.success("GitHub Copilot Project MCP → .vscode/mcp.json & .mcp.json");
+});
+// [LOCAL] Local LLM (Ollama / Private AI)
+registry.register({
+    id: "local",
+    frameworkDir: ".atabey",
+    shimFile: "LOCAL_AI.md",
+    shimTemplate: "src/cli/shims/local.md",
+    role: "commander",
+    templateDir: ".atabey",
+    nestedDirs: ["agents", "rules"],
+    agentsDir: ".atabey/agents",
+    agentsExt: ".md"
+}, (projectRoot, mcpBlock) => {
+    const frameworkDir = ".atabey";
+    writeJsonFile(path.join(projectRoot, frameworkDir, "mcp_config.json"), mcpBlock);
+    UI.success(`Local LLM MCP (Ollama/Private AI) registered → ${frameworkDir}/mcp_config.json`);
+});
+// [ANTIGRAVITY] Antigravity
+registry.register({
+    id: "antigravity-cli",
+    frameworkDir: ".agents",
+    shimFile: "AGENTS.md",
+    shimTemplate: "src/cli/shims/antigravity-cli.md",
+    role: "general",
+    templateDir: ".atabey",
+    nestedDirs: ["agents", "plugins", "rules"],
+    agentsDir: ".agents/agents",
+    agentsExt: ".md"
+}, (projectRoot, mcpBlock) => {
+    const frameworkDir = ".agents";
+    writeJsonFile(path.join(projectRoot, frameworkDir, "mcp_config.json"), mcpBlock);
+    UI.success(`Antigravity CLI MCP → ${frameworkDir}/mcp_config.json`);
+    registerGlobalAntigravityPlugins(mcpBlock);
+});
+/**
+ * Mapped Adapter Configurations
+ */
+export const ADAPTER_CONFIGS = registry.getConfigs();
+/**
+ * Post-Initialization Handlers for specific adapters
+ */
+export const POST_INIT_HANDLERS = registry.getHandlers();

package/framework-mcp/dist/src/modules/providers/registry.js

@@ -0,0 +1,18 @@
+class AdapterRegistry {
+    configs = new Map();
+    handlers = new Map();
+    register(config, handler) {
+        this.configs.set(config.id, config);
+        this.handlers.set(config.id, handler);
+    }
+    getConfigs() {
+        return Object.fromEntries(this.configs);
+    }
+    getHandlers() {
+        return Object.fromEntries(this.handlers);
+    }
+    getIds() {
+        return Array.from(this.configs.keys());
+    }
+}
+export const registry = new AdapterRegistry();

package/framework-mcp/dist/src/modules/providers/shared.js

@@ -0,0 +1,104 @@
+import path from "path";
+import fs from "fs";
+import os from "os";
+import { execSync } from "child_process";
+import { writeJsonFile, writeTextFile } from "../../shared/fs.js";
+import { ALL_AGENTS, toAntigravityJson } from "../agents/definitions.js";
+import { CORE_SKILLS } from "../skills/definitions.js";
+import { getPackageRoot } from "../../cli/utils/pkg.js";
+import { logger } from "../../shared/logger.js";
+export function registerGlobalAntigravityPlugins(mcpBlock) {
+    // Allow overriding via env var for different OS/Gemini CLI versions
+    const defaultGlobalDir = path.join(os.homedir(), ".gemini/antigravity-cli");
+    const customDir = process.env["ANTIGRAVITY_GLOBAL_DIR"];
+    const targets = [customDir ?? defaultGlobalDir];
+    const baseKnowledgeDir = path.join(getPackageRoot(), "templates/standards");
+    for (const globalDir of targets) {
+        try {
+            // Write directly under directory
+            fs.mkdirSync(globalDir, { recursive: true });
+            writeJsonFile(path.join(globalDir, "mcp.json"), mcpBlock);
+            writeJsonFile(path.join(globalDir, "mcp_config.json"), mcpBlock);
+            logger.info(`Antigravity MCP registered → ${globalDir}/`);
+            // Global Plugin configuration under plugins/atabey/
+            const globalPluginDir = path.join(globalDir, "plugins/atabey");
+            fs.mkdirSync(globalPluginDir, { recursive: true });
+            // plugin.json marker
+            writeJsonFile(path.join(globalPluginDir, "plugin.json"), {
+                name: "atabey",
+                version: "1.0.0",
+                description: "Agent Atabey AI Orchestration framework"
+            });
+            // MCP Server configs inside the plugin
+            writeJsonFile(path.join(globalPluginDir, "mcp.json"), mcpBlock);
+            writeJsonFile(path.join(globalPluginDir, "mcp_config.json"), mcpBlock);
+            // Scaffold 13 agents
+            const agentsBaseDir = path.join(globalPluginDir, "agents");
+            fs.mkdirSync(agentsBaseDir, { recursive: true });
+            // Also write directly to ~/.gemini/antigravity-cli/agents/ so the CLI's
+            // workspace/global agent discovery finds them without the plugins/ prefix.
+            const globalAgentsDir = path.join(globalDir, "agents");
+            fs.mkdirSync(globalAgentsDir, { recursive: true });
+            for (const ag of ALL_AGENTS) {
+                const agentJson = toAntigravityJson(ag, baseKnowledgeDir);
+                // 1. Nested format inside plugin: plugins/atabey/agents/{name}/agent.json
+                const nestedAgentDir = path.join(agentsBaseDir, ag.name);
+                fs.mkdirSync(nestedAgentDir, { recursive: true });
+                writeTextFile(path.join(nestedAgentDir, "agent.json"), agentJson);
+                // 2. Flat format inside plugin: plugins/atabey/agents/{name}.json
+                writeTextFile(path.join(agentsBaseDir, `${ag.name}.json`), agentJson);
+                // 3. Direct global agents root — THIS is what Antigravity CLI reads:
+                //    ~/.gemini/antigravity-cli/agents/{name}/agent.json
+                const globalNestedDir = path.join(globalAgentsDir, ag.name);
+                fs.mkdirSync(globalNestedDir, { recursive: true });
+                writeTextFile(path.join(globalNestedDir, "agent.json"), agentJson);
+                // 4. Flat global format as fallback
+                writeTextFile(path.join(globalAgentsDir, `${ag.name}.json`), agentJson);
+            }
+            // Scaffold skills
+            const skillsDir = path.join(globalPluginDir, "skills");
+            fs.mkdirSync(skillsDir, { recursive: true });
+            for (const [key, skill] of Object.entries(CORE_SKILLS)) {
+                const skillContent = `# [TOOL] Skill — ${skill.name}\n\n${skill.mandates.join("\n")}\n`;
+                writeTextFile(path.join(skillsDir, `${key.toLowerCase()}.md`), skillContent);
+            }
+            // Scaffold rules
+            const rulesDir = path.join(globalPluginDir, "rules");
+            fs.mkdirSync(rulesDir, { recursive: true });
+            for (const ag of ALL_AGENTS) {
+                const ruleContent = `# [ATABEY] Agent Atabey — @${ag.name} (${ag.displayName})
+
+You are the **${ag.displayName}** of the Agent Atabey AL.
+
+## [AI] Specialist Directive (Role: @${ag.name})
+${ag.instructions.identity}
+
+### [GOAL] Mission
+${ag.instructions.mission}
+
+### [DOCS] Discipline Rules
+${ag.instructions.rules.map(r => `- ${r}`).join("\n")}
+
+## [SECURITY] Core Mandates
+- **Surgical Precision:** Enforce replace_text / replace_file_content for all code modifications.
+- **Traceability:** Inherit and pass the active Trace ID across all delegations.
+- **Approval Signature:** High-risk actions require manager approval signature.
+`;
+                writeTextFile(path.join(rulesDir, `${ag.name}.md`), ruleContent);
+            }
+            // Scaffold optional empty hooks.json
+            writeJsonFile(path.join(globalPluginDir, "hooks.json"), {});
+            logger.info(`Antigravity Plugin registered → ${globalPluginDir}/`);
+            try {
+                execSync(`agy plugin install "${globalPluginDir}"`, { stdio: "ignore" });
+                logger.info("Antigravity Plugin installed in CLI.");
+            }
+            catch {
+                // Ignore if agy is not in PATH or fails
+            }
+        }
+        catch (e) {
+            logger.warn(`Failed to register plugin/MCP in ${globalDir}`, e);
+        }
+    }
+}

package/framework-mcp/dist/src/modules/providers/types.js

@@ -0,0 +1 @@
+export const ADAPTER_IDS = ["gemini", "claude", "grok", "cursor", "codex", "local", "antigravity-cli"];