力学笃行

1
import { MCPClient } from "@antigravity/mcp";
2
import { embed } from "./embed";
3

4
const client = new MCPClient({ serverUrl: "https://mcp.mycompany.com" });
5

6
interface Chunk {
7
  id: string;
8
  text: string;
9
  importance: number;
10
  tokens: number;
11
}
12

13
let coreMemory: Chunk[] = [];
14

15
export async function ingest(chunk: Chunk) {
16
  if (chunk.importance < 0.2) {
17
    await client.callTool("archive_store", { chunk });
18
    return;
19
  }
20

21
  const coreTokens = coreMemory.reduce((a, c) => a + c.tokens, 0);
22
  if (coreTokens + chunk.tokens > 400) await compactCore();
23
  coreMemory.push(chunk);
24
}
25

26
async function compactCore() {
27
  const oldest = coreMemory.splice(0, 3);
28
  const merged = oldest.map(c => c.text).join("\n");
29
  const summary = await client.callTool("summarize", {
30
    text: merged,
31
    maxTokens: 50,
32
  });
33
  coreMemory.push({
34
    id: `summary-${Date.now()}`,
35
    text: summary,
36
    importance: 0.7,
37
    tokens: 50,
38
  });
39
}
40

41
export async function retrieveAndInject(query: string) {
42
  const hits = coreMemory.filter(c => c.text.includes(query));
43
  if (hits.length) return hits;
44

45
  const vec = await embed(query);
46
  const results = await client.callTool("vector_search", {
47
    queryVector: vec,
48
    topK: 3,
49
  });
50

51
  for (const r of results) {
52
    await ingest({
53
      id: r.id,
54
      text: r.text,
55
      importance: 0.8,
56
      tokens: Math.round(r.text.split(/\s+/).length * 0.75),
57
    });
58
  }
59
  return results;
60
}