Mragank Shekhar
76 lines
2.0 KiB
TypeScript
76 lines
2.0 KiB
TypeScript
/**
|
|
* mem0 identify — declare which agent owns the current agent-mode key.
|
|
*
|
|
* Used when `mem0 init --agent` ran without --agent-caller, so the backend
|
|
* saved agent_caller=NULL. The agent re-runs `mem0 identify <name>` to PATCH
|
|
* its own row with its real identity. Idempotent.
|
|
*/
|
|
|
|
import { printError, printSuccess } from "../branding.js";
|
|
import { loadConfig, saveConfig } from "../config.js";
|
|
|
|
const SOURCE_HEADERS = {
|
|
"X-Mem0-Source": "cli",
|
|
"X-Mem0-Client-Language": "node",
|
|
} as const;
|
|
|
|
export async function runIdentify(name: string): Promise<void> {
|
|
const config = loadConfig();
|
|
if (!config.platform.apiKey) {
|
|
printError("No API key configured. Run `mem0 init --agent` first.");
|
|
process.exit(1);
|
|
}
|
|
if (!config.platform.agentMode) {
|
|
printError("This command only works on unclaimed agent-mode keys.");
|
|
process.exit(1);
|
|
}
|
|
|
|
const clean = (name ?? "").trim();
|
|
if (!clean) {
|
|
printError("Agent name is required.");
|
|
process.exit(1);
|
|
}
|
|
|
|
const baseUrl = (config.platform.baseUrl || "https://api.mem0.ai").replace(
|
|
/\/+$/,
|
|
"",
|
|
);
|
|
|
|
let resp: Response;
|
|
try {
|
|
resp = await fetch(`${baseUrl}/api/v1/auth/agent_mode/caller/`, {
|
|
method: "PATCH",
|
|
headers: {
|
|
...SOURCE_HEADERS,
|
|
Authorization: `Token ${config.platform.apiKey}`,
|
|
"Content-Type": "application/json",
|
|
},
|
|
body: JSON.stringify({ agent_caller: clean }),
|
|
signal: AbortSignal.timeout(30_000),
|
|
});
|
|
} catch (err) {
|
|
printError(
|
|
`Network error: ${err instanceof Error ? err.message : String(err)}`,
|
|
);
|
|
process.exit(1);
|
|
}
|
|
|
|
if (!resp.ok) {
|
|
let detail: string = resp.statusText;
|
|
try {
|
|
const body = (await resp.json()) as { error?: string };
|
|
if (body.error) detail = body.error;
|
|
} catch {
|
|
/* leave as statusText */
|
|
}
|
|
printError(`Identify failed: ${detail}`);
|
|
process.exit(1);
|
|
}
|
|
|
|
const body = (await resp.json()) as { agent_caller?: string };
|
|
const canonical = body.agent_caller ?? clean;
|
|
config.platform.agentCaller = canonical;
|
|
saveConfig(config);
|
|
printSuccess(`Identified as ${canonical}.`);
|
|
}
|