@simonfestl/husky-cli 1.9.1 → 1.9.2

package/dist/lib/biz/agent-brain.d.ts

@@ -1,4 +1,4 @@
-export declare const AGENT_TYPES: readonly ["support", "claude", "gotess", "supervisor", "worker"];
+export declare const AGENT_TYPES: readonly ["support", "claude", "gotess", "supervisor", "worker", "reviewer", "e2e_agent", "pr_agent"];
 export type AgentType = typeof AGENT_TYPES[number];
 export type MemoryVisibility = 'private' | 'team' | 'public';
 export interface Memory {
@@ -42,7 +42,9 @@ export declare class AgentBrain {
     getDatabaseInfo(): {
         agentType?: AgentType;
         databaseName: string;
+        collectionName: string;
     };
+    private getCollectionName;
     private ensureCollection;
     remember(content: string, tags?: string[], metadata?: Record<string, unknown>, visibility?: MemoryVisibility, allowPii?: boolean): Promise<string>;
     recall(query: string, limit?: number, minScore?: number): Promise<RecallResult[]>;

package/dist/lib/biz/agent-brain.js

@@ -3,9 +3,9 @@ import { EmbeddingService } from './embeddings.js';
 import { getConfig } from '../../commands/config.js';
 import { randomUUID } from 'crypto';
 import { sanitizeForEmbedding } from './pii-filter.js';
-const MEMORIES_COLLECTION = 'agent-memories';
+const DEFAULT_COLLECTION = 'agent-memories';
 const VECTOR_SIZE = 768;
-export const AGENT_TYPES = ['support', 'claude', 'gotess', 'supervisor', 'worker'];
+export const AGENT_TYPES = ['support', 'claude', 'gotess', 'supervisor', 'worker', 'reviewer', 'e2e_agent', 'pr_agent'];
 export function isValidAgentType(value) {
     return value !== undefined && AGENT_TYPES.includes(value);
 }
@@ -45,17 +45,25 @@ export class AgentBrain {
         });
     }
     getDatabaseInfo() {
+        const collectionName = this.getCollectionName();
         return {
             agentType: this.agentType,
-            databaseName: `qdrant:${MEMORIES_COLLECTION}`,
+            databaseName: `qdrant:${collectionName}`,
+            collectionName,
         };
     }
