mirror of https://github.com/openclaw/openclaw.git
64 lines
1.6 KiB
TypeScript
64 lines
1.6 KiB
TypeScript
import { expect } from "vitest";
|
|
import type { OpenClawConfig } from "../api.js";
|
|
import { createMemoryGetTool, createMemorySearchTool } from "./tools.js";
|
|
|
|
export function asOpenClawConfig(config: Partial<OpenClawConfig>): OpenClawConfig {
|
|
return config as OpenClawConfig;
|
|
}
|
|
|
|
export function createDefaultMemoryToolConfig(): OpenClawConfig {
|
|
return asOpenClawConfig({ agents: { list: [{ id: "main", default: true }] } });
|
|
}
|
|
|
|
export function createMemorySearchToolOrThrow(params?: {
|
|
config?: OpenClawConfig;
|
|
agentSessionKey?: string;
|
|
}) {
|
|
const tool = createMemorySearchTool({
|
|
config: params?.config ?? createDefaultMemoryToolConfig(),
|
|
...(params?.agentSessionKey ? { agentSessionKey: params.agentSessionKey } : {}),
|
|
});
|
|
if (!tool) {
|
|
throw new Error("tool missing");
|
|
}
|
|
return tool;
|
|
}
|
|
|
|
export function createMemoryGetToolOrThrow(
|
|
config: OpenClawConfig = createDefaultMemoryToolConfig(),
|
|
) {
|
|
const tool = createMemoryGetTool({ config });
|
|
if (!tool) {
|
|
throw new Error("tool missing");
|
|
}
|
|
return tool;
|
|
}
|
|
|
|
export function createAutoCitationsMemorySearchTool(agentSessionKey: string) {
|
|
return createMemorySearchToolOrThrow({
|
|
config: asOpenClawConfig({
|
|
memory: { citations: "auto" },
|
|
agents: { list: [{ id: "main", default: true }] },
|
|
}),
|
|
agentSessionKey,
|
|
});
|
|
}
|
|
|
|
export function expectUnavailableMemorySearchDetails(
|
|
details: unknown,
|
|
params: {
|
|
error: string;
|
|
warning: string;
|
|
action: string;
|
|
},
|
|
) {
|
|
expect(details).toEqual({
|
|
results: [],
|
|
disabled: true,
|
|
unavailable: true,
|
|
error: params.error,
|
|
warning: params.warning,
|
|
action: params.action,
|
|
});
|
|
}
|