opc-agent 1.1.3 → 1.2.1

package/tests/runtime.test.ts

@@ -1,42 +1,42 @@
-import { describe, it, expect } from 'vitest';
-import { AgentRuntime } from '../src/core/runtime';
-import { createCustomerServiceConfig } from '../src/templates/customer-service';
-
-describe('AgentRuntime', () => {
-  it('should initialize from config object', async () => {
-    const runtime = new AgentRuntime();
-    const config = createCustomerServiceConfig();
-    const agent = await runtime.initialize(config);
-    expect(agent.name).toBe('customer-service');
-    expect(agent.state).toBe('ready');
-  });
-
-  it('should throw if no config loaded', async () => {
-    const runtime = new AgentRuntime();
-    await expect(runtime.initialize()).rejects.toThrow('No config loaded');
-  });
-
-  it('should register skills after initialization', async () => {
-    const runtime = new AgentRuntime();
-    const config = createCustomerServiceConfig();
-    await runtime.initialize(config);
-
-    runtime.registerSkill({
-      name: 'test-skill',
-      description: 'Test',
-      execute: async () => ({ handled: false, confidence: 0 }),
-    });
-    // No throw = success
-  });
-
-  it('should throw registering skill before init', () => {
-    const runtime = new AgentRuntime();
-    expect(() =>
-      runtime.registerSkill({
-        name: 'test',
-        description: 'Test',
-        execute: async () => ({ handled: false, confidence: 0 }),
-      })
-    ).toThrow('Agent not initialized');
-  });
-});
+import { describe, it, expect } from 'vitest';
+import { AgentRuntime } from '../src/core/runtime';
+import { createCustomerServiceConfig } from '../src/templates/customer-service';
+
+describe('AgentRuntime', () => {
+  it('should initialize from config object', async () => {
+    const runtime = new AgentRuntime();
+    const config = createCustomerServiceConfig();
+    const agent = await runtime.initialize(config);
+    expect(agent.name).toBe('customer-service');
+    expect(agent.state).toBe('ready');
+  });
+
+  it('should throw if no config loaded', async () => {
+    const runtime = new AgentRuntime();
+    await expect(runtime.initialize()).rejects.toThrow('No config loaded');
+  });
+
+  it('should register skills after initialization', async () => {
+    const runtime = new AgentRuntime();
+    const config = createCustomerServiceConfig();
+    await runtime.initialize(config);
+
+    runtime.registerSkill({
+      name: 'test-skill',
+      description: 'Test',
+      execute: async () => ({ handled: false, confidence: 0 }),
+    });
+    // No throw = success
+  });
+
+  it('should throw registering skill before init', () => {
+    const runtime = new AgentRuntime();
+    expect(() =>
+      runtime.registerSkill({
+        name: 'test',
+        description: 'Test',
+        execute: async () => ({ handled: false, confidence: 0 }),
+      })
+    ).toThrow('Agent not initialized');
+  });
+});

package/tests/sandbox.test.ts

