feat(backends): add optional pi_embedded backend and config wiring

This commit is contained in:
William Valentin
2026-02-23 21:12:52 -08:00
parent 0af44330b5
commit ac61c9c3fb
9 changed files with 426 additions and 10 deletions
+100
View File
@@ -0,0 +1,100 @@
import { mkdtempSync, rmSync, writeFileSync } from 'fs';
import { join } from 'path';
import { tmpdir } from 'os';
import { pathToFileURL } from 'url';
import { describe, expect, it } from 'vitest';
import { PiEmbeddedBackend } from './piEmbedded.js';
function createModule(source: string): { moduleUrl: string; cleanup: () => void } {
const dir = mkdtempSync(join(tmpdir(), 'flynn-pi-embedded-'));
const file = join(dir, 'module.mjs');
writeFileSync(file, source, 'utf-8');
return {
moduleUrl: pathToFileURL(file).href,
cleanup: () => rmSync(dir, { recursive: true, force: true }),
};
}
describe('PiEmbeddedBackend', () => {
it('returns text from a createAgentSession/run response object', async () => {
const mod = createModule(`
export function createAgentSession() {
return {
run(payload) {
return { text: "pi says: " + payload.input };
},
};
}
`);
try {
const backend = new PiEmbeddedBackend({ module: mod.moduleUrl, timeoutMs: 2000 });
const result = await backend.process({ prompt: 'hello', history: [] });
expect(result).toBe('pi says: hello');
} finally {
mod.cleanup();
}
});
it('falls back to string payload when object payload is rejected', async () => {
const mod = createModule(`
export function createAgentSession() {
return {
run(payload) {
if (typeof payload !== "string") {
throw new Error("expected string payload");
}
return "echo " + payload;
},
};
}
`);
try {
const backend = new PiEmbeddedBackend({ module: mod.moduleUrl, timeoutMs: 2000 });
const result = await backend.process({ prompt: 'hello', history: [] });
expect(result).toContain('echo USER: hello');
} finally {
mod.cleanup();
}
});
it('throws when module has no supported session factory', async () => {
const mod = createModule('export const version = "0.0.0";');
try {
const backend = new PiEmbeddedBackend({ module: mod.moduleUrl, timeoutMs: 2000 });
await expect(backend.process({ prompt: 'hello', history: [] }))
.rejects.toThrow('supported session factory');
} finally {
mod.cleanup();
}
});
it('throws when module cannot be loaded', async () => {
const backend = new PiEmbeddedBackend({ module: '/definitely/missing/pi-module.mjs', timeoutMs: 2000 });
await expect(backend.process({ prompt: 'hello', history: [] }))
.rejects.toThrow('Failed to load Pi embedded runtime module');
});
it('times out slow Pi requests', async () => {
const mod = createModule(`
export function createAgentSession() {
return {
run() {
return new Promise(() => {});
},
};
}
`);
try {
const backend = new PiEmbeddedBackend({ module: mod.moduleUrl, timeoutMs: 10 });
await expect(backend.process({ prompt: 'hello', history: [] }))
.rejects.toThrow('timed out');
} finally {
mod.cleanup();
}
});
});