@oh-my-pi/cli 0.1.0 → 0.3.0

package/plugins/exa/tools/exa/shared.ts

@@ -0,0 +1,230 @@
+/**
+ * Shared utilities for Exa MCP tools
+ */
+
+import * as fs from "node:fs";
+import * as os from "node:os";
+import * as path from "node:path";
+import type { TSchema } from "@sinclair/typebox";
+import type { CustomAgentTool } from "@mariozechner/pi-coding-agent";
+
+// MCP endpoints
+export const EXA_MCP_URL = "https://mcp.exa.ai/mcp";
+export const WEBSETS_MCP_URL = "https://websetsmcp.exa.ai/mcp";
+
+export interface MCPTool {
+   name: string;
+   description: string;
+   inputSchema: TSchema;
+}
+
+interface MCPToolsResponse {
+   result?: {
+      tools: MCPTool[];
+   };
+   error?: {
+      code: number;
+      message: string;
+   };
+}
+
+/**
+ * Parse a .env file and return key-value pairs
+ */
+function parseEnvFile(filePath: string): Record<string, string> {
+   const result: Record<string, string> = {};
+   if (!fs.existsSync(filePath)) return result;
+
+   try {
+      const content = fs.readFileSync(filePath, "utf-8");
+      for (const line of content.split("\n")) {
+         const trimmed = line.trim();
+         if (!trimmed || trimmed.startsWith("#")) continue;
+
+         const eqIndex = trimmed.indexOf("=");
+         if (eqIndex === -1) continue;
+
+         const key = trimmed.slice(0, eqIndex).trim();
+         let value = trimmed.slice(eqIndex + 1).trim();
+
+         // Remove surrounding quotes
+         if ((value.startsWith('"') && value.endsWith('"')) || (value.startsWith("'") && value.endsWith("'"))) {
+            value = value.slice(1, -1);
+         }
+
+         result[key] = value;
+      }
+   } catch {
+      // Ignore read errors
+   }
+
+   return result;
+}
+
+/**
+ * Find EXA_API_KEY from environment or .env files
+ */
+export function findApiKey(): string | null {
+   // 1. Check environment variable
+   if (process.env.EXA_API_KEY) {
+      return process.env.EXA_API_KEY;
+   }
+
+   // 2. Check .env in current directory
+   const localEnv = parseEnvFile(path.join(process.cwd(), ".env"));
+   if (localEnv.EXA_API_KEY) {
+      return localEnv.EXA_API_KEY;
+   }
+
+   // 3. Check ~/.env
+   const homeEnv = parseEnvFile(path.join(os.homedir(), ".env"));
+   if (homeEnv.EXA_API_KEY) {
+      return homeEnv.EXA_API_KEY;
+   }
+
+   return null;
+}
+
+/**
+ * Call an MCP server endpoint
+ */
+async function callMCP(url: string, method: string, params?: Record<string, unknown>): Promise<unknown> {
+   const body = {
+      jsonrpc: "2.0",
+      method,
+      params: params ?? {},
+      id: 1,
+   };
+
+   const response = await fetch(url, {
+      method: "POST",
+      headers: {
+         "Content-Type": "application/json",
+         Accept: "application/json, text/event-stream",
+      },
+      body: JSON.stringify(body),
+   });
+
+   const text = await response.text();
+
+   // Parse SSE response format
+   let jsonData: string | null = null;
+   for (const line of text.split("\n")) {
+      if (line.startsWith("data: ")) {
+         jsonData = line.slice(6);
+         break;
+      }
+   }
+
+   if (!jsonData) {
+      // Try parsing as plain JSON
+      try {
+         return JSON.parse(text);
+      } catch {
+         throw new Error(`Failed to parse MCP response: ${text.slice(0, 500)}`);
+      }
+   }
+
+   return JSON.parse(jsonData);
+}
+
+/**
+ * Fetch available tools from Exa MCP server
+ */
+export async function fetchExaTools(apiKey: string, toolNames: string[]): Promise<MCPTool[]> {
+   const url = `${EXA_MCP_URL}?exaApiKey=${apiKey}&tools=${toolNames.join(",")}`;
+
+   try {
+      const response = (await callMCP(url, "tools/list")) as MCPToolsResponse;
+      if (response.error) {
+         throw new Error(response.error.message);
+      }
+      return response.result?.tools ?? [];
+   } catch (error) {
+      console.error(`Failed to fetch Exa tools:`, error);
+      return [];
+   }
+}
+
+/**
+ * Fetch available tools from Websets MCP server
+ */
+export async function fetchWebsetsTools(apiKey: string): Promise<MCPTool[]> {
+   const url = `${WEBSETS_MCP_URL}?exaApiKey=${apiKey}`;
+
+   try {
+      const response = (await callMCP(url, "tools/list")) as MCPToolsResponse;
+      if (response.error) {
+         throw new Error(response.error.message);
+      }
+      return response.result?.tools ?? [];
+   } catch (error) {
+      console.error(`Failed to fetch Websets tools:`, error);
+      return [];
+   }
+}
+
+/**
+ * Call a tool on Exa MCP server
+ */
+export async function callExaTool(apiKey: string, toolNames: string[], toolName: string, args: Record<string, unknown>): Promise<unknown> {
+   const url = `${EXA_MCP_URL}?exaApiKey=${apiKey}&tools=${toolNames.join(",")}`;
+   return callMCPTool(url, toolName, args);
+}
+
+/**
+ * Call a tool on Websets MCP server
+ */
+export async function callWebsetsTool(apiKey: string, toolName: string, args: Record<string, unknown>): Promise<unknown> {
+   const url = `${WEBSETS_MCP_URL}?exaApiKey=${apiKey}`;
+   return callMCPTool(url, toolName, args);
+}
+
+/**
+ * Call a tool on an MCP server
+ */
+async function callMCPTool(url: string, toolName: string, args: Record<string, unknown>): Promise<unknown> {
+   const response = (await callMCP(url, "tools/call", {
+      name: toolName,
+      arguments: args,
+   })) as { result?: { content?: Array<{ text?: string }> }; error?: { message: string } };
+
+   if (response.error) {
+      throw new Error(response.error.message);
+   }
+
+   // Extract text content from MCP response
+   const content = response.result?.content;
+   if (Array.isArray(content)) {
+      const texts = content.filter((c) => c.text).map((c) => c.text);
+      if (texts.length === 1) {
+         // Try to parse as JSON
+         try {
+            return JSON.parse(texts[0]!);
+         } catch {
+            return texts[0];
+         }
+      }
+      return texts.join("\n\n");
+   }
+
+   return response.result;
+}
+
+/**
+ * Create a tool wrapper for an MCP tool
+ */
+export function createToolWrapper(
+   mcpTool: MCPTool,
+   renamedName: string,
+   callFn: (toolName: string, args: Record<string, unknown>) => Promise<unknown>
+): CustomAgentTool<TSchema, unknown> {
+   return {
+      name: renamedName,
+      description: mcpTool.description,
+      parameters: mcpTool.inputSchema,
+      async execute(args) {
+         return callFn(mcpTool.name, args as Record<string, unknown>);
+      },
+   };
+}

