mu-core 0.15.0 → 0.16.1

package/src/provider/registry.test.ts

@@ -1,37 +0,0 @@
-import { describe, expect, it } from 'bun:test';
-import type { Provider } from './adapter';
-import { createProviderRegistry } from './registry';
-
-function fakeProvider(id: string): Provider {
-  return {
-    id,
-    async *streamChat() {
-      /* no chunks */
-    },
-    async listModels() {
-      return [];
-    },
-  };
-}
-
-describe('ProviderRegistry', () => {
-  it('register / get / list', () => {
-    const r = createProviderRegistry();
-    r.register(fakeProvider('openai'));
-    expect(r.get('openai')?.id).toBe('openai');
-    expect(r.list()).toHaveLength(1);
-  });
-
-  it('rejects duplicate ids', () => {
-    const r = createProviderRegistry();
-    r.register(fakeProvider('openai'));
-    expect(() => r.register(fakeProvider('openai'))).toThrow();
-  });
-
-  it('unregister callback removes', () => {
-    const r = createProviderRegistry();
-    const off = r.register(fakeProvider('a'));
-    off();
-    expect(r.get('a')).toBeUndefined();
-  });
-});

package/src/provider/registry.ts

@@ -1,26 +0,0 @@
-import type { Provider } from './adapter';
-
-export interface ProviderRegistry {
-  register: (provider: Provider) => () => void;
-  get: (id: string) => Provider | undefined;
-  list: () => Provider[];
-}
-
-export function createProviderRegistry(): ProviderRegistry {
-  const providers = new Map<string, Provider>();
-  return {
-    register(p) {
-      if (providers.has(p.id)) throw new Error(`Provider already registered: ${p.id}`);
-      providers.set(p.id, p);
-      return () => {
-        providers.delete(p.id);
-      };
-    },
-    get(id) {
-      return providers.get(id);
-    },
-    list() {
-      return Array.from(providers.values());
-    },
-  };
-}

package/src/provider/transport.test.ts

@@ -1,58 +0,0 @@
-import { describe, expect, it } from 'bun:test';
-import { fetchWithIdleTimeout, readNDJSON, readSSE } from './transport';
-
-function bodyResponse(text: string): Response {
-  return new Response(text, { status: 200, headers: { 'content-type': 'text/event-stream' } });
-}
-
-describe('readSSE', () => {
-  it('yields data lines from one event', async () => {
-    const r = bodyResponse('data: hello\n\n');
-    const out: string[] = [];
-    for await (const v of readSSE(r)) out.push(v);
-    expect(out).toEqual(['hello']);
-  });
-
-  it('handles multiple events split by blank lines', async () => {
-    const r = bodyResponse('data: a\n\ndata: b\n\n');
-    const out: string[] = [];
-    for await (const v of readSSE(r)) out.push(v);
-    expect(out).toEqual(['a', 'b']);
-  });
-
-  it('ignores non-data lines', async () => {
-    const r = bodyResponse(': comment\nevent: x\ndata: payload\n\n');
-    const out: string[] = [];
-    for await (const v of readSSE(r)) out.push(v);
-    expect(out).toEqual(['payload']);
-  });
-});
-
-describe('readNDJSON', () => {
-  it('yields one line per JSON record', async () => {
-    const r = bodyResponse('{"a":1}\n{"b":2}\n');
-    const out: string[] = [];
-    for await (const v of readNDJSON(r)) out.push(v);
-    expect(out).toEqual(['{"a":1}', '{"b":2}']);
-  });
-
-  it('emits trailing line without final newline', async () => {
-    const r = bodyResponse('{"a":1}');
-    const out: string[] = [];
-    for await (const v of readNDJSON(r)) out.push(v);
-    expect(out).toEqual(['{"a":1}']);
-  });
-});
-
-describe('fetchWithIdleTimeout', () => {
-  it('cancels idle timer when fetch itself rejects', async () => {
-    // Use a URL that rejects immediately. We can't directly observe the
-    // timer, but we can check that the call rejects synchronously and that
-    // the function does not leak a hanging timer (verified via the test
-    // process exiting promptly after this expect).
-    const start = Date.now();
-    await expect(fetchWithIdleTimeout('http://127.0.0.1:1', {}, 5000)).rejects.toBeDefined();
-    // If the timer leaked, the test would block until 5s; we assert <1s.
-    expect(Date.now() - start).toBeLessThan(1000);
-  });
-});

package/src/provider/transport.ts

