xiaoju
c3f3b822f1
- Use CF Workers AI bge-base-en-v1.5 for embeddings - Deploy stores capability embedding in KV - Query uses cosine similarity (find) and MMR (explore) - Query embedding cached in KV (1h TTL) - Fallback to string matching for capabilities without embeddings - Mock embedding service for unit tests
82 lines
2.4 KiB
TypeScript
82 lines
2.4 KiB
TypeScript
import { describe, it, expect, beforeEach } from 'vitest'
|
|
import { createMockKv, createMockCfApi, makeRequest, MockEmbeddingService } from './setup.js'
|
|
import { WorkerPool } from '../src/backend/worker-pool.js'
|
|
import { AuthModule } from '../src/auth.js'
|
|
import { KvStore } from '../src/kv.js'
|
|
import { handleRequest } from '../src/router.js'
|
|
|
|
describe('S1: 部署能力', () => {
|
|
let mockKv: KVNamespace
|
|
let mockCf: ReturnType<typeof createMockCfApi>
|
|
let mockEmbed: MockEmbeddingService
|
|
let pool: WorkerPool
|
|
let auth: AuthModule
|
|
let kv: KvStore
|
|
|
|
beforeEach(async () => {
|
|
mockKv = createMockKv()
|
|
mockCf = createMockCfApi()
|
|
mockEmbed = new MockEmbeddingService()
|
|
pool = new WorkerPool(mockKv, mockCf.cfApi, mockEmbed as any)
|
|
kv = new KvStore(mockKv)
|
|
auth = new AuthModule(kv)
|
|
|
|
// Set unified deploy token
|
|
await auth.setToken('deploy-token')
|
|
})
|
|
|
|
it('should deploy a capability via API', async () => {
|
|
const req = makeRequest('POST', '/_api/deploy', {
|
|
token: 'deploy-token',
|
|
body: {
|
|
name: 'ping',
|
|
code: "export default { fetch() { return new Response('pong') } }",
|
|
type: 'normal',
|
|
},
|
|
})
|
|
|
|
const resp = await handleRequest(req, { SIGIL_KV: mockKv, backend: pool, auth, kv })
|
|
expect(resp.status).toBe(201)
|
|
|
|
const body = await resp.json() as {
|
|
capability: string
|
|
url: string
|
|
cold_start: boolean
|
|
}
|
|
expect(body.capability).toBe('ping')
|
|
expect(body.url).toBe('https://sigil.shazhou.workers.dev/run/ping')
|
|
expect(body.cold_start).toBe(false)
|
|
})
|
|
|
|
it('should call CfApi.deployWorker', async () => {
|
|
await pool.deploy({
|
|
name: 'ping',
|
|
code: "export default { fetch() { return new Response('pong') } }",
|
|
type: 'normal',
|
|
})
|
|
|
|
expect(mockCf.deployCalls()).toContain('s-ping')
|
|
})
|
|
|
|
it('should write KV entries (code, meta, lru, route)', async () => {
|
|
await pool.deploy({
|
|
name: 'ping',
|
|
code: "export default { fetch() { return new Response('pong') } }",
|
|
type: 'normal',
|
|
})
|
|
|
|
const code = await kv.getCode('ping')
|
|
expect(code).toBeTruthy()
|
|
|
|
const meta = await kv.getMeta('ping')
|
|
expect(meta?.type).toBe('normal')
|
|
|
|
const lru = await kv.getLru('ping')
|
|
expect(lru?.deployed).toBe(true)
|
|
expect(lru?.access_count).toBe(0)
|
|
|
|
const route = await kv.getRoute('ping')
|
|
expect(route?.worker_name).toBe('s-ping')
|
|
})
|
|
})
|