package/plugins/exa/tools/exa/websets.ts

@@ -0,0 +1,62 @@
+/**
+ * Exa Websets Tools - Entity collection management
+ *
+ * Tools:
+ *   - webset_create: Create entity collections with search/enrichments
+ *   - webset_list: List all websets
+ *   - webset_get: Get webset details
+ *   - webset_update: Update webset metadata
+ *   - webset_delete: Delete a webset
+ *   - webset_items_list: List items in a webset
+ *   - webset_item_get: Get item details
+ *   - webset_search_create: Add search to webset
+ *   - webset_search_get: Check search status
+ *   - webset_search_cancel: Cancel running search
+ *   - webset_enrichment_create: Extract custom data from items
+ *   - webset_enrichment_get: Get enrichment details
+ *   - webset_enrichment_update: Update enrichment metadata
+ *   - webset_enrichment_delete: Delete enrichment
+ *   - webset_enrichment_cancel: Cancel running enrichment
+ *   - webset_monitor_create: Auto-update webset on schedule
+ */
+
+import type { TSchema } from "@sinclair/typebox";
+import type { CustomAgentTool, CustomToolFactory, ToolAPI } from "@mariozechner/pi-coding-agent";
+import { callWebsetsTool, createToolWrapper, fetchWebsetsTools, findApiKey } from "./shared";
+
+// Tool name mapping: MCP name -> exposed name
+const NAME_MAP: Record<string, string> = {
+   "create_webset": "webset_create",
+   "list_websets": "webset_list",
+   "get_webset": "webset_get",
+   "update_webset": "webset_update",
+   "delete_webset": "webset_delete",
+   "list_webset_items": "webset_items_list",
+   "get_item": "webset_item_get",
+   "create_search": "webset_search_create",
+   "get_search": "webset_search_get",
+   "cancel_search": "webset_search_cancel",
+   "create_enrichment": "webset_enrichment_create",
+   "get_enrichment": "webset_enrichment_get",
+   "update_enrichment": "webset_enrichment_update",
+   "delete_enrichment": "webset_enrichment_delete",
+   "cancel_enrichment": "webset_enrichment_cancel",
+   "create_monitor": "webset_monitor_create",
+};
+
+const factory: CustomToolFactory = async (_toolApi: ToolAPI): Promise<CustomAgentTool<TSchema, unknown>[] | null> => {
+   const apiKey = findApiKey();
+   if (!apiKey) return null;
+
+   const mcpTools = await fetchWebsetsTools(apiKey);
+   if (mcpTools.length === 0) return null;
+
+   const callFn = (toolName: string, args: Record<string, unknown>) =>
+      callWebsetsTool(apiKey, toolName, args);
+
+   return mcpTools.map((tool) =>
+      createToolWrapper(tool, NAME_MAP[tool.name] ?? tool.name, callFn)
+   );
+};
+
+export default factory;

