suemo 0.1.4 → 0.1.7

package/src/types.ts

@@ -131,19 +131,47 @@ export const SuemoStatsSchema = z.object({
 export type SuemoStats = z.infer<typeof SuemoStatsSchema>
 
 // ── query() input ─────────────────────────────────────────────────────────────
+export const QueryStrategySchema = z.enum(['vector', 'bm25', 'graph'])
+export type QueryStrategy = z.infer<typeof QueryStrategySchema>
+
 export const QueryInputSchema = z.object({
 	input: z.string().min(1),
 	scope: z.string().optional(),
 	kind: z.array(MemoryKindSchema).optional(),
 	topK: z.number().int().min(1).max(50).default(5).optional(),
 	activeOnly: z.boolean().default(true).optional(),
-	strategies: z
-		.array(z.enum(['vector', 'bm25', 'graph']))
-		.default(['vector', 'bm25', 'graph'])
-		.optional(),
+	strategies: z.array(QueryStrategySchema).default(['vector', 'bm25', 'graph']).optional(),
+	explain: z.boolean().default(false).optional(),
+	minScore: z.number().min(0).max(1).optional(),
+	relativeFloor: z.number().min(0).max(1).optional(),
 })
 export type QueryInput = z.infer<typeof QueryInputSchema>
 
+export interface QueryStrategyBreakdown {
+	raw: number
+	norm: number
+	weighted: number
+	rank: number
+}
+
+export interface QueryRetrievalExplain {
+	finalScore: number
+	matchedStrategies: QueryStrategy[]
+	agreementBonus: number
+	thresholds: {
+		minScore: number
+		relativeFloor: number
+		topScore: number
+		relativeMinScore: number
+		primaryEvidenceMin: number
+	}
+	vector?: QueryStrategyBreakdown
+	bm25?: QueryStrategyBreakdown
+	graph?: QueryStrategyBreakdown & { anchorCount: number }
+}
+
+export type QueryNode = MemoryNode & { _retrieval?: QueryRetrievalExplain }
+
 // ── Health report ─────────────────────────────────────────────────────────────
 export const HealthReportSchema = z.object({
 	nodes: z.object({