@squadrules/mcp 4.8.6

package/dist/tools/reward.js

@@ -0,0 +1,133 @@
+import { resolveToolDoc } from '../utils/mcp-tool-doc-runtime.js';
+import { getTenantId, getSpaceContextFromStorage } from '../utils/tenant-context.js';
+import { mcpToolCalls, mcpToolDuration, mcpToolErrors, mcpToolInputSize, mcpToolOutputSize } from '../services/metrics/mcp-metrics.js';
+import { applyRewardMetrics } from '../services/reward-metrics.js';
+import { rewardInputSchema, rewardOutputSchema } from './reward_schema.js';
+import { buildLayerUri, parseKairosUriOrThrow } from './kairos-uri.js';
+import { executionTraceStore } from '../services/execution-trace-store.js';
+import { evaluateReward } from '../services/reward-evals.js';
+import { KairosError } from '../types/index.js';
+import { mcpLooseToolInput } from './mcp-loose-input-schema.js';
+import { mcpToolInputValidationErrorResult } from './mcp-tool-input-teaching.js';
+function buildRewardPersistenceError(uri, error) {
+    return new KairosError(`Failed to record reward for ${uri}. No reward was stored. Retry the same reward call once reward storage is healthy.`, 'REWARD_FAILED', 503, error instanceof Error ? { cause: error.message } : undefined);
+}
+export async function executeReward(qdrantService, input) {
+    const parsed = parseKairosUriOrThrow(input.uri);
+    if (parsed.kind !== 'layer') {
+        throw new KairosError('reward requires a layer URI', 'INVALID_URI', 400);
+    }
+    const evaluation = evaluateReward({
+        outcome: input.outcome,
+        ...(input.score !== undefined ? { score: input.score } : {}),
+        ...(input.rater !== undefined ? { rater: input.rater } : {}),
+        ...(input.rubric_version !== undefined ? { rubricVersion: input.rubric_version } : {}),
+        ...(input.llm_model_id !== undefined ? { llmModelId: input.llm_model_id } : {})
+    });
+    const rewardMetricsInput = {
+        uri: buildLayerUri(parsed.id, parsed.executionId),
+        outcome: input.outcome,
+        feedback: input.feedback ?? `${input.outcome} reward`,
+        qualityBonus: evaluation.qualityBonus
+    };
+    const evaluatorId = input.llm_model_id ?? input.rater;
+    if (evaluatorId) {
+        rewardMetricsInput.evaluatorId = evaluatorId;
+    }
+    let rewardMetricsResult;
+    try {
+        rewardMetricsResult = await applyRewardMetrics(qdrantService, rewardMetricsInput);
+    }
+    catch (error) {
+        if (error instanceof KairosError) {
+            throw error;
+        }
+        throw buildRewardPersistenceError(input.uri, error);
+    }
+    const ratedAt = new Date().toISOString();
+    if (parsed.executionId) {
+        try {
+            await executionTraceStore.setReward(parsed.executionId, {
+                outcome: input.outcome,
+                score: evaluation.normalizedScore,
+                signed_score: evaluation.signedScore,
+                quality_bonus: evaluation.qualityBonus,
+                ...(input.feedback !== undefined && { feedback: input.feedback }),
+                ...(input.rater !== undefined && { rater: input.rater }),
+                ...(input.rubric_version !== undefined && { rubric_version: input.rubric_version }),
+                ...(input.llm_model_id !== undefined && { llm_model_id: input.llm_model_id }),
+                grader_kind: evaluation.graderKind,
+                evaluation_label: evaluation.label,
+                exportable_for_sft: evaluation.exportableForSft,
+                exportable_for_preference: evaluation.exportableForPreference,
+                sft_blockers: evaluation.sftEligibility.blockers,
+                preference_blockers: evaluation.preferenceEligibility.blockers,
+                rated_at: ratedAt
+            });
+        }
+        catch (error) {
+            throw buildRewardPersistenceError(input.uri, error);
+        }
+    }
+    return {
+        results: rewardMetricsResult.results.map((result) => ({
+            uri: input.uri,
+            outcome: result.outcome,
+            score: evaluation.normalizedScore,
+            feedback: input.feedback ?? null,
+            rater: input.rater ?? null,
+            rubric_version: input.rubric_version ?? null,
+            llm_model_id: input.llm_model_id ?? null,
+            grader_kind: evaluation.graderKind,
+            evaluation_label: evaluation.label,
+            exportable_for_sft: evaluation.exportableForSft,
+            exportable_for_preference: evaluation.exportableForPreference,
+            sft_blockers: evaluation.sftEligibility.blockers,
+            preference_blockers: evaluation.preferenceEligibility.blockers,
+            rated_at: ratedAt
+        })),
+        total_rated: rewardMetricsResult.total_rated,
+        total_failed: rewardMetricsResult.total_failed,
+        next_call: null
+    };
+}
+export function registerRewardTool(server, qdrantService, options = {}) {
+    const toolName = options.toolName || 'reward';
+    server.registerTool(toolName, {
+        title: 'Record adapter reward',
+        description: resolveToolDoc('reward') || 'Attach a reward signal after adapter execution completes.',
+        inputSchema: mcpLooseToolInput(rewardInputSchema),
+        outputSchema: rewardOutputSchema
+    }, async (params) => {
+        const tenantId = getTenantId();
+        const spaceId = getSpaceContextFromStorage()?.defaultWriteSpaceId ?? 'default';
+        void spaceId;
+        mcpToolInputSize.observe({ tool: toolName, tenant_id: tenantId }, JSON.stringify(params).length);
+        const timer = mcpToolDuration.startTimer({ tool: toolName, tenant_id: tenantId });
+        const parsedInput = rewardInputSchema.safeParse(params);
+        if (!parsedInput.success) {
+            mcpToolCalls.inc({ tool: toolName, status: 'error', tenant_id: tenantId });
+            mcpToolErrors.inc({ tool: toolName, status: 'error', tenant_id: tenantId });
+            timer({ tool: toolName, status: 'error', tenant_id: tenantId });
+            return mcpToolInputValidationErrorResult('reward', parsedInput.error, params);
+        }
+        const input = parsedInput.data;
+        try {
+            const output = await executeReward(qdrantService, input);
+            mcpToolCalls.inc({ tool: toolName, status: 'success', tenant_id: tenantId });
+            mcpToolOutputSize.observe({ tool: toolName, tenant_id: tenantId }, JSON.stringify(output).length);
+            timer({ tool: toolName, status: 'success', tenant_id: tenantId });
+            return {
+                content: [{ type: 'text', text: JSON.stringify(output) }],
+                structuredContent: output
+            };
+        }
+        catch (error) {
+            mcpToolCalls.inc({ tool: toolName, status: 'error', tenant_id: tenantId });
+            mcpToolErrors.inc({ tool: toolName, status: 'error', tenant_id: tenantId });
+            timer({ tool: toolName, status: 'error', tenant_id: tenantId });
+            throw error;
+        }
+    });
+}
+//# sourceMappingURL=reward.js.map