@@ -1,46 +1,46 @@
-import { describe, it, expect } from 'vitest';
-import { Sandbox } from '../src/core/sandbox';
-
-describe('Security Sandbox', () => {
-  it('should create sandbox with trust level', () => {
-    const sb = new Sandbox({ trustLevel: 'sandbox', agentDir: '/tmp/agent' });
-    expect(sb.trustLevel).toBe('sandbox');
-  });
-
-  it('should restrict shell in sandbox mode', () => {
-    const sb = new Sandbox({ trustLevel: 'sandbox', agentDir: '/tmp/agent' });
-    expect(sb.checkShellAccess()).toBe(false);
-  });
-
-  it('should allow shell in certified mode', () => {
-    const sb = new Sandbox({ trustLevel: 'certified', agentDir: '/tmp/agent' });
-    expect(sb.checkShellAccess()).toBe(true);
-  });
-
-  it('should restrict network in sandbox mode', () => {
-    const sb = new Sandbox({ trustLevel: 'sandbox', agentDir: '/tmp/agent' });
-    expect(sb.checkNetworkAccess('https://api.openai.com')).toBe(false);
-  });
-
-  it('should allow network with allowlist', () => {
-    const sb = new Sandbox({
-      trustLevel: 'sandbox',
-      agentDir: '/tmp/agent',
-      networkAllowlist: ['api.openai.com'],
-    });
-    expect(sb.checkNetworkAccess('https://api.openai.com/v1/chat')).toBe(true);
-    expect(sb.checkNetworkAccess('https://evil.com')).toBe(false);
-  });
-
-  it('should allow wildcard network in listed mode', () => {
-    const sb = new Sandbox({ trustLevel: 'listed', agentDir: '/tmp/agent' });
-    expect(sb.checkNetworkAccess('https://anything.com')).toBe(true);
-  });
-
-  it('should return restrictions snapshot', () => {
-    const sb = new Sandbox({ trustLevel: 'verified', agentDir: '/tmp/agent' });
-    const r = sb.getRestrictions();
-    expect(r.shell).toBe(false);
-    expect(r.network.allowed.length).toBeGreaterThan(0);
-  });
-});
+import { describe, it, expect } from 'vitest';
+import { Sandbox } from '../src/core/sandbox';
+
+describe('Security Sandbox', () => {
+  it('should create sandbox with trust level', () => {
+    const sb = new Sandbox({ trustLevel: 'sandbox', agentDir: '/tmp/agent' });
+    expect(sb.trustLevel).toBe('sandbox');
+  });
+
+  it('should restrict shell in sandbox mode', () => {
+    const sb = new Sandbox({ trustLevel: 'sandbox', agentDir: '/tmp/agent' });
+    expect(sb.checkShellAccess()).toBe(false);
+  });
+
+  it('should allow shell in certified mode', () => {
+    const sb = new Sandbox({ trustLevel: 'certified', agentDir: '/tmp/agent' });
+    expect(sb.checkShellAccess()).toBe(true);
+  });
+
+  it('should restrict network in sandbox mode', () => {
+    const sb = new Sandbox({ trustLevel: 'sandbox', agentDir: '/tmp/agent' });
+    expect(sb.checkNetworkAccess('https://api.openai.com')).toBe(false);
+  });
+
+  it('should allow network with allowlist', () => {
+    const sb = new Sandbox({
+      trustLevel: 'sandbox',
+      agentDir: '/tmp/agent',
+      networkAllowlist: ['api.openai.com'],
+    });
+    expect(sb.checkNetworkAccess('https://api.openai.com/v1/chat')).toBe(true);
+    expect(sb.checkNetworkAccess('https://evil.com')).toBe(false);
+  });
+
+  it('should allow wildcard network in listed mode', () => {
+    const sb = new Sandbox({ trustLevel: 'listed', agentDir: '/tmp/agent' });
+    expect(sb.checkNetworkAccess('https://anything.com')).toBe(true);
+  });
+
+  it('should return restrictions snapshot', () => {
+    const sb = new Sandbox({ trustLevel: 'verified', agentDir: '/tmp/agent' });
+    const r = sb.getRestrictions();
+    expect(r.shell).toBe(false);
+    expect(r.network.allowed.length).toBeGreaterThan(0);
+  });
+});

package/tests/security.test.ts

