claude-brain 0.14.2 → 0.14.4
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +191 -191
- package/VERSION +1 -1
- package/assets/CLAUDE-unified.md +11 -11
- package/assets/CLAUDE.md +11 -11
- package/bunfig.toml +8 -8
- package/package.json +80 -80
- package/packs/backend/node.json +173 -173
- package/packs/core/javascript.json +176 -176
- package/packs/core/typescript.json +222 -222
- package/packs/frontend/react.json +254 -254
- package/packs/meta/testing.json +172 -172
- package/src/automation/auto-context.ts +240 -240
- package/src/automation/decision-detector.ts +452 -452
- package/src/automation/index.ts +11 -11
- package/src/automation/phase12-manager.ts +456 -456
- package/src/automation/proactive-recall.ts +373 -373
- package/src/automation/project-detector.ts +310 -310
- package/src/automation/repo-scanner.ts +205 -205
- package/src/cli/auto-setup.ts +82 -82
- package/src/cli/bin.ts +202 -202
- package/src/cli/commands/chroma.ts +573 -573
- package/src/cli/commands/git-hook.ts +189 -189
- package/src/cli/commands/hooks.ts +213 -213
- package/src/cli/commands/init.ts +122 -122
- package/src/cli/commands/install-mcp.ts +92 -92
- package/src/cli/commands/pack.ts +197 -197
- package/src/cli/commands/serve.ts +167 -167
- package/src/cli/commands/start.ts +42 -42
- package/src/cli/commands/uninstall-mcp.ts +41 -41
- package/src/cli/commands/update.ts +121 -121
- package/src/cli/diagnose.ts +4 -4
- package/src/cli/health-check.ts +4 -4
- package/src/cli/migrate-chroma.ts +106 -106
- package/src/cli/setup.ts +4 -4
- package/src/cli/ui/animations.ts +80 -80
- package/src/cli/ui/components.ts +82 -82
- package/src/cli/ui/index.ts +4 -4
- package/src/cli/ui/logo.ts +36 -36
- package/src/cli/ui/theme.ts +55 -55
- package/src/config/defaults.ts +50 -50
- package/src/config/home.ts +55 -55
- package/src/config/index.ts +7 -7
- package/src/config/loader.ts +166 -166
- package/src/config/migration.ts +76 -76
- package/src/config/schema.ts +360 -360
- package/src/config/validator.ts +184 -184
- package/src/config/watcher.ts +86 -86
- package/src/context/assembler.ts +398 -398
- package/src/context/cache-manager.ts +101 -101
- package/src/context/formatter.ts +84 -84
- package/src/context/hierarchy.ts +85 -85
- package/src/context/index.ts +83 -83
- package/src/context/progress-tracker.ts +174 -174
- package/src/context/standards-manager.ts +287 -287
- package/src/context/types.ts +252 -252
- package/src/context/validator.ts +58 -58
- package/src/diagnostics/index.ts +123 -123
- package/src/health/index.ts +229 -229
- package/src/hooks/brain-hook.ts +112 -112
- package/src/hooks/capture.ts +168 -168
- package/src/hooks/deduplicator.ts +72 -72
- package/src/hooks/git-capture.ts +109 -109
- package/src/hooks/git-hook-installer.ts +207 -207
- package/src/hooks/index.ts +20 -20
- package/src/hooks/installer.ts +191 -194
- package/src/hooks/passive-classifier.ts +366 -366
- package/src/hooks/queue.ts +129 -129
- package/src/hooks/session-tracker.ts +275 -275
- package/src/hooks/types.ts +47 -47
- package/src/index.ts +7 -7
- package/src/intelligence/cross-project/affinity.ts +162 -162
- package/src/intelligence/cross-project/generalizer.ts +283 -283
- package/src/intelligence/cross-project/index.ts +13 -13
- package/src/intelligence/cross-project/transfer.ts +201 -201
- package/src/intelligence/index.ts +24 -24
- package/src/intelligence/optimization/index.ts +10 -10
- package/src/intelligence/optimization/precompute.ts +202 -202
- package/src/intelligence/optimization/semantic-cache.ts +207 -207
- package/src/intelligence/prediction/context-anticipator.ts +198 -198
- package/src/intelligence/prediction/decision-predictor.ts +184 -184
- package/src/intelligence/prediction/index.ts +13 -13
- package/src/intelligence/prediction/recommender.ts +268 -268
- package/src/intelligence/reasoning/chain-retrieval.ts +247 -247
- package/src/intelligence/reasoning/counterfactual.ts +248 -248
- package/src/intelligence/reasoning/index.ts +13 -13
- package/src/intelligence/reasoning/synthesizer.ts +169 -169
- package/src/intelligence/temporal/evolution.ts +197 -197
- package/src/intelligence/temporal/index.ts +16 -16
- package/src/intelligence/temporal/query-processor.ts +190 -190
- package/src/intelligence/temporal/timeline.ts +259 -259
- package/src/intelligence/temporal/trends.ts +263 -263
- package/src/knowledge/entity-extractor.ts +416 -416
- package/src/knowledge/graph/builder.ts +185 -185
- package/src/knowledge/graph/linker.ts +201 -201
- package/src/knowledge/graph/memory-graph.ts +359 -359
- package/src/knowledge/graph/schema.ts +99 -99
- package/src/knowledge/graph/search.ts +168 -168
- package/src/knowledge/relationship-extractor.ts +108 -108
- package/src/memory/chroma/client.ts +174 -174
- package/src/memory/chroma/collection-manager.ts +94 -94
- package/src/memory/chroma/config.ts +57 -57
- package/src/memory/chroma/embeddings.ts +153 -153
- package/src/memory/chroma/index.ts +82 -82
- package/src/memory/chroma/migration.ts +270 -270
- package/src/memory/chroma/schemas.ts +69 -69
- package/src/memory/chroma/search.ts +315 -315
- package/src/memory/chroma/store.ts +741 -741
- package/src/memory/consolidation/archiver.ts +164 -164
- package/src/memory/consolidation/merger.ts +186 -186
- package/src/memory/consolidation/scorer.ts +138 -138
- package/src/memory/context-builder.ts +236 -236
- package/src/memory/database.ts +169 -169
- package/src/memory/embedding-utils.ts +156 -156
- package/src/memory/embeddings.ts +226 -226
- package/src/memory/episodic/detector.ts +108 -108
- package/src/memory/episodic/manager.ts +351 -351
- package/src/memory/episodic/summarizer.ts +179 -179
- package/src/memory/episodic/types.ts +52 -52
- package/src/memory/index.ts +582 -582
- package/src/memory/knowledge-extractor.ts +455 -455
- package/src/memory/learning.ts +378 -378
- package/src/memory/patterns.ts +396 -396
- package/src/memory/schema.ts +88 -88
- package/src/memory/search.ts +309 -309
- package/src/memory/store.ts +787 -787
- package/src/memory/types.ts +121 -121
- package/src/orchestrator/coordinator.ts +272 -272
- package/src/orchestrator/decision-logger.ts +228 -228
- package/src/orchestrator/event-emitter.ts +198 -198
- package/src/orchestrator/event-queue.ts +184 -184
- package/src/orchestrator/handlers/base-handler.ts +70 -70
- package/src/orchestrator/handlers/context-handler.ts +73 -73
- package/src/orchestrator/handlers/decision-handler.ts +204 -204
- package/src/orchestrator/handlers/index.ts +10 -10
- package/src/orchestrator/handlers/status-handler.ts +131 -131
- package/src/orchestrator/handlers/task-handler.ts +171 -171
- package/src/orchestrator/index.ts +275 -275
- package/src/orchestrator/task-parser.ts +284 -284
- package/src/orchestrator/types.ts +98 -98
- package/src/packs/index.ts +9 -9
- package/src/packs/loader.ts +134 -134
- package/src/packs/manager.ts +204 -204
- package/src/packs/ranker.ts +78 -78
- package/src/packs/types.ts +81 -81
- package/src/phase12/index.ts +5 -5
- package/src/retrieval/bm25/index.ts +300 -300
- package/src/retrieval/bm25/tokenizer.ts +184 -184
- package/src/retrieval/feedback/adaptive.ts +223 -223
- package/src/retrieval/feedback/index.ts +16 -16
- package/src/retrieval/feedback/metrics.ts +223 -223
- package/src/retrieval/feedback/store.ts +283 -283
- package/src/retrieval/fusion/index.ts +194 -194
- package/src/retrieval/fusion/rrf.ts +163 -163
- package/src/retrieval/index.ts +12 -12
- package/src/retrieval/pipeline.ts +375 -375
- package/src/retrieval/query/expander.ts +198 -198
- package/src/retrieval/query/index.ts +27 -27
- package/src/retrieval/query/intent-classifier.ts +236 -236
- package/src/retrieval/query/temporal-parser.ts +295 -295
- package/src/retrieval/reranker/index.ts +188 -188
- package/src/retrieval/reranker/model.ts +95 -95
- package/src/retrieval/service.ts +125 -125
- package/src/retrieval/types.ts +162 -162
- package/src/routing/entity-extractor.ts +428 -428
- package/src/routing/intent-classifier.ts +436 -436
- package/src/routing/response-filter.ts +258 -254
- package/src/routing/router.ts +1322 -1314
- package/src/routing/search-engine.ts +475 -475
- package/src/routing/types.ts +94 -84
- package/src/scripts/health-check.ts +118 -118
- package/src/scripts/setup.ts +122 -122
- package/src/server/handlers/call-tool.ts +156 -156
- package/src/server/handlers/index.ts +9 -9
- package/src/server/handlers/list-tools.ts +35 -35
- package/src/server/handlers/tools/analyze-decision-evolution.ts +151 -151
- package/src/server/handlers/tools/auto-remember.ts +200 -200
- package/src/server/handlers/tools/brain.ts +85 -85
- package/src/server/handlers/tools/create-project.ts +135 -135
- package/src/server/handlers/tools/detect-trends.ts +144 -144
- package/src/server/handlers/tools/find-cross-project-patterns.ts +168 -168
- package/src/server/handlers/tools/get-activity-log.ts +194 -194
- package/src/server/handlers/tools/get-code-standards.ts +124 -124
- package/src/server/handlers/tools/get-corrections.ts +154 -154
- package/src/server/handlers/tools/get-decision-timeline.ts +172 -172
- package/src/server/handlers/tools/get-episode.ts +103 -103
- package/src/server/handlers/tools/get-patterns.ts +158 -158
- package/src/server/handlers/tools/get-phase12-status.ts +63 -63
- package/src/server/handlers/tools/get-project-context.ts +75 -75
- package/src/server/handlers/tools/get-recommendations.ts +145 -145
- package/src/server/handlers/tools/index.ts +31 -31
- package/src/server/handlers/tools/init-project.ts +757 -757
- package/src/server/handlers/tools/list-episodes.ts +90 -90
- package/src/server/handlers/tools/list-projects.ts +125 -125
- package/src/server/handlers/tools/rate-memory.ts +101 -101
- package/src/server/handlers/tools/recall-similar.ts +87 -87
- package/src/server/handlers/tools/recognize-pattern.ts +126 -126
- package/src/server/handlers/tools/record-correction.ts +125 -125
- package/src/server/handlers/tools/remember-decision.ts +153 -153
- package/src/server/handlers/tools/schemas.ts +253 -253
- package/src/server/handlers/tools/search-knowledge-graph.ts +102 -102
- package/src/server/handlers/tools/smart-context.ts +146 -146
- package/src/server/handlers/tools/update-progress.ts +131 -131
- package/src/server/handlers/tools/what-if-analysis.ts +135 -135
- package/src/server/http-api.ts +693 -693
- package/src/server/index.ts +40 -40
- package/src/server/mcp-server.ts +283 -283
- package/src/server/providers/index.ts +7 -7
- package/src/server/providers/prompts.ts +327 -327
- package/src/server/providers/resources.ts +622 -622
- package/src/server/services.ts +468 -468
- package/src/server/types.ts +39 -39
- package/src/server/utils/error-handler.ts +155 -155
- package/src/server/utils/index.ts +13 -13
- package/src/server/utils/memory-indicator.ts +83 -83
- package/src/server/utils/request-context.ts +122 -122
- package/src/server/utils/response-formatter.ts +129 -124
- package/src/server/utils/validators.ts +210 -210
- package/src/setup/index.ts +48 -48
- package/src/setup/wizard.ts +461 -461
- package/src/tools/index.ts +24 -24
- package/src/tools/registry.ts +115 -115
- package/src/tools/schemas.test.ts +30 -30
- package/src/tools/schemas.ts +617 -617
- package/src/tools/types.ts +412 -412
- package/src/utils/circuit-breaker.ts +130 -130
- package/src/utils/cleanup.ts +34 -34
- package/src/utils/error-handler.ts +132 -132
- package/src/utils/error-messages.ts +60 -60
- package/src/utils/fallback.ts +45 -45
- package/src/utils/index.ts +54 -54
- package/src/utils/logger-utils.ts +80 -80
- package/src/utils/logger.ts +88 -88
- package/src/utils/phase12-helper.ts +56 -56
- package/src/utils/retry.ts +94 -94
- package/src/utils/timing.ts +47 -47
- package/src/utils/transaction.ts +63 -63
- package/src/vault/frontmatter.ts +264 -264
- package/src/vault/index.ts +318 -318
- package/src/vault/paths.ts +106 -106
- package/src/vault/query.ts +422 -422
- package/src/vault/reader.ts +264 -264
- package/src/vault/templates.ts +186 -186
- package/src/vault/types.ts +73 -73
- package/src/vault/watcher.ts +277 -277
- package/src/vault/writer.ts +413 -413
- package/tsconfig.json +30 -30
|
@@ -1,131 +1,131 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Update Progress Handler
|
|
3
|
-
* Updates project progress tracking with completed tasks and next steps
|
|
4
|
-
*/
|
|
5
|
-
|
|
6
|
-
import type { Logger } from 'pino'
|
|
7
|
-
import type { ToolResponse } from '@/tools/types'
|
|
8
|
-
import { getVaultService, getContextService } from '@/server/services'
|
|
9
|
-
import { ToolValidator } from '@/server/utils/validators'
|
|
10
|
-
import { ResponseFormatter } from '@/server/utils/response-formatter'
|
|
11
|
-
import { ErrorHandler } from '@/server/utils/error-handler'
|
|
12
|
-
import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
|
|
13
|
-
import { UpdateProgressSchema } from './schemas'
|
|
14
|
-
|
|
15
|
-
export async function handleUpdateProgress(
|
|
16
|
-
args: unknown,
|
|
17
|
-
logger: Logger
|
|
18
|
-
): Promise<ToolResponse> {
|
|
19
|
-
try {
|
|
20
|
-
const input = ToolValidator.validate(args, UpdateProgressSchema)
|
|
21
|
-
const { project_name, completed_task, next_steps, notes } = input
|
|
22
|
-
|
|
23
|
-
logger.debug({
|
|
24
|
-
project_name,
|
|
25
|
-
completed_task,
|
|
26
|
-
next_steps,
|
|
27
|
-
hasNotes: !!notes
|
|
28
|
-
}, 'Updating progress')
|
|
29
|
-
|
|
30
|
-
const vault = getVaultService()
|
|
31
|
-
const context = getContextService()
|
|
32
|
-
|
|
33
|
-
const projectPaths = vault.getProjectPaths(project_name)
|
|
34
|
-
|
|
35
|
-
try {
|
|
36
|
-
const file = await vault.reader.readMarkdownFile(projectPaths.progress)
|
|
37
|
-
} catch {
|
|
38
|
-
throw new McpError(
|
|
39
|
-
ErrorCode.InvalidParams,
|
|
40
|
-
`Project "${project_name}" not found. Progress file does not exist. Use list_projects to see available projects.`
|
|
41
|
-
)
|
|
42
|
-
}
|
|
43
|
-
|
|
44
|
-
const taskId = generateTaskId(completed_task)
|
|
45
|
-
|
|
46
|
-
// Step 1: Append completed task to progress file
|
|
47
|
-
await context.progress.addCompletedTask(project_name, {
|
|
48
|
-
id: taskId,
|
|
49
|
-
title: completed_task,
|
|
50
|
-
status: 'done',
|
|
51
|
-
completedAt: new Date()
|
|
52
|
-
})
|
|
53
|
-
|
|
54
|
-
// Step 2: Read file bypassing cache, update next steps, write content
|
|
55
|
-
const progressFile = await vault.reader.readMarkdownFile(projectPaths.progress, false)
|
|
56
|
-
const updatedContent = updateNextStepsSection(progressFile.content, next_steps)
|
|
57
|
-
|
|
58
|
-
await vault.writer.writeMarkdownFile(
|
|
59
|
-
projectPaths.progress,
|
|
60
|
-
progressFile.frontmatter,
|
|
61
|
-
updatedContent,
|
|
62
|
-
true
|
|
63
|
-
)
|
|
64
|
-
|
|
65
|
-
// Step 3: Notes appended if present
|
|
66
|
-
if (notes) {
|
|
67
|
-
const notesEntry = `\n## Notes (${new Date().toLocaleDateString()})\n${notes}\n`
|
|
68
|
-
await vault.writer.appendContent(projectPaths.progress, notesEntry)
|
|
69
|
-
}
|
|
70
|
-
|
|
71
|
-
// Step 4: Calculate and update frontmatter LAST (so nothing overwrites it)
|
|
72
|
-
const progressState = await context.progress.getProgress(project_name)
|
|
73
|
-
const totalTasks = progressState.completedTasks.length + progressState.currentTasks.length
|
|
74
|
-
const completionPercentage = totalTasks > 0
|
|
75
|
-
? Math.round((progressState.completedTasks.length / totalTasks) * 100)
|
|
76
|
-
: 0
|
|
77
|
-
|
|
78
|
-
await context.progress.updateProgress(project_name, {
|
|
79
|
-
completionPercentage,
|
|
80
|
-
currentPhase: completionPercentage >= 100 ? 'complete' : 'active'
|
|
81
|
-
})
|
|
82
|
-
|
|
83
|
-
// Step 5: Invalidate cache so subsequent reads get fresh data
|
|
84
|
-
vault.reader.invalidateCache(projectPaths.progress)
|
|
85
|
-
context.invalidateContext(project_name)
|
|
86
|
-
|
|
87
|
-
logger.info({ projectName: project_name, taskId }, 'Progress updated successfully')
|
|
88
|
-
|
|
89
|
-
return ResponseFormatter.text(
|
|
90
|
-
`✅ Progress updated for ${project_name}\n\n` +
|
|
91
|
-
`**Completed:** ${completed_task}\n` +
|
|
92
|
-
`**Next Steps:** ${next_steps}\n` +
|
|
93
|
-
(notes ? `**Notes:** ${notes}\n` : '') +
|
|
94
|
-
`\nThe progress has been saved to your vault and context cache has been refreshed.`
|
|
95
|
-
)
|
|
96
|
-
|
|
97
|
-
} catch (error) {
|
|
98
|
-
ErrorHandler.logError(logger, error, { tool: 'update_progress', args })
|
|
99
|
-
|
|
100
|
-
if (error instanceof McpError) {
|
|
101
|
-
throw error
|
|
102
|
-
}
|
|
103
|
-
|
|
104
|
-
throw new McpError(
|
|
105
|
-
ErrorCode.InternalError,
|
|
106
|
-
`Failed to update progress: ${error instanceof Error ? error.message : 'Unknown error'}`
|
|
107
|
-
)
|
|
108
|
-
}
|
|
109
|
-
}
|
|
110
|
-
|
|
111
|
-
function generateTaskId(title: string): string {
|
|
112
|
-
return title
|
|
113
|
-
.toLowerCase()
|
|
114
|
-
.replace(/[^a-z0-9]+/g, '-')
|
|
115
|
-
.replace(/^-+|-+$/g, '')
|
|
116
|
-
.slice(0, 50)
|
|
117
|
-
}
|
|
118
|
-
|
|
119
|
-
function updateNextStepsSection(content: string, nextSteps: string): string {
|
|
120
|
-
const nextStepsRegex = /##\s+Next Steps\s+(.+?)(?=\n##|$)/s
|
|
121
|
-
const match = nextStepsRegex.exec(content)
|
|
122
|
-
|
|
123
|
-
if (match) {
|
|
124
|
-
return content.replace(
|
|
125
|
-
nextStepsRegex,
|
|
126
|
-
`## Next Steps\n\n- ${nextSteps}\n`
|
|
127
|
-
)
|
|
128
|
-
} else {
|
|
129
|
-
return content + `\n\n## Next Steps\n\n- ${nextSteps}\n`
|
|
130
|
-
}
|
|
131
|
-
}
|
|
1
|
+
/**
|
|
2
|
+
* Update Progress Handler
|
|
3
|
+
* Updates project progress tracking with completed tasks and next steps
|
|
4
|
+
*/
|
|
5
|
+
|
|
6
|
+
import type { Logger } from 'pino'
|
|
7
|
+
import type { ToolResponse } from '@/tools/types'
|
|
8
|
+
import { getVaultService, getContextService } from '@/server/services'
|
|
9
|
+
import { ToolValidator } from '@/server/utils/validators'
|
|
10
|
+
import { ResponseFormatter } from '@/server/utils/response-formatter'
|
|
11
|
+
import { ErrorHandler } from '@/server/utils/error-handler'
|
|
12
|
+
import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
|
|
13
|
+
import { UpdateProgressSchema } from './schemas'
|
|
14
|
+
|
|
15
|
+
export async function handleUpdateProgress(
|
|
16
|
+
args: unknown,
|
|
17
|
+
logger: Logger
|
|
18
|
+
): Promise<ToolResponse> {
|
|
19
|
+
try {
|
|
20
|
+
const input = ToolValidator.validate(args, UpdateProgressSchema)
|
|
21
|
+
const { project_name, completed_task, next_steps, notes } = input
|
|
22
|
+
|
|
23
|
+
logger.debug({
|
|
24
|
+
project_name,
|
|
25
|
+
completed_task,
|
|
26
|
+
next_steps,
|
|
27
|
+
hasNotes: !!notes
|
|
28
|
+
}, 'Updating progress')
|
|
29
|
+
|
|
30
|
+
const vault = getVaultService()
|
|
31
|
+
const context = getContextService()
|
|
32
|
+
|
|
33
|
+
const projectPaths = vault.getProjectPaths(project_name)
|
|
34
|
+
|
|
35
|
+
try {
|
|
36
|
+
const file = await vault.reader.readMarkdownFile(projectPaths.progress)
|
|
37
|
+
} catch {
|
|
38
|
+
throw new McpError(
|
|
39
|
+
ErrorCode.InvalidParams,
|
|
40
|
+
`Project "${project_name}" not found. Progress file does not exist. Use list_projects to see available projects.`
|
|
41
|
+
)
|
|
42
|
+
}
|
|
43
|
+
|
|
44
|
+
const taskId = generateTaskId(completed_task)
|
|
45
|
+
|
|
46
|
+
// Step 1: Append completed task to progress file
|
|
47
|
+
await context.progress.addCompletedTask(project_name, {
|
|
48
|
+
id: taskId,
|
|
49
|
+
title: completed_task,
|
|
50
|
+
status: 'done',
|
|
51
|
+
completedAt: new Date()
|
|
52
|
+
})
|
|
53
|
+
|
|
54
|
+
// Step 2: Read file bypassing cache, update next steps, write content
|
|
55
|
+
const progressFile = await vault.reader.readMarkdownFile(projectPaths.progress, false)
|
|
56
|
+
const updatedContent = updateNextStepsSection(progressFile.content, next_steps)
|
|
57
|
+
|
|
58
|
+
await vault.writer.writeMarkdownFile(
|
|
59
|
+
projectPaths.progress,
|
|
60
|
+
progressFile.frontmatter,
|
|
61
|
+
updatedContent,
|
|
62
|
+
true
|
|
63
|
+
)
|
|
64
|
+
|
|
65
|
+
// Step 3: Notes appended if present
|
|
66
|
+
if (notes) {
|
|
67
|
+
const notesEntry = `\n## Notes (${new Date().toLocaleDateString()})\n${notes}\n`
|
|
68
|
+
await vault.writer.appendContent(projectPaths.progress, notesEntry)
|
|
69
|
+
}
|
|
70
|
+
|
|
71
|
+
// Step 4: Calculate and update frontmatter LAST (so nothing overwrites it)
|
|
72
|
+
const progressState = await context.progress.getProgress(project_name)
|
|
73
|
+
const totalTasks = progressState.completedTasks.length + progressState.currentTasks.length
|
|
74
|
+
const completionPercentage = totalTasks > 0
|
|
75
|
+
? Math.round((progressState.completedTasks.length / totalTasks) * 100)
|
|
76
|
+
: 0
|
|
77
|
+
|
|
78
|
+
await context.progress.updateProgress(project_name, {
|
|
79
|
+
completionPercentage,
|
|
80
|
+
currentPhase: completionPercentage >= 100 ? 'complete' : 'active'
|
|
81
|
+
})
|
|
82
|
+
|
|
83
|
+
// Step 5: Invalidate cache so subsequent reads get fresh data
|
|
84
|
+
vault.reader.invalidateCache(projectPaths.progress)
|
|
85
|
+
context.invalidateContext(project_name)
|
|
86
|
+
|
|
87
|
+
logger.info({ projectName: project_name, taskId }, 'Progress updated successfully')
|
|
88
|
+
|
|
89
|
+
return ResponseFormatter.text(
|
|
90
|
+
`✅ Progress updated for ${project_name}\n\n` +
|
|
91
|
+
`**Completed:** ${completed_task}\n` +
|
|
92
|
+
`**Next Steps:** ${next_steps}\n` +
|
|
93
|
+
(notes ? `**Notes:** ${notes}\n` : '') +
|
|
94
|
+
`\nThe progress has been saved to your vault and context cache has been refreshed.`
|
|
95
|
+
)
|
|
96
|
+
|
|
97
|
+
} catch (error) {
|
|
98
|
+
ErrorHandler.logError(logger, error, { tool: 'update_progress', args })
|
|
99
|
+
|
|
100
|
+
if (error instanceof McpError) {
|
|
101
|
+
throw error
|
|
102
|
+
}
|
|
103
|
+
|
|
104
|
+
throw new McpError(
|
|
105
|
+
ErrorCode.InternalError,
|
|
106
|
+
`Failed to update progress: ${error instanceof Error ? error.message : 'Unknown error'}`
|
|
107
|
+
)
|
|
108
|
+
}
|
|
109
|
+
}
|
|
110
|
+
|
|
111
|
+
function generateTaskId(title: string): string {
|
|
112
|
+
return title
|
|
113
|
+
.toLowerCase()
|
|
114
|
+
.replace(/[^a-z0-9]+/g, '-')
|
|
115
|
+
.replace(/^-+|-+$/g, '')
|
|
116
|
+
.slice(0, 50)
|
|
117
|
+
}
|
|
118
|
+
|
|
119
|
+
function updateNextStepsSection(content: string, nextSteps: string): string {
|
|
120
|
+
const nextStepsRegex = /##\s+Next Steps\s+(.+?)(?=\n##|$)/s
|
|
121
|
+
const match = nextStepsRegex.exec(content)
|
|
122
|
+
|
|
123
|
+
if (match) {
|
|
124
|
+
return content.replace(
|
|
125
|
+
nextStepsRegex,
|
|
126
|
+
`## Next Steps\n\n- ${nextSteps}\n`
|
|
127
|
+
)
|
|
128
|
+
} else {
|
|
129
|
+
return content + `\n\n## Next Steps\n\n- ${nextSteps}\n`
|
|
130
|
+
}
|
|
131
|
+
}
|
|
@@ -1,135 +1,135 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* What-If Analysis Handler
|
|
3
|
-
* Counterfactual analysis using knowledge graph and decision history
|
|
4
|
-
*/
|
|
5
|
-
|
|
6
|
-
import type { Logger } from 'pino'
|
|
7
|
-
import type { ToolResponse } from '@/tools/types'
|
|
8
|
-
import { getMemoryService, getKnowledgeGraphService } from '@/server/services'
|
|
9
|
-
import { ToolValidator } from '@/server/utils/validators'
|
|
10
|
-
import { ResponseFormatter } from '@/server/utils/response-formatter'
|
|
11
|
-
import { withMemoryIndicator } from '@/server/utils/memory-indicator'
|
|
12
|
-
import { ErrorHandler } from '@/server/utils/error-handler'
|
|
13
|
-
import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
|
|
14
|
-
import { WhatIfAnalysisSchema } from './schemas'
|
|
15
|
-
import { CounterfactualAnalyzer } from '@/intelligence/reasoning/counterfactual'
|
|
16
|
-
|
|
17
|
-
export async function handleWhatIfAnalysis(
|
|
18
|
-
args: unknown,
|
|
19
|
-
logger: Logger
|
|
20
|
-
): Promise<ToolResponse> {
|
|
21
|
-
try {
|
|
22
|
-
const input = ToolValidator.validate(args, WhatIfAnalysisSchema)
|
|
23
|
-
const { change, project_name, max_results } = input
|
|
24
|
-
|
|
25
|
-
const memory = getMemoryService()
|
|
26
|
-
|
|
27
|
-
if (!memory.isChromaDBEnabled()) {
|
|
28
|
-
// SQLite fallback: use searchRaw for semantic search + knowledge graph if available
|
|
29
|
-
const results = await memory.searchRaw(change, {
|
|
30
|
-
project: project_name,
|
|
31
|
-
limit: max_results || 10,
|
|
32
|
-
minSimilarity: 0.2
|
|
33
|
-
})
|
|
34
|
-
|
|
35
|
-
const parts: string[] = []
|
|
36
|
-
parts.push(`## What-If: "${change}" (SQLite)`)
|
|
37
|
-
parts.push('')
|
|
38
|
-
|
|
39
|
-
if (results.length === 0) {
|
|
40
|
-
parts.push('No related decisions found to assess impact.')
|
|
41
|
-
return ResponseFormatter.text(parts.join('\n'))
|
|
42
|
-
}
|
|
43
|
-
|
|
44
|
-
// Assess impact based on similarity
|
|
45
|
-
const affectedDecisions = results.map(r => {
|
|
46
|
-
const similarity = r.similarity || 0
|
|
47
|
-
const impact = similarity > 0.7 ? 'high' : similarity > 0.4 ? 'medium' : 'low'
|
|
48
|
-
return {
|
|
49
|
-
decision: r.decision?.decision || r.memory?.content?.slice(0, 150) || '',
|
|
50
|
-
impact,
|
|
51
|
-
similarity,
|
|
52
|
-
reason: `${Math.round(similarity * 100)}% similarity — ${r.decision?.reasoning?.slice(0, 100) || 'related context found'}`
|
|
53
|
-
}
|
|
54
|
-
})
|
|
55
|
-
|
|
56
|
-
parts.push(`Analyzed ${results.length} related decisions.\n`)
|
|
57
|
-
|
|
58
|
-
parts.push('### Affected Decisions')
|
|
59
|
-
for (const d of affectedDecisions) {
|
|
60
|
-
const icon = d.impact === 'high' ? '🔴' : d.impact === 'medium' ? '🟡' : '🟢'
|
|
61
|
-
parts.push(`${icon} **[${d.impact}]** ${d.decision}`)
|
|
62
|
-
parts.push(` *${d.reason}*`)
|
|
63
|
-
}
|
|
64
|
-
|
|
65
|
-
// Try knowledge graph if available
|
|
66
|
-
try {
|
|
67
|
-
const kgService = getKnowledgeGraphService()
|
|
68
|
-
if (kgService?.graph) {
|
|
69
|
-
const graphResults = await kgService.graph.search(change, { limit: 5 })
|
|
70
|
-
if (graphResults && graphResults.length > 0) {
|
|
71
|
-
const technologies = graphResults
|
|
72
|
-
.filter((n: any) => n.type === 'technology')
|
|
73
|
-
.map((n: any) => n.name || n.label)
|
|
74
|
-
if (technologies.length > 0) {
|
|
75
|
-
parts.push(`\n### Connected Technologies`)
|
|
76
|
-
parts.push(technologies.map((t: string) => `- ${t}`).join('\n'))
|
|
77
|
-
}
|
|
78
|
-
}
|
|
79
|
-
}
|
|
80
|
-
} catch {
|
|
81
|
-
// Knowledge graph not available, skip
|
|
82
|
-
}
|
|
83
|
-
|
|
84
|
-
const content = parts.join('\n')
|
|
85
|
-
return ResponseFormatter.text(withMemoryIndicator(content, results.length))
|
|
86
|
-
}
|
|
87
|
-
|
|
88
|
-
const kgService = getKnowledgeGraphService()
|
|
89
|
-
const graph = kgService?.graph || null
|
|
90
|
-
|
|
91
|
-
const analyzer = new CounterfactualAnalyzer(
|
|
92
|
-
logger,
|
|
93
|
-
graph,
|
|
94
|
-
memory.chroma.collections,
|
|
95
|
-
memory.chroma.embeddings
|
|
96
|
-
)
|
|
97
|
-
|
|
98
|
-
const scenario = await analyzer.whatIf(change, {
|
|
99
|
-
project: project_name,
|
|
100
|
-
maxResults: max_results
|
|
101
|
-
})
|
|
102
|
-
|
|
103
|
-
const parts: string[] = []
|
|
104
|
-
parts.push(scenario.summary)
|
|
105
|
-
parts.push('')
|
|
106
|
-
|
|
107
|
-
if (scenario.affectedDecisions.length > 0) {
|
|
108
|
-
parts.push('### Affected Decisions')
|
|
109
|
-
for (const d of scenario.affectedDecisions.slice(0, 10)) {
|
|
110
|
-
const icon = d.impact === 'high' ? '🔴' : d.impact === 'medium' ? '🟡' : '🟢'
|
|
111
|
-
parts.push(`${icon} **[${d.impact}]** ${d.decision}`)
|
|
112
|
-
parts.push(` *${d.reason}*`)
|
|
113
|
-
}
|
|
114
|
-
}
|
|
115
|
-
|
|
116
|
-
if (scenario.affectedTechnologies.length > 0) {
|
|
117
|
-
parts.push(`\n### Connected Technologies`)
|
|
118
|
-
parts.push(scenario.affectedTechnologies.map(t => `- ${t}`).join('\n'))
|
|
119
|
-
}
|
|
120
|
-
|
|
121
|
-
const content = parts.join('\n')
|
|
122
|
-
const totalAffected = scenario.affectedDecisions.length
|
|
123
|
-
|
|
124
|
-
if (totalAffected > 0) {
|
|
125
|
-
return ResponseFormatter.text(withMemoryIndicator(content, totalAffected))
|
|
126
|
-
}
|
|
127
|
-
|
|
128
|
-
return ResponseFormatter.text(content)
|
|
129
|
-
|
|
130
|
-
} catch (error) {
|
|
131
|
-
ErrorHandler.logError(logger, error, { tool: 'what_if_analysis' })
|
|
132
|
-
if (error instanceof McpError) throw error
|
|
133
|
-
throw new McpError(ErrorCode.InternalError, `What-if analysis failed: ${error instanceof Error ? error.message : 'Unknown error'}`)
|
|
134
|
-
}
|
|
135
|
-
}
|
|
1
|
+
/**
|
|
2
|
+
* What-If Analysis Handler
|
|
3
|
+
* Counterfactual analysis using knowledge graph and decision history
|
|
4
|
+
*/
|
|
5
|
+
|
|
6
|
+
import type { Logger } from 'pino'
|
|
7
|
+
import type { ToolResponse } from '@/tools/types'
|
|
8
|
+
import { getMemoryService, getKnowledgeGraphService } from '@/server/services'
|
|
9
|
+
import { ToolValidator } from '@/server/utils/validators'
|
|
10
|
+
import { ResponseFormatter } from '@/server/utils/response-formatter'
|
|
11
|
+
import { withMemoryIndicator } from '@/server/utils/memory-indicator'
|
|
12
|
+
import { ErrorHandler } from '@/server/utils/error-handler'
|
|
13
|
+
import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
|
|
14
|
+
import { WhatIfAnalysisSchema } from './schemas'
|
|
15
|
+
import { CounterfactualAnalyzer } from '@/intelligence/reasoning/counterfactual'
|
|
16
|
+
|
|
17
|
+
export async function handleWhatIfAnalysis(
|
|
18
|
+
args: unknown,
|
|
19
|
+
logger: Logger
|
|
20
|
+
): Promise<ToolResponse> {
|
|
21
|
+
try {
|
|
22
|
+
const input = ToolValidator.validate(args, WhatIfAnalysisSchema)
|
|
23
|
+
const { change, project_name, max_results } = input
|
|
24
|
+
|
|
25
|
+
const memory = getMemoryService()
|
|
26
|
+
|
|
27
|
+
if (!memory.isChromaDBEnabled()) {
|
|
28
|
+
// SQLite fallback: use searchRaw for semantic search + knowledge graph if available
|
|
29
|
+
const results = await memory.searchRaw(change, {
|
|
30
|
+
project: project_name,
|
|
31
|
+
limit: max_results || 10,
|
|
32
|
+
minSimilarity: 0.2
|
|
33
|
+
})
|
|
34
|
+
|
|
35
|
+
const parts: string[] = []
|
|
36
|
+
parts.push(`## What-If: "${change}" (SQLite)`)
|
|
37
|
+
parts.push('')
|
|
38
|
+
|
|
39
|
+
if (results.length === 0) {
|
|
40
|
+
parts.push('No related decisions found to assess impact.')
|
|
41
|
+
return ResponseFormatter.text(parts.join('\n'))
|
|
42
|
+
}
|
|
43
|
+
|
|
44
|
+
// Assess impact based on similarity
|
|
45
|
+
const affectedDecisions = results.map(r => {
|
|
46
|
+
const similarity = r.similarity || 0
|
|
47
|
+
const impact = similarity > 0.7 ? 'high' : similarity > 0.4 ? 'medium' : 'low'
|
|
48
|
+
return {
|
|
49
|
+
decision: r.decision?.decision || r.memory?.content?.slice(0, 150) || '',
|
|
50
|
+
impact,
|
|
51
|
+
similarity,
|
|
52
|
+
reason: `${Math.round(similarity * 100)}% similarity — ${r.decision?.reasoning?.slice(0, 100) || 'related context found'}`
|
|
53
|
+
}
|
|
54
|
+
})
|
|
55
|
+
|
|
56
|
+
parts.push(`Analyzed ${results.length} related decisions.\n`)
|
|
57
|
+
|
|
58
|
+
parts.push('### Affected Decisions')
|
|
59
|
+
for (const d of affectedDecisions) {
|
|
60
|
+
const icon = d.impact === 'high' ? '🔴' : d.impact === 'medium' ? '🟡' : '🟢'
|
|
61
|
+
parts.push(`${icon} **[${d.impact}]** ${d.decision}`)
|
|
62
|
+
parts.push(` *${d.reason}*`)
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
// Try knowledge graph if available
|
|
66
|
+
try {
|
|
67
|
+
const kgService = getKnowledgeGraphService()
|
|
68
|
+
if (kgService?.graph) {
|
|
69
|
+
const graphResults = await kgService.graph.search(change, { limit: 5 })
|
|
70
|
+
if (graphResults && graphResults.length > 0) {
|
|
71
|
+
const technologies = graphResults
|
|
72
|
+
.filter((n: any) => n.type === 'technology')
|
|
73
|
+
.map((n: any) => n.name || n.label)
|
|
74
|
+
if (technologies.length > 0) {
|
|
75
|
+
parts.push(`\n### Connected Technologies`)
|
|
76
|
+
parts.push(technologies.map((t: string) => `- ${t}`).join('\n'))
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
} catch {
|
|
81
|
+
// Knowledge graph not available, skip
|
|
82
|
+
}
|
|
83
|
+
|
|
84
|
+
const content = parts.join('\n')
|
|
85
|
+
return ResponseFormatter.text(withMemoryIndicator(content, results.length))
|
|
86
|
+
}
|
|
87
|
+
|
|
88
|
+
const kgService = getKnowledgeGraphService()
|
|
89
|
+
const graph = kgService?.graph || null
|
|
90
|
+
|
|
91
|
+
const analyzer = new CounterfactualAnalyzer(
|
|
92
|
+
logger,
|
|
93
|
+
graph,
|
|
94
|
+
memory.chroma.collections,
|
|
95
|
+
memory.chroma.embeddings
|
|
96
|
+
)
|
|
97
|
+
|
|
98
|
+
const scenario = await analyzer.whatIf(change, {
|
|
99
|
+
project: project_name,
|
|
100
|
+
maxResults: max_results
|
|
101
|
+
})
|
|
102
|
+
|
|
103
|
+
const parts: string[] = []
|
|
104
|
+
parts.push(scenario.summary)
|
|
105
|
+
parts.push('')
|
|
106
|
+
|
|
107
|
+
if (scenario.affectedDecisions.length > 0) {
|
|
108
|
+
parts.push('### Affected Decisions')
|
|
109
|
+
for (const d of scenario.affectedDecisions.slice(0, 10)) {
|
|
110
|
+
const icon = d.impact === 'high' ? '🔴' : d.impact === 'medium' ? '🟡' : '🟢'
|
|
111
|
+
parts.push(`${icon} **[${d.impact}]** ${d.decision}`)
|
|
112
|
+
parts.push(` *${d.reason}*`)
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
|
|
116
|
+
if (scenario.affectedTechnologies.length > 0) {
|
|
117
|
+
parts.push(`\n### Connected Technologies`)
|
|
118
|
+
parts.push(scenario.affectedTechnologies.map(t => `- ${t}`).join('\n'))
|
|
119
|
+
}
|
|
120
|
+
|
|
121
|
+
const content = parts.join('\n')
|
|
122
|
+
const totalAffected = scenario.affectedDecisions.length
|
|
123
|
+
|
|
124
|
+
if (totalAffected > 0) {
|
|
125
|
+
return ResponseFormatter.text(withMemoryIndicator(content, totalAffected))
|
|
126
|
+
}
|
|
127
|
+
|
|
128
|
+
return ResponseFormatter.text(content)
|
|
129
|
+
|
|
130
|
+
} catch (error) {
|
|
131
|
+
ErrorHandler.logError(logger, error, { tool: 'what_if_analysis' })
|
|
132
|
+
if (error instanceof McpError) throw error
|
|
133
|
+
throw new McpError(ErrorCode.InternalError, `What-if analysis failed: ${error instanceof Error ? error.message : 'Unknown error'}`)
|
|
134
|
+
}
|
|
135
|
+
}
|