@@ -1,103 +0,0 @@
-/**
- * Transport primitives: SSE / NDJSON readers + fetch-with-idle-timeout.
- *
- * Used by `createProvider` to drive the lower-level HTTP plumbing without
- * each adapter having to re-implement framing.
- */
-
-export async function* readSSE(response: Response, signal?: AbortSignal): AsyncGenerator<string> {
-  if (!response.body) throw new Error('Response has no body');
-  const reader = response.body.getReader();
-  const decoder = new TextDecoder();
-  let buffer = '';
-  try {
-    while (true) {
-      if (signal?.aborted) throw new Error('aborted');
-      const { value, done } = await reader.read();
-      if (done) break;
-      buffer += decoder.decode(value, { stream: true });
-      // SSE events are separated by blank lines.
-      let idx: number;
-      // biome-ignore lint/suspicious/noAssignInExpressions: idiomatic
-      while ((idx = buffer.indexOf('\n\n')) !== -1) {
-        const event = buffer.slice(0, idx);
-        buffer = buffer.slice(idx + 2);
-        for (const line of event.split('\n')) {
-          if (line.startsWith('data:')) {
-            yield line.slice(5).trim();
-          }
-        }
-      }
-    }
-  } finally {
-    reader.releaseLock();
-  }
-}
-
-export async function* readNDJSON(response: Response, signal?: AbortSignal): AsyncGenerator<string> {
-  if (!response.body) throw new Error('Response has no body');
-  const reader = response.body.getReader();
-  const decoder = new TextDecoder();
-  let buffer = '';
-  try {
-    while (true) {
-      if (signal?.aborted) throw new Error('aborted');
-      const { value, done } = await reader.read();
-      if (done) break;
-      buffer += decoder.decode(value, { stream: true });
-      let nl: number;
-      // biome-ignore lint/suspicious/noAssignInExpressions: idiomatic
-      while ((nl = buffer.indexOf('\n')) !== -1) {
-        const line = buffer.slice(0, nl).trim();
-        buffer = buffer.slice(nl + 1);
-        if (line) yield line;
-      }
-    }
-    if (buffer.trim()) yield buffer.trim();
-  } finally {
-    reader.releaseLock();
-  }
-}
-
-/**
- * Fetch with an idle timeout: aborts if no bytes are received from the
- * server for `timeoutMs` consecutive milliseconds.
- *
- * Returns the Response *and* a `resetIdle()` callback the caller invokes
- * each time it consumes a chunk so the timer slides forward.
- */
-export async function fetchWithIdleTimeout(
-  url: string,
-  init: RequestInit,
-  timeoutMs: number,
-): Promise<{ response: Response; resetIdle: () => void; cancel: () => void }> {
-  const ctl = new AbortController();
-  const upstream = init.signal;
-  let timer: ReturnType<typeof setTimeout> | null = null;
-  const armed = { value: true };
-  const resetIdle = () => {
-    if (!armed.value) return;
-    if (timer) clearTimeout(timer);
-    timer = setTimeout(() => ctl.abort(), timeoutMs);
-  };
-  const cancel = () => {
-    armed.value = false;
-    if (timer) clearTimeout(timer);
-  };
-  if (upstream) {
-    if (upstream.aborted) ctl.abort();
-    else upstream.addEventListener('abort', () => ctl.abort(), { once: true });
-  }
-  resetIdle();
-  let response: Response;
-  try {
-    response = await fetch(url, { ...init, signal: ctl.signal });
-  } catch (err) {
-    // Cancel the idle timer so a failed fetch doesn't keep the event loop
-    // alive for `timeoutMs` after the rejection has propagated.
-    cancel();
-    throw err;
-  }
-  resetIdle();
-  return { response, resetIdle, cancel };
-}

package/src/registry.context.test.ts

@@ -1,71 +0,0 @@
-import { describe, expect, it } from 'bun:test';
-import type { AgentSourceRegistry, Plugin } from './plugin';
-import { PluginRegistry } from './registry';
-
-describe('PluginRegistry context propagation', () => {
-  it('setAgentsRegistry from one plugin reaches subsequent plugins', async () => {
-    const reg = new PluginRegistry({ cwd: '/tmp', config: {} });
-    const calls: Array<{ name: string; sawAgents: boolean }> = [];
-
-    const publisher: Plugin = {
-      name: 'publisher',
-      activate(ctx) {
-        const sourceReg: AgentSourceRegistry = {
-          registerSource: () =>
-            function unregister(): void {
-              /* noop */
-            },
-        };
-        ctx.setAgentsRegistry?.(sourceReg);
-        calls.push({ name: 'publisher', sawAgents: !!ctx.agents });
-      },
-    };
-
-    const consumer: Plugin = {
-      name: 'consumer',
-      activate(ctx) {
-        calls.push({ name: 'consumer', sawAgents: !!ctx.agents });
-        ctx.agents?.registerSource('/some/dir');
-      },
-    };
-
-    await reg.register(publisher);
-    await reg.register(consumer);
-
-    expect(calls.find((c) => c.name === 'consumer')?.sawAgents).toBe(true);
-  });
-
-  it('plugins activated before publisher do not see agents', async () => {
-    const reg = new PluginRegistry({ cwd: '/tmp', config: {} });
-    let earlySaw = false;
-    let lateSaw = false;
-    const early: Plugin = {
-      name: 'early',
-      activate(ctx) {
-        earlySaw = !!ctx.agents;
-      },
-    };
-    const publisher: Plugin = {
-      name: 'pub',
-      activate(ctx) {
-        ctx.setAgentsRegistry?.({
-          registerSource: () =>
-            function unregister(): void {
-              /* noop */
-            },
-        });
-      },
-    };
-    const late: Plugin = {
-      name: 'late',
-      activate(ctx) {
-        lateSaw = !!ctx.agents;
-      },
-    };
-    await reg.register(early);
-    await reg.register(publisher);
-    await reg.register(late);
-    expect(earlySaw).toBe(false);
-    expect(lateSaw).toBe(true);
-  });
-});