opc-agent 4.1.0 → 4.1.2

package/src/tools/image-generator.ts

@@ -1,150 +1,150 @@
-/**
- * Image Generator — multi-backend image generation tool.
- * Supports DALL·E (OpenAI), Stable Diffusion (local), and Replicate.
- */
-
-export interface ImageGenConfig {
-  provider?: 'dalle' | 'stable-diffusion' | 'replicate';
-  openaiApiKey?: string;
-  replicateApiKey?: string;
-  sdApiUrl?: string;
-  defaultModel?: string;
-  defaultSize?: string;
-}
-
-export interface ImageGenResult {
-  success: boolean;
-  url?: string;
-  base64?: string;
-  error?: string;
-  provider: string;
-}
-
-export class ImageGenerator {
-  private config: ImageGenConfig;
-
-  constructor(config?: ImageGenConfig) {
-    this.config = {
-      provider: config?.provider,
-      openaiApiKey: config?.openaiApiKey || process.env.OPENAI_API_KEY,
-      replicateApiKey: config?.replicateApiKey || process.env.REPLICATE_API_TOKEN,
-      sdApiUrl: config?.sdApiUrl || process.env.SD_API_URL,
-      defaultModel: config?.defaultModel || 'dall-e-3',
-      defaultSize: config?.defaultSize || '1024x1024',
-    };
-  }
-
-  /** Auto-detect best available provider */
-  detectProvider(): string | null {
-    if (this.config.openaiApiKey) return 'dalle';
-    if (this.config.sdApiUrl) return 'stable-diffusion';
-    if (this.config.replicateApiKey) return 'replicate';
-    return null;
-  }
-
-  /** Get configuration status for the settings UI */
-  getStatus(): { configured: boolean; providers: { name: string; configured: boolean }[] } {
-    return {
-      configured: !!this.detectProvider(),
-      providers: [
-        { name: 'dalle', configured: !!this.config.openaiApiKey },
-        { name: 'stable-diffusion', configured: !!this.config.sdApiUrl },
-        { name: 'replicate', configured: !!this.config.replicateApiKey },
-      ],
-    };
-  }
-
-  async generate(prompt: string, options?: { provider?: string; size?: string; model?: string }): Promise<ImageGenResult> {
-    const provider = options?.provider || this.config.provider || this.detectProvider();
-    if (!provider) {
-      return { success: false, error: 'No image generation provider configured. Please set OPENAI_API_KEY, SD_API_URL, or REPLICATE_API_TOKEN.', provider: 'none' };
-    }
-
-    switch (provider) {
-      case 'dalle': return this.generateDalle(prompt, options);
-      case 'stable-diffusion': return this.generateSD(prompt, options);
-      case 'replicate': return this.generateReplicate(prompt, options);
-      default: return { success: false, error: `Unknown provider: ${provider}`, provider };
-    }
-  }
-
-  private async generateDalle(prompt: string, options?: { size?: string; model?: string }): Promise<ImageGenResult> {
-    const apiKey = this.config.openaiApiKey;
-    if (!apiKey) return { success: false, error: 'OPENAI_API_KEY not configured', provider: 'dalle' };
-
-    try {
-      const res = await fetch('https://api.openai.com/v1/images/generations', {
-        method: 'POST',
-        headers: { 'Authorization': `Bearer ${apiKey}`, 'Content-Type': 'application/json' },
-        body: JSON.stringify({
-          model: options?.model || this.config.defaultModel || 'dall-e-3',
-          prompt,
-          size: options?.size || this.config.defaultSize || '1024x1024',
-          n: 1,
-        }),
-      });
-      const data = await res.json() as { data?: Array<{ url: string }>; error?: { message: string } };
-      if (data.error) return { success: false, error: data.error.message, provider: 'dalle' };
-      const url = data.data?.[0]?.url;
-      return url ? { success: true, url, provider: 'dalle' } : { success: false, error: 'No image returned', provider: 'dalle' };
-    } catch (err) {
-      return { success: false, error: (err as Error).message, provider: 'dalle' };
-    }
-  }
-
-  private async generateSD(prompt: string, options?: { size?: string }): Promise<ImageGenResult> {
-    const apiUrl = this.config.sdApiUrl;
-    if (!apiUrl) return { success: false, error: 'SD_API_URL not configured', provider: 'stable-diffusion' };
-
-    try {
-      const [w, h] = (options?.size || '1024x1024').split('x').map(Number);
-      const res = await fetch(`${apiUrl}/sdapi/v1/txt2img`, {
-        method: 'POST',
-        headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify({ prompt, width: w || 1024, height: h || 1024 }),
-      });
-      const data = await res.json() as { images?: string[] };
-      if (data.images?.length) {
-        return { success: true, base64: data.images[0], provider: 'stable-diffusion' };
-      }
-      return { success: false, error: 'No image generated', provider: 'stable-diffusion' };
-    } catch (err) {
-      return { success: false, error: (err as Error).message, provider: 'stable-diffusion' };
-    }
-  }
-
-  private async generateReplicate(prompt: string, _options?: { model?: string }): Promise<ImageGenResult> {
-    const apiKey = this.config.replicateApiKey;
-    if (!apiKey) return { success: false, error: 'REPLICATE_API_TOKEN not configured', provider: 'replicate' };
-
-    try {
-      const res = await fetch('https://api.replicate.com/v1/predictions', {
-        method: 'POST',
-        headers: { 'Authorization': `Token ${apiKey}`, 'Content-Type': 'application/json' },
-        body: JSON.stringify({
-          version: 'stability-ai/sdxl:39ed52f2a78e934b3ba6e2a89f5b1c712de7dfea535525255b1aa35c5565e08b',
-          input: { prompt },
-        }),
-      });
-      const prediction = await res.json() as { id: string; urls?: { get: string }; error?: string };
-      if (prediction.error) return { success: false, error: prediction.error, provider: 'replicate' };
-
-      // Poll for completion (max 60s)
-      const getUrl = prediction.urls?.get || `https://api.replicate.com/v1/predictions/${prediction.id}`;
-      for (let i = 0; i < 30; i++) {
-        await new Promise(r => setTimeout(r, 2000));
-        const poll = await fetch(getUrl, { headers: { 'Authorization': `Token ${apiKey}` } });
-        const result = await poll.json() as { status: string; output?: string[]; error?: string };
-        if (result.status === 'succeeded' && result.output?.length) {
-          return { success: true, url: result.output[0], provider: 'replicate' };
-        }
-        if (result.status === 'failed') {
-          return { success: false, error: result.error || 'Generation failed', provider: 'replicate' };
-        }
-      }
-      return { success: false, error: 'Timeout waiting for image generation', provider: 'replicate' };
-    } catch (err) {
-      return { success: false, error: (err as Error).message, provider: 'replicate' };
-    }
-  }
-}
+/**
+ * Image Generator — multi-backend image generation tool.
+ * Supports DALL·E (OpenAI), Stable Diffusion (local), and Replicate.
+ */
+
+export interface ImageGenConfig {
+  provider?: 'dalle' | 'stable-diffusion' | 'replicate';
+  openaiApiKey?: string;
+  replicateApiKey?: string;
+  sdApiUrl?: string;
+  defaultModel?: string;
+  defaultSize?: string;
+}
+
+export interface ImageGenResult {
+  success: boolean;
+  url?: string;
+  base64?: string;
+  error?: string;
+  provider: string;
+}
+
+export class ImageGenerator {
+  private config: ImageGenConfig;
+
+  constructor(config?: ImageGenConfig) {
+    this.config = {
+      provider: config?.provider,
+      openaiApiKey: config?.openaiApiKey || process.env.OPENAI_API_KEY,
+      replicateApiKey: config?.replicateApiKey || process.env.REPLICATE_API_TOKEN,
+      sdApiUrl: config?.sdApiUrl || process.env.SD_API_URL,
+      defaultModel: config?.defaultModel || 'dall-e-3',
+      defaultSize: config?.defaultSize || '1024x1024',
+    };
+  }
+
+  /** Auto-detect best available provider */
+  detectProvider(): string | null {
+    if (this.config.openaiApiKey) return 'dalle';
+    if (this.config.sdApiUrl) return 'stable-diffusion';
+    if (this.config.replicateApiKey) return 'replicate';
+    return null;
+  }
+
+  /** Get configuration status for the settings UI */
+  getStatus(): { configured: boolean; providers: { name: string; configured: boolean }[] } {
+    return {
+      configured: !!this.detectProvider(),
+      providers: [
+        { name: 'dalle', configured: !!this.config.openaiApiKey },
+        { name: 'stable-diffusion', configured: !!this.config.sdApiUrl },
+        { name: 'replicate', configured: !!this.config.replicateApiKey },
+      ],
+    };
+  }
+
+  async generate(prompt: string, options?: { provider?: string; size?: string; model?: string }): Promise<ImageGenResult> {
+    const provider = options?.provider || this.config.provider || this.detectProvider();
+    if (!provider) {
+      return { success: false, error: 'No image generation provider configured. Please set OPENAI_API_KEY, SD_API_URL, or REPLICATE_API_TOKEN.', provider: 'none' };
+    }
+
+    switch (provider) {
+      case 'dalle': return this.generateDalle(prompt, options);
+      case 'stable-diffusion': return this.generateSD(prompt, options);
+      case 'replicate': return this.generateReplicate(prompt, options);
+      default: return { success: false, error: `Unknown provider: ${provider}`, provider };
+    }
+  }
+
+  private async generateDalle(prompt: string, options?: { size?: string; model?: string }): Promise<ImageGenResult> {
+    const apiKey = this.config.openaiApiKey;
+    if (!apiKey) return { success: false, error: 'OPENAI_API_KEY not configured', provider: 'dalle' };
+
+    try {
+      const res = await fetch('https://api.openai.com/v1/images/generations', {
+        method: 'POST',
+        headers: { 'Authorization': `Bearer ${apiKey}`, 'Content-Type': 'application/json' },
+        body: JSON.stringify({
+          model: options?.model || this.config.defaultModel || 'dall-e-3',
+          prompt,
+          size: options?.size || this.config.defaultSize || '1024x1024',
+          n: 1,
+        }),
+      });
+      const data = await res.json() as { data?: Array<{ url: string }>; error?: { message: string } };
+      if (data.error) return { success: false, error: data.error.message, provider: 'dalle' };
+      const url = data.data?.[0]?.url;
+      return url ? { success: true, url, provider: 'dalle' } : { success: false, error: 'No image returned', provider: 'dalle' };
+    } catch (err) {
+      return { success: false, error: (err as Error).message, provider: 'dalle' };
+    }
+  }
+
+  private async generateSD(prompt: string, options?: { size?: string }): Promise<ImageGenResult> {
+    const apiUrl = this.config.sdApiUrl;
+    if (!apiUrl) return { success: false, error: 'SD_API_URL not configured', provider: 'stable-diffusion' };
+
+    try {
+      const [w, h] = (options?.size || '1024x1024').split('x').map(Number);
+      const res = await fetch(`${apiUrl}/sdapi/v1/txt2img`, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ prompt, width: w || 1024, height: h || 1024 }),
+      });
+      const data = await res.json() as { images?: string[] };
+      if (data.images?.length) {
+        return { success: true, base64: data.images[0], provider: 'stable-diffusion' };
+      }
+      return { success: false, error: 'No image generated', provider: 'stable-diffusion' };
+    } catch (err) {
+      return { success: false, error: (err as Error).message, provider: 'stable-diffusion' };
+    }
+  }
+
+  private async generateReplicate(prompt: string, _options?: { model?: string }): Promise<ImageGenResult> {
+    const apiKey = this.config.replicateApiKey;
+    if (!apiKey) return { success: false, error: 'REPLICATE_API_TOKEN not configured', provider: 'replicate' };
+
+    try {
+      const res = await fetch('https://api.replicate.com/v1/predictions', {
+        method: 'POST',
+        headers: { 'Authorization': `Token ${apiKey}`, 'Content-Type': 'application/json' },
+        body: JSON.stringify({
+          version: 'stability-ai/sdxl:39ed52f2a78e934b3ba6e2a89f5b1c712de7dfea535525255b1aa35c5565e08b',
+          input: { prompt },
+        }),
+      });
+      const prediction = await res.json() as { id: string; urls?: { get: string }; error?: string };
+      if (prediction.error) return { success: false, error: prediction.error, provider: 'replicate' };
+
+      // Poll for completion (max 60s)
+      const getUrl = prediction.urls?.get || `https://api.replicate.com/v1/predictions/${prediction.id}`;
+      for (let i = 0; i < 30; i++) {
+        await new Promise(r => setTimeout(r, 2000));
+        const poll = await fetch(getUrl, { headers: { 'Authorization': `Token ${apiKey}` } });
+        const result = await poll.json() as { status: string; output?: string[]; error?: string };
+        if (result.status === 'succeeded' && result.output?.length) {
+          return { success: true, url: result.output[0], provider: 'replicate' };
+        }
+        if (result.status === 'failed') {
+          return { success: false, error: result.error || 'Generation failed', provider: 'replicate' };
+        }
+      }
+      return { success: false, error: 'Timeout waiting for image generation', provider: 'replicate' };
+    } catch (err) {
+      return { success: false, error: (err as Error).message, provider: 'replicate' };
+    }
+  }
+}