package/dist/tools/reward.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reward.js","sourceRoot":"","sources":["../../src/tools/reward.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,0BAA0B,EAAE,MAAM,4BAA4B,CAAC;AACrF,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,aAAa,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AACvI,OAAO,EAAE,kBAAkB,EAA4B,MAAM,+BAA+B,CAAC;AAC7F,OAAO,EAAE,iBAAiB,EAAE,kBAAkB,EAAuC,MAAM,oBAAoB,CAAC;AAChH,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AACvE,OAAO,EAAE,mBAAmB,EAAE,MAAM,sCAAsC,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAChE,OAAO,EAAE,iCAAiC,EAAE,MAAM,8BAA8B,CAAC;AAMjF,SAAS,2BAA2B,CAAC,GAAW,EAAE,KAAc;IAC9D,OAAO,IAAI,WAAW,CACpB,+BAA+B,GAAG,oFAAoF,EACtH,eAAe,EACf,GAAG,EACH,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAC9D,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,aAA4B,EAC5B,KAAkB;IAElB,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAChD,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC5B,MAAM,IAAI,WAAW,CAAC,6BAA6B,EAAE,aAAa,EAAE,GAAG,CAAC,CAAC;IAC3E,CAAC;IAED,MAAM,UAAU,GAAG,cAAc,CAAC;QAChC,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,GAAG,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5D,GAAG,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5D,GAAG,CAAC,KAAK,CAAC,cAAc,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACtF,GAAG,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAChF,CAAC,CAAC;IAEH,MAAM,kBAAkB,GAA6C;QACnE,GAAG,EAAE,aAAa,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,WAAW,CAAC;QACjD,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,GAAG,KAAK,CAAC,OAAO,SAAS;QACrD,YAAY,EAAE,UAAU,CAAC,YAAY;KACtC,CAAC;IACF,MAAM,WAAW,GAAG,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,KAAK,CAAC;IACtD,IAAI,WAAW,EAAE,CAAC;QAChB,kBAAkB,CAAC,WAAW,GAAG,WAAW,CAAC;IAC/C,CAAC;IACD,IAAI,mBAAwC,CAAC;IAC7C,IAAI,CAAC;QACH,mBAAmB,GAAG,MAAM,kBAAkB,CAAC,aAAa,EAAE,kBAAkB,CAAC,CAAC;IACpF,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,WAAW,EAAE,CAAC;YACjC,MAAM,KAAK,CAAC;QACd,CAAC;QACD,MAAM,2BAA2B,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IACtD,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACzC,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,mBAAmB,CAAC,SAAS,CAAC,MAAM,CAAC,WAAW,EAAE;gBACtD,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,KAAK,EAAE,UAAU,CAAC,eAAe;gBACjC,YAAY,EAAE,UAAU,CAAC,WAAW;gBACpC,aAAa,EAAE,UAAU,CAAC,YAAY;gBACtC,GAAG,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACjE,GAAG,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;gBACxD,GAAG,CAAC,KAAK,CAAC,cAAc,KAAK,SAAS,IAAI,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC;gBACnF,GAAG,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,IAAI,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;gBAC7E,WAAW,EAAE,UAAU,CAAC,UAAU;gBAClC,gBAAgB,EAAE,UAAU,CAAC,KAAK;gBAClC,kBAAkB,EAAE,UAAU,CAAC,gBAAgB;gBAC/C,yBAAyB,EAAE,UAAU,CAAC,uBAAuB;gBAC7D,YAAY,EAAE,UAAU,CAAC,cAAc,CAAC,QAAQ;gBAChD,mBAAmB,EAAE,UAAU,CAAC,qBAAqB,CAAC,QAAQ;gBAC9D,QAAQ,EAAE,OAAO;aAClB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,2BAA2B,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,OAAO;QACL,OAAO,EAAE,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACpD,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,OAAO,EAAE,MAAM,CAAC,OAAgC;YAChD,KAAK,EAAE,UAAU,CAAC,eAAe;YACjC,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,IAAI;YAChC,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,IAAI;YAC1B,cAAc,EAAE,KAAK,CAAC,cAAc,IAAI,IAAI;YAC5C,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,IAAI;YACxC,WAAW,EAAE,UAAU,CAAC,UAAU;YAClC,gBAAgB,EAAE,UAAU,CAAC,KAAK;YAClC,kBAAkB,EAAE,UAAU,CAAC,gBAAgB;YAC/C,yBAAyB,EAAE,UAAU,CAAC,uBAAuB;YAC7D,YAAY,EAAE,UAAU,CAAC,cAAc,CAAC,QAAQ;YAChD,mBAAmB,EAAE,UAAU,CAAC,qBAAqB,CAAC,QAAQ;YAC9D,QAAQ,EAAE,OAAO;SAClB,CAAC,CAAC;QACH,WAAW,EAAE,mBAAmB,CAAC,WAAW;QAC5C,YAAY,EAAE,mBAAmB,CAAC,YAAY;QAC9C,SAAS,EAAE,IAAI;KAChB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAAW,EAAE,aAA4B,EAAE,UAAiC,EAAE;IAC/G,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,QAAQ,CAAC;IAC9C,MAAM,CAAC,YAAY,CACjB,QAAQ,EACR;QACE,KAAK,EAAE,uBAAuB;QAC9B,WAAW,EAAE,cAAc,CAAC,QAAQ,CAAC,IAAI,2DAA2D;QACpG,WAAW,EAAE,iBAAiB,CAAC,iBAAiB,CAAC;QACjD,YAAY,EAAE,kBAAkB;KACjC,EACD,KAAK,EAAE,MAAe,EAAE,EAAE;QACxB,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,0BAA0B,EAAE,EAAE,mBAAmB,IAAI,SAAS,CAAC;QAC/E,KAAK,OAAO,CAAC;QACb,gBAAgB,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC;QACjG,MAAM,KAAK,GAAG,eAAe,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;QAElF,MAAM,WAAW,GAAG,iBAAiB,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACxD,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,YAAY,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC3E,aAAa,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC5E,KAAK,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;YAChE,OAAO,iCAAiC,CAAC,QAAQ,EAAE,WAAW,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAChF,CAAC;QACD,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC;QAE/B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;YACzD,YAAY,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC7E,iBAAiB,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC;YAClG,KAAK,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;YAClE,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;gBAClE,iBAAiB,EAAE,MAAM;aAC1B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,YAAY,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC3E,aAAa,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC5E,KAAK,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;YAChE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}