package/plugins/metal-theme/package.json

@@ -1,10 +1,15 @@
 {
   "name": "@oh-my-pi/metal-theme",
-  "version": "0.1.0",
+  "version": "0.3.0",
   "description": "Metal theme for pi",
   "keywords": ["omp-plugin", "theme", "metal"],
-  "author": "",
+  "author": "Can Bölük <me@can.ac>",
   "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/can1357/oh-my-pi.git",
+    "directory": "plugins/metal-theme"
+  },
   "omp": {
     "install": [
       { "src": "themes/metal.json", "dest": "agent/themes/metal.json" }

package/plugins/subagents/package.json

@@ -1,10 +1,15 @@
 {
   "name": "@oh-my-pi/subagents",
-  "version": "0.1.0",
+  "version": "0.3.0",
   "description": "Task delegation system with specialized subagents (task, planner, explore, reviewer)",
   "keywords": ["omp-plugin", "agents", "task-delegation"],
-  "author": "",
+  "author": "Can Bölük <me@can.ac>",
   "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/can1357/oh-my-pi.git",
+    "directory": "plugins/subagents"
+  },
   "omp": {
     "install": [
       { "src": "tools/task/index.ts", "dest": "agent/tools/task/index.ts" },

package/plugins/user-prompt/README.md

@@ -0,0 +1,130 @@
+# User Prompt Plugin
+
+Interactive user prompting tool for gathering user input during agent execution.
+
+## Installation
+
+```bash
+omp install oh-my-pi/plugins/user-prompt
+```
+
+## Tool
+
+### `user_prompt`
+
+Asks the user questions during execution and returns their response. Useful for:
+
+- Gathering user preferences or requirements
+- Clarifying ambiguous instructions
+- Getting decisions on implementation choices
+- Offering choices about what direction to take
+
+## Features
+
+### Enhanced UI (when available)
+
+The plugin provides custom TUI components that integrate directly into pi's interface:
+
+**Single-select with inline "Other" input:**
+```
+─────────────────────────────────────────────
+  Which database would you like to use?
+
+→ PostgreSQL (Recommended)
+  MySQL
+  SQLite
+  MongoDB
+  Other (type your own)
+
+  ↑↓ navigate · enter select · esc cancel
+─────────────────────────────────────────────
+```
+
+When "Other" is selected, an inline text input appears - no separate dialog needed.
+
+**Multi-select with checkboxes:**
+```
+─────────────────────────────────────────────
+  Which features should I implement?
+
+→ [X] Authentication
+  [X] API endpoints
+  [ ] Database models
+  [ ] Unit tests
+  [ ] Documentation
+
+  ↑↓ navigate · space toggle · enter confirm · esc cancel
+─────────────────────────────────────────────
+```
+
+Space toggles selection, Enter confirms. Selected items show `[X]` in green with white text.
+
+### Fallback Mode
+
+If the enhanced UI cannot be loaded, the plugin gracefully falls back to using pi's built-in `select()` and `input()` methods.
+
+## Parameters
+
+| Parameter | Type | Required | Description |
+|-----------|------|----------|-------------|
+| `question` | string | Yes | The question to ask the user |
+| `options` | array | Yes | Array of `{label: string}` options to present |
+| `multiSelect` | boolean | No | Allow multiple selections (default: false) |
+
+## Usage Notes
+
+- Users can always select "Other" to provide custom text input
+- Use `multiSelect: true` to allow multiple answers to be selected
+- If you recommend a specific option, make that the first option and add "(Recommended)" at the end of the label
+
+## Examples
+
+### Single-choice question
+
+```json
+{
+  "question": "Which database would you like to use?",
+  "options": [
+    {"label": "PostgreSQL (Recommended)"},
+    {"label": "MySQL"},
+    {"label": "SQLite"},
+    {"label": "MongoDB"}
+  ]
+}
+```
+
+### Multi-select question
+
+```json
+{
+  "question": "Which features should I implement?",
+  "options": [
+    {"label": "Authentication"},
+    {"label": "API endpoints"},
+    {"label": "Database models"},
+    {"label": "Unit tests"},
+    {"label": "Documentation"}
+  ],
+  "multiSelect": true
+}
+```
+
+## Response Format
+
+The tool returns the user's selection in a structured format:
+
+- **Single selection**: `"User selected: PostgreSQL (Recommended)"`
+- **Multi-selection**: `"User selected: Authentication, API endpoints, Unit tests"`
+- **Custom input**: `"User provided custom input: Use Redis for caching"`
+- **Cancelled**: `"User cancelled the selection"`
+
+## How It Works
+
+The plugin hooks into pi's interactive mode at runtime to provide custom TUI components. It:
+
+1. Dynamically imports pi's theme for consistent styling
+2. Locates the InteractiveMode instance to access the editor container
+3. Swaps in custom components (MultiSelectList, SelectWithInput) when prompting
+4. Restores the normal editor when done
+
+This approach provides a seamless, native-feeling UI without requiring upstream changes to pi.

package/plugins/user-prompt/package.json

@@ -0,0 +1,19 @@
+{
+  "name": "@oh-my-pi/user-prompt",
+  "version": "0.3.0",
+  "description": "Interactive user prompting tool for gathering user input during execution",
+  "keywords": ["omp-plugin", "user-prompt", "interactive", "questions", "input"],
+  "author": "Can Bölük <me@can.ac>",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/can1357/oh-my-pi.git",
+    "directory": "plugins/user-prompt"
+  },
+  "omp": {
+    "install": [
+      { "src": "tools/user-prompt/index.ts", "dest": "agent/tools/user-prompt/index.ts" }
+    ]
+  },
+  "files": ["tools"]
+}