@@ -1,60 +1,60 @@
-import { describe, it, expect } from 'vitest';
-import { sanitizeInput, detectInjection, APIKeyManager } from '../src/core/security';
-
-describe('Security', () => {
-  describe('sanitizeInput', () => {
-    it('strips script tags', () => {
-      expect(sanitizeInput('<script>alert(1)</script>hello')).not.toContain('<script');
-    });
-    it('encodes HTML entities', () => {
-      const result = sanitizeInput('a < b & c > d');
-      expect(result).toContain('&lt;');
-      expect(result).toContain('&amp;');
-      expect(result).toContain('&gt;');
-    });
-  });
-
-  describe('detectInjection', () => {
-    it('detects XSS', () => {
-      const r = detectInjection('<script>alert(1)</script>');
-      expect(r.safe).toBe(false);
-      expect(r.threats).toContain('xss');
-    });
-    it('passes clean input', () => {
-      expect(detectInjection('Hello world')).toEqual({ safe: true, threats: [] });
-    });
-  });
-
-  describe('APIKeyManager', () => {
-    it('add, validate, revoke', () => {
-      const mgr = new APIKeyManager();
-      mgr.addKey('key1', { label: 'test' });
-      expect(mgr.isValid('key1')).toBe(true);
-      expect(mgr.isValid('key2')).toBe(false);
-      mgr.revokeKey('key1');
-      expect(mgr.isValid('key1')).toBe(false);
-    });
-
-    it('rotate key', () => {
-      const mgr = new APIKeyManager();
-      mgr.addKey('old');
-      expect(mgr.rotateKey('old', 'new')).toBe(true);
-      expect(mgr.isValid('old')).toBe(false);
-      expect(mgr.isValid('new')).toBe(true);
-    });
-
-    it('expires keys', () => {
-      const mgr = new APIKeyManager();
-      mgr.addKey('expired', { expiresAt: Date.now() - 1000 });
-      expect(mgr.isValid('expired')).toBe(false);
-    });
-
-    it('listActive filters', () => {
-      const mgr = new APIKeyManager();
-      mgr.addKey('a');
-      mgr.addKey('b');
-      mgr.revokeKey('b');
-      expect(mgr.listActive().length).toBe(1);
-    });
-  });
-});
+import { describe, it, expect } from 'vitest';
+import { sanitizeInput, detectInjection, APIKeyManager } from '../src/core/security';
+
+describe('Security', () => {
+  describe('sanitizeInput', () => {
+    it('strips script tags', () => {
+      expect(sanitizeInput('<script>alert(1)</script>hello')).not.toContain('<script');
+    });
+    it('encodes HTML entities', () => {
+      const result = sanitizeInput('a < b & c > d');
+      expect(result).toContain('&lt;');
+      expect(result).toContain('&amp;');
+      expect(result).toContain('&gt;');
+    });
+  });
+
+  describe('detectInjection', () => {
+    it('detects XSS', () => {
+      const r = detectInjection('<script>alert(1)</script>');
+      expect(r.safe).toBe(false);
+      expect(r.threats).toContain('xss');
+    });
+    it('passes clean input', () => {
+      expect(detectInjection('Hello world')).toEqual({ safe: true, threats: [] });
+    });
+  });
+
+  describe('APIKeyManager', () => {
+    it('add, validate, revoke', () => {
+      const mgr = new APIKeyManager();
+      mgr.addKey('key1', { label: 'test' });
+      expect(mgr.isValid('key1')).toBe(true);
+      expect(mgr.isValid('key2')).toBe(false);
+      mgr.revokeKey('key1');
+      expect(mgr.isValid('key1')).toBe(false);
+    });
+
+    it('rotate key', () => {
+      const mgr = new APIKeyManager();
+      mgr.addKey('old');
+      expect(mgr.rotateKey('old', 'new')).toBe(true);
+      expect(mgr.isValid('old')).toBe(false);
+      expect(mgr.isValid('new')).toBe(true);
+    });
+
+    it('expires keys', () => {
+      const mgr = new APIKeyManager();
+      mgr.addKey('expired', { expiresAt: Date.now() - 1000 });
+      expect(mgr.isValid('expired')).toBe(false);
+    });
+
+    it('listActive filters', () => {
+      const mgr = new APIKeyManager();
+      mgr.addKey('a');
+      mgr.addKey('b');
+      mgr.revokeKey('b');
+      expect(mgr.listActive().length).toBe(1);
+    });
+  });
+});