package/src/tools/integrations/calendar.ts

@@ -1,73 +1,73 @@
-import type { MCPTool, MCPToolResult } from '../mcp';
-
-export const CalendarTool: MCPTool = {
-  name: 'calendar',
-  description: 'Google Calendar integration: create events, list events, check availability. Requires GOOGLE_CALENDAR_API_KEY or GOOGLE_ACCESS_TOKEN env var.',
-  inputSchema: {
-    type: 'object',
-    properties: {
-      action: { type: 'string', enum: ['create_event', 'list_events', 'check_availability'], description: 'Action' },
-      calendar_id: { type: 'string', description: 'Calendar ID (default: primary)' },
-      summary: { type: 'string', description: 'Event title' },
-      start: { type: 'string', description: 'Start time (ISO 8601)' },
-      end: { type: 'string', description: 'End time (ISO 8601)' },
-      description: { type: 'string', description: 'Event description' },
-      time_min: { type: 'string', description: 'Start of time range (ISO 8601)' },
-      time_max: { type: 'string', description: 'End of time range (ISO 8601)' },
-    },
-    required: ['action'],
-  },
-
-  async execute(input: Record<string, unknown>): Promise<MCPToolResult> {
-    const token = process.env.GOOGLE_ACCESS_TOKEN;
-    if (!token) return { content: 'Error: GOOGLE_ACCESS_TOKEN required', isError: true };
-
-    const calId = encodeURIComponent(String(input.calendar_id ?? 'primary'));
-    const headers = { 'Authorization': `Bearer ${token}`, 'Content-Type': 'application/json' };
-    const action = String(input.action ?? '');
-
-    try {
-      if (action === 'create_event') {
-        if (!input.summary || !input.start || !input.end) return { content: 'Error: summary, start, end required', isError: true };
-        const res = await fetch(`https://www.googleapis.com/calendar/v3/calendars/${calId}/events`, {
-          method: 'POST', headers,
-          body: JSON.stringify({
-            summary: input.summary,
-            description: input.description ?? '',
-            start: { dateTime: input.start },
-            end: { dateTime: input.end },
-          }),
-        });
-        const data = await res.json() as Record<string, unknown>;
-        return { content: `Event created: ${data.htmlLink ?? data.id}` };
-      }
-
-      if (action === 'list_events') {
-        const now = new Date().toISOString();
-        const timeMin = String(input.time_min ?? now);
-        const timeMax = input.time_max ? `&timeMax=${encodeURIComponent(String(input.time_max))}` : '';
-        const res = await fetch(`https://www.googleapis.com/calendar/v3/calendars/${calId}/events?timeMin=${encodeURIComponent(timeMin)}${timeMax}&maxResults=20&singleEvents=true&orderBy=startTime`, { headers });
-        const data = await res.json() as Record<string, unknown>;
-        return { content: JSON.stringify(data, null, 2).slice(0, 4000) };
-      }
-
-      if (action === 'check_availability') {
-        if (!input.time_min || !input.time_max) return { content: 'Error: time_min, time_max required', isError: true };
-        const res = await fetch('https://www.googleapis.com/calendar/v3/freeBusy', {
-          method: 'POST', headers,
-          body: JSON.stringify({
-            timeMin: input.time_min,
-            timeMax: input.time_max,
-            items: [{ id: String(input.calendar_id ?? 'primary') }],
-          }),
-        });
-        const data = await res.json() as Record<string, unknown>;
-        return { content: JSON.stringify(data, null, 2).slice(0, 4000) };
-      }
-
-      return { content: `Unknown action: ${action}`, isError: true };
-    } catch (err) {
-      return { content: `Calendar error: ${(err as Error).message}`, isError: true };
-    }
-  },
-};
+import type { MCPTool, MCPToolResult } from '../mcp';
+
+export const CalendarTool: MCPTool = {
+  name: 'calendar',
+  description: 'Google Calendar integration: create events, list events, check availability. Requires GOOGLE_CALENDAR_API_KEY or GOOGLE_ACCESS_TOKEN env var.',
+  inputSchema: {
+    type: 'object',
+    properties: {
+      action: { type: 'string', enum: ['create_event', 'list_events', 'check_availability'], description: 'Action' },
+      calendar_id: { type: 'string', description: 'Calendar ID (default: primary)' },
+      summary: { type: 'string', description: 'Event title' },
+      start: { type: 'string', description: 'Start time (ISO 8601)' },
+      end: { type: 'string', description: 'End time (ISO 8601)' },
+      description: { type: 'string', description: 'Event description' },
+      time_min: { type: 'string', description: 'Start of time range (ISO 8601)' },
+      time_max: { type: 'string', description: 'End of time range (ISO 8601)' },
+    },
+    required: ['action'],
+  },
+
+  async execute(input: Record<string, unknown>): Promise<MCPToolResult> {
+    const token = process.env.GOOGLE_ACCESS_TOKEN;
+    if (!token) return { content: 'Error: GOOGLE_ACCESS_TOKEN required', isError: true };
+
+    const calId = encodeURIComponent(String(input.calendar_id ?? 'primary'));
+    const headers = { 'Authorization': `Bearer ${token}`, 'Content-Type': 'application/json' };
+    const action = String(input.action ?? '');
+
+    try {
+      if (action === 'create_event') {
+        if (!input.summary || !input.start || !input.end) return { content: 'Error: summary, start, end required', isError: true };
+        const res = await fetch(`https://www.googleapis.com/calendar/v3/calendars/${calId}/events`, {
+          method: 'POST', headers,
+          body: JSON.stringify({
+            summary: input.summary,
+            description: input.description ?? '',
+            start: { dateTime: input.start },
+            end: { dateTime: input.end },
+          }),
+        });
+        const data = await res.json() as Record<string, unknown>;
+        return { content: `Event created: ${data.htmlLink ?? data.id}` };
+      }
+
+      if (action === 'list_events') {
+        const now = new Date().toISOString();
+        const timeMin = String(input.time_min ?? now);
+        const timeMax = input.time_max ? `&timeMax=${encodeURIComponent(String(input.time_max))}` : '';
+        const res = await fetch(`https://www.googleapis.com/calendar/v3/calendars/${calId}/events?timeMin=${encodeURIComponent(timeMin)}${timeMax}&maxResults=20&singleEvents=true&orderBy=startTime`, { headers });
+        const data = await res.json() as Record<string, unknown>;
+        return { content: JSON.stringify(data, null, 2).slice(0, 4000) };
+      }
+
+      if (action === 'check_availability') {
+        if (!input.time_min || !input.time_max) return { content: 'Error: time_min, time_max required', isError: true };
+        const res = await fetch('https://www.googleapis.com/calendar/v3/freeBusy', {
+          method: 'POST', headers,
+          body: JSON.stringify({
+            timeMin: input.time_min,
+            timeMax: input.time_max,
+            items: [{ id: String(input.calendar_id ?? 'primary') }],
+          }),
+        });
+        const data = await res.json() as Record<string, unknown>;
+        return { content: JSON.stringify(data, null, 2).slice(0, 4000) };
+      }
+
+      return { content: `Unknown action: ${action}`, isError: true };
+    } catch (err) {
+      return { content: `Calendar error: ${(err as Error).message}`, isError: true };
+    }
+  },
+};

