@contractspec/module.ai-chat 4.3.17 → 4.3.18

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.
@@ -1,690 +1,40 @@
1
- var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
2
- get: (a, b) => (typeof require !== "undefined" ? require : a)[b]
3
- }) : x)(function(x) {
4
- if (typeof require !== "undefined")
5
- return require.apply(this, arguments);
6
- throw Error('Dynamic require of "' + x + '" is not supported');
7
- });
8
-
9
- // src/presentation/hooks/index.ts
10
- import { useCompletion } from "@ai-sdk/react";
11
-
12
- // src/presentation/hooks/useChat.tsx
13
- import {
14
- createProvider
15
- } from "@contractspec/lib.ai-providers";
16
- import { tool as tool4 } from "ai";
17
- import * as React from "react";
18
- import { z as z4 } from "zod";
19
-
20
- // src/core/chat-service.ts
21
- import { generateText, streamText } from "ai";
22
- import { compilePlannerPrompt } from "@contractspec/lib.surface-runtime/runtime/planner-prompt";
23
-
24
- // src/core/agent-tools-adapter.ts
25
- import { tool } from "ai";
26
- import { z } from "zod";
27
- function getInputSchema(_schema) {
28
- return z.object({}).passthrough();
29
- }
30
- function agentToolConfigsToToolSet(configs, handlers) {
31
- const result = {};
32
- for (const config of configs) {
33
- const handler = handlers?.[config.name];
34
- const inputSchema = getInputSchema(config.schema);
35
- result[config.name] = tool({
36
- description: config.description ?? config.name,
37
- inputSchema,
38
- execute: async (input) => {
39
- if (!handler) {
40
- return {
41
- status: "unimplemented",
42
- message: "Wire handler in host",
43
- toolName: config.name
44
- };
45
- }
46
- try {
47
- const output = await Promise.resolve(handler(input));
48
- return typeof output === "string" ? output : output;
49
- } catch (err) {
50
- return {
51
- status: "error",
52
- error: err instanceof Error ? err.message : String(err),
53
- toolName: config.name
54
- };
55
- }
56
- }
57
- });
58
- }
59
- return result;
60
- }
61
-
62
- // src/core/contracts-context.ts
63
- function buildContractsContextPrompt(config) {
64
- const parts = [];
65
- if (!config.agentSpecs?.length && !config.dataViewSpecs?.length && !config.formSpecs?.length && !config.presentationSpecs?.length && !config.operationRefs?.length) {
66
- return "";
67
- }
68
- parts.push(`
69
-
70
- ## Available resources`);
71
- if (config.agentSpecs?.length) {
72
- parts.push(`
73
- ### Agent tools`);
74
- for (const agent of config.agentSpecs) {
75
- const toolNames = agent.tools?.map((t) => t.name).join(", ") ?? "none";
76
- parts.push(`- **${agent.key}**: tools: ${toolNames}`);
77
- }
78
- }
79
- if (config.dataViewSpecs?.length) {
80
- parts.push(`
81
- ### Data views`);
82
- for (const dv of config.dataViewSpecs) {
83
- parts.push(`- **${dv.key}**: ${dv.meta.title ?? dv.key}`);
84
- }
85
- }
86
- if (config.formSpecs?.length) {
87
- parts.push(`
88
- ### Forms`);
89
- for (const form of config.formSpecs) {
90
- parts.push(`- **${form.key}**: ${form.meta.title ?? form.key}`);
91
- }
92
- }
93
- if (config.presentationSpecs?.length) {
94
- parts.push(`
95
- ### Presentations`);
96
- for (const pres of config.presentationSpecs) {
97
- parts.push(`- **${pres.key}**: ${pres.meta.title ?? pres.key} (targets: ${pres.targets?.join(", ") ?? "react"})`);
98
- }
99
- }
100
- if (config.operationRefs?.length) {
101
- parts.push(`
102
- ### Operations`);
103
- for (const op of config.operationRefs) {
104
- parts.push(`- **${op.key}@${op.version}**`);
105
- }
106
- }
107
- parts.push(`
108
- Use the available tools to invoke operations, query data views, or propose surface changes when appropriate.`);
109
- return parts.join(`
110
- `);
111
- }
112
-
113
- // src/core/conversation-store.ts
114
- function generateId(prefix) {
115
- return `${prefix}_${Date.now()}_${Math.random().toString(36).slice(2, 11)}`;
116
- }
117
-
118
- class InMemoryConversationStore {
119
- conversations = new Map;
120
- async get(conversationId) {
121
- return this.conversations.get(conversationId) ?? null;
122
- }
123
- async create(conversation) {
124
- const now = new Date;
125
- const fullConversation = {
126
- ...conversation,
127
- id: generateId("conv"),
128
- createdAt: now,
129
- updatedAt: now
130
- };
131
- this.conversations.set(fullConversation.id, fullConversation);
132
- return fullConversation;
133
- }
134
- async update(conversationId, updates) {
135
- const conversation = this.conversations.get(conversationId);
136
- if (!conversation)
137
- return null;
138
- const updated = {
139
- ...conversation,
140
- ...updates,
141
- updatedAt: new Date
142
- };
143
- this.conversations.set(conversationId, updated);
144
- return updated;
145
- }
146
- async appendMessage(conversationId, message) {
147
- const conversation = this.conversations.get(conversationId);
148
- if (!conversation) {
149
- throw new Error(`Conversation ${conversationId} not found`);
150
- }
151
- const now = new Date;
152
- const fullMessage = {
153
- ...message,
154
- id: generateId("msg"),
155
- conversationId,
156
- createdAt: now,
157
- updatedAt: now
158
- };
159
- conversation.messages.push(fullMessage);
160
- conversation.updatedAt = now;
161
- return fullMessage;
162
- }
163
- async updateMessage(conversationId, messageId, updates) {
164
- const conversation = this.conversations.get(conversationId);
165
- if (!conversation)
166
- return null;
167
- const messageIndex = conversation.messages.findIndex((m) => m.id === messageId);
168
- if (messageIndex === -1)
169
- return null;
170
- const message = conversation.messages[messageIndex];
171
- if (!message)
172
- return null;
173
- const updated = {
174
- ...message,
175
- ...updates,
176
- updatedAt: new Date
177
- };
178
- conversation.messages[messageIndex] = updated;
179
- conversation.updatedAt = new Date;
180
- return updated;
181
- }
182
- async delete(conversationId) {
183
- return this.conversations.delete(conversationId);
184
- }
185
- async list(options) {
186
- let results = Array.from(this.conversations.values());
187
- if (options?.status) {
188
- results = results.filter((c) => c.status === options.status);
189
- }
190
- if (options?.projectId) {
191
- results = results.filter((c) => c.projectId === options.projectId);
192
- }
193
- if (options?.tags && options.tags.length > 0) {
194
- const tagSet = new Set(options.tags);
195
- results = results.filter((c) => c.tags && c.tags.some((t) => tagSet.has(t)));
196
- }
197
- results.sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
198
- const offset = options?.offset ?? 0;
199
- const limit = options?.limit ?? 100;
200
- return results.slice(offset, offset + limit);
201
- }
202
- async fork(conversationId, upToMessageId) {
203
- const source = this.conversations.get(conversationId);
204
- if (!source) {
205
- throw new Error(`Conversation ${conversationId} not found`);
206
- }
207
- let messagesToCopy = source.messages;
208
- if (upToMessageId) {
209
- const idx = source.messages.findIndex((m) => m.id === upToMessageId);
210
- if (idx === -1) {
211
- throw new Error(`Message ${upToMessageId} not found`);
212
- }
213
- messagesToCopy = source.messages.slice(0, idx + 1);
214
- }
215
- const now = new Date;
216
- const forkedMessages = messagesToCopy.map((m) => ({
217
- ...m,
218
- id: generateId("msg"),
219
- conversationId: "",
220
- createdAt: new Date(m.createdAt),
221
- updatedAt: new Date(m.updatedAt)
222
- }));
223
- const forked = {
224
- ...source,
225
- id: generateId("conv"),
226
- title: source.title ? `${source.title} (fork)` : undefined,
227
- forkedFromId: source.id,
228
- createdAt: now,
229
- updatedAt: now,
230
- messages: forkedMessages
231
- };
232
- for (const m of forked.messages) {
233
- m.conversationId = forked.id;
234
- }
235
- this.conversations.set(forked.id, forked);
236
- return forked;
237
- }
238
- async truncateAfter(conversationId, messageId) {
239
- const conv = this.conversations.get(conversationId);
240
- if (!conv)
241
- return null;
242
- const idx = conv.messages.findIndex((m) => m.id === messageId);
243
- if (idx === -1)
244
- return null;
245
- conv.messages = conv.messages.slice(0, idx + 1);
246
- conv.updatedAt = new Date;
247
- return conv;
248
- }
249
- async search(query, limit = 20) {
250
- const lowerQuery = query.toLowerCase();
251
- const results = [];
252
- for (const conversation of this.conversations.values()) {
253
- if (conversation.title?.toLowerCase().includes(lowerQuery)) {
254
- results.push(conversation);
255
- continue;
256
- }
257
- const hasMatch = conversation.messages.some((m) => m.content.toLowerCase().includes(lowerQuery));
258
- if (hasMatch) {
259
- results.push(conversation);
260
- }
261
- if (results.length >= limit)
262
- break;
263
- }
264
- return results;
265
- }
266
- clear() {
267
- this.conversations.clear();
268
- }
269
- }
270
- function createInMemoryConversationStore() {
271
- return new InMemoryConversationStore;
272
- }
273
-
274
- // src/core/surface-planner-tools.ts
275
- import { buildSurfacePatchProposal } from "@contractspec/lib.surface-runtime/runtime/planner-tools";
276
- import {
277
- validatePatchProposal
278
- } from "@contractspec/lib.surface-runtime/spec/validate-surface-patch";
279
- import { tool as tool2 } from "ai";
280
- import { z as z2 } from "zod";
281
- var VALID_OPS = [
282
- "insert-node",
283
- "replace-node",
284
- "remove-node",
285
- "move-node",
286
- "resize-panel",
287
- "set-layout",
288
- "reveal-field",
289
- "hide-field",
290
- "promote-action",
291
- "set-focus"
292
- ];
293
- var DEFAULT_NODE_KINDS = [
294
- "entity-section",
295
- "entity-card",
296
- "data-view",
297
- "assistant-panel",
298
- "chat-thread",
299
- "action-bar",
300
- "timeline",
301
- "table",
302
- "rich-doc",
303
- "form",
304
- "chart",
305
- "custom-widget"
306
- ];
307
- function collectSlotIdsFromRegion(node) {
308
- const ids = [];
309
- if (node.type === "slot") {
310
- ids.push(node.slotId);
311
- }
312
- if (node.type === "panel-group" || node.type === "stack") {
313
- for (const child of node.children) {
314
- ids.push(...collectSlotIdsFromRegion(child));
315
- }
316
- }
317
- if (node.type === "tabs") {
318
- for (const tab of node.tabs) {
319
- ids.push(...collectSlotIdsFromRegion(tab.child));
320
- }
321
- }
322
- if (node.type === "floating") {
323
- ids.push(node.anchorSlotId);
324
- ids.push(...collectSlotIdsFromRegion(node.child));
325
- }
326
- return ids;
327
- }
328
- function deriveConstraints(plan) {
329
- const slotIds = collectSlotIdsFromRegion(plan.layoutRoot);
330
- const uniqueSlots = [...new Set(slotIds)];
331
- return {
332
- allowedOps: VALID_OPS,
333
- allowedSlots: uniqueSlots.length > 0 ? uniqueSlots : ["assistant", "primary"],
334
- allowedNodeKinds: DEFAULT_NODE_KINDS
335
- };
336
- }
337
- var ProposePatchInputSchema = z2.object({
338
- proposalId: z2.string().describe("Unique proposal identifier"),
339
- ops: z2.array(z2.object({
340
- op: z2.enum([
341
- "insert-node",
342
- "replace-node",
343
- "remove-node",
344
- "move-node",
345
- "resize-panel",
346
- "set-layout",
347
- "reveal-field",
348
- "hide-field",
349
- "promote-action",
350
- "set-focus"
351
- ]),
352
- slotId: z2.string().optional(),
353
- nodeId: z2.string().optional(),
354
- toSlotId: z2.string().optional(),
355
- index: z2.number().optional(),
356
- node: z2.object({
357
- nodeId: z2.string(),
358
- kind: z2.string(),
359
- title: z2.string().optional(),
360
- props: z2.record(z2.string(), z2.unknown()).optional(),
361
- children: z2.array(z2.unknown()).optional()
362
- }).optional(),
363
- persistKey: z2.string().optional(),
364
- sizes: z2.array(z2.number()).optional(),
365
- layoutId: z2.string().optional(),
366
- fieldId: z2.string().optional(),
367
- actionId: z2.string().optional(),
368
- placement: z2.enum(["header", "inline", "context", "assistant"]).optional(),
369
- targetId: z2.string().optional()
370
- }))
371
- });
372
- function createSurfacePlannerTools(config) {
373
- const { plan, constraints, onPatchProposal } = config;
374
- const resolvedConstraints = constraints ?? deriveConstraints(plan);
375
- const proposePatchTool = tool2({
376
- description: "Propose surface patches (layout changes, node insertions, etc.) for user approval. " + "Only use allowed ops, slots, and node kinds from the planner context.",
377
- inputSchema: ProposePatchInputSchema,
378
- execute: async (input) => {
379
- const ops = input.ops;
380
- try {
381
- validatePatchProposal(ops, resolvedConstraints);
382
- const proposal = buildSurfacePatchProposal(input.proposalId, ops);
383
- onPatchProposal?.(proposal);
384
- return {
385
- success: true,
386
- proposalId: proposal.proposalId,
387
- opsCount: proposal.ops.length,
388
- message: "Patch proposal validated; awaiting user approval"
389
- };
390
- } catch (err) {
391
- return {
392
- success: false,
393
- error: err instanceof Error ? err.message : String(err),
394
- proposalId: input.proposalId
395
- };
396
- }
397
- }
398
- });
399
- return {
400
- "propose-patch": proposePatchTool
401
- };
402
- }
403
- function buildPlannerPromptInput(plan) {
404
- const constraints = deriveConstraints(plan);
405
- return {
406
- bundleMeta: {
407
- key: plan.bundleKey,
408
- version: "0.0.0",
409
- title: plan.bundleKey
410
- },
411
- surfaceId: plan.surfaceId,
412
- allowedPatchOps: constraints.allowedOps,
413
- allowedSlots: [...constraints.allowedSlots],
414
- allowedNodeKinds: [...constraints.allowedNodeKinds],
415
- actions: plan.actions.map((a) => ({
416
- actionId: a.actionId,
417
- title: a.title
418
- })),
419
- preferences: {
420
- guidance: "hints",
421
- density: "standard",
422
- dataDepth: "detailed",
423
- control: "standard",
424
- media: "text",
425
- pace: "balanced",
426
- narrative: "top-down"
427
- }
428
- };
429
- }
430
-
431
- // src/core/thinking-levels.ts
432
- var THINKING_LEVEL_LABELS = {
433
- instant: "Instant",
434
- thinking: "Thinking",
435
- extra_thinking: "Extra Thinking",
436
- max: "Max"
437
- };
438
- var THINKING_LEVEL_DESCRIPTIONS = {
439
- instant: "Fast responses, minimal reasoning",
440
- thinking: "Standard reasoning depth",
441
- extra_thinking: "More thorough reasoning",
442
- max: "Maximum reasoning depth"
443
- };
444
- function getProviderOptions(level, providerName) {
445
- if (!level || level === "instant") {
446
- return {};
447
- }
448
- switch (providerName) {
449
- case "anthropic": {
450
- const budgetMap = {
451
- thinking: 8000,
452
- extra_thinking: 16000,
453
- max: 32000
454
- };
455
- return {
456
- anthropic: {
457
- thinking: { type: "enabled", budgetTokens: budgetMap[level] }
458
- }
459
- };
460
- }
461
- case "openai": {
462
- const effortMap = {
463
- thinking: "low",
464
- extra_thinking: "medium",
465
- max: "high"
466
- };
467
- return {
468
- openai: {
469
- reasoningEffort: effortMap[level]
470
- }
471
- };
472
- }
473
- case "ollama":
474
- case "mistral":
475
- case "gemini":
476
- return {};
477
- default:
478
- return {};
479
- }
480
- }
481
-
482
- // src/core/workflow-tools.ts
483
- import {
484
- validateExtension,
485
- WorkflowComposer
486
- } from "@contractspec/lib.workflow-composer";
487
- import { tool as tool3 } from "ai";
488
- import { z as z3 } from "zod";
489
- var StepTypeSchema = z3.enum(["human", "automation", "decision"]);
490
- var StepActionSchema = z3.object({
491
- operation: z3.object({
492
- name: z3.string(),
493
- version: z3.number()
494
- }).optional(),
495
- form: z3.object({
496
- key: z3.string(),
497
- version: z3.number()
498
- }).optional()
499
- }).optional();
500
- var StepSchema = z3.object({
501
- id: z3.string(),
502
- type: StepTypeSchema,
503
- label: z3.string(),
504
- description: z3.string().optional(),
505
- action: StepActionSchema
506
- });
507
- var StepInjectionSchema = z3.object({
508
- after: z3.string().optional(),
509
- before: z3.string().optional(),
510
- inject: StepSchema,
511
- transitionTo: z3.string().optional(),
512
- transitionFrom: z3.string().optional(),
513
- when: z3.string().optional()
514
- });
515
- var WorkflowExtensionInputSchema = z3.object({
516
- workflow: z3.string(),
517
- tenantId: z3.string().optional(),
518
- role: z3.string().optional(),
519
- priority: z3.number().optional(),
520
- customSteps: z3.array(StepInjectionSchema).optional(),
521
- hiddenSteps: z3.array(z3.string()).optional()
522
- });
523
- function createWorkflowTools(config) {
524
- const { baseWorkflows, composer } = config;
525
- const baseByKey = new Map(baseWorkflows.map((b) => [b.meta.key, b]));
526
- const createWorkflowExtensionTool = tool3({
527
- description: "Create or validate a workflow extension. Use when the user asks to add steps, modify a workflow, or create a tenant-specific extension. The extension targets an existing base workflow.",
528
- inputSchema: WorkflowExtensionInputSchema,
529
- execute: async (input) => {
530
- const extension = {
531
- workflow: input.workflow,
532
- tenantId: input.tenantId,
533
- role: input.role,
534
- priority: input.priority,
535
- customSteps: input.customSteps,
536
- hiddenSteps: input.hiddenSteps
537
- };
538
- const base = baseByKey.get(input.workflow);
539
- if (!base) {
540
- return {
541
- success: false,
542
- error: `Base workflow "${input.workflow}" not found. Available: ${Array.from(baseByKey.keys()).join(", ")}`,
543
- extension
544
- };
545
- }
546
- try {
547
- validateExtension(extension, base);
548
- return {
549
- success: true,
550
- message: "Extension validated successfully",
551
- extension
552
- };
553
- } catch (err) {
554
- return {
555
- success: false,
556
- error: err instanceof Error ? err.message : String(err),
557
- extension
558
- };
559
- }
560
- }
561
- });
562
- const composeWorkflowInputSchema = z3.object({
563
- workflowKey: z3.string().describe("Base workflow meta.key"),
564
- tenantId: z3.string().optional(),
565
- role: z3.string().optional(),
566
- extensions: z3.array(WorkflowExtensionInputSchema).optional().describe("Extensions to register before composing")
567
- });
568
- const composeWorkflowTool = tool3({
569
- description: "Compose a workflow by applying registered extensions to a base workflow. Returns the composed WorkflowSpec.",
570
- inputSchema: composeWorkflowInputSchema,
571
- execute: async (input) => {
572
- const base = baseByKey.get(input.workflowKey);
573
- if (!base) {
574
- return {
575
- success: false,
576
- error: `Base workflow "${input.workflowKey}" not found. Available: ${Array.from(baseByKey.keys()).join(", ")}`
577
- };
578
- }
579
- const comp = composer ?? new WorkflowComposer;
580
- if (input.extensions?.length) {
581
- for (const ext of input.extensions) {
582
- comp.register({
583
- workflow: ext.workflow,
584
- tenantId: ext.tenantId,
585
- role: ext.role,
586
- priority: ext.priority,
587
- customSteps: ext.customSteps,
588
- hiddenSteps: ext.hiddenSteps
589
- });
590
- }
591
- }
592
- try {
593
- const composed = comp.compose({
594
- base,
595
- tenantId: input.tenantId,
596
- role: input.role
597
- });
598
- return {
599
- success: true,
600
- workflow: composed,
601
- meta: composed.meta,
602
- stepIds: composed.definition.steps.map((s) => s.id)
603
- };
604
- } catch (err) {
605
- return {
606
- success: false,
607
- error: err instanceof Error ? err.message : String(err)
608
- };
609
- }
610
- }
611
- });
612
- const generateWorkflowSpecCodeInputSchema = z3.object({
613
- workflowKey: z3.string().describe("Workflow meta.key"),
614
- composedSteps: z3.array(z3.object({
615
- id: z3.string(),
616
- type: z3.enum(["human", "automation", "decision"]),
617
- label: z3.string(),
618
- description: z3.string().optional()
619
- })).optional().describe("Steps to include; if omitted, uses the base workflow")
620
- });
621
- const generateWorkflowSpecCodeTool = tool3({
622
- description: "Generate TypeScript code for a workflow spec. Use after composing a workflow to output the spec as code the user can save.",
623
- inputSchema: generateWorkflowSpecCodeInputSchema,
624
- execute: async (input) => {
625
- const base = baseByKey.get(input.workflowKey);
626
- if (!base) {
627
- return {
628
- success: false,
629
- error: `Base workflow "${input.workflowKey}" not found. Available: ${Array.from(baseByKey.keys()).join(", ")}`,
630
- code: null
631
- };
632
- }
633
- const steps = input.composedSteps ?? base.definition.steps;
634
- const specVarName = toPascalCase((base.meta.key.split(".").pop() ?? "Workflow") + "") + "Workflow";
635
- const stepsCode = steps.map((s) => ` {
636
- id: '${s.id}',
637
- type: '${s.type}',
638
- label: '${escapeString(s.label)}',${s.description ? `
639
- description: '${escapeString(s.description)}',` : ""}
1
+ var PX=((X)=>typeof require<"u"?require:typeof Proxy<"u"?new Proxy(X,{get:(Z,$)=>(typeof require<"u"?require:Z)[$]}):X)(function(X){if(typeof require<"u")return require.apply(this,arguments);throw Error('Dynamic require of "'+X+'" is not supported')});import{useCompletion as cZ}from"@ai-sdk/react";import{createProvider as nX}from"@contractspec/lib.ai-providers";import{tool as tX}from"ai";import*as E from"react";import{z as sX}from"zod";import{generateText as pX,streamText as cX}from"ai";import{compilePlannerPrompt as iX}from"@contractspec/lib.surface-runtime/runtime/planner-prompt";import{tool as xX}from"ai";import{z as bX}from"zod";function kX(X){return bX.object({}).passthrough()}function qX(X,Z){let $={};for(let J of X){let Y=Z?.[J.name],G=kX(J.schema);$[J.name]=xX({description:J.description??J.name,inputSchema:G,execute:async(j)=>{if(!Y)return{status:"unimplemented",message:"Wire handler in host",toolName:J.name};try{let H=await Promise.resolve(Y(j));return typeof H==="string"?H:H}catch(H){return{status:"error",error:H instanceof Error?H.message:String(H),toolName:J.name}}}})}return $}function BX(X){let Z=[];if(!X.agentSpecs?.length&&!X.dataViewSpecs?.length&&!X.formSpecs?.length&&!X.presentationSpecs?.length&&!X.operationRefs?.length)return"";if(Z.push(`
2
+
3
+ ## Available resources`),X.agentSpecs?.length){Z.push(`
4
+ ### Agent tools`);for(let $ of X.agentSpecs){let J=$.tools?.map((Y)=>Y.name).join(", ")??"none";Z.push(`- **${$.key}**: tools: ${J}`)}}if(X.dataViewSpecs?.length){Z.push(`
5
+ ### Data views`);for(let $ of X.dataViewSpecs)Z.push(`- **${$.key}**: ${$.meta.title??$.key}`)}if(X.formSpecs?.length){Z.push(`
6
+ ### Forms`);for(let $ of X.formSpecs)Z.push(`- **${$.key}**: ${$.meta.title??$.key}`)}if(X.presentationSpecs?.length){Z.push(`
7
+ ### Presentations`);for(let $ of X.presentationSpecs)Z.push(`- **${$.key}**: ${$.meta.title??$.key} (targets: ${$.targets?.join(", ")??"react"})`)}if(X.operationRefs?.length){Z.push(`
8
+ ### Operations`);for(let $ of X.operationRefs)Z.push(`- **${$.key}@${$.version}**`)}return Z.push(`
9
+ Use the available tools to invoke operations, query data views, or propose surface changes when appropriate.`),Z.join(`
10
+ `)}function s(X){return`${X}_${Date.now()}_${Math.random().toString(36).slice(2,11)}`}class o{conversations=new Map;async get(X){return this.conversations.get(X)??null}async create(X){let Z=new Date,$={...X,id:s("conv"),createdAt:Z,updatedAt:Z};return this.conversations.set($.id,$),$}async update(X,Z){let $=this.conversations.get(X);if(!$)return null;let J={...$,...Z,updatedAt:new Date};return this.conversations.set(X,J),J}async appendMessage(X,Z){let $=this.conversations.get(X);if(!$)throw Error(`Conversation ${X} not found`);let J=new Date,Y={...Z,id:s("msg"),conversationId:X,createdAt:J,updatedAt:J};return $.messages.push(Y),$.updatedAt=J,Y}async updateMessage(X,Z,$){let J=this.conversations.get(X);if(!J)return null;let Y=J.messages.findIndex((H)=>H.id===Z);if(Y===-1)return null;let G=J.messages[Y];if(!G)return null;let j={...G,...$,updatedAt:new Date};return J.messages[Y]=j,J.updatedAt=new Date,j}async delete(X){return this.conversations.delete(X)}async list(X){let Z=Array.from(this.conversations.values());if(X?.status)Z=Z.filter((Y)=>Y.status===X.status);if(X?.projectId)Z=Z.filter((Y)=>Y.projectId===X.projectId);if(X?.tags&&X.tags.length>0){let Y=new Set(X.tags);Z=Z.filter((G)=>G.tags&&G.tags.some((j)=>Y.has(j)))}Z.sort((Y,G)=>G.updatedAt.getTime()-Y.updatedAt.getTime());let $=X?.offset??0,J=X?.limit??100;return Z.slice($,$+J)}async fork(X,Z){let $=this.conversations.get(X);if(!$)throw Error(`Conversation ${X} not found`);let J=$.messages;if(Z){let H=$.messages.findIndex((q)=>q.id===Z);if(H===-1)throw Error(`Message ${Z} not found`);J=$.messages.slice(0,H+1)}let Y=new Date,G=J.map((H)=>({...H,id:s("msg"),conversationId:"",createdAt:new Date(H.createdAt),updatedAt:new Date(H.updatedAt)})),j={...$,id:s("conv"),title:$.title?`${$.title} (fork)`:void 0,forkedFromId:$.id,createdAt:Y,updatedAt:Y,messages:G};for(let H of j.messages)H.conversationId=j.id;return this.conversations.set(j.id,j),j}async truncateAfter(X,Z){let $=this.conversations.get(X);if(!$)return null;let J=$.messages.findIndex((Y)=>Y.id===Z);if(J===-1)return null;return $.messages=$.messages.slice(0,J+1),$.updatedAt=new Date,$}async search(X,Z=20){let $=X.toLowerCase(),J=[];for(let Y of this.conversations.values()){if(Y.title?.toLowerCase().includes($)){J.push(Y);continue}if(Y.messages.some((j)=>j.content.toLowerCase().includes($)))J.push(Y);if(J.length>=Z)break}return J}clear(){this.conversations.clear()}}function qZ(){return new o}import{buildSurfacePatchProposal as TX}from"@contractspec/lib.surface-runtime/runtime/planner-tools";import{validatePatchProposal as SX}from"@contractspec/lib.surface-runtime/spec/validate-surface-patch";import{tool as hX}from"ai";import{z as K}from"zod";var CX=["insert-node","replace-node","remove-node","move-node","resize-panel","set-layout","reveal-field","hide-field","promote-action","set-focus"],MX=["entity-section","entity-card","data-view","assistant-panel","chat-thread","action-bar","timeline","table","rich-doc","form","chart","custom-widget"];function e(X){let Z=[];if(X.type==="slot")Z.push(X.slotId);if(X.type==="panel-group"||X.type==="stack")for(let $ of X.children)Z.push(...e($));if(X.type==="tabs")for(let $ of X.tabs)Z.push(...e($.child));if(X.type==="floating")Z.push(X.anchorSlotId),Z.push(...e(X.child));return Z}function _X(X){let Z=e(X.layoutRoot),$=[...new Set(Z)];return{allowedOps:CX,allowedSlots:$.length>0?$:["assistant","primary"],allowedNodeKinds:MX}}var IX=K.object({proposalId:K.string().describe("Unique proposal identifier"),ops:K.array(K.object({op:K.enum(["insert-node","replace-node","remove-node","move-node","resize-panel","set-layout","reveal-field","hide-field","promote-action","set-focus"]),slotId:K.string().optional(),nodeId:K.string().optional(),toSlotId:K.string().optional(),index:K.number().optional(),node:K.object({nodeId:K.string(),kind:K.string(),title:K.string().optional(),props:K.record(K.string(),K.unknown()).optional(),children:K.array(K.unknown()).optional()}).optional(),persistKey:K.string().optional(),sizes:K.array(K.number()).optional(),layoutId:K.string().optional(),fieldId:K.string().optional(),actionId:K.string().optional(),placement:K.enum(["header","inline","context","assistant"]).optional(),targetId:K.string().optional()}))});function NX(X){let{plan:Z,constraints:$,onPatchProposal:J}=X,Y=$??_X(Z);return{"propose-patch":hX({description:"Propose surface patches (layout changes, node insertions, etc.) for user approval. Only use allowed ops, slots, and node kinds from the planner context.",inputSchema:IX,execute:async(j)=>{let H=j.ops;try{SX(H,Y);let q=TX(j.proposalId,H);return J?.(q),{success:!0,proposalId:q.proposalId,opsCount:q.ops.length,message:"Patch proposal validated; awaiting user approval"}}catch(q){return{success:!1,error:q instanceof Error?q.message:String(q),proposalId:j.proposalId}}}})}}function UX(X){let Z=_X(X);return{bundleMeta:{key:X.bundleKey,version:"0.0.0",title:X.bundleKey},surfaceId:X.surfaceId,allowedPatchOps:Z.allowedOps,allowedSlots:[...Z.allowedSlots],allowedNodeKinds:[...Z.allowedNodeKinds],actions:X.actions.map(($)=>({actionId:$.actionId,title:$.title})),preferences:{guidance:"hints",density:"standard",dataDepth:"detailed",control:"standard",media:"text",pace:"balanced",narrative:"top-down"}}}var FZ={instant:"Instant",thinking:"Thinking",extra_thinking:"Extra Thinking",max:"Max"},AZ={instant:"Fast responses, minimal reasoning",thinking:"Standard reasoning depth",extra_thinking:"More thorough reasoning",max:"Maximum reasoning depth"};function GX(X,Z){if(!X||X==="instant")return{};switch(Z){case"anthropic":return{anthropic:{thinking:{type:"enabled",budgetTokens:{thinking:8000,extra_thinking:16000,max:32000}[X]}}};case"openai":return{openai:{reasoningEffort:{thinking:"low",extra_thinking:"medium",max:"high"}[X]}};case"ollama":case"mistral":case"gemini":return{};default:return{}}}import{validateExtension as fX,WorkflowComposer as mX}from"@contractspec/lib.workflow-composer";import{tool as QX}from"ai";import{z as B}from"zod";var uX=B.enum(["human","automation","decision"]),vX=B.object({operation:B.object({name:B.string(),version:B.number()}).optional(),form:B.object({key:B.string(),version:B.number()}).optional()}).optional(),gX=B.object({id:B.string(),type:uX,label:B.string(),description:B.string().optional(),action:vX}),dX=B.object({after:B.string().optional(),before:B.string().optional(),inject:gX,transitionTo:B.string().optional(),transitionFrom:B.string().optional(),when:B.string().optional()}),WX=B.object({workflow:B.string(),tenantId:B.string().optional(),role:B.string().optional(),priority:B.number().optional(),customSteps:B.array(dX).optional(),hiddenSteps:B.array(B.string()).optional()});function DX(X){let{baseWorkflows:Z,composer:$}=X,J=new Map(Z.map((Q)=>[Q.meta.key,Q])),Y=QX({description:"Create or validate a workflow extension. Use when the user asks to add steps, modify a workflow, or create a tenant-specific extension. The extension targets an existing base workflow.",inputSchema:WX,execute:async(Q)=>{let _={workflow:Q.workflow,tenantId:Q.tenantId,role:Q.role,priority:Q.priority,customSteps:Q.customSteps,hiddenSteps:Q.hiddenSteps},A=J.get(Q.workflow);if(!A)return{success:!1,error:`Base workflow "${Q.workflow}" not found. Available: ${Array.from(J.keys()).join(", ")}`,extension:_};try{return fX(_,A),{success:!0,message:"Extension validated successfully",extension:_}}catch(N){return{success:!1,error:N instanceof Error?N.message:String(N),extension:_}}}}),G=B.object({workflowKey:B.string().describe("Base workflow meta.key"),tenantId:B.string().optional(),role:B.string().optional(),extensions:B.array(WX).optional().describe("Extensions to register before composing")}),j=QX({description:"Compose a workflow by applying registered extensions to a base workflow. Returns the composed WorkflowSpec.",inputSchema:G,execute:async(Q)=>{let _=J.get(Q.workflowKey);if(!_)return{success:!1,error:`Base workflow "${Q.workflowKey}" not found. Available: ${Array.from(J.keys()).join(", ")}`};let A=$??new mX;if(Q.extensions?.length)for(let N of Q.extensions)A.register({workflow:N.workflow,tenantId:N.tenantId,role:N.role,priority:N.priority,customSteps:N.customSteps,hiddenSteps:N.hiddenSteps});try{let N=A.compose({base:_,tenantId:Q.tenantId,role:Q.role});return{success:!0,workflow:N,meta:N.meta,stepIds:N.definition.steps.map((k)=>k.id)}}catch(N){return{success:!1,error:N instanceof Error?N.message:String(N)}}}}),H=B.object({workflowKey:B.string().describe("Workflow meta.key"),composedSteps:B.array(B.object({id:B.string(),type:B.enum(["human","automation","decision"]),label:B.string(),description:B.string().optional()})).optional().describe("Steps to include; if omitted, uses the base workflow")}),q=QX({description:"Generate TypeScript code for a workflow spec. Use after composing a workflow to output the spec as code the user can save.",inputSchema:H,execute:async(Q)=>{let _=J.get(Q.workflowKey);if(!_)return{success:!1,error:`Base workflow "${Q.workflowKey}" not found. Available: ${Array.from(J.keys()).join(", ")}`,code:null};let A=Q.composedSteps??_.definition.steps,N=lX((_.meta.key.split(".").pop()??"Workflow")+"")+"Workflow",k=A.map((U)=>` {
11
+ id: '${U.id}',
12
+ type: '${U.type}',
13
+ label: '${XX(U.label)}',${U.description?`
14
+ description: '${XX(U.description)}',`:""}
640
15
  }`).join(`,
641
- `);
642
- const meta = base.meta;
643
- const transitionsJson = JSON.stringify(base.definition.transitions, null, 6);
644
- const code = `import type { WorkflowSpec } from '@contractspec/lib.contracts-spec/workflow/spec';
16
+ `),P=_.meta,I=JSON.stringify(_.definition.transitions,null,6);return{success:!0,code:`import type { WorkflowSpec } from '@contractspec/lib.contracts-spec/workflow/spec';
645
17
 
646
18
  /**
647
- * Workflow: ${base.meta.key}
19
+ * Workflow: ${_.meta.key}
648
20
  * Generated via AI chat workflow tools.
649
21
  */
650
- export const ${specVarName}: WorkflowSpec = {
22
+ export const ${N}: WorkflowSpec = {
651
23
  meta: {
652
- key: '${base.meta.key}',
653
- version: '${String(base.meta.version)}',
654
- title: '${escapeString(meta.title ?? base.meta.key)}',
655
- description: '${escapeString(meta.description ?? "")}',
24
+ key: '${_.meta.key}',
25
+ version: '${String(_.meta.version)}',
26
+ title: '${XX(P.title??_.meta.key)}',
27
+ description: '${XX(P.description??"")}',
656
28
  },
657
29
  definition: {
658
- entryStepId: '${base.definition.entryStepId ?? base.definition.steps[0]?.id ?? ""}',
30
+ entryStepId: '${_.definition.entryStepId??_.definition.steps[0]?.id??""}',
659
31
  steps: [
660
- ${stepsCode}
32
+ ${k}
661
33
  ],
662
- transitions: ${transitionsJson},
34
+ transitions: ${I},
663
35
  },
664
36
  };
665
- `;
666
- return {
667
- success: true,
668
- code,
669
- workflowKey: input.workflowKey
670
- };
671
- }
672
- });
673
- return {
674
- create_workflow_extension: createWorkflowExtensionTool,
675
- compose_workflow: composeWorkflowTool,
676
- generate_workflow_spec_code: generateWorkflowSpecCodeTool
677
- };
678
- }
679
- function toPascalCase(value) {
680
- return value.split(/[-_.]/).filter(Boolean).map((part) => part.charAt(0).toUpperCase() + part.slice(1)).join("");
681
- }
682
- function escapeString(value) {
683
- return value.replace(/\\/g, "\\\\").replace(/'/g, "\\'");
684
- }
685
-
686
- // src/core/chat-service.ts
687
- var DEFAULT_SYSTEM_PROMPT = `You are ContractSpec AI, an expert coding assistant specialized in ContractSpec development.
37
+ `,workflowKey:Q.workflowKey}}});return{create_workflow_extension:Y,compose_workflow:j,generate_workflow_spec_code:q}}function lX(X){return X.split(/[-_.]/).filter(Boolean).map((Z)=>Z.charAt(0).toUpperCase()+Z.slice(1)).join("")}function XX(X){return X.replace(/\\/g,"\\\\").replace(/'/g,"\\'")}var aX=`You are ContractSpec AI, an expert coding assistant specialized in ContractSpec development.
688
38
 
689
39
  Your capabilities:
690
40
  - Help users create, modify, and understand ContractSpec specifications
@@ -697,971 +47,15 @@ Guidelines:
697
47
  - Provide code examples when helpful
698
48
  - Reference relevant ContractSpec concepts and patterns
699
49
  - Ask clarifying questions when the user's intent is unclear
700
- - When suggesting code changes, explain the rationale`;
701
- var WORKFLOW_TOOLS_PROMPT = `
702
-
703
- Workflow creation: You can create and modify workflows. Use create_workflow_extension when the user asks to add steps, change a workflow, or create a tenant-specific extension. Use compose_workflow to apply extensions to a base workflow. Use generate_workflow_spec_code to output TypeScript for the user to save.`;
50
+ - When suggesting code changes, explain the rationale`,rX=`
704
51
 
705
- class ChatService {
706
- provider;
707
- context;
708
- store;
709
- systemPrompt;
710
- maxHistoryMessages;
711
- onUsage;
712
- tools;
713
- thinkingLevel;
714
- sendReasoning;
715
- sendSources;
716
- modelSelector;
717
- constructor(config) {
718
- this.provider = config.provider;
719
- this.context = config.context;
720
- this.store = config.store ?? new InMemoryConversationStore;
721
- this.systemPrompt = this.buildSystemPrompt(config);
722
- this.maxHistoryMessages = config.maxHistoryMessages ?? 20;
723
- this.onUsage = config.onUsage;
724
- this.tools = this.mergeTools(config);
725
- this.thinkingLevel = config.thinkingLevel;
726
- this.modelSelector = config.modelSelector;
727
- this.sendReasoning = config.sendReasoning ?? (config.thinkingLevel != null && config.thinkingLevel !== "instant");
728
- this.sendSources = config.sendSources ?? false;
729
- }
730
- buildSystemPrompt(config) {
731
- let base = config.systemPrompt ?? DEFAULT_SYSTEM_PROMPT;
732
- if (config.workflowToolsConfig?.baseWorkflows?.length) {
733
- base += WORKFLOW_TOOLS_PROMPT;
734
- }
735
- const contractsPrompt = buildContractsContextPrompt(config.contractsContext ?? {});
736
- if (contractsPrompt) {
737
- base += contractsPrompt;
738
- }
739
- if (config.surfacePlanConfig?.plan) {
740
- const plannerInput = buildPlannerPromptInput(config.surfacePlanConfig.plan);
741
- base += `
52
+ Workflow creation: You can create and modify workflows. Use create_workflow_extension when the user asks to add steps, change a workflow, or create a tenant-specific extension. Use compose_workflow to apply extensions to a base workflow. Use generate_workflow_spec_code to output TypeScript for the user to save.`;class ZX{provider;context;store;systemPrompt;maxHistoryMessages;onUsage;tools;thinkingLevel;sendReasoning;sendSources;modelSelector;constructor(X){this.provider=X.provider,this.context=X.context,this.store=X.store??new o,this.systemPrompt=this.buildSystemPrompt(X),this.maxHistoryMessages=X.maxHistoryMessages??20,this.onUsage=X.onUsage,this.tools=this.mergeTools(X),this.thinkingLevel=X.thinkingLevel,this.modelSelector=X.modelSelector,this.sendReasoning=X.sendReasoning??(X.thinkingLevel!=null&&X.thinkingLevel!=="instant"),this.sendSources=X.sendSources??!1}buildSystemPrompt(X){let Z=X.systemPrompt??aX;if(X.workflowToolsConfig?.baseWorkflows?.length)Z+=rX;let $=BX(X.contractsContext??{});if($)Z+=$;if(X.surfacePlanConfig?.plan){let J=UX(X.surfacePlanConfig.plan);Z+=`
742
53
 
743
- ` + compilePlannerPrompt(plannerInput);
744
- }
745
- return base;
746
- }
747
- mergeTools(config) {
748
- let merged = config.tools ?? {};
749
- const wfConfig = config.workflowToolsConfig;
750
- if (wfConfig?.baseWorkflows?.length) {
751
- const workflowTools = createWorkflowTools({
752
- baseWorkflows: wfConfig.baseWorkflows,
753
- composer: wfConfig.composer
754
- });
755
- merged = { ...merged, ...workflowTools };
756
- }
757
- const contractsCtx = config.contractsContext;
758
- if (contractsCtx?.agentSpecs?.length) {
759
- const allTools = [];
760
- for (const agent of contractsCtx.agentSpecs) {
761
- if (agent.tools?.length)
762
- allTools.push(...agent.tools);
763
- }
764
- if (allTools.length > 0) {
765
- const agentTools = agentToolConfigsToToolSet(allTools);
766
- merged = { ...merged, ...agentTools };
767
- }
768
- }
769
- const surfaceConfig = config.surfacePlanConfig;
770
- if (surfaceConfig?.plan) {
771
- const plannerTools = createSurfacePlannerTools({
772
- plan: surfaceConfig.plan,
773
- onPatchProposal: surfaceConfig.onPatchProposal
774
- });
775
- merged = { ...merged, ...plannerTools };
776
- }
777
- if (config.mcpTools && Object.keys(config.mcpTools).length > 0) {
778
- merged = { ...merged, ...config.mcpTools };
779
- }
780
- return Object.keys(merged).length > 0 ? merged : undefined;
781
- }
782
- async resolveModel() {
783
- if (this.modelSelector) {
784
- const dimension = this.thinkingLevelToDimension(this.thinkingLevel);
785
- const { model, selection } = await this.modelSelector.selectAndCreate({
786
- taskDimension: dimension
787
- });
788
- return { model, providerName: selection.providerKey };
789
- }
790
- return {
791
- model: this.provider.getModel(),
792
- providerName: this.provider.name
793
- };
794
- }
795
- thinkingLevelToDimension(level) {
796
- if (!level || level === "instant")
797
- return "latency";
798
- return "reasoning";
799
- }
800
- async send(options) {
801
- let conversation;
802
- if (options.conversationId) {
803
- const existing = await this.store.get(options.conversationId);
804
- if (!existing) {
805
- throw new Error(`Conversation ${options.conversationId} not found`);
806
- }
807
- conversation = existing;
808
- } else {
809
- conversation = await this.store.create({
810
- status: "active",
811
- provider: this.provider.name,
812
- model: this.provider.model,
813
- messages: [],
814
- workspacePath: this.context?.workspacePath
815
- });
816
- }
817
- if (!options.skipUserAppend) {
818
- await this.store.appendMessage(conversation.id, {
819
- role: "user",
820
- content: options.content,
821
- status: "completed",
822
- attachments: options.attachments
823
- });
824
- }
825
- conversation = await this.store.get(conversation.id) ?? conversation;
826
- const messages = this.buildMessages(conversation, options);
827
- const { model, providerName } = await this.resolveModel();
828
- const providerOptions = getProviderOptions(this.thinkingLevel, providerName);
829
- try {
830
- const result = await generateText({
831
- model,
832
- messages,
833
- system: this.systemPrompt,
834
- tools: this.tools,
835
- providerOptions: Object.keys(providerOptions).length > 0 ? providerOptions : undefined
836
- });
837
- const assistantMessage = await this.store.appendMessage(conversation.id, {
838
- role: "assistant",
839
- content: result.text,
840
- status: "completed"
841
- });
842
- const updatedConversation = await this.store.get(conversation.id);
843
- if (!updatedConversation) {
844
- throw new Error("Conversation lost after update");
845
- }
846
- return {
847
- message: assistantMessage,
848
- conversation: updatedConversation
849
- };
850
- } catch (error) {
851
- await this.store.appendMessage(conversation.id, {
852
- role: "assistant",
853
- content: "",
854
- status: "error",
855
- error: {
856
- code: "generation_failed",
857
- message: error instanceof Error ? error.message : String(error)
858
- }
859
- });
860
- throw error;
861
- }
862
- }
863
- async stream(options) {
864
- let conversation;
865
- if (options.conversationId) {
866
- const existing = await this.store.get(options.conversationId);
867
- if (!existing) {
868
- throw new Error(`Conversation ${options.conversationId} not found`);
869
- }
870
- conversation = existing;
871
- } else {
872
- conversation = await this.store.create({
873
- status: "active",
874
- provider: this.provider.name,
875
- model: this.provider.model,
876
- messages: [],
877
- workspacePath: this.context?.workspacePath
878
- });
879
- }
880
- if (!options.skipUserAppend) {
881
- await this.store.appendMessage(conversation.id, {
882
- role: "user",
883
- content: options.content,
884
- status: "completed",
885
- attachments: options.attachments
886
- });
887
- }
888
- conversation = await this.store.get(conversation.id) ?? conversation;
889
- const assistantMessage = await this.store.appendMessage(conversation.id, {
890
- role: "assistant",
891
- content: "",
892
- status: "streaming"
893
- });
894
- const messages = this.buildMessages(conversation, options);
895
- const { model, providerName } = await this.resolveModel();
896
- const systemPrompt = this.systemPrompt;
897
- const tools = this.tools;
898
- const store = this.store;
899
- const onUsage = this.onUsage;
900
- const streamProviderOptions = getProviderOptions(this.thinkingLevel, providerName);
901
- async function* streamGenerator() {
902
- let fullContent = "";
903
- let fullReasoning = "";
904
- const toolCallsMap = new Map;
905
- const sources = [];
906
- try {
907
- const result = streamText({
908
- model,
909
- messages,
910
- system: systemPrompt,
911
- tools,
912
- providerOptions: Object.keys(streamProviderOptions).length > 0 ? streamProviderOptions : undefined
913
- });
914
- for await (const part of result.fullStream) {
915
- if (part.type === "text-delta") {
916
- const text = part.text ?? "";
917
- if (text) {
918
- fullContent += text;
919
- yield { type: "text", content: text };
920
- }
921
- } else if (part.type === "reasoning-delta") {
922
- const text = part.text ?? "";
923
- if (text) {
924
- fullReasoning += text;
925
- yield { type: "reasoning", content: text };
926
- }
927
- } else if (part.type === "source") {
928
- const src = part;
929
- const source = {
930
- id: src.id,
931
- title: src.title ?? "",
932
- url: src.url,
933
- type: "web"
934
- };
935
- sources.push(source);
936
- yield { type: "source", source };
937
- } else if (part.type === "tool-call") {
938
- const toolCall = {
939
- id: part.toolCallId,
940
- name: part.toolName,
941
- args: part.input ?? {},
942
- status: "running"
943
- };
944
- toolCallsMap.set(part.toolCallId, toolCall);
945
- yield { type: "tool_call", toolCall };
946
- } else if (part.type === "tool-result") {
947
- const tc = toolCallsMap.get(part.toolCallId);
948
- if (tc) {
949
- tc.result = part.output;
950
- tc.status = "completed";
951
- }
952
- yield {
953
- type: "tool_result",
954
- toolResult: {
955
- toolCallId: part.toolCallId,
956
- toolName: part.toolName,
957
- result: part.output
958
- }
959
- };
960
- } else if (part.type === "tool-error") {
961
- const tc = toolCallsMap.get(part.toolCallId);
962
- if (tc) {
963
- tc.status = "error";
964
- tc.error = part.error ?? "Tool execution failed";
965
- }
966
- } else if (part.type === "finish") {
967
- const usage = part.usage;
968
- const inputTokens = usage?.inputTokens ?? 0;
969
- const outputTokens = usage?.completionTokens ?? 0;
970
- await store.updateMessage(conversation.id, assistantMessage.id, {
971
- content: fullContent,
972
- status: "completed",
973
- reasoning: fullReasoning || undefined,
974
- sources: sources.length > 0 ? sources : undefined,
975
- toolCalls: toolCallsMap.size > 0 ? Array.from(toolCallsMap.values()) : undefined,
976
- usage: usage ? { inputTokens, outputTokens } : undefined
977
- });
978
- onUsage?.({ inputTokens, outputTokens });
979
- yield {
980
- type: "done",
981
- usage: usage ? { inputTokens, outputTokens } : undefined
982
- };
983
- return;
984
- }
985
- }
986
- await store.updateMessage(conversation.id, assistantMessage.id, {
987
- content: fullContent,
988
- status: "completed",
989
- reasoning: fullReasoning || undefined,
990
- sources: sources.length > 0 ? sources : undefined,
991
- toolCalls: toolCallsMap.size > 0 ? Array.from(toolCallsMap.values()) : undefined
992
- });
993
- yield { type: "done" };
994
- } catch (error) {
995
- await store.updateMessage(conversation.id, assistantMessage.id, {
996
- content: fullContent,
997
- status: "error",
998
- error: {
999
- code: "stream_failed",
1000
- message: error instanceof Error ? error.message : String(error)
1001
- }
1002
- });
1003
- yield {
1004
- type: "error",
1005
- error: {
1006
- code: "stream_failed",
1007
- message: error instanceof Error ? error.message : String(error)
1008
- }
1009
- };
1010
- }
1011
- }
1012
- return {
1013
- conversationId: conversation.id,
1014
- messageId: assistantMessage.id,
1015
- stream: streamGenerator()
1016
- };
1017
- }
1018
- async getConversation(conversationId) {
1019
- return this.store.get(conversationId);
1020
- }
1021
- async listConversations(options) {
1022
- return this.store.list({
1023
- status: "active",
1024
- ...options
1025
- });
1026
- }
1027
- async updateConversation(conversationId, updates) {
1028
- return this.store.update(conversationId, updates);
1029
- }
1030
- async forkConversation(conversationId, upToMessageId) {
1031
- return this.store.fork(conversationId, upToMessageId);
1032
- }
1033
- async updateMessage(conversationId, messageId, updates) {
1034
- return this.store.updateMessage(conversationId, messageId, updates);
1035
- }
1036
- async truncateAfter(conversationId, messageId) {
1037
- return this.store.truncateAfter(conversationId, messageId);
1038
- }
1039
- async deleteConversation(conversationId) {
1040
- return this.store.delete(conversationId);
1041
- }
1042
- buildMessages(conversation, _options) {
1043
- const historyStart = Math.max(0, conversation.messages.length - this.maxHistoryMessages);
1044
- const messages = [];
1045
- for (let i = historyStart;i < conversation.messages.length; i++) {
1046
- const msg = conversation.messages[i];
1047
- if (!msg)
1048
- continue;
1049
- if (msg.role === "user") {
1050
- let content = msg.content;
1051
- if (msg.attachments?.length) {
1052
- const attachmentInfo = msg.attachments.map((a) => {
1053
- if (a.type === "file" || a.type === "code") {
1054
- return `
54
+ `+iX(J)}return Z}mergeTools(X){let Z=X.tools??{},$=X.workflowToolsConfig;if($?.baseWorkflows?.length){let G=DX({baseWorkflows:$.baseWorkflows,composer:$.composer});Z={...Z,...G}}let J=X.contractsContext;if(J?.agentSpecs?.length){let G=[];for(let j of J.agentSpecs)if(j.tools?.length)G.push(...j.tools);if(G.length>0){let j=qX(G);Z={...Z,...j}}}let Y=X.surfacePlanConfig;if(Y?.plan){let G=NX({plan:Y.plan,onPatchProposal:Y.onPatchProposal});Z={...Z,...G}}if(X.mcpTools&&Object.keys(X.mcpTools).length>0)Z={...Z,...X.mcpTools};return Object.keys(Z).length>0?Z:void 0}async resolveModel(){if(this.modelSelector){let X=this.thinkingLevelToDimension(this.thinkingLevel),{model:Z,selection:$}=await this.modelSelector.selectAndCreate({taskDimension:X});return{model:Z,providerName:$.providerKey}}return{model:this.provider.getModel(),providerName:this.provider.name}}thinkingLevelToDimension(X){if(!X||X==="instant")return"latency";return"reasoning"}async send(X){let Z;if(X.conversationId){let j=await this.store.get(X.conversationId);if(!j)throw Error(`Conversation ${X.conversationId} not found`);Z=j}else Z=await this.store.create({status:"active",provider:this.provider.name,model:this.provider.model,messages:[],workspacePath:this.context?.workspacePath});if(!X.skipUserAppend)await this.store.appendMessage(Z.id,{role:"user",content:X.content,status:"completed",attachments:X.attachments});Z=await this.store.get(Z.id)??Z;let $=this.buildMessages(Z,X),{model:J,providerName:Y}=await this.resolveModel(),G=GX(this.thinkingLevel,Y);try{let j=await pX({model:J,messages:$,system:this.systemPrompt,tools:this.tools,providerOptions:Object.keys(G).length>0?G:void 0}),H=await this.store.appendMessage(Z.id,{role:"assistant",content:j.text,status:"completed"}),q=await this.store.get(Z.id);if(!q)throw Error("Conversation lost after update");return{message:H,conversation:q}}catch(j){throw await this.store.appendMessage(Z.id,{role:"assistant",content:"",status:"error",error:{code:"generation_failed",message:j instanceof Error?j.message:String(j)}}),j}}async stream(X){let Z;if(X.conversationId){let N=await this.store.get(X.conversationId);if(!N)throw Error(`Conversation ${X.conversationId} not found`);Z=N}else Z=await this.store.create({status:"active",provider:this.provider.name,model:this.provider.model,messages:[],workspacePath:this.context?.workspacePath});if(!X.skipUserAppend)await this.store.appendMessage(Z.id,{role:"user",content:X.content,status:"completed",attachments:X.attachments});Z=await this.store.get(Z.id)??Z;let $=await this.store.appendMessage(Z.id,{role:"assistant",content:"",status:"streaming"}),J=this.buildMessages(Z,X),{model:Y,providerName:G}=await this.resolveModel(),j=this.systemPrompt,H=this.tools,q=this.store,Q=this.onUsage,_=GX(this.thinkingLevel,G);async function*A(){let N="",k="",P=new Map,I=[];try{let h=cX({model:Y,messages:J,system:j,tools:H,providerOptions:Object.keys(_).length>0?_:void 0});for await(let U of h.fullStream)if(U.type==="text-delta"){let W=U.text??"";if(W)N+=W,yield{type:"text",content:W}}else if(U.type==="reasoning-delta"){let W=U.text??"";if(W)k+=W,yield{type:"reasoning",content:W}}else if(U.type==="source"){let W=U,f={id:W.id,title:W.title??"",url:W.url,type:"web"};I.push(f),yield{type:"source",source:f}}else if(U.type==="tool-call"){let W={id:U.toolCallId,name:U.toolName,args:U.input??{},status:"running"};P.set(U.toolCallId,W),yield{type:"tool_call",toolCall:W}}else if(U.type==="tool-result"){let W=P.get(U.toolCallId);if(W)W.result=U.output,W.status="completed";yield{type:"tool_result",toolResult:{toolCallId:U.toolCallId,toolName:U.toolName,result:U.output}}}else if(U.type==="tool-error"){let W=P.get(U.toolCallId);if(W)W.status="error",W.error=U.error??"Tool execution failed"}else if(U.type==="finish"){let W=U.usage,f=W?.inputTokens??0,g=W?.completionTokens??0;await q.updateMessage(Z.id,$.id,{content:N,status:"completed",reasoning:k||void 0,sources:I.length>0?I:void 0,toolCalls:P.size>0?Array.from(P.values()):void 0,usage:W?{inputTokens:f,outputTokens:g}:void 0}),Q?.({inputTokens:f,outputTokens:g}),yield{type:"done",usage:W?{inputTokens:f,outputTokens:g}:void 0};return}await q.updateMessage(Z.id,$.id,{content:N,status:"completed",reasoning:k||void 0,sources:I.length>0?I:void 0,toolCalls:P.size>0?Array.from(P.values()):void 0}),yield{type:"done"}}catch(h){await q.updateMessage(Z.id,$.id,{content:N,status:"error",error:{code:"stream_failed",message:h instanceof Error?h.message:String(h)}}),yield{type:"error",error:{code:"stream_failed",message:h instanceof Error?h.message:String(h)}}}}return{conversationId:Z.id,messageId:$.id,stream:A()}}async getConversation(X){return this.store.get(X)}async listConversations(X){return this.store.list({status:"active",...X})}async updateConversation(X,Z){return this.store.update(X,Z)}async forkConversation(X,Z){return this.store.fork(X,Z)}async updateMessage(X,Z,$){return this.store.updateMessage(X,Z,$)}async truncateAfter(X,Z){return this.store.truncateAfter(X,Z)}async deleteConversation(X){return this.store.delete(X)}buildMessages(X,Z){let $=Math.max(0,X.messages.length-this.maxHistoryMessages),J=[];for(let Y=$;Y<X.messages.length;Y++){let G=X.messages[Y];if(!G)continue;if(G.role==="user"){let j=G.content;if(G.attachments?.length){let H=G.attachments.map((q)=>{if(q.type==="file"||q.type==="code")return`
1055
55
 
1056
- ### ${a.name}
56
+ ### ${q.name}
1057
57
  \`\`\`
1058
- ${a.content ?? ""}
1059
- \`\`\``;
1060
- }
1061
- return `
58
+ ${q.content??""}
59
+ \`\`\``;return`
1062
60
 
1063
- [Attachment: ${a.name}]`;
1064
- }).join("");
1065
- content += attachmentInfo;
1066
- }
1067
- messages.push({ role: "user", content });
1068
- } else if (msg.role === "assistant") {
1069
- if (msg.toolCalls?.length) {
1070
- messages.push({
1071
- role: "assistant",
1072
- content: msg.content || "",
1073
- toolCalls: msg.toolCalls.map((tc) => ({
1074
- type: "tool-call",
1075
- toolCallId: tc.id,
1076
- toolName: tc.name,
1077
- args: tc.args
1078
- }))
1079
- });
1080
- messages.push({
1081
- role: "tool",
1082
- content: msg.toolCalls.map((tc) => ({
1083
- type: "tool-result",
1084
- toolCallId: tc.id,
1085
- toolName: tc.name,
1086
- output: tc.result
1087
- }))
1088
- });
1089
- } else {
1090
- messages.push({ role: "assistant", content: msg.content });
1091
- }
1092
- }
1093
- }
1094
- return messages;
1095
- }
1096
- }
1097
- function createChatService(config) {
1098
- return new ChatService(config);
1099
- }
1100
-
1101
- // src/presentation/hooks/useChat.tsx
1102
- "use client";
1103
- function toolsToToolSet(defs) {
1104
- const result = {};
1105
- for (const def of defs) {
1106
- result[def.name] = tool4({
1107
- description: def.description ?? def.name,
1108
- inputSchema: z4.object({}).passthrough(),
1109
- execute: async () => ({})
1110
- });
1111
- }
1112
- return result;
1113
- }
1114
- function useChat(options = {}) {
1115
- const {
1116
- provider = "openai",
1117
- mode = "byok",
1118
- model,
1119
- apiKey,
1120
- proxyUrl,
1121
- conversationId: initialConversationId,
1122
- store,
1123
- systemPrompt,
1124
- streaming = true,
1125
- onSend,
1126
- onResponse,
1127
- onError,
1128
- onUsage,
1129
- tools: toolsDefs,
1130
- thinkingLevel,
1131
- workflowToolsConfig,
1132
- modelSelector,
1133
- contractsContext,
1134
- surfacePlanConfig,
1135
- mcpServers,
1136
- agentMode
1137
- } = options;
1138
- const [messages, setMessages] = React.useState([]);
1139
- const [mcpTools, setMcpTools] = React.useState(null);
1140
- const mcpCleanupRef = React.useRef(null);
1141
- const [conversation, setConversation] = React.useState(null);
1142
- const [isLoading, setIsLoading] = React.useState(false);
1143
- const [error, setError] = React.useState(null);
1144
- const [conversationId, setConversationId] = React.useState(initialConversationId ?? null);
1145
- const abortControllerRef = React.useRef(null);
1146
- const chatServiceRef = React.useRef(null);
1147
- React.useEffect(() => {
1148
- if (!mcpServers?.length) {
1149
- setMcpTools(null);
1150
- return;
1151
- }
1152
- let cancelled = false;
1153
- import("@contractspec/lib.ai-agent/tools/mcp-client.browser").then(({ createMcpToolsets }) => {
1154
- createMcpToolsets(mcpServers).then(({ tools, cleanup }) => {
1155
- if (!cancelled) {
1156
- setMcpTools(tools);
1157
- mcpCleanupRef.current = cleanup;
1158
- } else {
1159
- cleanup().catch(() => {
1160
- return;
1161
- });
1162
- }
1163
- }).catch(() => {
1164
- if (!cancelled)
1165
- setMcpTools(null);
1166
- });
1167
- });
1168
- return () => {
1169
- cancelled = true;
1170
- const cleanup = mcpCleanupRef.current;
1171
- mcpCleanupRef.current = null;
1172
- if (cleanup)
1173
- cleanup().catch(() => {
1174
- return;
1175
- });
1176
- setMcpTools(null);
1177
- };
1178
- }, [mcpServers]);
1179
- React.useEffect(() => {
1180
- const chatProvider = createProvider({
1181
- provider,
1182
- model,
1183
- apiKey,
1184
- proxyUrl
1185
- });
1186
- chatServiceRef.current = new ChatService({
1187
- provider: chatProvider,
1188
- store,
1189
- systemPrompt,
1190
- onUsage,
1191
- tools: toolsDefs?.length ? toolsToToolSet(toolsDefs) : undefined,
1192
- thinkingLevel,
1193
- workflowToolsConfig,
1194
- modelSelector,
1195
- contractsContext,
1196
- surfacePlanConfig,
1197
- mcpTools
1198
- });
1199
- }, [
1200
- provider,
1201
- mode,
1202
- model,
1203
- apiKey,
1204
- proxyUrl,
1205
- store,
1206
- systemPrompt,
1207
- onUsage,
1208
- toolsDefs,
1209
- thinkingLevel,
1210
- workflowToolsConfig,
1211
- modelSelector,
1212
- contractsContext,
1213
- surfacePlanConfig,
1214
- mcpTools
1215
- ]);
1216
- React.useEffect(() => {
1217
- if (!conversationId || !chatServiceRef.current)
1218
- return;
1219
- const loadConversation = async () => {
1220
- if (!chatServiceRef.current)
1221
- return;
1222
- const conv = await chatServiceRef.current.getConversation(conversationId);
1223
- if (conv) {
1224
- setConversation(conv);
1225
- setMessages(conv.messages);
1226
- }
1227
- };
1228
- loadConversation().catch(console.error);
1229
- }, [conversationId]);
1230
- const sendMessage = React.useCallback(async (content, attachments, opts) => {
1231
- if (agentMode?.agent) {
1232
- setIsLoading(true);
1233
- setError(null);
1234
- abortControllerRef.current = new AbortController;
1235
- try {
1236
- if (!opts?.skipUserAppend) {
1237
- const userMessage = {
1238
- id: `msg_${Date.now()}`,
1239
- conversationId: conversationId ?? "",
1240
- role: "user",
1241
- content,
1242
- status: "completed",
1243
- createdAt: new Date,
1244
- updatedAt: new Date,
1245
- attachments
1246
- };
1247
- setMessages((prev) => [...prev, userMessage]);
1248
- onSend?.(userMessage);
1249
- }
1250
- const result = await agentMode.agent.generate({
1251
- prompt: content,
1252
- signal: abortControllerRef.current.signal
1253
- });
1254
- const toolCallsMap = new Map;
1255
- for (const tc of result.toolCalls ?? []) {
1256
- const tr = result.toolResults?.find((r) => r.toolCallId === tc.toolCallId);
1257
- toolCallsMap.set(tc.toolCallId, {
1258
- id: tc.toolCallId,
1259
- name: tc.toolName,
1260
- args: tc.args ?? {},
1261
- result: tr?.output,
1262
- status: "completed"
1263
- });
1264
- }
1265
- const assistantMessage = {
1266
- id: `msg_${Date.now()}_a`,
1267
- conversationId: conversationId ?? "",
1268
- role: "assistant",
1269
- content: result.text,
1270
- status: "completed",
1271
- createdAt: new Date,
1272
- updatedAt: new Date,
1273
- toolCalls: toolCallsMap.size ? Array.from(toolCallsMap.values()) : undefined,
1274
- usage: result.usage
1275
- };
1276
- setMessages((prev) => [...prev, assistantMessage]);
1277
- onResponse?.(assistantMessage);
1278
- onUsage?.(result.usage ?? { inputTokens: 0, outputTokens: 0 });
1279
- if (store && !conversationId) {
1280
- const conv = await store.create({
1281
- status: "active",
1282
- provider: "agent",
1283
- model: "agent",
1284
- messages: []
1285
- });
1286
- if (!opts?.skipUserAppend) {
1287
- await store.appendMessage(conv.id, {
1288
- role: "user",
1289
- content,
1290
- status: "completed",
1291
- attachments
1292
- });
1293
- }
1294
- await store.appendMessage(conv.id, {
1295
- role: "assistant",
1296
- content: result.text,
1297
- status: "completed",
1298
- toolCalls: assistantMessage.toolCalls,
1299
- usage: result.usage
1300
- });
1301
- const updated = await store.get(conv.id);
1302
- if (updated)
1303
- setConversation(updated);
1304
- setConversationId(conv.id);
1305
- }
1306
- } catch (err) {
1307
- setError(err instanceof Error ? err : new Error(String(err)));
1308
- onError?.(err instanceof Error ? err : new Error(String(err)));
1309
- } finally {
1310
- setIsLoading(false);
1311
- }
1312
- return;
1313
- }
1314
- if (!chatServiceRef.current) {
1315
- throw new Error("Chat service not initialized");
1316
- }
1317
- setIsLoading(true);
1318
- setError(null);
1319
- abortControllerRef.current = new AbortController;
1320
- try {
1321
- if (!opts?.skipUserAppend) {
1322
- const userMessage = {
1323
- id: `msg_${Date.now()}`,
1324
- conversationId: conversationId ?? "",
1325
- role: "user",
1326
- content,
1327
- status: "completed",
1328
- createdAt: new Date,
1329
- updatedAt: new Date,
1330
- attachments
1331
- };
1332
- setMessages((prev) => [...prev, userMessage]);
1333
- onSend?.(userMessage);
1334
- }
1335
- if (streaming) {
1336
- const result = await chatServiceRef.current.stream({
1337
- conversationId: conversationId ?? undefined,
1338
- content,
1339
- attachments,
1340
- skipUserAppend: opts?.skipUserAppend
1341
- });
1342
- if (!conversationId && !opts?.skipUserAppend) {
1343
- setConversationId(result.conversationId);
1344
- }
1345
- const assistantMessage = {
1346
- id: result.messageId,
1347
- conversationId: result.conversationId,
1348
- role: "assistant",
1349
- content: "",
1350
- status: "streaming",
1351
- createdAt: new Date,
1352
- updatedAt: new Date
1353
- };
1354
- setMessages((prev) => [...prev, assistantMessage]);
1355
- let fullContent = "";
1356
- let fullReasoning = "";
1357
- const toolCallsMap = new Map;
1358
- const sources = [];
1359
- for await (const chunk of result.stream) {
1360
- if (chunk.type === "text" && chunk.content) {
1361
- fullContent += chunk.content;
1362
- setMessages((prev) => prev.map((m) => m.id === result.messageId ? {
1363
- ...m,
1364
- content: fullContent,
1365
- reasoning: fullReasoning || undefined,
1366
- sources: sources.length ? sources : undefined,
1367
- toolCalls: toolCallsMap.size ? Array.from(toolCallsMap.values()) : undefined
1368
- } : m));
1369
- } else if (chunk.type === "reasoning" && chunk.content) {
1370
- fullReasoning += chunk.content;
1371
- setMessages((prev) => prev.map((m) => m.id === result.messageId ? { ...m, reasoning: fullReasoning } : m));
1372
- } else if (chunk.type === "source" && chunk.source) {
1373
- sources.push(chunk.source);
1374
- setMessages((prev) => prev.map((m) => m.id === result.messageId ? { ...m, sources: [...sources] } : m));
1375
- } else if (chunk.type === "tool_call" && chunk.toolCall) {
1376
- const tc = chunk.toolCall;
1377
- const chatTc = {
1378
- id: tc.id,
1379
- name: tc.name,
1380
- args: tc.args,
1381
- status: "running"
1382
- };
1383
- toolCallsMap.set(tc.id, chatTc);
1384
- setMessages((prev) => prev.map((m) => m.id === result.messageId ? { ...m, toolCalls: Array.from(toolCallsMap.values()) } : m));
1385
- } else if (chunk.type === "tool_result" && chunk.toolResult) {
1386
- const tr = chunk.toolResult;
1387
- const tc = toolCallsMap.get(tr.toolCallId);
1388
- if (tc) {
1389
- tc.result = tr.result;
1390
- tc.status = "completed";
1391
- }
1392
- setMessages((prev) => prev.map((m) => m.id === result.messageId ? { ...m, toolCalls: Array.from(toolCallsMap.values()) } : m));
1393
- } else if (chunk.type === "done") {
1394
- setMessages((prev) => prev.map((m) => m.id === result.messageId ? {
1395
- ...m,
1396
- content: fullContent,
1397
- reasoning: fullReasoning || undefined,
1398
- sources: sources.length ? sources : undefined,
1399
- toolCalls: toolCallsMap.size ? Array.from(toolCallsMap.values()) : undefined,
1400
- status: "completed",
1401
- usage: chunk.usage,
1402
- updatedAt: new Date
1403
- } : m));
1404
- onResponse?.(messages.find((m) => m.id === result.messageId) ?? assistantMessage);
1405
- } else if (chunk.type === "error") {
1406
- setMessages((prev) => prev.map((m) => m.id === result.messageId ? {
1407
- ...m,
1408
- status: "error",
1409
- error: chunk.error,
1410
- updatedAt: new Date
1411
- } : m));
1412
- if (chunk.error) {
1413
- const err = new Error(chunk.error.message);
1414
- setError(err);
1415
- onError?.(err);
1416
- }
1417
- }
1418
- }
1419
- } else {
1420
- const result = await chatServiceRef.current.send({
1421
- conversationId: conversationId ?? undefined,
1422
- content,
1423
- attachments,
1424
- skipUserAppend: opts?.skipUserAppend
1425
- });
1426
- setConversation(result.conversation);
1427
- setMessages(result.conversation.messages);
1428
- if (!conversationId) {
1429
- setConversationId(result.conversation.id);
1430
- }
1431
- onResponse?.(result.message);
1432
- }
1433
- } catch (err) {
1434
- const error2 = err instanceof Error ? err : new Error(String(err));
1435
- setError(error2);
1436
- onError?.(error2);
1437
- } finally {
1438
- setIsLoading(false);
1439
- abortControllerRef.current = null;
1440
- }
1441
- }, [
1442
- conversationId,
1443
- streaming,
1444
- onSend,
1445
- onResponse,
1446
- onError,
1447
- onUsage,
1448
- messages,
1449
- agentMode,
1450
- store
1451
- ]);
1452
- const clearConversation = React.useCallback(() => {
1453
- setMessages([]);
1454
- setConversation(null);
1455
- setConversationId(null);
1456
- setError(null);
1457
- }, []);
1458
- const regenerate = React.useCallback(async () => {
1459
- let lastUserMessageIndex = -1;
1460
- for (let i = messages.length - 1;i >= 0; i--) {
1461
- const m = messages[i];
1462
- if (m?.role === "user") {
1463
- lastUserMessageIndex = i;
1464
- break;
1465
- }
1466
- }
1467
- if (lastUserMessageIndex === -1)
1468
- return;
1469
- const lastUserMessage = messages[lastUserMessageIndex];
1470
- if (!lastUserMessage)
1471
- return;
1472
- setMessages((prev) => prev.slice(0, lastUserMessageIndex + 1));
1473
- await sendMessage(lastUserMessage.content, lastUserMessage.attachments);
1474
- }, [messages, sendMessage]);
1475
- const stop = React.useCallback(() => {
1476
- abortControllerRef.current?.abort();
1477
- setIsLoading(false);
1478
- }, []);
1479
- const createNewConversation = clearConversation;
1480
- const editMessage = React.useCallback(async (messageId, newContent) => {
1481
- if (!chatServiceRef.current || !conversationId)
1482
- return;
1483
- const msg = messages.find((m) => m.id === messageId);
1484
- if (!msg || msg.role !== "user")
1485
- return;
1486
- await chatServiceRef.current.updateMessage(conversationId, messageId, {
1487
- content: newContent
1488
- });
1489
- const truncated = await chatServiceRef.current.truncateAfter(conversationId, messageId);
1490
- if (truncated) {
1491
- setMessages(truncated.messages);
1492
- }
1493
- await sendMessage(newContent, undefined, { skipUserAppend: true });
1494
- }, [conversationId, messages, sendMessage]);
1495
- const forkConversation = React.useCallback(async (upToMessageId) => {
1496
- if (!chatServiceRef.current)
1497
- return null;
1498
- const idToFork = conversationId ?? conversation?.id;
1499
- if (!idToFork)
1500
- return null;
1501
- try {
1502
- const forked = await chatServiceRef.current.forkConversation(idToFork, upToMessageId);
1503
- setConversationId(forked.id);
1504
- setConversation(forked);
1505
- setMessages(forked.messages);
1506
- return forked.id;
1507
- } catch {
1508
- return null;
1509
- }
1510
- }, [conversationId, conversation]);
1511
- const updateConversationFn = React.useCallback(async (updates) => {
1512
- if (!chatServiceRef.current || !conversationId)
1513
- return null;
1514
- const updated = await chatServiceRef.current.updateConversation(conversationId, updates);
1515
- if (updated)
1516
- setConversation(updated);
1517
- return updated;
1518
- }, [conversationId]);
1519
- const addToolApprovalResponse = React.useCallback((_toolCallId, _result) => {
1520
- throw new Error(`addToolApprovalResponse: Tool approval requires server route with toUIMessageStreamResponse. ` + `Use createChatRoute and @ai-sdk/react useChat for tools with requireApproval.`);
1521
- }, []);
1522
- const hasApprovalTools = toolsDefs?.some((t) => t.requireApproval) ?? false;
1523
- return {
1524
- messages,
1525
- conversation,
1526
- isLoading,
1527
- error,
1528
- sendMessage,
1529
- clearConversation,
1530
- setConversationId,
1531
- regenerate,
1532
- stop,
1533
- createNewConversation,
1534
- editMessage,
1535
- forkConversation,
1536
- updateConversation: updateConversationFn,
1537
- ...hasApprovalTools && { addToolApprovalResponse }
1538
- };
1539
- }
1540
- // src/presentation/hooks/useConversations.ts
1541
- import * as React2 from "react";
1542
- "use client";
1543
- function useConversations(options) {
1544
- const { store, projectId, tags, limit = 50 } = options;
1545
- const [conversations, setConversations] = React2.useState([]);
1546
- const [isLoading, setIsLoading] = React2.useState(true);
1547
- const refresh = React2.useCallback(async () => {
1548
- setIsLoading(true);
1549
- try {
1550
- const list = await store.list({
1551
- status: "active",
1552
- projectId,
1553
- tags,
1554
- limit
1555
- });
1556
- setConversations(list);
1557
- } finally {
1558
- setIsLoading(false);
1559
- }
1560
- }, [store, projectId, tags, limit]);
1561
- React2.useEffect(() => {
1562
- refresh();
1563
- }, [refresh]);
1564
- const deleteConversation = React2.useCallback(async (id) => {
1565
- const ok = await store.delete(id);
1566
- if (ok) {
1567
- setConversations((prev) => prev.filter((c) => c.id !== id));
1568
- }
1569
- return ok;
1570
- }, [store]);
1571
- return {
1572
- conversations,
1573
- isLoading,
1574
- refresh,
1575
- deleteConversation
1576
- };
1577
- }
1578
- // src/presentation/hooks/useMessageSelection.ts
1579
- import * as React3 from "react";
1580
- "use client";
1581
- function useMessageSelection(messageIds) {
1582
- const [selectedIds, setSelectedIds] = React3.useState(() => new Set);
1583
- const idSet = React3.useMemo(() => new Set(messageIds), [messageIds.join(",")]);
1584
- React3.useEffect(() => {
1585
- setSelectedIds((prev) => {
1586
- const next = new Set;
1587
- for (const id of prev) {
1588
- if (idSet.has(id))
1589
- next.add(id);
1590
- }
1591
- return next.size === prev.size ? prev : next;
1592
- });
1593
- }, [idSet]);
1594
- const toggle = React3.useCallback((id) => {
1595
- setSelectedIds((prev) => {
1596
- const next = new Set(prev);
1597
- if (next.has(id))
1598
- next.delete(id);
1599
- else
1600
- next.add(id);
1601
- return next;
1602
- });
1603
- }, []);
1604
- const selectAll = React3.useCallback(() => {
1605
- setSelectedIds(new Set(messageIds));
1606
- }, [messageIds.join(",")]);
1607
- const clearSelection = React3.useCallback(() => {
1608
- setSelectedIds(new Set);
1609
- }, []);
1610
- const isSelected = React3.useCallback((id) => selectedIds.has(id), [selectedIds]);
1611
- const selectedCount = selectedIds.size;
1612
- return {
1613
- selectedIds,
1614
- toggle,
1615
- selectAll,
1616
- clearSelection,
1617
- isSelected,
1618
- selectedCount
1619
- };
1620
- }
1621
- // src/presentation/hooks/useProviders.tsx
1622
- import {
1623
- getAvailableProviders,
1624
- getModelsForProvider
1625
- } from "@contractspec/lib.ai-providers";
1626
- import * as React4 from "react";
1627
- "use client";
1628
- function useProviders() {
1629
- const [providers, setProviders] = React4.useState([]);
1630
- const [isLoading, setIsLoading] = React4.useState(true);
1631
- const loadProviders = React4.useCallback(async () => {
1632
- setIsLoading(true);
1633
- try {
1634
- const available = getAvailableProviders();
1635
- const providersWithModels = available.map((p) => ({
1636
- ...p,
1637
- models: getModelsForProvider(p.provider)
1638
- }));
1639
- setProviders(providersWithModels);
1640
- } catch (error) {
1641
- console.error("Failed to load providers:", error);
1642
- } finally {
1643
- setIsLoading(false);
1644
- }
1645
- }, []);
1646
- React4.useEffect(() => {
1647
- loadProviders();
1648
- }, [loadProviders]);
1649
- const availableProviders = React4.useMemo(() => providers.filter((p) => p.available), [providers]);
1650
- const isAvailable = React4.useCallback((provider) => providers.some((p) => p.provider === provider && p.available), [providers]);
1651
- const getModelsCallback = React4.useCallback((provider) => providers.find((p) => p.provider === provider)?.models ?? [], [providers]);
1652
- return {
1653
- providers,
1654
- availableProviders,
1655
- isAvailable,
1656
- getModels: getModelsCallback,
1657
- isLoading,
1658
- refresh: loadProviders
1659
- };
1660
- }
1661
- export {
1662
- useProviders,
1663
- useMessageSelection,
1664
- useConversations,
1665
- useCompletion,
1666
- useChat
1667
- };
61
+ [Attachment: ${q.name}]`}).join("");j+=H}J.push({role:"user",content:j})}else if(G.role==="assistant")if(G.toolCalls?.length)J.push({role:"assistant",content:G.content||"",toolCalls:G.toolCalls.map((j)=>({type:"tool-call",toolCallId:j.id,toolName:j.name,args:j.args}))}),J.push({role:"tool",content:G.toolCalls.map((j)=>({type:"tool-result",toolCallId:j.id,toolName:j.name,output:j.result}))});else J.push({role:"assistant",content:G.content})}return J}}function SZ(X){return new ZX(X)}function oX(X){let Z={};for(let $ of X)Z[$.name]=tX({description:$.description??$.name,inputSchema:sX.object({}).passthrough(),execute:async()=>({})});return Z}function eX(X={}){let{provider:Z="openai",mode:$="byok",model:J,apiKey:Y,proxyUrl:G,conversationId:j,store:H,systemPrompt:q,streaming:Q=!0,onSend:_,onResponse:A,onError:N,onUsage:k,tools:P,thinkingLevel:I,workflowToolsConfig:h,modelSelector:U,contractsContext:W,surfacePlanConfig:f,mcpServers:g,agentMode:$X}=X,[m,x]=E.useState([]),[VX,t]=E.useState(null),JX=E.useRef(null),[YX,p]=E.useState(null),[FX,a]=E.useState(!1),[AX,c]=E.useState(null),[w,i]=E.useState(j??null),r=E.useRef(null),T=E.useRef(null);E.useEffect(()=>{if(!g?.length){t(null);return}let D=!1;return import("@contractspec/lib.ai-agent/tools/mcp-client.browser").then(({createMcpToolsets:F})=>{F(g).then(({tools:L,cleanup:V})=>{if(!D)t(L),JX.current=V;else V().catch(()=>{return})}).catch(()=>{if(!D)t(null)})}),()=>{D=!0;let F=JX.current;if(JX.current=null,F)F().catch(()=>{return});t(null)}},[g]),E.useEffect(()=>{let D=nX({provider:Z,model:J,apiKey:Y,proxyUrl:G});T.current=new ZX({provider:D,store:H,systemPrompt:q,onUsage:k,tools:P?.length?oX(P):void 0,thinkingLevel:I,workflowToolsConfig:h,modelSelector:U,contractsContext:W,surfacePlanConfig:f,mcpTools:VX})},[Z,$,J,Y,G,H,q,k,P,I,h,U,W,f,VX]),E.useEffect(()=>{if(!w||!T.current)return;(async()=>{if(!T.current)return;let F=await T.current.getConversation(w);if(F)p(F),x(F.messages)})().catch(console.error)},[w]);let n=E.useCallback(async(D,F,L)=>{if($X?.agent){a(!0),c(null),r.current=new AbortController;try{if(!L?.skipUserAppend){let R={id:`msg_${Date.now()}`,conversationId:w??"",role:"user",content:D,status:"completed",createdAt:new Date,updatedAt:new Date,attachments:F};x((b)=>[...b,R]),_?.(R)}let V=await $X.agent.generate({prompt:D,signal:r.current.signal}),S=new Map;for(let R of V.toolCalls??[]){let b=V.toolResults?.find((u)=>u.toolCallId===R.toolCallId);S.set(R.toolCallId,{id:R.toolCallId,name:R.toolName,args:R.args??{},result:b?.output,status:"completed"})}let d={id:`msg_${Date.now()}_a`,conversationId:w??"",role:"assistant",content:V.text,status:"completed",createdAt:new Date,updatedAt:new Date,toolCalls:S.size?Array.from(S.values()):void 0,usage:V.usage};if(x((R)=>[...R,d]),A?.(d),k?.(V.usage??{inputTokens:0,outputTokens:0}),H&&!w){let R=await H.create({status:"active",provider:"agent",model:"agent",messages:[]});if(!L?.skipUserAppend)await H.appendMessage(R.id,{role:"user",content:D,status:"completed",attachments:F});await H.appendMessage(R.id,{role:"assistant",content:V.text,status:"completed",toolCalls:d.toolCalls,usage:V.usage});let b=await H.get(R.id);if(b)p(b);i(R.id)}}catch(V){c(V instanceof Error?V:Error(String(V))),N?.(V instanceof Error?V:Error(String(V)))}finally{a(!1)}return}if(!T.current)throw Error("Chat service not initialized");a(!0),c(null),r.current=new AbortController;try{if(!L?.skipUserAppend){let V={id:`msg_${Date.now()}`,conversationId:w??"",role:"user",content:D,status:"completed",createdAt:new Date,updatedAt:new Date,attachments:F};x((S)=>[...S,V]),_?.(V)}if(Q){let V=await T.current.stream({conversationId:w??void 0,content:D,attachments:F,skipUserAppend:L?.skipUserAppend});if(!w&&!L?.skipUserAppend)i(V.conversationId);let S={id:V.messageId,conversationId:V.conversationId,role:"assistant",content:"",status:"streaming",createdAt:new Date,updatedAt:new Date};x((y)=>[...y,S]);let d="",R="",b=new Map,u=[];for await(let y of V.stream)if(y.type==="text"&&y.content)d+=y.content,x((z)=>z.map((O)=>O.id===V.messageId?{...O,content:d,reasoning:R||void 0,sources:u.length?u:void 0,toolCalls:b.size?Array.from(b.values()):void 0}:O));else if(y.type==="reasoning"&&y.content)R+=y.content,x((z)=>z.map((O)=>O.id===V.messageId?{...O,reasoning:R}:O));else if(y.type==="source"&&y.source)u.push(y.source),x((z)=>z.map((O)=>O.id===V.messageId?{...O,sources:[...u]}:O));else if(y.type==="tool_call"&&y.toolCall){let z=y.toolCall,O={id:z.id,name:z.name,args:z.args,status:"running"};b.set(z.id,O),x((jX)=>jX.map((l)=>l.id===V.messageId?{...l,toolCalls:Array.from(b.values())}:l))}else if(y.type==="tool_result"&&y.toolResult){let z=y.toolResult,O=b.get(z.toolCallId);if(O)O.result=z.result,O.status="completed";x((jX)=>jX.map((l)=>l.id===V.messageId?{...l,toolCalls:Array.from(b.values())}:l))}else if(y.type==="done")x((z)=>z.map((O)=>O.id===V.messageId?{...O,content:d,reasoning:R||void 0,sources:u.length?u:void 0,toolCalls:b.size?Array.from(b.values()):void 0,status:"completed",usage:y.usage,updatedAt:new Date}:O)),A?.(m.find((z)=>z.id===V.messageId)??S);else if(y.type==="error"){if(x((z)=>z.map((O)=>O.id===V.messageId?{...O,status:"error",error:y.error,updatedAt:new Date}:O)),y.error){let z=Error(y.error.message);c(z),N?.(z)}}}else{let V=await T.current.send({conversationId:w??void 0,content:D,attachments:F,skipUserAppend:L?.skipUserAppend});if(p(V.conversation),x(V.conversation.messages),!w)i(V.conversation.id);A?.(V.message)}}catch(V){let S=V instanceof Error?V:Error(String(V));c(S),N?.(S)}finally{a(!1),r.current=null}},[w,Q,_,A,N,k,m,$X,H]),HX=E.useCallback(()=>{x([]),p(null),i(null),c(null)},[]),KX=E.useCallback(async()=>{let D=-1;for(let L=m.length-1;L>=0;L--)if(m[L]?.role==="user"){D=L;break}if(D===-1)return;let F=m[D];if(!F)return;x((L)=>L.slice(0,D+1)),await n(F.content,F.attachments)},[m,n]),OX=E.useCallback(()=>{r.current?.abort(),a(!1)},[]),EX=HX,LX=E.useCallback(async(D,F)=>{if(!T.current||!w)return;let L=m.find((S)=>S.id===D);if(!L||L.role!=="user")return;await T.current.updateMessage(w,D,{content:F});let V=await T.current.truncateAfter(w,D);if(V)x(V.messages);await n(F,void 0,{skipUserAppend:!0})},[w,m,n]),yX=E.useCallback(async(D)=>{if(!T.current)return null;let F=w??YX?.id;if(!F)return null;try{let L=await T.current.forkConversation(F,D);return i(L.id),p(L),x(L.messages),L.id}catch{return null}},[w,YX]),zX=E.useCallback(async(D)=>{if(!T.current||!w)return null;let F=await T.current.updateConversation(w,D);if(F)p(F);return F},[w]),wX=E.useCallback((D,F)=>{throw Error("addToolApprovalResponse: Tool approval requires server route with toUIMessageStreamResponse. Use createChatRoute and @ai-sdk/react useChat for tools with requireApproval.")},[]),RX=P?.some((D)=>D.requireApproval)??!1;return{messages:m,conversation:YX,isLoading:FX,error:AX,sendMessage:n,clearConversation:HX,setConversationId:i,regenerate:KX,stop:OX,createNewConversation:EX,editMessage:LX,forkConversation:yX,updateConversation:zX,...RX&&{addToolApprovalResponse:wX}}}import*as v from"react";function XZ(X){let{store:Z,projectId:$,tags:J,limit:Y=50}=X,[G,j]=v.useState([]),[H,q]=v.useState(!0),Q=v.useCallback(async()=>{q(!0);try{let A=await Z.list({status:"active",projectId:$,tags:J,limit:Y});j(A)}finally{q(!1)}},[Z,$,J,Y]);v.useEffect(()=>{Q()},[Q]);let _=v.useCallback(async(A)=>{let N=await Z.delete(A);if(N)j((k)=>k.filter((P)=>P.id!==A));return N},[Z]);return{conversations:G,isLoading:H,refresh:Q,deleteConversation:_}}import*as C from"react";function ZZ(X){let[Z,$]=C.useState(()=>new Set),J=C.useMemo(()=>new Set(X),[X.join(",")]);C.useEffect(()=>{$((Q)=>{let _=new Set;for(let A of Q)if(J.has(A))_.add(A);return _.size===Q.size?Q:_})},[J]);let Y=C.useCallback((Q)=>{$((_)=>{let A=new Set(_);if(A.has(Q))A.delete(Q);else A.add(Q);return A})},[]),G=C.useCallback(()=>{$(new Set(X))},[X.join(",")]),j=C.useCallback(()=>{$(new Set)},[]),H=C.useCallback((Q)=>Z.has(Q),[Z]),q=Z.size;return{selectedIds:Z,toggle:Y,selectAll:G,clearSelection:j,isSelected:H,selectedCount:q}}import{getAvailableProviders as $Z,getModelsForProvider as JZ}from"@contractspec/lib.ai-providers";import*as M from"react";function YZ(){let[X,Z]=M.useState([]),[$,J]=M.useState(!0),Y=M.useCallback(async()=>{J(!0);try{let Q=$Z().map((_)=>({..._,models:JZ(_.provider)}));Z(Q)}catch(q){console.error("Failed to load providers:",q)}finally{J(!1)}},[]);M.useEffect(()=>{Y()},[Y]);let G=M.useMemo(()=>X.filter((q)=>q.available),[X]),j=M.useCallback((q)=>X.some((Q)=>Q.provider===q&&Q.available),[X]),H=M.useCallback((q)=>X.find((Q)=>Q.provider===q)?.models??[],[X]);return{providers:X,availableProviders:G,isAvailable:j,getModels:H,isLoading:$,refresh:Y}}export{YZ as useProviders,ZZ as useMessageSelection,XZ as useConversations,cZ as useCompletion,eX as useChat};