package/tests/streaming.test.ts

@@ -0,0 +1,109 @@
+import { describe, it, expect, vi } from 'vitest';
+import { StreamingManager, StreamableResponse } from '../src/core/streaming';
+import type { StreamChunk } from '../src/core/streaming';
+
+describe('StreamableResponse', () => {
+  it('should collect chunks and build text', () => {
+    const stream = new StreamableResponse('test-1');
+    stream.push({ id: '0', type: 'text', data: 'Hello ', timestamp: Date.now() });
+    stream.push({ id: '1', type: 'text', data: 'World', timestamp: Date.now() });
+    expect(stream.getText()).toBe('Hello World');
+    expect(stream.length).toBe(2);
+  });
+
+  it('should emit chunk events', () => {
+    const stream = new StreamableResponse('test-2');
+    const received: StreamChunk[] = [];
+    stream.on('chunk', (c: StreamChunk) => received.push(c));
+    stream.push({ id: '0', type: 'text', data: 'hi', timestamp: Date.now() });
+    expect(received.length).toBe(1);
+  });
+
+  it('should emit end event', () => {
+    const stream = new StreamableResponse('test-3');
+    let ended = false;
+    stream.on('end', () => { ended = true; });
+    stream.end();
+    expect(ended).toBe(true);
+    expect(stream.isEnded).toBe(true);
+  });
+
+  it('should apply backpressure at highWaterMark', () => {
+    const stream = new StreamableResponse('test-4', { highWaterMark: 2 });
+    const chunk = (): StreamChunk => ({ id: 'c', type: 'text', data: 'x', timestamp: Date.now() });
+    stream.push(chunk()); // 1 — ok
+    const ok = stream.push(chunk()); // 2 — triggers backpressure
+    expect(ok).toBe(false);
+    expect(stream.isPaused).toBe(true);
+  });
+
+  it('should flush buffer on resume', () => {
+    const stream = new StreamableResponse('test-5', { highWaterMark: 1 });
+    const received: StreamChunk[] = [];
+    stream.on('chunk', (c: StreamChunk) => received.push(c));
+    stream.push({ id: '0', type: 'text', data: 'a', timestamp: Date.now() });
+    // Now paused — next chunk goes to buffer
+    stream.push({ id: '1', type: 'text', data: 'b', timestamp: Date.now() });
+    expect(received.length).toBe(1);
+    stream.resume();
+    expect(received.length).toBe(2);
+    expect(stream.isPaused).toBe(false);
+  });
+
+  it('should reject pushes after end', () => {
+    const stream = new StreamableResponse('test-6');
+    stream.end();
+    const ok = stream.push({ id: '0', type: 'text', data: 'late', timestamp: Date.now() });
+    expect(ok).toBe(false);
+    expect(stream.length).toBe(0);
+  });
+});
+
+describe('StreamingManager', () => {
+  it('should create and manage streams', () => {
+    const mgr = new StreamingManager();
+    const stream = mgr.createStream();
+    expect(stream.id).toMatch(/^stream_/);
+    expect(mgr.activeCount).toBe(1);
+  });
+
+  it('should write chunks and end stream', () => {
+    const mgr = new StreamingManager();
+    const stream = mgr.createStream();
+    mgr.writeChunk(stream.id, 'hello');
+    mgr.writeChunk(stream.id, ' world');
+    mgr.endStream(stream.id);
+    expect(stream.getText()).toBe('hello world');
+    expect(stream.isEnded).toBe(true);
+  });
+
+  it('should return false for unknown stream writes', () => {
+    const mgr = new StreamingManager();
+    expect(mgr.writeChunk('nonexistent', 'data')).toBe(false);
+  });
+
+  it('should format SSE correctly', () => {
+    const chunk: StreamChunk = { id: 'c1', type: 'text', data: 'hi', timestamp: 123 };
+    const sse = StreamingManager.formatSSE(chunk);
+    expect(sse).toContain('event: text');
+    expect(sse).toContain('id: c1');
+    expect(sse).toContain('"data":"hi"');
+  });
+
+  it('should pipe to SSE response', () => {
+    const mgr = new StreamingManager();
+    const stream = mgr.createStream();
+    const written: string[] = [];
+    const mockRes = {
+      write: (d: string) => { written.push(d); return true; },
+      end: vi.fn(),
+      setHeader: vi.fn(),
+    };
+    StreamingManager.pipeSSE(stream, mockRes, { heartbeatInterval: 100_000 });
+    mgr.writeChunk(stream.id, 'data');
+    mgr.endStream(stream.id);
+    expect(written.length).toBeGreaterThan(0);
+    expect(mockRes.end).toHaveBeenCalled();
+    expect(mockRes.setHeader).toHaveBeenCalledWith('Content-Type', 'text/event-stream');
+  });
+});