package/src/tools/integrations/code-exec.ts

@@ -1,39 +1,39 @@
-import type { MCPTool, MCPToolResult } from '../mcp';
-import { execSync } from 'child_process';
-
-export const CodeExecutionTool: MCPTool = {
-  name: 'code-exec',
-  description: 'Execute JavaScript or Python code in a sandboxed subprocess.',
-  inputSchema: {
-    type: 'object',
-    properties: {
-      language: { type: 'string', enum: ['javascript', 'python'], description: 'Language to execute' },
-      code: { type: 'string', description: 'Code to execute' },
-      timeout: { type: 'number', description: 'Timeout in seconds (default: 10)' },
-    },
-    required: ['language', 'code'],
-  },
-
-  async execute(input: Record<string, unknown>): Promise<MCPToolResult> {
-    const language = String(input.language ?? '');
-    const code = String(input.code ?? '');
-    if (!code) return { content: 'Error: code required', isError: true };
-
-    const timeout = Math.min(Number(input.timeout ?? 10), 30) * 1000;
-
-    try {
-      let result: string;
-      if (language === 'javascript') {
-        result = execSync(`node -e ${JSON.stringify(code)}`, { timeout, encoding: 'utf-8', maxBuffer: 1024 * 1024 });
-      } else if (language === 'python') {
-        result = execSync(`python -c ${JSON.stringify(code)}`, { timeout, encoding: 'utf-8', maxBuffer: 1024 * 1024 });
-      } else {
-        return { content: `Unsupported language: ${language}`, isError: true };
-      }
-      return { content: result.slice(0, 4000) || '(no output)' };
-    } catch (err) {
-      const msg = (err as Error & { stderr?: string }).stderr || (err as Error).message;
-      return { content: `Execution error: ${String(msg).slice(0, 2000)}`, isError: true };
-    }
-  },
-};
+import type { MCPTool, MCPToolResult } from '../mcp';
+import { execSync } from 'child_process';
+
+export const CodeExecutionTool: MCPTool = {
+  name: 'code-exec',
+  description: 'Execute JavaScript or Python code in a sandboxed subprocess.',
+  inputSchema: {
+    type: 'object',
+    properties: {
+      language: { type: 'string', enum: ['javascript', 'python'], description: 'Language to execute' },
+      code: { type: 'string', description: 'Code to execute' },
+      timeout: { type: 'number', description: 'Timeout in seconds (default: 10)' },
+    },
+    required: ['language', 'code'],
+  },
+
+  async execute(input: Record<string, unknown>): Promise<MCPToolResult> {
+    const language = String(input.language ?? '');
+    const code = String(input.code ?? '');
+    if (!code) return { content: 'Error: code required', isError: true };
+
+    const timeout = Math.min(Number(input.timeout ?? 10), 30) * 1000;
+
+    try {
+      let result: string;
+      if (language === 'javascript') {
+        result = execSync(`node -e ${JSON.stringify(code)}`, { timeout, encoding: 'utf-8', maxBuffer: 1024 * 1024 });
+      } else if (language === 'python') {
+        result = execSync(`python -c ${JSON.stringify(code)}`, { timeout, encoding: 'utf-8', maxBuffer: 1024 * 1024 });
+      } else {
+        return { content: `Unsupported language: ${language}`, isError: true };
+      }
+      return { content: result.slice(0, 4000) || '(no output)' };
+    } catch (err) {
+      const msg = (err as Error & { stderr?: string }).stderr || (err as Error).message;
+      return { content: `Execution error: ${String(msg).slice(0, 2000)}`, isError: true };
+    }
+  },
+};