@duckcodeailabs/dql-cli 1.4.3 → 1.5.0
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/LICENSE +123 -0
- package/README.md +72 -0
- package/dist/apps-api.d.ts +79 -0
- package/dist/apps-api.d.ts.map +1 -0
- package/dist/apps-api.js +934 -0
- package/dist/apps-api.js.map +1 -0
- package/dist/apps-api.test.d.ts +2 -0
- package/dist/apps-api.test.d.ts.map +1 -0
- package/dist/apps-api.test.js +111 -0
- package/dist/apps-api.test.js.map +1 -0
- package/dist/args.d.ts +30 -0
- package/dist/args.d.ts.map +1 -0
- package/dist/args.js +105 -0
- package/dist/args.js.map +1 -0
- package/dist/args.test.d.ts +2 -0
- package/dist/args.test.d.ts.map +1 -0
- package/dist/args.test.js +41 -0
- package/dist/args.test.js.map +1 -0
- package/dist/assets/dql-notebook/assets/codemirror-DJYUkPr1.js +11 -0
- package/dist/assets/dql-notebook/assets/index-R3UrqjLQ.css +1 -0
- package/dist/assets/dql-notebook/assets/index-mlfOQ2me.js +857 -0
- package/dist/assets/dql-notebook/assets/react-CRB3T2We.js +32 -0
- package/dist/assets/dql-notebook/index.html +18 -0
- package/dist/assets/notebook-browser/app.js +548 -0
- package/dist/assets/notebook-browser/index.html +83 -0
- package/dist/assets/notebook-browser/styles.css +336 -0
- package/dist/block-studio-import.d.ts +58 -0
- package/dist/block-studio-import.d.ts.map +1 -0
- package/dist/block-studio-import.js +390 -0
- package/dist/block-studio-import.js.map +1 -0
- package/dist/block-studio-import.test.d.ts +2 -0
- package/dist/block-studio-import.test.d.ts.map +1 -0
- package/dist/block-studio-import.test.js +106 -0
- package/dist/block-studio-import.test.js.map +1 -0
- package/dist/block-templates.d.ts +8 -0
- package/dist/block-templates.d.ts.map +1 -0
- package/dist/block-templates.js +60 -0
- package/dist/block-templates.js.map +1 -0
- package/dist/commands/agent.d.ts +19 -0
- package/dist/commands/agent.d.ts.map +1 -0
- package/dist/commands/agent.js +258 -0
- package/dist/commands/agent.js.map +1 -0
- package/dist/commands/app.d.ts +32 -0
- package/dist/commands/app.d.ts.map +1 -0
- package/dist/commands/app.js +307 -0
- package/dist/commands/app.js.map +1 -0
- package/dist/commands/build.d.ts +3 -0
- package/dist/commands/build.d.ts.map +1 -0
- package/dist/commands/build.js +69 -0
- package/dist/commands/build.js.map +1 -0
- package/dist/commands/build.test.d.ts +2 -0
- package/dist/commands/build.test.d.ts.map +1 -0
- package/dist/commands/build.test.js +44 -0
- package/dist/commands/build.test.js.map +1 -0
- package/dist/commands/certify.d.ts +3 -0
- package/dist/commands/certify.d.ts.map +1 -0
- package/dist/commands/certify.js +228 -0
- package/dist/commands/certify.js.map +1 -0
- package/dist/commands/compile.d.ts +21 -0
- package/dist/commands/compile.d.ts.map +1 -0
- package/dist/commands/compile.js +198 -0
- package/dist/commands/compile.js.map +1 -0
- package/dist/commands/compile.test.d.ts +2 -0
- package/dist/commands/compile.test.d.ts.map +1 -0
- package/dist/commands/compile.test.js +115 -0
- package/dist/commands/compile.test.js.map +1 -0
- package/dist/commands/diff.d.ts +3 -0
- package/dist/commands/diff.d.ts.map +1 -0
- package/dist/commands/diff.js +52 -0
- package/dist/commands/diff.js.map +1 -0
- package/dist/commands/doctor.d.ts +3 -0
- package/dist/commands/doctor.d.ts.map +1 -0
- package/dist/commands/doctor.js +191 -0
- package/dist/commands/doctor.js.map +1 -0
- package/dist/commands/doctor.test.d.ts +2 -0
- package/dist/commands/doctor.test.d.ts.map +1 -0
- package/dist/commands/doctor.test.js +43 -0
- package/dist/commands/doctor.test.js.map +1 -0
- package/dist/commands/fmt.d.ts +3 -0
- package/dist/commands/fmt.d.ts.map +1 -0
- package/dist/commands/fmt.js +53 -0
- package/dist/commands/fmt.js.map +1 -0
- package/dist/commands/import.d.ts +3 -0
- package/dist/commands/import.d.ts.map +1 -0
- package/dist/commands/import.js +50 -0
- package/dist/commands/import.js.map +1 -0
- package/dist/commands/info.d.ts +3 -0
- package/dist/commands/info.d.ts.map +1 -0
- package/dist/commands/info.js +56 -0
- package/dist/commands/info.js.map +1 -0
- package/dist/commands/init.d.ts +3 -0
- package/dist/commands/init.d.ts.map +1 -0
- package/dist/commands/init.js +250 -0
- package/dist/commands/init.js.map +1 -0
- package/dist/commands/init.test.d.ts +2 -0
- package/dist/commands/init.test.d.ts.map +1 -0
- package/dist/commands/init.test.js +118 -0
- package/dist/commands/init.test.js.map +1 -0
- package/dist/commands/lineage.d.ts +24 -0
- package/dist/commands/lineage.d.ts.map +1 -0
- package/dist/commands/lineage.js +634 -0
- package/dist/commands/lineage.js.map +1 -0
- package/dist/commands/mcp.d.ts +7 -0
- package/dist/commands/mcp.d.ts.map +1 -0
- package/dist/commands/mcp.js +16 -0
- package/dist/commands/mcp.js.map +1 -0
- package/dist/commands/migrate.d.ts +12 -0
- package/dist/commands/migrate.d.ts.map +1 -0
- package/dist/commands/migrate.js +197 -0
- package/dist/commands/migrate.js.map +1 -0
- package/dist/commands/new.d.ts +3 -0
- package/dist/commands/new.d.ts.map +1 -0
- package/dist/commands/new.js +490 -0
- package/dist/commands/new.js.map +1 -0
- package/dist/commands/new.test.d.ts +2 -0
- package/dist/commands/new.test.d.ts.map +1 -0
- package/dist/commands/new.test.js +191 -0
- package/dist/commands/new.test.js.map +1 -0
- package/dist/commands/notebook.d.ts +3 -0
- package/dist/commands/notebook.d.ts.map +1 -0
- package/dist/commands/notebook.js +46 -0
- package/dist/commands/notebook.js.map +1 -0
- package/dist/commands/parse.d.ts +3 -0
- package/dist/commands/parse.d.ts.map +1 -0
- package/dist/commands/parse.js +63 -0
- package/dist/commands/parse.js.map +1 -0
- package/dist/commands/preview.d.ts +3 -0
- package/dist/commands/preview.d.ts.map +1 -0
- package/dist/commands/preview.js +42 -0
- package/dist/commands/preview.js.map +1 -0
- package/dist/commands/schedule.d.ts +3 -0
- package/dist/commands/schedule.d.ts.map +1 -0
- package/dist/commands/schedule.js +215 -0
- package/dist/commands/schedule.js.map +1 -0
- package/dist/commands/semantic.d.ts +12 -0
- package/dist/commands/semantic.d.ts.map +1 -0
- package/dist/commands/semantic.js +356 -0
- package/dist/commands/semantic.js.map +1 -0
- package/dist/commands/serve.d.ts +3 -0
- package/dist/commands/serve.d.ts.map +1 -0
- package/dist/commands/serve.js +30 -0
- package/dist/commands/serve.js.map +1 -0
- package/dist/commands/slack.d.ts +13 -0
- package/dist/commands/slack.d.ts.map +1 -0
- package/dist/commands/slack.js +53 -0
- package/dist/commands/slack.js.map +1 -0
- package/dist/commands/sync.d.ts +3 -0
- package/dist/commands/sync.d.ts.map +1 -0
- package/dist/commands/sync.js +192 -0
- package/dist/commands/sync.js.map +1 -0
- package/dist/commands/sync.test.d.ts +2 -0
- package/dist/commands/sync.test.d.ts.map +1 -0
- package/dist/commands/sync.test.js +147 -0
- package/dist/commands/sync.test.js.map +1 -0
- package/dist/commands/test.d.ts +3 -0
- package/dist/commands/test.d.ts.map +1 -0
- package/dist/commands/test.js +167 -0
- package/dist/commands/test.js.map +1 -0
- package/dist/commands/validate.d.ts +3 -0
- package/dist/commands/validate.d.ts.map +1 -0
- package/dist/commands/validate.js +163 -0
- package/dist/commands/validate.js.map +1 -0
- package/dist/commands/validate.test.d.ts +2 -0
- package/dist/commands/validate.test.d.ts.map +1 -0
- package/dist/commands/validate.test.js +55 -0
- package/dist/commands/validate.test.js.map +1 -0
- package/dist/commands/verify.d.ts +11 -0
- package/dist/commands/verify.d.ts.map +1 -0
- package/dist/commands/verify.js +74 -0
- package/dist/commands/verify.js.map +1 -0
- package/dist/digest.d.ts +10 -0
- package/dist/digest.d.ts.map +1 -0
- package/dist/digest.js +83 -0
- package/dist/digest.js.map +1 -0
- package/dist/git-service.d.ts +17 -0
- package/dist/git-service.d.ts.map +1 -0
- package/dist/git-service.js +54 -0
- package/dist/git-service.js.map +1 -0
- package/dist/governance-runtime.d.ts +15 -0
- package/dist/governance-runtime.d.ts.map +1 -0
- package/dist/governance-runtime.js +50 -0
- package/dist/governance-runtime.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/{index.js → dist/index.js} +5 -0
- package/dist/index.js.map +1 -0
- package/dist/llm/index.d.ts +4 -0
- package/dist/llm/index.d.ts.map +1 -0
- package/dist/llm/index.js +20 -0
- package/dist/llm/index.js.map +1 -0
- package/dist/llm/providers/claude-agent-sdk.d.ts +3 -0
- package/dist/llm/providers/claude-agent-sdk.d.ts.map +1 -0
- package/dist/llm/providers/claude-agent-sdk.js +174 -0
- package/dist/llm/providers/claude-agent-sdk.js.map +1 -0
- package/dist/llm/providers/claude-code.d.ts +8 -0
- package/dist/llm/providers/claude-code.d.ts.map +1 -0
- package/dist/llm/providers/claude-code.js +171 -0
- package/dist/llm/providers/claude-code.js.map +1 -0
- package/dist/llm/providers/dql-agent-provider.d.ts +5 -0
- package/dist/llm/providers/dql-agent-provider.d.ts.map +1 -0
- package/dist/llm/providers/dql-agent-provider.js +287 -0
- package/dist/llm/providers/dql-agent-provider.js.map +1 -0
- package/dist/llm/tools.d.ts +9 -0
- package/dist/llm/tools.d.ts.map +1 -0
- package/dist/llm/tools.js +112 -0
- package/dist/llm/tools.js.map +1 -0
- package/dist/llm/types.d.ts +72 -0
- package/dist/llm/types.d.ts.map +1 -0
- package/dist/llm/types.js +2 -0
- package/dist/llm/types.js.map +1 -0
- package/dist/local-runtime.d.ts +142 -0
- package/dist/local-runtime.d.ts.map +1 -0
- package/dist/local-runtime.js +4859 -0
- package/dist/local-runtime.js.map +1 -0
- package/dist/local-runtime.test.d.ts +2 -0
- package/dist/local-runtime.test.d.ts.map +1 -0
- package/dist/local-runtime.test.js +241 -0
- package/dist/local-runtime.test.js.map +1 -0
- package/dist/metricflow.d.ts +35 -0
- package/dist/metricflow.d.ts.map +1 -0
- package/dist/metricflow.js +122 -0
- package/dist/metricflow.js.map +1 -0
- package/dist/metricflow.test.d.ts +2 -0
- package/dist/metricflow.test.d.ts.map +1 -0
- package/dist/metricflow.test.js +54 -0
- package/dist/metricflow.test.js.map +1 -0
- package/dist/open-browser.d.ts +2 -0
- package/dist/open-browser.d.ts.map +1 -0
- package/dist/open-browser.js +29 -0
- package/dist/open-browser.js.map +1 -0
- package/dist/schedule/alerts.d.ts +5 -0
- package/dist/schedule/alerts.d.ts.map +1 -0
- package/dist/schedule/alerts.js +54 -0
- package/dist/schedule/alerts.js.map +1 -0
- package/dist/schedule/discovery.d.ts +4 -0
- package/dist/schedule/discovery.d.ts.map +1 -0
- package/dist/schedule/discovery.js +36 -0
- package/dist/schedule/discovery.js.map +1 -0
- package/dist/schedule/notifiers/email.d.ts +3 -0
- package/dist/schedule/notifiers/email.d.ts.map +1 -0
- package/dist/schedule/notifiers/email.js +76 -0
- package/dist/schedule/notifiers/email.js.map +1 -0
- package/dist/schedule/notifiers/file.d.ts +3 -0
- package/dist/schedule/notifiers/file.d.ts.map +1 -0
- package/dist/schedule/notifiers/file.js +50 -0
- package/dist/schedule/notifiers/file.js.map +1 -0
- package/dist/schedule/notifiers/index.d.ts +10 -0
- package/dist/schedule/notifiers/index.d.ts.map +1 -0
- package/dist/schedule/notifiers/index.js +33 -0
- package/dist/schedule/notifiers/index.js.map +1 -0
- package/dist/schedule/notifiers/slack.d.ts +3 -0
- package/dist/schedule/notifiers/slack.d.ts.map +1 -0
- package/dist/schedule/notifiers/slack.js +58 -0
- package/dist/schedule/notifiers/slack.js.map +1 -0
- package/dist/schedule/runner.d.ts +14 -0
- package/dist/schedule/runner.d.ts.map +1 -0
- package/dist/schedule/runner.js +225 -0
- package/dist/schedule/runner.js.map +1 -0
- package/dist/schedule/runs.d.ts +5 -0
- package/dist/schedule/runs.d.ts.map +1 -0
- package/dist/schedule/runs.js +41 -0
- package/dist/schedule/runs.js.map +1 -0
- package/dist/schedule/service.d.ts +13 -0
- package/dist/schedule/service.d.ts.map +1 -0
- package/dist/schedule/service.js +87 -0
- package/dist/schedule/service.js.map +1 -0
- package/dist/schedule/types.d.ts +70 -0
- package/dist/schedule/types.d.ts.map +1 -0
- package/dist/schedule/types.js +2 -0
- package/dist/schedule/types.js.map +1 -0
- package/dist/semantic-import.d.ts +135 -0
- package/dist/semantic-import.d.ts.map +1 -0
- package/dist/semantic-import.js +979 -0
- package/dist/semantic-import.js.map +1 -0
- package/dist/semantic-import.test.d.ts +2 -0
- package/dist/semantic-import.test.d.ts.map +1 -0
- package/dist/semantic-import.test.js +95 -0
- package/dist/semantic-import.test.js.map +1 -0
- package/dist/settings/provider-settings.d.ts +33 -0
- package/dist/settings/provider-settings.d.ts.map +1 -0
- package/dist/settings/provider-settings.js +91 -0
- package/dist/settings/provider-settings.js.map +1 -0
- package/package.json +29 -21
|
@@ -0,0 +1,287 @@
|
|
|
1
|
+
import { KGStore, MemoryStore, defaultKgPath, defaultMemoryPath, GeminiProvider, OllamaProvider, OpenAIProvider, answer, loadSkills, reindexProject, } from '@duckcodeailabs/dql-agent';
|
|
2
|
+
import { existsSync } from 'node:fs';
|
|
3
|
+
import { getEffectiveProviderConfig } from '../../settings/provider-settings.js';
|
|
4
|
+
const SPECS = {
|
|
5
|
+
openai: {
|
|
6
|
+
label: 'OpenAI',
|
|
7
|
+
setup: 'Configure OpenAI in Settings or set OPENAI_API_KEY. Optional: OPENAI_MODEL and OPENAI_BASE_URL.',
|
|
8
|
+
create: (projectRoot) => {
|
|
9
|
+
const config = getEffectiveProviderConfig(projectRoot, 'openai');
|
|
10
|
+
return new OpenAIProvider({ apiKey: config.apiKey, model: config.model, baseUrl: config.baseUrl });
|
|
11
|
+
},
|
|
12
|
+
},
|
|
13
|
+
gemini: {
|
|
14
|
+
label: 'Gemini',
|
|
15
|
+
setup: 'Configure Gemini in Settings or set GEMINI_API_KEY. Optional: GEMINI_MODEL.',
|
|
16
|
+
create: (projectRoot) => {
|
|
17
|
+
const config = getEffectiveProviderConfig(projectRoot, 'gemini');
|
|
18
|
+
return new GeminiProvider({ apiKey: config.apiKey, model: config.model });
|
|
19
|
+
},
|
|
20
|
+
},
|
|
21
|
+
ollama: {
|
|
22
|
+
label: 'Ollama',
|
|
23
|
+
setup: 'Start Ollama and configure OLLAMA_BASE_URL / OLLAMA_MODEL in Settings or env.',
|
|
24
|
+
create: (projectRoot) => {
|
|
25
|
+
const config = getEffectiveProviderConfig(projectRoot, 'ollama');
|
|
26
|
+
return new OllamaProvider({ model: config.model, baseUrl: config.baseUrl });
|
|
27
|
+
},
|
|
28
|
+
},
|
|
29
|
+
'custom-openai': {
|
|
30
|
+
label: 'Custom OpenAI-compatible',
|
|
31
|
+
setup: 'Configure a custom OpenAI-compatible endpoint in Settings with base URL, model, and optional API key.',
|
|
32
|
+
create: (projectRoot) => {
|
|
33
|
+
const config = getEffectiveProviderConfig(projectRoot, 'custom-openai');
|
|
34
|
+
return new OpenAIProvider({ apiKey: config.apiKey, model: config.model, baseUrl: config.baseUrl, allowNoApiKey: true });
|
|
35
|
+
},
|
|
36
|
+
},
|
|
37
|
+
};
|
|
38
|
+
function emitProposalFromText(text, emit) {
|
|
39
|
+
const match = text.match(/DQL_BLOCK_PROPOSAL\s*[:=]?\s*(\{[\s\S]*\})\s*$/);
|
|
40
|
+
if (!match)
|
|
41
|
+
return;
|
|
42
|
+
try {
|
|
43
|
+
const raw = JSON.parse(match[1]);
|
|
44
|
+
if (!raw.name || !raw.sql)
|
|
45
|
+
return;
|
|
46
|
+
emit({
|
|
47
|
+
kind: 'proposal',
|
|
48
|
+
proposal: {
|
|
49
|
+
name: String(raw.name),
|
|
50
|
+
domain: String(raw.domain ?? ''),
|
|
51
|
+
owner: String(raw.owner ?? ''),
|
|
52
|
+
description: String(raw.description ?? ''),
|
|
53
|
+
sql: String(raw.sql),
|
|
54
|
+
tags: Array.isArray(raw.tags) ? raw.tags.map(String) : undefined,
|
|
55
|
+
chartType: typeof raw.chartType === 'string' ? raw.chartType : undefined,
|
|
56
|
+
},
|
|
57
|
+
governance: { certified: false, errors: [], warnings: ['Generated by a non-tool provider; review before saving.'] },
|
|
58
|
+
});
|
|
59
|
+
}
|
|
60
|
+
catch {
|
|
61
|
+
// Ignore malformed proposal text. The visible assistant response still streams as text.
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
export function createDqlAgentProviderRunner(id) {
|
|
65
|
+
return {
|
|
66
|
+
async run(req, emit, signal) {
|
|
67
|
+
const spec = SPECS[id];
|
|
68
|
+
const provider = spec.create(req.projectRoot);
|
|
69
|
+
const available = await provider.available().catch(() => false);
|
|
70
|
+
if (!available) {
|
|
71
|
+
emit({ kind: 'error', message: `${spec.label} is not configured or reachable. ${spec.setup}` });
|
|
72
|
+
return;
|
|
73
|
+
}
|
|
74
|
+
try {
|
|
75
|
+
emit({ kind: 'thinking', text: `Using ${spec.label} through the governed DQL agent.` });
|
|
76
|
+
const kgPath = defaultKgPath(req.projectRoot);
|
|
77
|
+
if (!existsSync(kgPath)) {
|
|
78
|
+
emit({ kind: 'thinking', text: 'Building the local agent knowledge graph from blocks, apps, dashboards, dbt, and semantic metadata.' });
|
|
79
|
+
await reindexProject(req.projectRoot, { kgPath });
|
|
80
|
+
}
|
|
81
|
+
const question = lastUserMessage(req);
|
|
82
|
+
if (!question) {
|
|
83
|
+
emit({ kind: 'error', message: 'No user question found.' });
|
|
84
|
+
return;
|
|
85
|
+
}
|
|
86
|
+
const memory = new MemoryStore(defaultMemoryPath(req.projectRoot));
|
|
87
|
+
const kg = new KGStore(kgPath);
|
|
88
|
+
try {
|
|
89
|
+
const memoryContext = memory.search({
|
|
90
|
+
query: question,
|
|
91
|
+
scopes: ['thread', 'notebook', 'project', 'user', 'artifact'],
|
|
92
|
+
scopeId: req.upstream?.cellId,
|
|
93
|
+
limit: 6,
|
|
94
|
+
});
|
|
95
|
+
const skills = loadSkills(req.projectRoot).skills;
|
|
96
|
+
const blockHints = inferFollowUpBlockHints(req, question);
|
|
97
|
+
const result = await answer({
|
|
98
|
+
question,
|
|
99
|
+
extraContext: renderExtraContext(req, blockHints),
|
|
100
|
+
provider,
|
|
101
|
+
kg,
|
|
102
|
+
skills,
|
|
103
|
+
blockHints,
|
|
104
|
+
memoryContext,
|
|
105
|
+
signal,
|
|
106
|
+
executeCertifiedBlock: req.executeCertifiedBlock,
|
|
107
|
+
});
|
|
108
|
+
emit({ kind: 'tool_result', id: 'governed_answer', output: result });
|
|
109
|
+
emit({ kind: 'text', text: formatAgentAnswer(result) });
|
|
110
|
+
if (result.proposedSql) {
|
|
111
|
+
emitDraftProposal(result, question, emit);
|
|
112
|
+
}
|
|
113
|
+
else {
|
|
114
|
+
emitProposalFromText(result.text, emit);
|
|
115
|
+
}
|
|
116
|
+
persistThreadMemory(memory, req, question, result);
|
|
117
|
+
}
|
|
118
|
+
finally {
|
|
119
|
+
kg.close();
|
|
120
|
+
memory.close();
|
|
121
|
+
}
|
|
122
|
+
emit({ kind: 'done', stopReason: 'stop' });
|
|
123
|
+
}
|
|
124
|
+
catch (err) {
|
|
125
|
+
const message = err instanceof Error ? err.message : String(err);
|
|
126
|
+
const setupHint = shouldShowProviderSetupHint(message) ? ` ${spec.setup}` : '';
|
|
127
|
+
emit({ kind: 'error', message: `${spec.label} failed: ${message}.${setupHint}` });
|
|
128
|
+
}
|
|
129
|
+
},
|
|
130
|
+
};
|
|
131
|
+
}
|
|
132
|
+
function shouldShowProviderSetupHint(message) {
|
|
133
|
+
return /api key|not configured|not reachable|connection refused|ECONNREFUSED|fetch failed|network error|model .*not found/i
|
|
134
|
+
.test(message);
|
|
135
|
+
}
|
|
136
|
+
function lastUserMessage(req) {
|
|
137
|
+
for (let i = req.messages.length - 1; i >= 0; i--) {
|
|
138
|
+
const msg = req.messages[i];
|
|
139
|
+
if (msg.role === 'user' && msg.content.trim())
|
|
140
|
+
return msg.content.trim();
|
|
141
|
+
}
|
|
142
|
+
return '';
|
|
143
|
+
}
|
|
144
|
+
function renderExtraContext(req, blockHints) {
|
|
145
|
+
const parts = [];
|
|
146
|
+
if (req.upstream?.sql)
|
|
147
|
+
parts.push(`Current upstream SQL:\n${req.upstream.sql}`);
|
|
148
|
+
if (blockHints.length > 0) {
|
|
149
|
+
parts.push(`Follow-up context: the user is referring to certified block "${blockHints[0]}".`);
|
|
150
|
+
}
|
|
151
|
+
return parts.length > 0 ? parts.join('\n\n') : undefined;
|
|
152
|
+
}
|
|
153
|
+
function inferFollowUpBlockHints(req, question) {
|
|
154
|
+
if (!isGenericFollowUp(question))
|
|
155
|
+
return [];
|
|
156
|
+
for (let i = req.messages.length - 1; i >= 0; i--) {
|
|
157
|
+
const msg = req.messages[i];
|
|
158
|
+
if (msg.role !== 'assistant')
|
|
159
|
+
continue;
|
|
160
|
+
const fromAnswer = msg.content.match(/Answered by certified block \*\*([^*]+)\*\*/i)?.[1];
|
|
161
|
+
const fromCitation = msg.content.match(/^- block:\s*([A-Za-z0-9_.-]+)/im)?.[1];
|
|
162
|
+
const block = fromAnswer ?? fromCitation;
|
|
163
|
+
if (block)
|
|
164
|
+
return [block.trim()];
|
|
165
|
+
}
|
|
166
|
+
return [];
|
|
167
|
+
}
|
|
168
|
+
const GENERIC_FOLLOW_UP_WORDS = new Set([
|
|
169
|
+
'a', 'about', 'again', 'all', 'also', 'and', 'as', 'be', 'block', 'can', 'could', 'data', 'do',
|
|
170
|
+
'execute', 'for', 'from', 'get', 'give', 'import', 'it', 'its', 'let', 'lets', 'me', 'metrics',
|
|
171
|
+
'more', 'now', 'of', 'output', 'please', 'result', 'results', 'run', 'show', 'solution', 'summary',
|
|
172
|
+
'that', 'the', 'them', 'this', 'to', 'use', 'with', 'you',
|
|
173
|
+
]);
|
|
174
|
+
function isGenericFollowUp(question) {
|
|
175
|
+
const lower = question.toLowerCase();
|
|
176
|
+
if (!/\b(block|data|execute|import|it|result|results|run|solution|that|this)\b/.test(lower))
|
|
177
|
+
return false;
|
|
178
|
+
const tokens = lower.match(/[a-z0-9_]+/g) ?? [];
|
|
179
|
+
const meaningful = tokens.filter((token) => token.length > 1 && !GENERIC_FOLLOW_UP_WORDS.has(token));
|
|
180
|
+
return meaningful.length === 0;
|
|
181
|
+
}
|
|
182
|
+
function formatAgentAnswer(result) {
|
|
183
|
+
const badge = result.certification === 'certified'
|
|
184
|
+
? 'Certified'
|
|
185
|
+
: result.sourceTier === 'semantic_layer'
|
|
186
|
+
? 'AI generated from semantic layer - analyst review required'
|
|
187
|
+
: result.kind === 'no_answer'
|
|
188
|
+
? 'No answer'
|
|
189
|
+
: 'AI generated from dbt manifest - analyst review required';
|
|
190
|
+
const citations = result.citations.length > 0
|
|
191
|
+
? '\n\nCitations:\n' + result.citations.map((c) => `- ${c.kind}: ${c.name}${c.provenance ? ` (${c.provenance})` : ''}`).join('\n')
|
|
192
|
+
: '';
|
|
193
|
+
const resultPreview = formatResultPreview(result.result);
|
|
194
|
+
const sql = result.proposedSql ? `\n\nProposed SQL:\n\`\`\`sql\n${result.proposedSql}\n\`\`\`` : '';
|
|
195
|
+
return `[${badge}]\n\n${result.text}${resultPreview}${citations}${sql}`;
|
|
196
|
+
}
|
|
197
|
+
function formatResultPreview(result) {
|
|
198
|
+
if (!result)
|
|
199
|
+
return '';
|
|
200
|
+
const columns = normalizeColumns(result.columns).slice(0, 8);
|
|
201
|
+
const rows = Array.isArray(result.rows) ? result.rows.slice(0, 8) : [];
|
|
202
|
+
const shown = rows.length;
|
|
203
|
+
const timing = typeof result.executionTime === 'number' && result.executionTime > 0
|
|
204
|
+
? ` in ${Math.round(result.executionTime)} ms`
|
|
205
|
+
: '';
|
|
206
|
+
if (columns.length === 0 || rows.length === 0) {
|
|
207
|
+
return `\n\nResults: ${result.rowCount} row${result.rowCount === 1 ? '' : 's'}${timing}.`;
|
|
208
|
+
}
|
|
209
|
+
const tableRows = rows.map((row) => {
|
|
210
|
+
const record = row && typeof row === 'object' ? row : {};
|
|
211
|
+
return `| ${columns.map((col) => formatCell(record[col])).join(' | ')} |`;
|
|
212
|
+
});
|
|
213
|
+
const omittedRows = result.rowCount > shown ? ` Showing first ${shown} rows.` : '';
|
|
214
|
+
return [
|
|
215
|
+
`\n\nResults: ${result.rowCount} row${result.rowCount === 1 ? '' : 's'}${timing}.${omittedRows}`,
|
|
216
|
+
`| ${columns.map(escapeMarkdownTable).join(' | ')} |`,
|
|
217
|
+
`| ${columns.map(() => '---').join(' | ')} |`,
|
|
218
|
+
...tableRows,
|
|
219
|
+
].join('\n');
|
|
220
|
+
}
|
|
221
|
+
function normalizeColumns(columns) {
|
|
222
|
+
return columns.map((column) => {
|
|
223
|
+
if (typeof column === 'string')
|
|
224
|
+
return column;
|
|
225
|
+
if (column && typeof column === 'object' && typeof column.name === 'string') {
|
|
226
|
+
return column.name;
|
|
227
|
+
}
|
|
228
|
+
return String(column);
|
|
229
|
+
});
|
|
230
|
+
}
|
|
231
|
+
function formatCell(value) {
|
|
232
|
+
if (value === null || typeof value === 'undefined')
|
|
233
|
+
return '';
|
|
234
|
+
const raw = typeof value === 'object' ? JSON.stringify(value) : String(value);
|
|
235
|
+
return escapeMarkdownTable(raw.length > 80 ? `${raw.slice(0, 77)}...` : raw);
|
|
236
|
+
}
|
|
237
|
+
function escapeMarkdownTable(value) {
|
|
238
|
+
return value.replace(/\|/g, '\\|').replace(/\r?\n/g, ' ');
|
|
239
|
+
}
|
|
240
|
+
function emitDraftProposal(result, question, emit) {
|
|
241
|
+
const proposal = {
|
|
242
|
+
name: slugify(question).slice(0, 56) || 'ai_generated_analysis',
|
|
243
|
+
domain: '',
|
|
244
|
+
owner: `${process.env.USER ?? 'analyst'}@local`,
|
|
245
|
+
description: result.text.slice(0, 240),
|
|
246
|
+
sql: result.proposedSql,
|
|
247
|
+
tags: ['ai-generated', 'needs-review', result.sourceTier ?? 'dbt_manifest'],
|
|
248
|
+
chartType: result.suggestedViz,
|
|
249
|
+
};
|
|
250
|
+
emit({
|
|
251
|
+
kind: 'proposal',
|
|
252
|
+
proposal,
|
|
253
|
+
governance: {
|
|
254
|
+
certified: false,
|
|
255
|
+
errors: [],
|
|
256
|
+
warnings: ['AI generated. Analyst review and certification are required before reuse as governed content.'],
|
|
257
|
+
},
|
|
258
|
+
});
|
|
259
|
+
}
|
|
260
|
+
function persistThreadMemory(memory, req, question, result) {
|
|
261
|
+
const cellId = req.upstream?.cellId;
|
|
262
|
+
if (!cellId)
|
|
263
|
+
return;
|
|
264
|
+
memory.upsert({
|
|
265
|
+
id: `thread:${cellId}`,
|
|
266
|
+
scope: 'thread',
|
|
267
|
+
scopeId: cellId,
|
|
268
|
+
title: 'Notebook chat summary',
|
|
269
|
+
content: [
|
|
270
|
+
`Latest question: ${question}`,
|
|
271
|
+
`Latest route: ${result.sourceTier ?? 'unknown'} / ${result.certification ?? 'unknown'}`,
|
|
272
|
+
`Latest answer: ${result.text.slice(0, 500)}`,
|
|
273
|
+
].join('\n'),
|
|
274
|
+
tags: ['chat', result.sourceTier ?? 'unknown', result.certification ?? 'unknown'],
|
|
275
|
+
source: 'notebook-chat',
|
|
276
|
+
confidence: result.confidence ?? 0.5,
|
|
277
|
+
importance: 0.45,
|
|
278
|
+
});
|
|
279
|
+
}
|
|
280
|
+
function slugify(input) {
|
|
281
|
+
return input
|
|
282
|
+
.toLowerCase()
|
|
283
|
+
.replace(/[^a-z0-9]+/g, '_')
|
|
284
|
+
.replace(/^_+|_+$/g, '')
|
|
285
|
+
.replace(/_{2,}/g, '_');
|
|
286
|
+
}
|
|
287
|
+
//# sourceMappingURL=dql-agent-provider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"dql-agent-provider.js","sourceRoot":"","sources":["../../../src/llm/providers/dql-agent-provider.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,OAAO,EACP,WAAW,EACX,aAAa,EACb,iBAAiB,EACjB,cAAc,EACd,cAAc,EACd,cAAc,EACd,MAAM,EACN,UAAU,EACV,cAAc,GAIf,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAErC,OAAO,EAAE,0BAA0B,EAAE,MAAM,qCAAqC,CAAC;AAUjF,MAAM,KAAK,GAA2C;IACpD,MAAM,EAAE;QACN,KAAK,EAAE,QAAQ;QACf,KAAK,EAAE,iGAAiG;QACxG,MAAM,EAAE,CAAC,WAAW,EAAE,EAAE;YACtB,MAAM,MAAM,GAAG,0BAA0B,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YACjE,OAAO,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QACrG,CAAC;KACF;IACD,MAAM,EAAE;QACN,KAAK,EAAE,QAAQ;QACf,KAAK,EAAE,6EAA6E;QACpF,MAAM,EAAE,CAAC,WAAW,EAAE,EAAE;YACtB,MAAM,MAAM,GAAG,0BAA0B,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YACjE,OAAO,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;QAC5E,CAAC;KACF;IACD,MAAM,EAAE;QACN,KAAK,EAAE,QAAQ;QACf,KAAK,EAAE,+EAA+E;QACtF,MAAM,EAAE,CAAC,WAAW,EAAE,EAAE;YACtB,MAAM,MAAM,GAAG,0BAA0B,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YACjE,OAAO,IAAI,cAAc,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QAC9E,CAAC;KACF;IACD,eAAe,EAAE;QACf,KAAK,EAAE,0BAA0B;QACjC,KAAK,EAAE,uGAAuG;QAC9G,MAAM,EAAE,CAAC,WAAW,EAAE,EAAE;YACtB,MAAM,MAAM,GAAG,0BAA0B,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;YACxE,OAAO,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1H,CAAC;KACF;CACF,CAAC;AAEF,SAAS,oBAAoB,CAAC,IAAY,EAAE,IAA+B;IACzE,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;IAC3E,IAAI,CAAC,KAAK;QAAE,OAAO;IACnB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAA2B,CAAC;QAC3D,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG;YAAE,OAAO;QAClC,IAAI,CAAC;YACH,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACR,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC;gBAChC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;gBAC9B,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,IAAI,EAAE,CAAC;gBAC1C,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;gBACpB,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;gBAChE,SAAS,EAAE,OAAO,GAAG,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;aACzE;YACD,UAAU,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,yDAAyD,CAAC,EAAE;SACpH,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,wFAAwF;IAC1F,CAAC;AACH,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,EAAoB;IAC/D,OAAO;QACL,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,EAAE,MAAM;YACzB,MAAM,IAAI,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC;YACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAC9C,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;YAChE,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,oCAAoC,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAChG,OAAO;YACT,CAAC;YAED,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,IAAI,CAAC,KAAK,kCAAkC,EAAE,CAAC,CAAC;gBACxF,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC9C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;oBACxB,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,qGAAqG,EAAE,CAAC,CAAC;oBACxI,MAAM,cAAc,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;gBACpD,CAAC;gBAED,MAAM,QAAQ,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC;gBACtC,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,yBAAyB,EAAE,CAAC,CAAC;oBAC5D,OAAO;gBACT,CAAC;gBAED,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,iBAAiB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC;gBACnE,MAAM,EAAE,GAAG,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;gBAC/B,IAAI,CAAC;oBACH,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC;wBAClC,KAAK,EAAE,QAAQ;wBACf,MAAM,EAAE,CAAC,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,CAAC;wBAC7D,OAAO,EAAE,GAAG,CAAC,QAAQ,EAAE,MAAM;wBAC7B,KAAK,EAAE,CAAC;qBACT,CAAC,CAAC;oBACH,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC;oBAClD,MAAM,UAAU,GAAG,uBAAuB,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;oBAC1D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC;wBAC1B,QAAQ;wBACR,YAAY,EAAE,kBAAkB,CAAC,GAAG,EAAE,UAAU,CAAC;wBACjD,QAAQ;wBACR,EAAE;wBACF,MAAM;wBACN,UAAU;wBACV,aAAa;wBACb,MAAM;wBACN,qBAAqB,EAAE,GAAG,CAAC,qBAAqB;qBACjD,CAAC,CAAC;oBACH,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;oBACrE,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBACxD,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;wBACvB,iBAAiB,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;oBAC5C,CAAC;yBAAM,CAAC;wBACN,oBAAoB,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;oBAC1C,CAAC;oBACD,mBAAmB,CAAC,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;gBACrD,CAAC;wBAAS,CAAC;oBACT,EAAE,CAAC,KAAK,EAAE,CAAC;oBACX,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,CAAC;gBACD,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACjE,MAAM,SAAS,GAAG,2BAA2B,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC/E,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,YAAY,OAAO,IAAI,SAAS,EAAE,EAAE,CAAC,CAAC;YACpF,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,2BAA2B,CAAC,OAAe;IAClD,OAAO,oHAAoH;SACxH,IAAI,CAAC,OAAO,CAAC,CAAC;AACnB,CAAC;AAED,SAAS,eAAe,CAAC,GAAoB;IAC3C,KAAK,IAAI,CAAC,GAAG,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAClD,MAAM,GAAG,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,IAAI,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE;YAAE,OAAO,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;IAC3E,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,kBAAkB,CAAC,GAAoB,EAAE,UAAoB;IACpE,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,GAAG,CAAC,QAAQ,EAAE,GAAG;QAAE,KAAK,CAAC,IAAI,CAAC,0BAA0B,GAAG,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC;IAChF,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1B,KAAK,CAAC,IAAI,CAAC,gEAAgE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAChG,CAAC;IACD,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAC3D,CAAC;AAED,SAAS,uBAAuB,CAAC,GAAoB,EAAE,QAAgB;IACrE,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC;QAAE,OAAO,EAAE,CAAC;IAC5C,KAAK,IAAI,CAAC,GAAG,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAClD,MAAM,GAAG,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW;YAAE,SAAS;QACvC,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,8CAA8C,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1F,MAAM,YAAY,GAAG,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,iCAAiC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/E,MAAM,KAAK,GAAG,UAAU,IAAI,YAAY,CAAC;QACzC,IAAI,KAAK;YAAE,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IACnC,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,MAAM,uBAAuB,GAAG,IAAI,GAAG,CAAC;IACtC,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI;IAC9F,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS;IAC9F,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS;IAClG,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK;CAC1D,CAAC,CAAC;AAEH,SAAS,iBAAiB,CAAC,QAAgB;IACzC,MAAM,KAAK,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;IACrC,IAAI,CAAC,0EAA0E,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAC1G,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;IAChD,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IACrG,OAAO,UAAU,CAAC,MAAM,KAAK,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAmB;IAC5C,MAAM,KAAK,GAAG,MAAM,CAAC,aAAa,KAAK,WAAW;QAChD,CAAC,CAAC,WAAW;QACb,CAAC,CAAC,MAAM,CAAC,UAAU,KAAK,gBAAgB;YACtC,CAAC,CAAC,4DAA4D;YAC9D,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,WAAW;gBAC3B,CAAC,CAAC,WAAW;gBACb,CAAC,CAAC,0DAA0D,CAAC;IACnE,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC;QAC3C,CAAC,CAAC,kBAAkB,GAAG,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QAClI,CAAC,CAAC,EAAE,CAAC;IACP,MAAM,aAAa,GAAG,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACzD,MAAM,GAAG,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,iCAAiC,MAAM,CAAC,WAAW,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;IACpG,OAAO,IAAI,KAAK,QAAQ,MAAM,CAAC,IAAI,GAAG,aAAa,GAAG,SAAS,GAAG,GAAG,EAAE,CAAC;AAC1E,CAAC;AAED,SAAS,mBAAmB,CAAC,MAA2B;IACtD,IAAI,CAAC,MAAM;QAAE,OAAO,EAAE,CAAC;IACvB,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7D,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACvE,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC;IAC1B,MAAM,MAAM,GAAG,OAAO,MAAM,CAAC,aAAa,KAAK,QAAQ,IAAI,MAAM,CAAC,aAAa,GAAG,CAAC;QACjF,CAAC,CAAC,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK;QAC9C,CAAC,CAAC,EAAE,CAAC;IACP,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9C,OAAO,gBAAgB,MAAM,CAAC,QAAQ,OAAO,MAAM,CAAC,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,MAAM,GAAG,CAAC;IAC5F,CAAC;IACD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACjC,MAAM,MAAM,GAAG,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAA8B,CAAC,CAAC,CAAC,EAAE,CAAC;QACpF,OAAO,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IAC5E,CAAC,CAAC,CAAC;IACH,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,kBAAkB,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;IACnF,OAAO;QACL,gBAAgB,MAAM,CAAC,QAAQ,OAAO,MAAM,CAAC,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,MAAM,IAAI,WAAW,EAAE;QAChG,KAAK,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI;QACrD,KAAK,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI;QAC7C,GAAG,SAAS;KACb,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAkB;IAC1C,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QAC5B,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,OAAO,MAAM,CAAC;QAC9C,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,OAAQ,MAA6B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACpG,OAAQ,MAA2B,CAAC,IAAI,CAAC;QAC3C,CAAC;QACD,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC;IACxB,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,UAAU,CAAC,KAAc;IAChC,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,WAAW;QAAE,OAAO,EAAE,CAAC;IAC9D,MAAM,GAAG,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9E,OAAO,mBAAmB,CAAC,GAAG,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;AAC/E,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAa;IACxC,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;AAC5D,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAmB,EAAE,QAAgB,EAAE,IAA+B;IAC/F,MAAM,QAAQ,GAAkB;QAC9B,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,uBAAuB;QAC/D,MAAM,EAAE,EAAE;QACV,KAAK,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,SAAS,QAAQ;QAC/C,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;QACtC,GAAG,EAAE,MAAM,CAAC,WAAY;QACxB,IAAI,EAAE,CAAC,cAAc,EAAE,cAAc,EAAE,MAAM,CAAC,UAAU,IAAI,cAAc,CAAC;QAC3E,SAAS,EAAE,MAAM,CAAC,YAAY;KAC/B,CAAC;IACF,IAAI,CAAC;QACH,IAAI,EAAE,UAAU;QAChB,QAAQ;QACR,UAAU,EAAE;YACV,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,EAAE;YACV,QAAQ,EAAE,CAAC,+FAA+F,CAAC;SAC5G;KACF,CAAC,CAAC;AACL,CAAC;AAED,SAAS,mBAAmB,CAAC,MAAmB,EAAE,GAAoB,EAAE,QAAgB,EAAE,MAAmB;IAC3G,MAAM,MAAM,GAAG,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC;IACpC,IAAI,CAAC,MAAM;QAAE,OAAO;IACpB,MAAM,CAAC,MAAM,CAAC;QACZ,EAAE,EAAE,UAAU,MAAM,EAAE;QACtB,KAAK,EAAE,QAAQ;QACf,OAAO,EAAE,MAAM;QACf,KAAK,EAAE,uBAAuB;QAC9B,OAAO,EAAE;YACP,oBAAoB,QAAQ,EAAE;YAC9B,iBAAiB,MAAM,CAAC,UAAU,IAAI,SAAS,MAAM,MAAM,CAAC,aAAa,IAAI,SAAS,EAAE;YACxF,kBAAkB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE;SAC9C,CAAC,IAAI,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,UAAU,IAAI,SAAS,EAAE,MAAM,CAAC,aAAa,IAAI,SAAS,CAAC;QACjF,MAAM,EAAE,eAAe;QACvB,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,GAAG;QACpC,UAAU,EAAE,IAAI;KACjB,CAAC,CAAC;AACL,CAAC;AAED,SAAS,OAAO,CAAC,KAAa;IAC5B,OAAO,KAAK;SACT,WAAW,EAAE;SACb,OAAO,CAAC,aAAa,EAAE,GAAG,CAAC;SAC3B,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;SACvB,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;AAC5B,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { DQLContext } from '@duckcodeailabs/dql-mcp';
|
|
2
|
+
export interface AgentTool {
|
|
3
|
+
name: string;
|
|
4
|
+
description: string;
|
|
5
|
+
inputSchema: Record<string, unknown>;
|
|
6
|
+
run(args: unknown): Promise<unknown>;
|
|
7
|
+
}
|
|
8
|
+
export declare function buildAgentTools(ctx: DQLContext): AgentTool[];
|
|
9
|
+
//# sourceMappingURL=tools.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../../src/llm/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,UAAU,EAQX,MAAM,yBAAyB,CAAC;AAEjC,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,GAAG,CAAC,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;CACtC;AAuED,wBAAgB,eAAe,CAAC,GAAG,EAAE,UAAU,GAAG,SAAS,EAAE,CA+C5D"}
|
|
@@ -0,0 +1,112 @@
|
|
|
1
|
+
import { searchBlocks, getBlock, listMetrics, listDimensions, lineageImpact, certify, suggestBlock, } from '@duckcodeailabs/dql-mcp';
|
|
2
|
+
const SEARCH_BLOCKS_SCHEMA = {
|
|
3
|
+
type: 'object',
|
|
4
|
+
additionalProperties: false,
|
|
5
|
+
properties: {
|
|
6
|
+
query: { type: 'string', description: 'Substring matched against name, description, or tags.' },
|
|
7
|
+
domain: { type: 'string', description: 'Filter to a single business domain.' },
|
|
8
|
+
status: {
|
|
9
|
+
type: 'string',
|
|
10
|
+
enum: ['draft', 'review', 'certified', 'deprecated', 'pending_recertification'],
|
|
11
|
+
description: 'Filter by certification status.',
|
|
12
|
+
},
|
|
13
|
+
limit: { type: 'number', description: 'Max results (default 50).' },
|
|
14
|
+
},
|
|
15
|
+
};
|
|
16
|
+
const GET_BLOCK_SCHEMA = {
|
|
17
|
+
type: 'object',
|
|
18
|
+
additionalProperties: false,
|
|
19
|
+
required: ['name'],
|
|
20
|
+
properties: {
|
|
21
|
+
name: { type: 'string', description: 'Block name (as shown in search_blocks).' },
|
|
22
|
+
includeSource: { type: 'boolean', description: 'Include full .dql source text. Default true.' },
|
|
23
|
+
},
|
|
24
|
+
};
|
|
25
|
+
const DOMAIN_FILTER_SCHEMA = {
|
|
26
|
+
type: 'object',
|
|
27
|
+
additionalProperties: false,
|
|
28
|
+
properties: {
|
|
29
|
+
domain: { type: 'string', description: 'Filter to a single domain.' },
|
|
30
|
+
},
|
|
31
|
+
};
|
|
32
|
+
const LINEAGE_IMPACT_SCHEMA = {
|
|
33
|
+
type: 'object',
|
|
34
|
+
additionalProperties: false,
|
|
35
|
+
required: ['focus'],
|
|
36
|
+
properties: {
|
|
37
|
+
focus: { type: 'string', description: 'Node id ("block:revenue") or bare name.' },
|
|
38
|
+
upstreamDepth: { type: 'number' },
|
|
39
|
+
downstreamDepth: { type: 'number' },
|
|
40
|
+
paths: { type: 'boolean', description: 'Include full source→leaf paths (slower on large graphs).' },
|
|
41
|
+
},
|
|
42
|
+
};
|
|
43
|
+
const CERTIFY_SCHEMA = {
|
|
44
|
+
type: 'object',
|
|
45
|
+
additionalProperties: false,
|
|
46
|
+
required: ['name'],
|
|
47
|
+
properties: {
|
|
48
|
+
name: { type: 'string', description: 'Block name to certify.' },
|
|
49
|
+
},
|
|
50
|
+
};
|
|
51
|
+
const SUGGEST_BLOCK_SCHEMA = {
|
|
52
|
+
type: 'object',
|
|
53
|
+
additionalProperties: false,
|
|
54
|
+
required: ['name', 'domain', 'owner', 'description', 'sql'],
|
|
55
|
+
properties: {
|
|
56
|
+
name: { type: 'string', description: 'Proposed block name (kebab_case).' },
|
|
57
|
+
domain: { type: 'string', description: 'Business domain (finance, product, …).' },
|
|
58
|
+
owner: { type: 'string', description: 'Block owner identity (email or team handle).' },
|
|
59
|
+
description: { type: 'string', description: 'One-line description.' },
|
|
60
|
+
sql: { type: 'string', description: 'The block body SQL.' },
|
|
61
|
+
tags: { type: 'array', items: { type: 'string' } },
|
|
62
|
+
chartType: { type: 'string', description: 'Optional visualization type.' },
|
|
63
|
+
},
|
|
64
|
+
};
|
|
65
|
+
export function buildAgentTools(ctx) {
|
|
66
|
+
return [
|
|
67
|
+
{
|
|
68
|
+
name: 'search_blocks',
|
|
69
|
+
description: 'Find certified DQL blocks by keyword, domain, or status.',
|
|
70
|
+
inputSchema: SEARCH_BLOCKS_SCHEMA,
|
|
71
|
+
run: async (args) => searchBlocks(ctx, args),
|
|
72
|
+
},
|
|
73
|
+
{
|
|
74
|
+
name: 'get_block',
|
|
75
|
+
description: 'Return full metadata, dependencies, and SQL for a block.',
|
|
76
|
+
inputSchema: GET_BLOCK_SCHEMA,
|
|
77
|
+
run: async (args) => getBlock(ctx, args),
|
|
78
|
+
},
|
|
79
|
+
{
|
|
80
|
+
name: 'list_metrics',
|
|
81
|
+
description: 'List semantic-layer metrics, optionally filtered by domain.',
|
|
82
|
+
inputSchema: DOMAIN_FILTER_SCHEMA,
|
|
83
|
+
run: async (args) => listMetrics(ctx, args),
|
|
84
|
+
},
|
|
85
|
+
{
|
|
86
|
+
name: 'list_dimensions',
|
|
87
|
+
description: 'List semantic-layer dimensions, optionally filtered by domain.',
|
|
88
|
+
inputSchema: DOMAIN_FILTER_SCHEMA,
|
|
89
|
+
run: async (args) => listDimensions(ctx, args),
|
|
90
|
+
},
|
|
91
|
+
{
|
|
92
|
+
name: 'lineage_impact',
|
|
93
|
+
description: 'Return upstream/downstream lineage for a block, metric, or model.',
|
|
94
|
+
inputSchema: LINEAGE_IMPACT_SCHEMA,
|
|
95
|
+
run: async (args) => lineageImpact(ctx, args),
|
|
96
|
+
},
|
|
97
|
+
{
|
|
98
|
+
name: 'certify',
|
|
99
|
+
description: 'Run governance rules against a block and report pass/fail.',
|
|
100
|
+
inputSchema: CERTIFY_SCHEMA,
|
|
101
|
+
run: async (args) => certify(ctx, args),
|
|
102
|
+
},
|
|
103
|
+
{
|
|
104
|
+
name: 'suggest_block',
|
|
105
|
+
description: 'Write a proposed block to blocks/_drafts/ and return governance results. ' +
|
|
106
|
+
'Use this at the end of a conversation to hand the user a reviewable draft.',
|
|
107
|
+
inputSchema: SUGGEST_BLOCK_SCHEMA,
|
|
108
|
+
run: async (args) => suggestBlock(ctx, args),
|
|
109
|
+
},
|
|
110
|
+
];
|
|
111
|
+
}
|
|
112
|
+
//# sourceMappingURL=tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../src/llm/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,YAAY,EACZ,QAAQ,EACR,WAAW,EACX,cAAc,EACd,aAAa,EACb,OAAO,EACP,YAAY,GACb,MAAM,yBAAyB,CAAC;AASjC,MAAM,oBAAoB,GAAG;IAC3B,IAAI,EAAE,QAAQ;IACd,oBAAoB,EAAE,KAAK;IAC3B,UAAU,EAAE;QACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,uDAAuD,EAAE;QAC/F,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,qCAAqC,EAAE;QAC9E,MAAM,EAAE;YACN,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,CAAC,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,yBAAyB,CAAC;YAC/E,WAAW,EAAE,iCAAiC;SAC/C;QACD,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,2BAA2B,EAAE;KACpE;CACO,CAAC;AAEX,MAAM,gBAAgB,GAAG;IACvB,IAAI,EAAE,QAAQ;IACd,oBAAoB,EAAE,KAAK;IAC3B,QAAQ,EAAE,CAAC,MAAM,CAAC;IAClB,UAAU,EAAE;QACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,yCAAyC,EAAE;QAChF,aAAa,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,8CAA8C,EAAE;KAChG;CACO,CAAC;AAEX,MAAM,oBAAoB,GAAG;IAC3B,IAAI,EAAE,QAAQ;IACd,oBAAoB,EAAE,KAAK;IAC3B,UAAU,EAAE;QACV,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,4BAA4B,EAAE;KACtE;CACO,CAAC;AAEX,MAAM,qBAAqB,GAAG;IAC5B,IAAI,EAAE,QAAQ;IACd,oBAAoB,EAAE,KAAK;IAC3B,QAAQ,EAAE,CAAC,OAAO,CAAC;IACnB,UAAU,EAAE;QACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,yCAAyC,EAAE;QACjF,aAAa,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QACjC,eAAe,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QACnC,KAAK,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,0DAA0D,EAAE;KACpG;CACO,CAAC;AAEX,MAAM,cAAc,GAAG;IACrB,IAAI,EAAE,QAAQ;IACd,oBAAoB,EAAE,KAAK;IAC3B,QAAQ,EAAE,CAAC,MAAM,CAAC;IAClB,UAAU,EAAE;QACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,wBAAwB,EAAE;KAChE;CACO,CAAC;AAEX,MAAM,oBAAoB,GAAG;IAC3B,IAAI,EAAE,QAAQ;IACd,oBAAoB,EAAE,KAAK;IAC3B,QAAQ,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,EAAE,KAAK,CAAC;IAC3D,UAAU,EAAE;QACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,mCAAmC,EAAE;QAC1E,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,wCAAwC,EAAE;QACjF,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,8CAA8C,EAAE;QACtF,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,uBAAuB,EAAE;QACrE,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,qBAAqB,EAAE;QAC3D,IAAI,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;QAClD,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,8BAA8B,EAAE;KAC3E;CACO,CAAC;AAEX,MAAM,UAAU,eAAe,CAAC,GAAe;IAC7C,OAAO;QACL;YACE,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,0DAA0D;YACvE,WAAW,EAAE,oBAAoB;YACjC,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,EAAE,IAA0C,CAAC;SACnF;QACD;YACE,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,0DAA0D;YACvE,WAAW,EAAE,gBAAgB;YAC7B,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,EAAE,IAAsC,CAAC;SAC3E;QACD;YACE,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,6DAA6D;YAC1E,WAAW,EAAE,oBAAoB;YACjC,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE,IAAyC,CAAC;SACjF;QACD;YACE,IAAI,EAAE,iBAAiB;YACvB,WAAW,EAAE,gEAAgE;YAC7E,WAAW,EAAE,oBAAoB;YACjC,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,cAAc,CAAC,GAAG,EAAE,IAA4C,CAAC;SACvF;QACD;YACE,IAAI,EAAE,gBAAgB;YACtB,WAAW,EAAE,mEAAmE;YAChF,WAAW,EAAE,qBAAqB;YAClC,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,EAAE,IAA2C,CAAC;SACrF;QACD;YACE,IAAI,EAAE,SAAS;YACf,WAAW,EAAE,4DAA4D;YACzE,WAAW,EAAE,cAAc;YAC3B,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,IAAqC,CAAC;SACzE;QACD;YACE,IAAI,EAAE,eAAe;YACrB,WAAW,EACT,2EAA2E;gBAC3E,4EAA4E;YAC9E,WAAW,EAAE,oBAAoB;YACjC,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,EAAE,IAA0C,CAAC;SACnF;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,72 @@
|
|
|
1
|
+
import type { AgentResultPayload, KGNode } from '@duckcodeailabs/dql-agent';
|
|
2
|
+
export type ProviderId = 'claude-agent-sdk' | 'claude-code' | 'openai' | 'gemini' | 'ollama' | 'custom-openai';
|
|
3
|
+
export interface ChatTurn {
|
|
4
|
+
role: 'user' | 'assistant';
|
|
5
|
+
content: string;
|
|
6
|
+
}
|
|
7
|
+
export interface BlockProposal {
|
|
8
|
+
name: string;
|
|
9
|
+
domain: string;
|
|
10
|
+
owner: string;
|
|
11
|
+
description: string;
|
|
12
|
+
sql: string;
|
|
13
|
+
tags?: string[];
|
|
14
|
+
chartType?: string;
|
|
15
|
+
}
|
|
16
|
+
/**
|
|
17
|
+
* Normalized event a provider streams back for each step of an agent run.
|
|
18
|
+
* The UI renders these in order; the final `proposal` (if any) routes through
|
|
19
|
+
* the governance gate before `/api/blocks/save-from-cell`.
|
|
20
|
+
*/
|
|
21
|
+
export type AgentTurn = {
|
|
22
|
+
kind: 'thinking';
|
|
23
|
+
text: string;
|
|
24
|
+
} | {
|
|
25
|
+
kind: 'text';
|
|
26
|
+
text: string;
|
|
27
|
+
} | {
|
|
28
|
+
kind: 'tool_call';
|
|
29
|
+
id: string;
|
|
30
|
+
name: string;
|
|
31
|
+
input: unknown;
|
|
32
|
+
} | {
|
|
33
|
+
kind: 'tool_result';
|
|
34
|
+
id: string;
|
|
35
|
+
output: unknown;
|
|
36
|
+
isError?: boolean;
|
|
37
|
+
} | {
|
|
38
|
+
kind: 'proposal';
|
|
39
|
+
proposal: BlockProposal;
|
|
40
|
+
governance: {
|
|
41
|
+
certified: boolean;
|
|
42
|
+
errors: string[];
|
|
43
|
+
warnings: string[];
|
|
44
|
+
};
|
|
45
|
+
} | {
|
|
46
|
+
kind: 'error';
|
|
47
|
+
message: string;
|
|
48
|
+
} | {
|
|
49
|
+
kind: 'done';
|
|
50
|
+
stopReason?: string;
|
|
51
|
+
};
|
|
52
|
+
export interface AgentRunRequest {
|
|
53
|
+
provider: ProviderId;
|
|
54
|
+
messages: ChatTurn[];
|
|
55
|
+
upstream?: {
|
|
56
|
+
cellId?: string;
|
|
57
|
+
sql?: string;
|
|
58
|
+
preview?: unknown;
|
|
59
|
+
};
|
|
60
|
+
projectRoot: string;
|
|
61
|
+
executeCertifiedBlock?: (block: KGNode) => Promise<AgentResultPayload>;
|
|
62
|
+
}
|
|
63
|
+
export interface AgentRunner {
|
|
64
|
+
run(req: AgentRunRequest, emit: (turn: AgentTurn) => void, signal: AbortSignal): Promise<void>;
|
|
65
|
+
}
|
|
66
|
+
/**
|
|
67
|
+
* Public-facing name for the adapter contract a provider must implement.
|
|
68
|
+
* `AgentRunner` is the internal call site; `LLMProvider` is what community
|
|
69
|
+
* authors target when they add a new `providers/<name>.ts`. Keep in sync.
|
|
70
|
+
*/
|
|
71
|
+
export type LLMProvider = AgentRunner;
|
|
72
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/llm/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,EAAE,MAAM,2BAA2B,CAAC;AAE5E,MAAM,MAAM,UAAU,GAAG,kBAAkB,GAAG,aAAa,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,GAAG,eAAe,CAAC;AAE/G,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;;;GAIG;AACH,MAAM,MAAM,SAAS,GACjB;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAClC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAC9B;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,EAAE,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,OAAO,CAAA;CAAE,GAC/D;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,EAAE,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,OAAO,CAAC;IAAC,OAAO,CAAC,EAAE,OAAO,CAAA;CAAE,GACvE;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,QAAQ,EAAE,aAAa,CAAC;IAAC,UAAU,EAAE;QAAE,SAAS,EAAE,OAAO,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAC;QAAC,QAAQ,EAAE,MAAM,EAAE,CAAA;KAAE,CAAA;CAAE,GACvH;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAClC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAE1C,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,UAAU,CAAC;IACrB,QAAQ,EAAE,QAAQ,EAAE,CAAC;IACrB,QAAQ,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC;IAChE,WAAW,EAAE,MAAM,CAAC;IACpB,qBAAqB,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,kBAAkB,CAAC,CAAC;CACxE;AAED,MAAM,WAAW,WAAW;IAC1B,GAAG,CAAC,GAAG,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,IAAI,EAAE,SAAS,KAAK,IAAI,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAChG;AAED;;;;GAIG;AACH,MAAM,MAAM,WAAW,GAAG,WAAW,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/llm/types.ts"],"names":[],"mappings":""}
|