package/dist/tools/reward_schema.d.ts

@@ -0,0 +1,61 @@
+import { z } from 'zod';
+export declare const rewardInputSchema: z.ZodObject<{
+    uri: z.ZodString;
+    outcome: z.ZodEnum<{
+        success: "success";
+        failure: "failure";
+    }>;
+    score: z.ZodOptional<z.ZodNumber>;
+    feedback: z.ZodOptional<z.ZodString>;
+    rater: z.ZodOptional<z.ZodString>;
+    rubric_version: z.ZodOptional<z.ZodString>;
+    llm_model_id: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>;
+export declare const rewardOutputSchema: z.ZodObject<{
+    results: z.ZodArray<z.ZodObject<{
+        uri: z.ZodString;
+        outcome: z.ZodEnum<{
+            success: "success";
+            failure: "failure";
+        }>;
+        score: z.ZodNullable<z.ZodNumber>;
+        feedback: z.ZodNullable<z.ZodString>;
+        rater: z.ZodNullable<z.ZodString>;
+        rubric_version: z.ZodNullable<z.ZodString>;
+        llm_model_id: z.ZodNullable<z.ZodString>;
+        grader_kind: z.ZodEnum<{
+            unknown: "unknown";
+            human: "human";
+            model: "model";
+        }>;
+        evaluation_label: z.ZodEnum<{
+            gold: "gold";
+            silver: "silver";
+            bronze: "bronze";
+            rejected: "rejected";
+        }>;
+        exportable_for_sft: z.ZodBoolean;
+        exportable_for_preference: z.ZodBoolean;
+        sft_blockers: z.ZodArray<z.ZodEnum<{
+            missing_rubric_version: "missing_rubric_version";
+            missing_evaluator_identity: "missing_evaluator_identity";
+            outcome_not_success: "outcome_not_success";
+            score_below_sft_threshold: "score_below_sft_threshold";
+            score_below_preference_threshold: "score_below_preference_threshold";
+        }>>;
+        preference_blockers: z.ZodArray<z.ZodEnum<{
+            missing_rubric_version: "missing_rubric_version";
+            missing_evaluator_identity: "missing_evaluator_identity";
+            outcome_not_success: "outcome_not_success";
+            score_below_sft_threshold: "score_below_sft_threshold";
+            score_below_preference_threshold: "score_below_preference_threshold";
+        }>>;
+        rated_at: z.ZodString;
+    }, z.core.$strip>>;
+    total_rated: z.ZodNumber;
+    total_failed: z.ZodNumber;
+    next_call: z.ZodNull;
+}, z.core.$strict>;
+export type RewardInput = z.infer<typeof rewardInputSchema>;
+export type RewardOutput = z.infer<typeof rewardOutputSchema>;
+//# sourceMappingURL=reward_schema.d.ts.map

package/dist/tools/reward_schema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"reward_schema.d.ts","sourceRoot":"","sources":["../../src/tools/reward_schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AASxB,eAAO,MAAM,iBAAiB;;;;;;;;;;;iBAQ5B,CAAC;AAUH,eAAO,MAAM,kBAAkB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;kBAoBpB,CAAC;AAEZ,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAC5D,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC"}

package/dist/tools/reward_schema.js

@@ -0,0 +1,42 @@
+import { z } from 'zod';
+const layerUriSchema = z
+    .string()
+    .regex(/^(?:kairos|squadrules):\/\/layer\/[0-9a-f-]{36}(?:\?execution_id=[0-9a-f-]{36})?$/i, 'must match kairos://layer/{uuid}[?execution_id={uuid}] (squadrules:// also accepted)');
+export const rewardInputSchema = z.object({
+    uri: layerUriSchema.describe('URI of the final layer in the adapter execution'),
+    outcome: z.enum(['success', 'failure']).describe('Execution outcome'),
+    score: z.number().min(0).max(1).optional().describe('Optional normalized reward score'),
+    feedback: z.string().min(1).optional().describe('Optional evaluator feedback'),
+    rater: z.string().optional().describe('Human evaluator identifier. Combined with llm_model_id provides evaluator identity — omitting both blocks exportable_for_sft and exportable_for_preference.'),
+    rubric_version: z.string().optional().describe('Rubric or policy version tag (e.g. "v1"). Required for SFT and preference export eligibility — omitting blocks exportable_for_sft and exportable_for_preference.'),
+    llm_model_id: z.string().optional().describe('LLM model identifier for evaluator attribution. Required for evaluator identity — omitting blocks exportable_for_sft and exportable_for_preference.')
+});
+const rewardEligibilityBlockerSchema = z.enum([
+    'missing_rubric_version',
+    'missing_evaluator_identity',
+    'outcome_not_success',
+    'score_below_sft_threshold',
+    'score_below_preference_threshold'
+]);
+export const rewardOutputSchema = z.object({
+    results: z.array(z.object({
+        uri: layerUriSchema,
+        outcome: z.enum(['success', 'failure']),
+        score: z.number().nullable(),
+        feedback: z.string().nullable(),
+        rater: z.string().nullable(),
+        rubric_version: z.string().nullable(),
+        llm_model_id: z.string().nullable(),
+        grader_kind: z.enum(['human', 'model', 'unknown']),
+        evaluation_label: z.enum(['gold', 'silver', 'bronze', 'rejected']),
+        exportable_for_sft: z.boolean(),
+        exportable_for_preference: z.boolean(),
+        sft_blockers: z.array(rewardEligibilityBlockerSchema),
+        preference_blockers: z.array(rewardEligibilityBlockerSchema),
+        rated_at: z.string()
+    })),
+    total_rated: z.number(),
+    total_failed: z.number(),
+    next_call: z.null()
+}).strict();
+//# sourceMappingURL=reward_schema.js.map

package/dist/tools/reward_schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reward_schema.js","sourceRoot":"","sources":["../../src/tools/reward_schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,cAAc,GAAG,CAAC;KACrB,MAAM,EAAE;KACR,KAAK,CACJ,oFAAoF,EACpF,sFAAsF,CACvF,CAAC;AAEJ,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,GAAG,EAAE,cAAc,CAAC,QAAQ,CAAC,iDAAiD,CAAC;IAC/E,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,mBAAmB,CAAC;IACrE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kCAAkC,CAAC;IACvF,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;IAC9E,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6JAA6J,CAAC;IACpM,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kKAAkK,CAAC;IAClN,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qJAAqJ,CAAC;CACpM,CAAC,CAAC;AAEH,MAAM,8BAA8B,GAAG,CAAC,CAAC,IAAI,CAAC;IAC5C,wBAAwB;IACxB,4BAA4B;IAC5B,qBAAqB;IACrB,2BAA2B;IAC3B,kCAAkC;CACnC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;QACxB,GAAG,EAAE,cAAc;QACnB,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QACvC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAC5B,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAC/B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAC5B,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACrC,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACnC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QAClD,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;QAClE,kBAAkB,EAAE,CAAC,CAAC,OAAO,EAAE;QAC/B,yBAAyB,EAAE,CAAC,CAAC,OAAO,EAAE;QACtC,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,8BAA8B,CAAC;QACrD,mBAAmB,EAAE,CAAC,CAAC,KAAK,CAAC,8BAA8B,CAAC;QAC5D,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;KACrB,CAAC,CAAC;IACH,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;IACvB,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;IACxB,SAAS,EAAE,CAAC,CAAC,IAAI,EAAE;CACpB,CAAC,CAAC,MAAM,EAAE,CAAC"}

package/dist/tools/search.d.ts

@@ -0,0 +1,11 @@
+import type { MemoryQdrantStore } from '../services/memory/store.js';
+import type { QdrantService } from '../services/qdrant/service.js';
+import { type SearchInput, type SearchOutput } from './search_schema.js';
+/**
+ * Shared execute: search for protocol adapters. Used by MCP tool and HTTP route.
+ * When runInSpace is provided and input has space/space_id, the search runs in that space context.
+ */
+export declare function executeSearch(memoryStore: MemoryQdrantStore, qdrantService: QdrantService | undefined, input: SearchInput, options?: {
+    runInSpace?: (fn: () => Promise<SearchOutput>) => Promise<SearchOutput>;
+}): Promise<SearchOutput>;
+//# sourceMappingURL=search.d.ts.map

package/dist/tools/search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.d.ts","sourceRoot":"","sources":["../../src/tools/search.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAkBnE,OAAO,EAAsB,KAAK,WAAW,EAAE,KAAK,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAmL7F;;;GAGG;AACH,wBAAsB,aAAa,CACjC,WAAW,EAAE,iBAAiB,EAC9B,aAAa,EAAE,aAAa,GAAG,SAAS,EACxC,KAAK,EAAE,WAAW,EAClB,OAAO,CAAC,EAAE;IAAE,UAAU,CAAC,EAAE,CAAC,EAAE,EAAE,MAAM,OAAO,CAAC,YAAY,CAAC,KAAK,OAAO,CAAC,YAAY,CAAC,CAAA;CAAE,GACpF,OAAO,CAAC,YAAY,CAAC,CAwDvB"}

package/dist/tools/search.js

@@ -0,0 +1,197 @@
+import { redisCacheService } from '../services/redis-cache.js';
+import { getRequestIdFromStorage, getTenantId, getSpaceIdFromStorage, getSpaceContextFromStorage } from '../utils/tenant-context.js';
+import { SCORE_THRESHOLD, KAIROS_SEARCH_MAX_CHOICES, KAIROS_SEARCH_LIMIT_CAP, KAIROS_SEARCH_LIMIT_MIN, KAIROS_ENABLE_GROUP_COLLAPSE, isRedisConfigured } from '../config.js';
+import { createResults, generateUnifiedOutput } from './search_output.js';
+import { searchOutputSchema } from './search_schema.js';
+import { logSearchAnomaly } from '../services/embedding/audit.js';
+import { structuredLogger } from '../utils/structured-logger.js';
+import { KAIROS_CREATION_FOOTER_NEXT_ACTION, KAIROS_CREATION_PROTOCOL_SLUG, KAIROS_REFINING_PROTOCOL_SLUG } from '../constants/builtin-search-meta.js';
+import { KAIROS_APP_SPACE_ID } from '../config.js';
+import { buildAdapterUri } from './kairos-uri.js';
+const CREATION_PROTOCOL_URI = buildAdapterUri(KAIROS_CREATION_PROTOCOL_SLUG);
+const REFINING_PROTOCOL_URI = buildAdapterUri(KAIROS_REFINING_PROTOCOL_SLUG);
+const REFINING_NEXT_ACTION = `call forward with ${REFINING_PROTOCOL_URI} to execute the refine adapter`;
+const CREATE_NEXT_ACTION = KAIROS_CREATION_FOOTER_NEXT_ACTION;
+/** Strip built-in protocol URIs and slugs from query so they are not used for search or cache key. */
+function queryForSearch(query) {
+    let q = (query || '').trim();
+    for (const token of [
+        REFINING_PROTOCOL_URI,
+        KAIROS_REFINING_PROTOCOL_SLUG,
+        CREATION_PROTOCOL_URI,
+        KAIROS_CREATION_PROTOCOL_SLUG
+    ]) {
+        q = q.replace(new RegExp(escapeRegex(token), 'gi'), ' ');
+    }
+    return q.replace(/\s+/g, ' ').trim();
+}
+function escapeRegex(s) {
+    return s.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+}
+const SCORE_TIE_EPS = 1e-9;
+/** TTL (seconds) for activate/search unified cache entries written to the key-value store. */
+const ACTIVATE_SEARCH_CACHE_TTL_SEC = 300;
+function addCandidate(candidateMap, memory, score) {
+    if (!memory)
+        return;
+    const key = memory.adapter?.id || memory.memory_uuid;
+    const existing = candidateMap.get(key);
+    const incomingIsHead = memory.adapter?.layer_index === 1;
+    const defaultWrite = getSpaceContextFromStorage().defaultWriteSpaceId;
+    if (!existing) {
+        candidateMap.set(key, { memory, score });
+        return;
+    }
+    const existingIsHead = existing.memory.adapter?.layer_index === 1;
+    if (incomingIsHead && !existingIsHead) {
+        candidateMap.set(key, { memory, score });
+        return;
+    }
+    if (!incomingIsHead && existingIsHead) {
+        return;
+    }
+    if (score > existing.score + SCORE_TIE_EPS) {
+        candidateMap.set(key, { memory, score });
+        return;
+    }
+    if (Math.abs(score - existing.score) <= SCORE_TIE_EPS) {
+        const incomingPersonal = memory.space_id === defaultWrite;
+        const existingPersonal = existing.memory.space_id === defaultWrite;
+        if (incomingPersonal && !existingPersonal) {
+            candidateMap.set(key, { memory, score });
+        }
+    }
+}
+async function searchAndBuildCandidates(memoryStore, query, enableGroupCollapse, maxChoices) {
+    const candidateMap = new Map();
+    const firstLimit = Math.min(200, Math.max(40, maxChoices * 3));
+    const { memories, scores } = await memoryStore.searchMemories(query, firstLimit, enableGroupCollapse);
+    memories.forEach((memory, idx) => addCandidate(candidateMap, memory, scores[idx] ?? 0));
+    if (candidateMap.size < Math.max(10, maxChoices) && enableGroupCollapse) {
+        const secondLimit = Math.min(200, Math.max(80, maxChoices * 3));
+        const { memories: moreMemories, scores: moreScores } = await memoryStore.searchMemories(query, secondLimit, false);
+        moreMemories.forEach((memory, idx) => addCandidate(candidateMap, memory, moreScores[idx] ?? 0));
+    }
+    return candidateMap;
+}
+/** Never throws: footer versions are optional; Qdrant errors must not break search or the error fallback. */
+async function resolveFooterProtocolVersions(memoryStore) {
+    try {
+        const { client, collection } = memoryStore.getQdrantAccess();
+        const slugs = [KAIROS_REFINING_PROTOCOL_SLUG, KAIROS_CREATION_PROTOCOL_SLUG];
+        const page = await client.scroll(collection, {
+            filter: {
+                must: [
+                    { key: 'slug', match: { any: slugs } },
+                    { key: 'space_id', match: { value: KAIROS_APP_SPACE_ID } },
+                    { key: 'adapter.layer_index', match: { value: 1 } }
+                ]
+            },
+            limit: 10,
+            with_payload: { include: ['slug', 'adapter'] },
+            with_vector: false
+        });
+        let refine = null;
+        let create = null;
+        for (const point of page?.points ?? []) {
+            const payload = (point.payload ?? {});
+            const slug = typeof payload['slug'] === 'string' ? payload['slug'] : '';
+            const version = typeof payload['adapter']?.protocol_version === 'string'
+                ? payload['adapter'].protocol_version
+                : null;
+            if (slug === KAIROS_REFINING_PROTOCOL_SLUG)
+                refine = version;
+            if (slug === KAIROS_CREATION_PROTOCOL_SLUG)
+                create = version;
+        }
+        return { refine, create };
+    }
+    catch (err) {
+        structuredLogger.warn(`search resolveFooterProtocolVersions: ${err instanceof Error ? err.message : String(err)}`);
+        return { refine: null, create: null };
+    }
+}
+/** Core search: build candidates and return unified output. Used by executeSearch. */
+async function doSearch(memoryStore, qdrantService, searchQuery, effectiveLimit) {
+    const tenantId = getTenantId();
+    const requestId = getRequestIdFromStorage();
+    const footerVersions = await resolveFooterProtocolVersions(memoryStore);
+    const candidateMap = await searchAndBuildCandidates(memoryStore, searchQuery, KAIROS_ENABLE_GROUP_COLLAPSE, effectiveLimit);
+    const defaultWriteForSort = getSpaceContextFromStorage().defaultWriteSpaceId;
+    let headCandidates = Array.from(candidateMap.values()).sort((a, b) => {
+        const byScore = b.score - a.score;
+        if (Math.abs(byScore) > SCORE_TIE_EPS)
+            return byScore;
+        const aPersonal = a.memory.space_id === defaultWriteForSort ? 1 : 0;
+        const bPersonal = b.memory.space_id === defaultWriteForSort ? 1 : 0;
+        return bPersonal - aPersonal;
+    });
+    if (headCandidates.length > effectiveLimit) {
+        headCandidates = headCandidates.slice(0, effectiveLimit);
+    }
+    const results = headCandidates.length > 0 ? createResults(headCandidates, SCORE_THRESHOLD) : [];
+    logSearchAnomaly({
+        tenantId,
+        requestId,
+        resultCount: results.length,
+        queryLength: searchQuery.length,
+        topScore: results[0]?.score ?? null
+    });
+    return generateUnifiedOutput(results, qdrantService, {
+        refiningUri: REFINING_PROTOCOL_URI,
+        refiningNextAction: REFINING_NEXT_ACTION,
+        createUri: CREATION_PROTOCOL_URI,
+        createNextAction: CREATE_NEXT_ACTION,
+        refiningProtocolVersion: footerVersions.refine,
+        createProtocolVersion: footerVersions.create
+    });
+}
+/**
+ * Shared execute: search for protocol adapters. Used by MCP tool and HTTP route.
+ * When runInSpace is provided and input has space/space_id, the search runs in that space context.
+ */
+export async function executeSearch(memoryStore, qdrantService, input, options) {
+    const { query, space, space_id, max_choices } = input;
+    const spaceParam = space ?? space_id;
+    const effectiveLimit = Math.max(KAIROS_SEARCH_LIMIT_MIN, Math.min(KAIROS_SEARCH_LIMIT_CAP, max_choices ?? KAIROS_SEARCH_MAX_CHOICES));
+    const searchQuery = queryForSearch(query);
+    const runWithCache = async () => {
+        const effectiveSpaceId = spaceParam ?? getSpaceIdFromStorage();
+        const cacheKey = `activate:v6:${effectiveSpaceId}:${searchQuery}:${KAIROS_ENABLE_GROUP_COLLAPSE}:${effectiveLimit}`;
+        const tenantId = getTenantId();
+        const requestId = getRequestIdFromStorage();
+        const cacheDebugBase = {
+            component: 'activate_search_cache',
+            cache_backend: isRedisConfigured ? 'redis' : 'memory',
+            cache_key_version: 'v6',
+            group_collapse: KAIROS_ENABLE_GROUP_COLLAPSE,
+            effective_limit: effectiveLimit,
+            normalized_query_len: searchQuery.length,
+            effective_space_id: effectiveSpaceId,
+            explicit_space_param: spaceParam != null,
+            tenant_id: tenantId,
+            request_id: requestId
+        };
+        const cachedResult = await redisCacheService.get(cacheKey);
+        if (cachedResult) {
+            structuredLogger.debug({
+                ...cacheDebugBase,
+                cache_lookup: 'hit',
+                configured_store_ttl_seconds: ACTIVATE_SEARCH_CACHE_TTL_SEC
+            }, 'activate search cache hit');
+            return searchOutputSchema.parse(JSON.parse(cachedResult));
+        }
+        const result = await doSearch(memoryStore, qdrantService, searchQuery, effectiveLimit);
+        await redisCacheService.set(cacheKey, JSON.stringify(result), ACTIVATE_SEARCH_CACHE_TTL_SEC);
+        structuredLogger.debug({
+            ...cacheDebugBase,
+            cache_lookup: 'miss',
+            configured_store_ttl_seconds: ACTIVATE_SEARCH_CACHE_TTL_SEC
+        }, 'activate search cache miss; stored');
+        return result;
+    };
+    return options?.runInSpace && spaceParam != null
+        ? await options.runInSpace(runWithCache)
+        : await runWithCache();
+}
+//# sourceMappingURL=search.js.map

package/dist/tools/search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.js","sourceRoot":"","sources":["../../src/tools/search.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,EACL,uBAAuB,EACvB,WAAW,EACX,qBAAqB,EACrB,0BAA0B,EAC3B,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EACL,eAAe,EACf,yBAAyB,EACzB,uBAAuB,EACvB,uBAAuB,EACvB,4BAA4B,EAC5B,iBAAiB,EAClB,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC1E,OAAO,EAAE,kBAAkB,EAAuC,MAAM,oBAAoB,CAAC;AAC7F,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAClE,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AACjE,OAAO,EACL,kCAAkC,EAClC,6BAA6B,EAC7B,6BAA6B,EAC9B,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAElD,MAAM,qBAAqB,GAAG,eAAe,CAAC,6BAA6B,CAAC,CAAC;AAC7E,MAAM,qBAAqB,GAAG,eAAe,CAAC,6BAA6B,CAAC,CAAC;AAC7E,MAAM,oBAAoB,GAAG,qBAAqB,qBAAqB,gCAAgC,CAAC;AACxG,MAAM,kBAAkB,GAAG,kCAAkC,CAAC;AAE9D,sGAAsG;AACtG,SAAS,cAAc,CAAC,KAAa;IACnC,IAAI,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAC7B,KAAK,MAAM,KAAK,IAAI;QAClB,qBAAqB;QACrB,6BAA6B;QAC7B,qBAAqB;QACrB,6BAA6B;KAC9B,EAAE,CAAC;QACF,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC;IAC3D,CAAC;IACD,OAAO,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;AACvC,CAAC;AAED,SAAS,WAAW,CAAC,CAAS;IAC5B,OAAO,CAAC,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,aAAa,GAAG,IAAI,CAAC;AAE3B,8FAA8F;AAC9F,MAAM,6BAA6B,GAAG,GAAG,CAAC;AAE1C,SAAS,YAAY,CACnB,YAA4D,EAC5D,MAAc,EACd,KAAa;IAEb,IAAI,CAAC,MAAM;QAAE,OAAO;IACpB,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,EAAE,EAAE,IAAI,MAAM,CAAC,WAAW,CAAC;IACrD,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACvC,MAAM,cAAc,GAAG,MAAM,CAAC,OAAO,EAAE,WAAW,KAAK,CAAC,CAAC;IACzD,MAAM,YAAY,GAAG,0BAA0B,EAAE,CAAC,mBAAmB,CAAC;IACtE,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QACzC,OAAO;IACT,CAAC;IACD,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,WAAW,KAAK,CAAC,CAAC;IAClE,IAAI,cAAc,IAAI,CAAC,cAAc,EAAE,CAAC;QACtC,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QACzC,OAAO;IACT,CAAC;IACD,IAAI,CAAC,cAAc,IAAI,cAAc,EAAE,CAAC;QACtC,OAAO;IACT,CAAC;IACD,IAAI,KAAK,GAAG,QAAQ,CAAC,KAAK,GAAG,aAAa,EAAE,CAAC;QAC3C,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QACzC,OAAO;IACT,CAAC;IACD,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,aAAa,EAAE,CAAC;QACtD,MAAM,gBAAgB,GAAG,MAAM,CAAC,QAAQ,KAAK,YAAY,CAAC;QAC1D,MAAM,gBAAgB,GAAG,QAAQ,CAAC,MAAM,CAAC,QAAQ,KAAK,YAAY,CAAC;QACnE,IAAI,gBAAgB,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1C,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;AACH,CAAC;AAED,KAAK,UAAU,wBAAwB,CACrC,WAA8B,EAC9B,KAAa,EACb,mBAA4B,EAC5B,UAAkB;IAElB,MAAM,YAAY,GAAG,IAAI,GAAG,EAA6C,CAAC;IAC1E,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC;IAC/D,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,MAAM,WAAW,CAAC,cAAc,CAAC,KAAK,EAAE,UAAU,EAAE,mBAAmB,CAAC,CAAC;IACtG,QAAQ,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAExF,IAAI,YAAY,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,IAAI,mBAAmB,EAAE,CAAC;QACxE,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC;QAChE,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,WAAW,CAAC,cAAc,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;QACnH,YAAY,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClG,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,6GAA6G;AAC7G,KAAK,UAAU,6BAA6B,CAC1C,WAA8B;IAE9B,IAAI,CAAC;QACH,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,WAAW,CAAC,eAAe,EAAE,CAAC;QAC7D,MAAM,KAAK,GAAG,CAAC,6BAA6B,EAAE,6BAA6B,CAAC,CAAC;QAC7E,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE;YAC3C,MAAM,EAAE;gBACN,IAAI,EAAE;oBACJ,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;oBACtC,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,mBAAmB,EAAE,EAAE;oBAC1D,EAAE,GAAG,EAAE,qBAAqB,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;iBACpD;aACF;YACD,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,EAAE,OAAO,EAAE,CAAC,MAAM,EAAE,SAAS,CAAC,EAAE;YAC9C,WAAW,EAAE,KAAK;SACZ,CAAC,CAAC;QAEV,IAAI,MAAM,GAAkB,IAAI,CAAC;QACjC,IAAI,MAAM,GAAkB,IAAI,CAAC;QACjC,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE,MAAM,IAAI,EAAE,EAAE,CAAC;YACvC,MAAM,OAAO,GAAG,CAAC,KAAK,CAAC,OAAO,IAAI,EAAE,CAAwB,CAAC;YAC7D,MAAM,IAAI,GAAG,OAAO,OAAO,CAAC,MAAM,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACxE,MAAM,OAAO,GAAG,OAAO,OAAO,CAAC,SAAS,CAAC,EAAE,gBAAgB,KAAK,QAAQ;gBACtE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,gBAAgB;gBACrC,CAAC,CAAC,IAAI,CAAC;YACT,IAAI,IAAI,KAAK,6BAA6B;gBAAE,MAAM,GAAG,OAAO,CAAC;YAC7D,IAAI,IAAI,KAAK,6BAA6B;gBAAE,MAAM,GAAG,OAAO,CAAC;QAC/D,CAAC;QACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;IAC5B,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,gBAAgB,CAAC,IAAI,CACnB,yCAAyC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAC5F,CAAC;QACF,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACxC,CAAC;AACH,CAAC;AAED,sFAAsF;AACtF,KAAK,UAAU,QAAQ,CACrB,WAA8B,EAC9B,aAAwC,EACxC,WAAmB,EACnB,cAAsB;IAEtB,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAC;IAC/B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;IAC5C,MAAM,cAAc,GAAG,MAAM,6BAA6B,CAAC,WAAW,CAAC,CAAC;IACxE,MAAM,YAAY,GAAG,MAAM,wBAAwB,CACjD,WAAW,EACX,WAAW,EACX,4BAA4B,EAC5B,cAAc,CACf,CAAC;IACF,MAAM,mBAAmB,GAAG,0BAA0B,EAAE,CAAC,mBAAmB,CAAC;IAC7E,IAAI,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACnE,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;QAClC,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,aAAa;YAAE,OAAO,OAAO,CAAC;QACtD,MAAM,SAAS,GAAG,CAAC,CAAC,MAAM,CAAC,QAAQ,KAAK,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACpE,MAAM,SAAS,GAAG,CAAC,CAAC,MAAM,CAAC,QAAQ,KAAK,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACpE,OAAO,SAAS,GAAG,SAAS,CAAC;IAC/B,CAAC,CAAC,CAAC;IACH,IAAI,cAAc,CAAC,MAAM,GAAG,cAAc,EAAE,CAAC;QAC3C,cAAc,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC;IAC3D,CAAC;IACD,MAAM,OAAO,GAAG,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,cAAc,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAChG,gBAAgB,CAAC;QACf,QAAQ;QACR,SAAS;QACT,WAAW,EAAE,OAAO,CAAC,MAAM;QAC3B,WAAW,EAAE,WAAW,CAAC,MAAM;QAC/B,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,IAAI;KACpC,CAAC,CAAC;IACH,OAAO,qBAAqB,CAAC,OAAO,EAAE,aAAa,EAAE;QACnD,WAAW,EAAE,qBAAqB;QAClC,kBAAkB,EAAE,oBAAoB;QACxC,SAAS,EAAE,qBAAqB;QAChC,gBAAgB,EAAE,kBAAkB;QACpC,uBAAuB,EAAE,cAAc,CAAC,MAAM;QAC9C,qBAAqB,EAAE,cAAc,CAAC,MAAM;KAC7C,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,WAA8B,EAC9B,aAAwC,EACxC,KAAkB,EAClB,OAAqF;IAErF,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IACtD,MAAM,UAAU,GAAG,KAAK,IAAI,QAAQ,CAAC;IACrC,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAC7B,uBAAuB,EACvB,IAAI,CAAC,GAAG,CAAC,uBAAuB,EAAE,WAAW,IAAI,yBAAyB,CAAC,CAC5E,CAAC;IACF,MAAM,WAAW,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;IAE1C,MAAM,YAAY,GAAG,KAAK,IAA2B,EAAE;QACrD,MAAM,gBAAgB,GAAG,UAAU,IAAI,qBAAqB,EAAE,CAAC;QAC/D,MAAM,QAAQ,GAAG,eAAe,gBAAgB,IAAI,WAAW,IAAI,4BAA4B,IAAI,cAAc,EAAE,CAAC;QACpH,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,cAAc,GAA4B;YAC9C,SAAS,EAAE,uBAAuB;YAClC,aAAa,EAAE,iBAAiB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ;YACrD,iBAAiB,EAAE,IAAI;YACvB,cAAc,EAAE,4BAA4B;YAC5C,eAAe,EAAE,cAAc;YAC/B,oBAAoB,EAAE,WAAW,CAAC,MAAM;YACxC,kBAAkB,EAAE,gBAAgB;YACpC,oBAAoB,EAAE,UAAU,IAAI,IAAI;YACxC,SAAS,EAAE,QAAQ;YACnB,UAAU,EAAE,SAAS;SACtB,CAAC;QAEF,MAAM,YAAY,GAAG,MAAM,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC3D,IAAI,YAAY,EAAE,CAAC;YACjB,gBAAgB,CAAC,KAAK,CACpB;gBACE,GAAG,cAAc;gBACjB,YAAY,EAAE,KAAK;gBACnB,4BAA4B,EAAE,6BAA6B;aAC5D,EACD,2BAA2B,CAC5B,CAAC;YACF,OAAO,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAiB,CAAC;QAC5E,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,EAAE,aAAa,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;QACvF,MAAM,iBAAiB,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,6BAA6B,CAAC,CAAC;QAC7F,gBAAgB,CAAC,KAAK,CACpB;YACE,GAAG,cAAc;YACjB,YAAY,EAAE,MAAM;YACpB,4BAA4B,EAAE,6BAA6B;SAC5D,EACD,oCAAoC,CACrC,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;IAEF,OAAO,OAAO,EAAE,UAAU,IAAI,UAAU,IAAI,IAAI;QAC9C,CAAC,CAAC,MAAM,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC;QACxC,CAAC,CAAC,MAAM,YAAY,EAAE,CAAC;AAC3B,CAAC"}

package/dist/tools/search_output.d.ts

@@ -0,0 +1,55 @@
+/**
+ * Output building for search: createResults, resolveHead, generateUnifiedOutput.
+ * Extracted to keep `search.ts` under the max-lines limit.
+ */
+import type { Memory } from '../types/memory.js';
+import type { QdrantService } from '../services/qdrant/service.js';
+export interface Candidate {
+    memory: Memory;
+    score: number;
+    uri: string;
+    label: string;
+    tags: string[];
+    layer_count: number;
+}
+export interface UnifiedChoice {
+    uri: string;
+    label: string;
+    adapter_name: string | null;
+    score: number | null;
+    role: 'match' | 'refine' | 'create';
+    tags: string[];
+    next_action: string;
+    adapter_version: string | null;
+    activation_patterns?: string[];
+    /** Display name of the space where the adapter is stored (match rows only). */
+    space_name: string | null;
+    /** Stored adapter routing slug when present; null for refine/create or adapters without slug. */
+    slug: string | null;
+}
+export declare function normalizePublicSearchScore(score: number | null | undefined): number;
+export declare function createResults(headCandidates: Array<{
+    memory: Memory;
+    score: number;
+}>, scoreThreshold: number): Candidate[];
+export declare function resolveHead(memory: Memory, qdrantService?: QdrantService): Promise<{
+    uri: string;
+    label: string;
+}>;
+export interface GenerateUnifiedOutputOpts {
+    refiningUri: string;
+    refiningNextAction: string;
+    createUri: string;
+    createNextAction: string;
+    /** Stored adapter version for refine protocol (from Qdrant), when resolvable. */
+    refiningProtocolVersion?: string | null;
+    /** Stored adapter version for create protocol (from Qdrant), when resolvable. */
+    createProtocolVersion?: string | null;
+}
+export declare function generateUnifiedOutput(results: Candidate[], qdrantService: QdrantService | undefined, opts: GenerateUnifiedOutputOpts): Promise<{
+    must_obey: boolean;
+    message: string;
+    next_action: string;
+    choices: UnifiedChoice[];
+}>;
+//# sourceMappingURL=search_output.d.ts.map

package/dist/tools/search_output.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"search_output.d.ts","sourceRoot":"","sources":["../../src/tools/search_output.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAoCnE,MAAM,WAAW,SAAS;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC5B,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,IAAI,EAAE,OAAO,GAAG,QAAQ,GAAG,QAAQ,CAAC;IACpC,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,mBAAmB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC/B,+EAA+E;IAC/E,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,iGAAiG;IACjG,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;CACrB;AAID,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,MAAM,CAKnF;AAED,wBAAgB,aAAa,CAC3B,cAAc,EAAE,KAAK,CAAC;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC,EACxD,cAAc,EAAE,MAAM,GACrB,SAAS,EAAE,CAYb;AAED,wBAAsB,WAAW,CAC/B,MAAM,EAAE,MAAM,EACd,aAAa,CAAC,EAAE,aAAa,GAC5B,OAAO,CAAC;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC,CAOzC;AAED,MAAM,WAAW,yBAAyB;IACxC,WAAW,EAAE,MAAM,CAAC;IACpB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;IACzB,iFAAiF;IACjF,uBAAuB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxC,iFAAiF;IACjF,qBAAqB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACvC;AAED,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,SAAS,EAAE,EACpB,aAAa,EAAE,aAAa,GAAG,SAAS,EACxC,IAAI,EAAE,yBAAyB,GAC9B,OAAO,CAAC;IAAE,SAAS,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,aAAa,EAAE,CAAA;CAAE,CAAC,CAqHjG"}