+    getCollectionName() {
+        if (!this.agentType)
+            return DEFAULT_COLLECTION;
+        return `${this.agentType}-memories`;
+    }
     async ensureCollection() {
+        const collection = this.getCollectionName();
         try {
-            await this.qdrant.getCollection(MEMORIES_COLLECTION);
+            await this.qdrant.getCollection(collection);
         }
         catch {
-            await this.qdrant.createCollection(MEMORIES_COLLECTION, VECTOR_SIZE);
+            await this.qdrant.createCollection(collection, VECTOR_SIZE);
         }
     }
     async remember(content, tags = [], metadata, visibility = 'private', allowPii = false) {
@@ -69,7 +77,7 @@ export class AgentBrain {
         const embedding = await this.embeddings.embed(sanitizeResult.sanitized);
         const id = randomUUID();
         const now = new Date().toISOString();
-        await this.qdrant.upsertOne(MEMORIES_COLLECTION, id, embedding, {
+        await this.qdrant.upsertOne(this.getCollectionName(), id, embedding, {
             agent: this.agentId,
             agentType: this.agentType || 'default',
             content: sanitizeResult.sanitized, // Store sanitized content
@@ -108,7 +116,7 @@ export class AgentBrain {
         if (this.agentType) {
             filter.must.push({ key: 'agentType', match: { value: this.agentType } });
         }
-        const results = await this.qdrant.search(MEMORIES_COLLECTION, queryEmbedding, limit, {
+        const results = await this.qdrant.search(this.getCollectionName(), queryEmbedding, limit, {
             filter,
             scoreThreshold: minScore,
         });
@@ -142,7 +150,7 @@ export class AgentBrain {
                 }
             ]
         };
-        const results = await this.qdrant.scroll(MEMORIES_COLLECTION, {
+        const results = await this.qdrant.scroll(this.getCollectionName(), {
             filter,
             limit,
             with_payload: true,
@@ -162,7 +170,7 @@ export class AgentBrain {
     }
     async forget(memoryId) {
         await this.ensureCollection();
-        await this.qdrant.deletePoints(MEMORIES_COLLECTION, [memoryId]);
+        await this.qdrant.deletePoints(this.getCollectionName(), [memoryId]);
     }
     async listMemories(limit = 20) {
         await this.ensureCollection();
@@ -174,7 +182,7 @@ export class AgentBrain {
         if (this.agentType) {
             filter.must.push({ key: 'agentType', match: { value: this.agentType } });
         }
-        const results = await this.qdrant.scroll(MEMORIES_COLLECTION, {
+        const results = await this.qdrant.scroll(this.getCollectionName(), {
             filter,
             limit,
             with_payload: true,
@@ -202,7 +210,7 @@ export class AgentBrain {
         if (this.agentType) {
             filter.must.push({ key: 'agentType', match: { value: this.agentType } });
         }
-        const results = await this.qdrant.scroll(MEMORIES_COLLECTION, {
+        const results = await this.qdrant.scroll(this.getCollectionName(), {
             filter,
             limit: 1000,
             with_payload: true,
@@ -228,7 +236,7 @@ export class AgentBrain {
     async publish(memoryId, visibility) {
         await this.ensureCollection();
         const now = new Date().toISOString();
-        await this.qdrant.setPayload(MEMORIES_COLLECTION, memoryId, {
+        await this.qdrant.setPayload(this.getCollectionName(), memoryId, {
             visibility,
             publishedBy: this.agentId,
             publishedAt: now,
@@ -241,7 +249,7 @@ export class AgentBrain {
     async unpublish(memoryId) {
         await this.ensureCollection();
         const now = new Date().toISOString();
-        await this.qdrant.setPayload(MEMORIES_COLLECTION, memoryId, {
+        await this.qdrant.setPayload(this.getCollectionName(), memoryId, {
             visibility: 'private',
             publishedBy: undefined,
             publishedAt: undefined,
@@ -269,7 +277,7 @@ export class AgentBrain {
                 { key: 'status', match: { value: 'active' } },
             ],
         };
-        const results = await this.qdrant.search(MEMORIES_COLLECTION, queryEmbedding, limit * 3, {
+        const results = await this.qdrant.search(this.getCollectionName(), queryEmbedding, limit * 3, {
             filter,
             scoreThreshold: minScore,
         });
@@ -312,7 +320,7 @@ export class AgentBrain {
                 { key: 'status', match: { value: 'active' } },
             ],
         };
-        const results = await this.qdrant.scroll(MEMORIES_COLLECTION, {
+        const results = await this.qdrant.scroll(this.getCollectionName(), {
             filter,
             limit,
             with_payload: true,
@@ -340,13 +348,13 @@ export class AgentBrain {
      */
     async boost(memoryId) {
         await this.ensureCollection();
-        const point = await this.qdrant.getPoint(MEMORIES_COLLECTION, memoryId);
+        const point = await this.qdrant.getPoint(this.getCollectionName(), memoryId);
         if (!point)
             throw new Error('Memory not found');
         const boostCount = Number(point.payload?.boostCount || 0) + 1;
         const downvoteCount = Number(point.payload?.downvoteCount || 0);
         const qualityScore = this.calculateQualityScore(boostCount, downvoteCount);
-        await this.qdrant.setPayload(MEMORIES_COLLECTION, memoryId, {
+        await this.qdrant.setPayload(this.getCollectionName(), memoryId, {
             boostCount,
             qualityScore,
             updatedAt: new Date().toISOString(),
@@ -357,13 +365,13 @@ export class AgentBrain {
      */
     async downvote(memoryId) {
         await this.ensureCollection();
-        const point = await this.qdrant.getPoint(MEMORIES_COLLECTION, memoryId);
+        const point = await this.qdrant.getPoint(this.getCollectionName(), memoryId);
         if (!point)
             throw new Error('Memory not found');
         const boostCount = Number(point.payload?.boostCount || 0);
         const downvoteCount = Number(point.payload?.downvoteCount || 0) + 1;
         const qualityScore = this.calculateQualityScore(boostCount, downvoteCount);
-        await this.qdrant.setPayload(MEMORIES_COLLECTION, memoryId, {
+        await this.qdrant.setPayload(this.getCollectionName(), memoryId, {
             downvoteCount,
             qualityScore,
             updatedAt: new Date().toISOString(),
@@ -374,7 +382,7 @@ export class AgentBrain {
      */
     async getQuality(memoryId) {
         await this.ensureCollection();
-        const point = await this.qdrant.getPoint(MEMORIES_COLLECTION, memoryId);
+        const point = await this.qdrant.getPoint(this.getCollectionName(), memoryId);
         if (!point)
             throw new Error('Memory not found');
         return {
@@ -434,7 +442,7 @@ export class AgentBrain {
                 }))
             });
         }
-        const results = await this.qdrant.scroll(MEMORIES_COLLECTION, {
+        const results = await this.qdrant.scroll(this.getCollectionName(), {
             filter,
             limit: 1000,
             with_payload: true,
@@ -468,7 +476,7 @@ export class AgentBrain {
                     boostCount,
                 });
                 if (!dryRun) {
-                    await this.qdrant.setPayload(MEMORIES_COLLECTION, String(r.id), {
+                    await this.qdrant.setPayload(this.getCollectionName(), String(r.id), {
                         status: 'archived',
                         updatedAt: new Date().toISOString(),
                     });
@@ -488,7 +496,7 @@ export class AgentBrain {
                 { key: 'status', match: { value: 'archived' } },
             ],
         };
-        const results = await this.qdrant.scroll(MEMORIES_COLLECTION, {
+        const results = await this.qdrant.scroll(this.getCollectionName(), {
             filter,
             limit: 1000,
             with_payload: true,
@@ -503,7 +511,7 @@ export class AgentBrain {
             }
         }
         if (toPurge.length > 0) {
-            await this.qdrant.deletePoints(MEMORIES_COLLECTION, toPurge);
+            await this.qdrant.deletePoints(this.getCollectionName(), toPurge);
         }
         return toPurge.length;
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@simonfestl/husky-cli",
-  "version": "1.9.1",
+  "version": "1.9.2",
   "description": "CLI for Huskyv0 Task Orchestration with Claude Agent SDK",
   "type": "module",
   "bin": {