package/tests/templates.test.ts

@@ -1,77 +1,77 @@
-import { describe, it, expect } from 'vitest';
-import { createLegalAssistantConfig, ContractReviewSkill, ComplianceCheckSkill } from '../src/templates/legal-assistant';
-import { createFinancialAdvisorConfig, BudgetAnalysisSkill, FinancialPlanningSkill } from '../src/templates/financial-advisor';
-import { createExecutiveAssistantConfig, CalendarSkill, EmailDraftSkill, MeetingPrepSkill } from '../src/templates/executive-assistant';
-import type { AgentContext, Message, MemoryStore } from '../src/core/types';
-
-const mockMemory: MemoryStore = {
-  get: async () => null, set: async () => {}, getConversation: async () => [],
-  addMessage: async () => {}, clear: async () => {},
-};
-const ctx: AgentContext = { agentName: 'test', sessionId: 's1', messages: [], memory: mockMemory, metadata: {} };
-const msg = (content: string): Message => ({ id: '1', role: 'user', content, timestamp: Date.now() });
-
-describe('Legal Assistant Template', () => {
-  it('should create valid config', () => {
-    const config = createLegalAssistantConfig();
-    expect(config.metadata.name).toBe('legal-assistant');
-    expect(config.spec.skills).toHaveLength(2);
-  });
-
-  it('ContractReviewSkill matches contract terms', async () => {
-    const skill = new ContractReviewSkill();
-    const r = await skill.execute(ctx, msg('What is force majeure?'));
-    expect(r.handled).toBe(true);
-  });
-
-  it('ComplianceCheckSkill matches GDPR', async () => {
-    const skill = new ComplianceCheckSkill();
-    const r = await skill.execute(ctx, msg('GDPR requirements'));
-    expect(r.handled).toBe(true);
-  });
-});
-
-describe('Financial Advisor Template', () => {
-  it('should create valid config', () => {
-    const config = createFinancialAdvisorConfig();
-    expect(config.metadata.name).toBe('financial-advisor');
-  });
-
-  it('BudgetAnalysisSkill matches budget queries', async () => {
-    const skill = new BudgetAnalysisSkill();
-    const r = await skill.execute(ctx, msg('Help with my budget'));
-    expect(r.handled).toBe(true);
-  });
-
-  it('FinancialPlanningSkill matches investment queries', async () => {
-    const skill = new FinancialPlanningSkill();
-    const r = await skill.execute(ctx, msg('How to invest'));
-    expect(r.handled).toBe(true);
-  });
-});
-
-describe('Executive Assistant Template', () => {
-  it('should create valid config', () => {
-    const config = createExecutiveAssistantConfig();
-    expect(config.metadata.name).toBe('executive-assistant');
-    expect(config.spec.skills).toHaveLength(3);
-  });
-
-  it('CalendarSkill matches scheduling', async () => {
-    const skill = new CalendarSkill();
-    const r = await skill.execute(ctx, msg('Schedule a meeting'));
-    expect(r.handled).toBe(true);
-  });
-
-  it('EmailDraftSkill matches email queries', async () => {
-    const skill = new EmailDraftSkill();
-    const r = await skill.execute(ctx, msg('Draft an email'));
-    expect(r.handled).toBe(true);
-  });
-
-  it('MeetingPrepSkill matches prep queries', async () => {
-    const skill = new MeetingPrepSkill();
-    const r = await skill.execute(ctx, msg('Prepare the agenda'));
-    expect(r.handled).toBe(true);
-  });
-});
+import { describe, it, expect } from 'vitest';
+import { createLegalAssistantConfig, ContractReviewSkill, ComplianceCheckSkill } from '../src/templates/legal-assistant';
+import { createFinancialAdvisorConfig, BudgetAnalysisSkill, FinancialPlanningSkill } from '../src/templates/financial-advisor';
+import { createExecutiveAssistantConfig, CalendarSkill, EmailDraftSkill, MeetingPrepSkill } from '../src/templates/executive-assistant';
+import type { AgentContext, Message, MemoryStore } from '../src/core/types';
+
+const mockMemory: MemoryStore = {
+  get: async () => null, set: async () => {}, getConversation: async () => [],
+  addMessage: async () => {}, clear: async () => {},
+};
+const ctx: AgentContext = { agentName: 'test', sessionId: 's1', messages: [], memory: mockMemory, metadata: {} };
+const msg = (content: string): Message => ({ id: '1', role: 'user', content, timestamp: Date.now() });
+
+describe('Legal Assistant Template', () => {
+  it('should create valid config', () => {
+    const config = createLegalAssistantConfig();
+    expect(config.metadata.name).toBe('legal-assistant');
+    expect(config.spec.skills).toHaveLength(2);
+  });
+
+  it('ContractReviewSkill matches contract terms', async () => {
+    const skill = new ContractReviewSkill();
+    const r = await skill.execute(ctx, msg('What is force majeure?'));
+    expect(r.handled).toBe(true);
+  });
+
+  it('ComplianceCheckSkill matches GDPR', async () => {
+    const skill = new ComplianceCheckSkill();
+    const r = await skill.execute(ctx, msg('GDPR requirements'));
+    expect(r.handled).toBe(true);
+  });
+});
+
+describe('Financial Advisor Template', () => {
+  it('should create valid config', () => {
+    const config = createFinancialAdvisorConfig();
+    expect(config.metadata.name).toBe('financial-advisor');
+  });
+
+  it('BudgetAnalysisSkill matches budget queries', async () => {
+    const skill = new BudgetAnalysisSkill();
+    const r = await skill.execute(ctx, msg('Help with my budget'));
+    expect(r.handled).toBe(true);
+  });
+
+  it('FinancialPlanningSkill matches investment queries', async () => {
+    const skill = new FinancialPlanningSkill();
+    const r = await skill.execute(ctx, msg('How to invest'));
+    expect(r.handled).toBe(true);
+  });
+});
+
+describe('Executive Assistant Template', () => {
+  it('should create valid config', () => {
+    const config = createExecutiveAssistantConfig();
+    expect(config.metadata.name).toBe('executive-assistant');
+    expect(config.spec.skills).toHaveLength(3);
+  });
+
+  it('CalendarSkill matches scheduling', async () => {
+    const skill = new CalendarSkill();
+    const r = await skill.execute(ctx, msg('Schedule a meeting'));
+    expect(r.handled).toBe(true);
+  });
+
+  it('EmailDraftSkill matches email queries', async () => {
+    const skill = new EmailDraftSkill();
+    const r = await skill.execute(ctx, msg('Draft an email'));
+    expect(r.handled).toBe(true);
+  });
+
+  it('MeetingPrepSkill matches prep queries', async () => {
+    const skill = new MeetingPrepSkill();
+    const r = await skill.execute(ctx, msg('Prepare the agenda'));
+    expect(r.handled).toBe(true);
+  });
+});