psyche-ai 5.0.0 → 7.1.0

package/dist/cli.js

@@ -3,21 +3,24 @@
 // psyche — Artificial Psyche CLI (v0.2)
 //
 // Usage:
-//   psyche init <dir> [--mbti TYPE] [--name NAME] [--lang LOCALE]
+//   psyche init <dir> [--mbti TYPE] [--name NAME] [--lang LOCALE] [--mode MODE] [--traits "O:80,C:40,E:90,A:60,N:30"] [--no-persist]
 //   psyche status <dir> [--json] [--user ID]
 //   psyche inject <dir> [--protocol] [--json] [--lang LOCALE] [--user ID]
 //   psyche decay <dir>
 //   psyche update <dir> '{"DA":80,"CORT":45}' [--user ID]
-//   psyche reset <dir>
+//   psyche mode <dir> <natural|work|companion>
+//   psyche intensity              Show info about personality intensity config
+//   psyche reset <dir> [--full]
 //   psyche profiles [--json] [--mbti TYPE]
 // ============================================================
 import { resolve } from "node:path";
 import { parseArgs } from "node:util";
 import { loadState, saveState, decayAndSave, initializeState, mergeUpdates, generatePsycheMd, getRelationship, } from "./psyche-file.js";
-import { describeEmotionalState, getExpressionHint } from "./chemistry.js";
-import { getBaseline, getTemperament, getSensitivity, getDefaultSelfModel } from "./profiles.js";
+import { describeEmotionalState, getExpressionHint, detectEmotions } from "./chemistry.js";
+import { getBaseline, getTemperament, getSensitivity, getDefaultSelfModel, traitsToBaseline } from "./profiles.js";
 import { buildDynamicContext, buildProtocolContext } from "./prompt.js";
-import { CHEMICAL_KEYS, CHEMICAL_NAMES_ZH } from "./types.js";
+import { t } from "./i18n.js";
+import { CHEMICAL_KEYS, CHEMICAL_NAMES_ZH, DRIVE_KEYS, DRIVE_NAMES_ZH } from "./types.js";
 import { isMBTIType, isChemicalKey, isLocale } from "./guards.js";
 // ── Logger ───────────────────────────────────────────────────
 const cliLogger = {
@@ -49,12 +52,32 @@ function printChemistry(state) {
         console.log(`  ${label} ${nameZh} ${bar(val)} ${String(val).padStart(3)} (base:${base} ${a})`);
     }
 }
+function printDrives(state) {
+    const icons = { survival: "🛡️", safety: "🏠", connection: "🤝", esteem: "⭐", curiosity: "🔍" };
+    for (const key of DRIVE_KEYS) {
+        const val = Math.round(state.drives[key]);
+        const icon = icons[key] ?? "·";
+        const status = val >= 60 ? "🟢" : val >= 40 ? "🟡" : "🔴";
+        const nameZh = DRIVE_NAMES_ZH[key].padEnd(6);
+        console.log(`  ${icon} ${nameZh} ${bar(val, 20)} ${String(val).padStart(3)} ${status}`);
+    }
+}
+function printEmotions(state) {
+    const emotions = detectEmotions(state.current);
+    if (emotions.length > 0) {
+        const names = emotions.map((e) => e.nameZh).join("、");
+        console.log(`  涌现情绪: ${names}`);
+    }
+    else {
+        console.log(`  涌现情绪: 平静`);
+    }
+}
 function die(msg) {
     console.error(`error: ${msg}`);
     process.exit(1);
 }
 // ── Commands ─────────────────────────────────────────────────
