chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,71 @@
|
||||
import { describe, expect, it } from "vitest";
|
||||
import { readFileSync } from "node:fs";
|
||||
|
||||
const expectedHermesHooks = [
|
||||
"prefetch",
|
||||
"sync_turn",
|
||||
"on_session_end",
|
||||
"on_pre_compress",
|
||||
"on_memory_write",
|
||||
"system_prompt_block",
|
||||
];
|
||||
|
||||
function readHermesPluginHooks(): string[] {
|
||||
const manifest = readFileSync("integrations/hermes/plugin.yaml", "utf8");
|
||||
const hooks: string[] = [];
|
||||
let inHooks = false;
|
||||
|
||||
for (const line of manifest.split(/\r?\n/)) {
|
||||
if (line.trim() === "hooks:") {
|
||||
inHooks = true;
|
||||
continue;
|
||||
}
|
||||
if (!inHooks) continue;
|
||||
if (line.trim() === "") continue;
|
||||
if (!line.startsWith(" ")) break;
|
||||
|
||||
const match = line.match(/^\s*-\s*([A-Za-z_][A-Za-z0-9_]*)\s*$/);
|
||||
if (match) hooks.push(match[1]);
|
||||
}
|
||||
|
||||
return hooks;
|
||||
}
|
||||
|
||||
function isHermesLifecycleHook(methodName: string): boolean {
|
||||
return (
|
||||
methodName === "prefetch" ||
|
||||
methodName === "sync_turn" ||
|
||||
methodName === "system_prompt_block" ||
|
||||
methodName.startsWith("on_")
|
||||
);
|
||||
}
|
||||
|
||||
function readAgentMemoryProviderHookMethods(): string[] {
|
||||
const source = readFileSync("integrations/hermes/__init__.py", "utf8");
|
||||
const methods: string[] = [];
|
||||
const providerMethodPattern = /^ def ([a-z_][a-z0-9_]*)\(/gm;
|
||||
|
||||
for (const match of source.matchAll(providerMethodPattern)) {
|
||||
const methodName = match[1];
|
||||
if (isHermesLifecycleHook(methodName)) methods.push(methodName);
|
||||
}
|
||||
|
||||
return methods;
|
||||
}
|
||||
|
||||
describe("Hermes plugin manifest", () => {
|
||||
it("declares every implemented lifecycle hook", () => {
|
||||
const declaredHooks = readHermesPluginHooks();
|
||||
const implementedHooks = readAgentMemoryProviderHookMethods();
|
||||
|
||||
expect([...declaredHooks].sort()).toEqual([...implementedHooks].sort());
|
||||
expect(declaredHooks).toEqual(expectedHermesHooks);
|
||||
});
|
||||
|
||||
it("preloads AGENTMEMORY_URL default at import time", () => {
|
||||
const source = readFileSync("integrations/hermes/__init__.py", "utf8");
|
||||
expect(source).toMatch(
|
||||
/os\.environ\.setdefault\(\s*["']AGENTMEMORY_URL["']\s*,\s*DEFAULT_BASE_URL\s*\)/,
|
||||
);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user