-async function cmdInit(dir, mbti, name, lang) {
+async function cmdInit(dir, mbti, name, lang, mode, traits, noPersist) {
     const absDir = resolve(dir);
     const opts = {};
     if (mbti) {
@@ -70,7 +93,46 @@ async function cmdInit(dir, mbti, name, lang) {
             die(`invalid locale: ${lang}. Valid: zh, en`);
         opts.locale = lang;
     }
+    if (mode) {
+        const validModes = ["natural", "work", "companion"];
+        if (!validModes.includes(mode))
+            die(`invalid mode: ${mode}. Valid: ${validModes.join(", ")}`);
+    }
+    // Parse traits string like "O:80,C:40,E:90,A:60,N:30"
+    let parsedTraits;
+    if (traits) {
+        const traitMap = {
+            O: "openness", C: "conscientiousness", E: "extraversion", A: "agreeableness", N: "neuroticism",
+        };
+        const parsed = {};
+        for (const pair of traits.split(",")) {
+            const [key, val] = pair.trim().split(":");
+            const traitKey = traitMap[key?.toUpperCase()];
+            if (!traitKey)
+                die(`invalid trait key: ${key}. Valid: O, C, E, A, N`);
+            const num = parseInt(val, 10);
+            if (isNaN(num) || num < 0 || num > 100)
+                die(`trait value must be 0-100: ${val}`);
+            parsed[traitKey] = num;
+        }
+        if (Object.keys(parsed).length !== 5)
+            die("all 5 traits required: O, C, E, A, N");
+        parsedTraits = parsed;
+    }
     const state = await initializeState(absDir, opts, cliLogger);
+    // Apply parsed traits to override baseline
+    if (parsedTraits) {
+        const { baseline } = traitsToBaseline(parsedTraits);
+        state.baseline = baseline;
+        state.current = { ...baseline };
+        await saveState(absDir, state);
+        await generatePsycheMd(absDir, state);
+    }
+    // Apply mode after initialization
+    if (mode) {
+        state.meta.mode = mode;
+        await saveState(absDir, state);
+    }
     console.log(`\nPsyche initialized for ${state.meta.agentName} (${state.mbti})\n`);
     printChemistry(state);
     console.log(`\nFiles created:`);
@@ -98,6 +160,10 @@ async function cmdStatus(dir, json, userId) {
     const elapsed = ((Date.now() - new Date(state.updatedAt).getTime()) / 60000).toFixed(1);
     console.log(`\n${state.meta.agentName} (${state.mbti}) — ${emotion}\n`);
     printChemistry(state);
+    console.log();
+    printDrives(state);
+    printEmotions(state);
+    console.log(`  Mode: ${state.meta.mode ?? "natural"} — ${t(`mode.${state.meta.mode ?? "natural"}`, locale)}`);
     console.log(`\n  Expression: ${hint}`);
     console.log(`  Relationship (${userId ?? "_default"}): trust ${relationship.trust}, intimacy ${relationship.intimacy} (${relationship.phase})`);
     console.log(`  Interactions: ${state.meta.totalInteractions}`);
@@ -176,17 +242,22 @@ async function cmdUpdate(dir, updateJson, userId) {
     printChemistry(merged);
     console.log();
 }
-async function cmdReset(dir) {
+async function cmdReset(dir, full) {
     const absDir = resolve(dir);
     const state = await loadState(absDir, cliLogger);
     state.current = { ...state.baseline };
+    state.drives = { survival: 80, safety: 70, connection: 60, esteem: 60, curiosity: 70 };
     state.updatedAt = new Date().toISOString();
     state.empathyLog = null;
     state.agreementStreak = 0;
     state.lastDisagreement = null;
+    state.emotionalHistory = [];
+    if (full) {
+        state.relationships = { _default: { trust: 50, intimacy: 30, phase: "acquaintance" } };
+    }
     await saveState(absDir, state);
     await generatePsycheMd(absDir, state);
-    console.log(`\n${state.meta.agentName} reset to baseline (${state.mbti})\n`);
+    console.log(`\n${state.meta.agentName} reset to baseline (${state.mbti})${full ? " [full reset including relationships]" : ""}\n`);
     printChemistry(state);
     console.log();
 }
@@ -256,6 +327,32 @@ function cmdProfiles(json, mbti) {
         console.log();
     }
 }
+async function cmdMode(dir, newMode) {
+    const absDir = resolve(dir);
+    const validModes = ["natural", "work", "companion"];
+    if (!validModes.includes(newMode))
+        die(`invalid mode: ${newMode}. Valid: ${validModes.join(", ")}`);
+    const state = await loadState(absDir, cliLogger);
+    state.meta.mode = newMode;
+    await saveState(absDir, state);
+    console.log(`\n${state.meta.agentName} mode set to: ${newMode}\n`);
+}
+function cmdIntensity() {
+    console.log(`
+Personality intensity controls how strongly Psyche affects AI behavior.
+
+  0.0 = Traditional warm AI (no emotional variation)
+  0.5 = Subtle emotional responses
+  0.7 = Default
+  1.0 = Full Psyche experience
+
+Set via PsycheEngine config:
+  new PsycheEngine({ personalityIntensity: 0.7 }, storage)
+
+Or in OpenClaw plugin config:
+  { "personalityIntensity": 0.7 }
+`);
+}
 // ── Usage ────────────────────────────────────────────────────
 function usage() {
     console.log(`
@@ -263,12 +360,14 @@ psyche — Artificial Psyche CLI (v0.2)
 Plug-and-play emotional intelligence for any AI agent.
 
 Usage:
-  psyche init <dir> [--mbti TYPE] [--name NAME] [--lang LOCALE]
+  psyche init <dir> [--mbti TYPE] [--name NAME] [--lang LOCALE] [--mode MODE] [--traits "O:80,C:40,..."] [--no-persist]
   psyche status <dir> [--json] [--user ID]
   psyche inject <dir> [--protocol] [--json] [--lang LOCALE] [--user ID]
   psyche decay <dir>
   psyche update <dir> '{"DA":80,"CORT":45}' [--user ID]
-  psyche reset <dir>
+  psyche mode <dir> <natural|work|companion>
+  psyche intensity              Show info about personality intensity config
+  psyche reset <dir> [--full]
   psyche profiles [--mbti TYPE] [--json]
 
 Options:
@@ -318,12 +417,15 @@ async function main() {
                         mbti: { type: "string" },
                         name: { type: "string" },
                         lang: { type: "string" },
+                        mode: { type: "string" },
+                        traits: { type: "string" },
+                        "no-persist": { type: "boolean", default: false },
                     },
                     allowPositionals: true,
                 });
                 if (positionals.length === 0)
                     die("missing <dir> argument");
-                await cmdInit(positionals[0], values.mbti, values.name, values.lang);
+                await cmdInit(positionals[0], values.mbti, values.name, values.lang, values.mode, values.traits, values["no-persist"]);
                 break;
             }
             case "status": {
@@ -377,9 +479,16 @@ async function main() {
                 break;
             }
             case "reset": {
-                if (rest.length === 0)
+                const { values: resetVals, positionals: resetPos } = parseArgs({
+                    args: rest,
+                    options: {
+                        full: { type: "boolean", default: false },
+                    },
+                    allowPositionals: true,
+                });
+                if (resetPos.length === 0)
                     die("missing <dir> argument");
-                await cmdReset(rest[0]);
+                await cmdReset(resetPos[0], resetVals.full ?? false);
                 break;
             }
             case "profiles": {
@@ -394,6 +503,16 @@ async function main() {
                 cmdProfiles(values.json ?? false, values.mbti);
                 break;
             }
+            case "mode": {
+                if (rest.length < 2)
+                    die("usage: psyche mode <dir> <natural|work|companion>");
+                await cmdMode(rest[0], rest[1]);
+                break;
+            }
+            case "intensity": {
+                cmdIntensity();
+                break;
+            }
             default:
                 die(`unknown command: ${command}. Run 'psyche help' for usage.`);
         }

package/dist/core.d.ts

@@ -1,4 +1,4 @@
-import type { PsycheState, StimulusType, Locale, MBTIType, OutcomeScore } from "./types.js";
+import type { PsycheState, StimulusType, Locale, MBTIType, OutcomeScore, PsycheMode, PersonalityTraits } from "./types.js";
 import type { StorageAdapter } from "./storage.js";
 export interface PsycheEngineConfig {
     mbti?: MBTIType;
@@ -9,6 +9,14 @@ export interface PsycheEngineConfig {
     maxChemicalDelta?: number;
     /** Compact mode: algorithms handle chemistry, LLM only sees behavioral output. Default: true */
     compactMode?: boolean;
+    /** Operating mode: "natural" (default), "work" (minimal emotions), "companion" (full emotions) */
+    mode?: PsycheMode;
+    /** Personality intensity: 0.0 (traditional warm AI) to 1.0 (full Psyche). Default: 0.7 */
+    personalityIntensity?: number;
+    /** Whether to persist state to disk. Default: true. When false, uses in-memory storage. */
+    persist?: boolean;
+    /** Big Five traits. If provided, overrides MBTI for baseline calculation. */
+    traits?: PersonalityTraits;
 }
 export interface ProcessInputResult {
     /** Cacheable protocol prompt (stable across turns) */
@@ -32,7 +40,10 @@ export interface ProcessOutcomeResult {
 }
 export declare class PsycheEngine {
     private state;
-    private readonly storage;
+    private storage;
+    /** Whether the algorithm applied a stimulus in the last processInput call */
+    private _lastAlgorithmApplied;
+    private readonly traits;
     private readonly cfg;
     private readonly protocolCache;
     /** Pending prediction from last processInput for auto-learning */
@@ -77,6 +88,24 @@ export declare class PsycheEngine {
      * Get the cacheable protocol prompt for a locale.
      */
     getProtocol(locale?: Locale): string;
+    /**
+     * End the current session: compress emotionalHistory into a rich summary
+     * stored in relationship.memory[], then clear the history.
+     * No-op if history has fewer than 2 entries.
+     */
+    endSession(opts?: {
+        userId?: string;
+    }): Promise<void>;
     private ensureInitialized;
     private createDefaultState;
+    /**
+     * Reset state to baseline. Optionally preserves relationships.
+     */
+    resetState(opts?: {
+        preserveRelationships?: boolean;
+    }): Promise<void>;
+    /**
+     * Get a single-line status summary with emoji.
+     */
+    getStatusSummary(): string;
 }