@codeworksh/aikit 0.2.0 → 0.2.1

package/index.mjs

@@ -1,5 +1,6 @@
 import { Filesystem, NamedError, lazy } from "@codeworksh/utils";
 import { Type } from "@sinclair/typebox";
+import { uuidv7 } from "uuidv7";
 import { keys, mapValues, pick, pipe } from "remeda";
 import Anthropic from "@anthropic-ai/sdk";
 import { parse } from "partial-json";
@@ -347,6 +348,7 @@ let Message;
 		Type.Literal("aborted")
 	]);
 	const UserMessageSchema = _Message.UserMessageSchema = Type.Object({
+		messageId: Type.String(),
 		role: Type.Literal("user"),
 		time: Type.Object({ created: Type.Number() }),
 		parts: Type.Array(Type.Union([TextContentSchema, ImageContentSchema]))
@@ -369,9 +371,30 @@ let Message;
 			ThinkingContentSchema,
 			ToolCallSchema
 		])),
-		responseID: Type.Optional(Type.String())
+		responseId: Type.Optional(Type.String()),
+		messageId: Type.String()
 	});
 	const MessageSchema = _Message.MessageSchema = Type.Union([UserMessageSchema, AssistantMessageSchema]);
+	function createMessageId() {
+		return uuidv7();
+	}
+	_Message.createMessageId = createMessageId;
+	function createUserMessage(message) {
+		const { messageId = createMessageId(), ...rest } = message;
+		return {
+			messageId,
+			...rest
+		};
+	}
+	_Message.createUserMessage = createUserMessage;
+	function createAssistantMessage(message) {
+		const { messageId = createMessageId(), ...rest } = message;
+		return {
+			messageId,
+			...rest
+		};
+	}
+	_Message.createAssistantMessage = createAssistantMessage;
 	const ToolSchema = _Message.ToolSchema = Type.Object({
 		name: Type.String(),
 		description: Type.String(),
@@ -1028,7 +1051,7 @@ function buildParams(model, context, isOAuthToken, options) {
 const streamAnthropic = (model, context, options) => {
 	const stream = new AssistantMessageEventStream();
 	(async () => {
-		const output = {
+		const output = Message.createAssistantMessage({
 			role: "assistant",
 			parts: [],
 			protocol: model.protocol,
@@ -1053,7 +1076,7 @@ const streamAnthropic = (model, context, options) => {
 				created: Date.now(),
 				completed: Date.now()
 			}
-		};
+		});
 		try {
 			let client;
 			let isOAuthToken;
@@ -1080,7 +1103,7 @@ const streamAnthropic = (model, context, options) => {
 			});
 			const blocks = output.parts;
 			for await (const event of anthropicStream) if (event.type === "message_start") {
-				output.responseID = event.message.id;
+				output.responseId = event.message.id;
 				output.usage.input = event.message.usage.input_tokens || 0;
 				output.usage.output = event.message.usage.output_tokens || 0;
 				output.usage.cacheRead = event.message.usage.cache_read_input_tokens || 0;
@@ -1278,14 +1301,6 @@ const ajv = new Ajv({
 });
 addFormats(ajv);
 /**
-* Finds a tool by name and validates the tool call arguments against its TypeBox schema
-*/
-function validateToolCall(tools, toolExecution) {
-	const tool = tools.find((t) => t.name === toolExecution.name);
-	if (!tool) throw new Error(`Tool "${toolExecution.name}" not found`);
-	return validateToolArguments(tool, toolExecution);
-}
-/**
 * Validates tool call arguments against the tool's TypeBox schema
 */
 function validateToolArguments(tool, toolExecution) {
@@ -1849,7 +1864,6 @@ let Agent;
 		_state;
 		listeners = /* @__PURE__ */ new Set();
 		abortController;
-		activeAssistantIndex;
 		convertToLlm;
 		transformContext;
 		steeringQueue = [];
@@ -2001,19 +2015,34 @@ let Agent;
 		}
 		replaceMessages(ms) {
 			this._state.messages = ms.slice();
-			this.activeAssistantIndex = void 0;
 		}
 		appendMessage(m) {
 			this._state.messages = [...this._state.messages, m];
 		}
-		patchStoredAssistantPart(partIndex, part) {
-			if (this.activeAssistantIndex === void 0) return false;
+		findMessageIndexById(messageId) {
+			return this._state.messages.findIndex((message) => message.messageId === messageId);
+		}
+		replaceMessageById(message) {
+			const messageIndex = this.findMessageIndexById(message.messageId);
+			if (messageIndex === -1) return false;
 			const messages = this._state.messages.slice();
-			const existing = messages[this.activeAssistantIndex];
+			messages[messageIndex] = message;
+			this._state.messages = messages;
+			return true;
+		}
+		upsertMessage(message) {
+			if (this.replaceMessageById(message)) return;
+			this.appendMessage(message);
+		}
+		patchStoredAssistantPart(messageId, partIndex, part) {
+			const messageIndex = this.findMessageIndexById(messageId);
+			if (messageIndex === -1) return false;
+			const messages = this._state.messages.slice();
+			const existing = messages[messageIndex];
 			if (!existing || existing.role !== "assistant") return false;
 			const nextParts = existing.parts.slice();
 			nextParts[partIndex] = part;
-			messages[this.activeAssistantIndex] = {
+			messages[messageIndex] = {
 				...existing,
 				parts: nextParts
 			};
@@ -2076,7 +2105,6 @@ let Agent;
 		}
 		clearMessages() {
 			this._state.messages = [];
-			this.activeAssistantIndex = void 0;
 		}
 		abort() {
 			this.abortController?.abort();
@@ -2090,7 +2118,6 @@ let Agent;
 			this._state.streamMessage = null;
 			this._state.pendingToolCalls = /* @__PURE__ */ new Set();
 			this._state.error = void 0;
-			this.activeAssistantIndex = void 0;
 			this.steeringQueue = [];
 			this.followUpQueue = [];
 		}
@@ -2147,7 +2174,7 @@ let Agent;
 				if (messages) await Loop.runAgentLoop(config, context, messages, async (e) => this._processLoopEvent(e), this.streamFn, this.abortController.signal);
 				else await Loop.runAgentLoopContinue(config, context, async (e) => this._processLoopEvent(e), this.streamFn, this.abortController.signal);
 			} catch (err) {
-				const message = {
+				const message = Message.createAssistantMessage({
 					role: "assistant",
 					protocol: model.protocol,
 					provider: model.provider,
@@ -2173,7 +2200,7 @@ let Agent;
 						completed: Date.now()
 					},
 					parts: []
-				};
+				});
 				this.appendMessage(message);
 				this._state.error = err?.message || String(err);
 				this.emit({
@@ -2201,32 +2228,31 @@ let Agent;
 		async prompt(messages, images) {
 			this.ensureIdle();
 			if (!this._state.model) throw new ModelNotConfiguredErr({ message: "no model configured yet." });
-			const message = {
+			const message = Message.createUserMessage({
 				role: "user",
 				time: { created: Date.now() },
 				parts: [...messages, ...images ?? []]
-			};
+			});
 			await this.loop([message]);
 		}
 		_processLoopEvent(event) {
 			switch (event.type) {
 				case "message.start":
 					this._state.streamMessage = event.message;
-					if (event.message.role === "assistant") this.activeAssistantIndex = void 0;
 					break;
 				case "message.part.start":
 				case "message.part.update":
 				case "message.part.end":
 					this._state.streamMessage = event.message;
-					if (event.message.role === "assistant") this.patchStoredAssistantPart(event.partIndex, event.part);
+					if (event.message.role === "assistant") this.patchStoredAssistantPart(event.message.messageId, event.partIndex, event.part);
 					break;
 				case "message.update":
 					this._state.streamMessage = event.message;
+					this.replaceMessageById(event.message);
 					break;
 				case "message.end":
 					this._state.streamMessage = null;
-					this.appendMessage(event.message);
-					if (event.message.role === "assistant") this.activeAssistantIndex = this._state.messages.length - 1;
+					this.upsertMessage(event.message);
 					break;
 				case "tool.execution.start": {
 					const pendingToolCalls = new Set(this._state.pendingToolCalls);
@@ -2273,206 +2299,10 @@ let Agent;
 //#endregion
 //#region src/agent.ts
 const agent = {
-	create: Agent.create,
 	loop: Loop.run,
 	loopContinue: Loop.runContinue
 };
 //#endregion
-//#region src/event/event.ts
-let Event;
-(function(_Event) {
-	const LLMMessageEventType = _Event.LLMMessageEventType = {
-		start: "start",
-		textStart: "text.start",
-		textDelta: "text.delta",
-		textEnd: "text.end",
-		thinkingStart: "thinking.start",
-		thinkingDelta: "thinking.delta",
-		thinkingEnd: "thinking.end",
-		toolcallStart: "toolcall.start",
-		toolcallDelta: "toolcall.delta",
-		toolcallEnd: "toolcall.end",
-		done: "done",
-		error: "error"
-	};
-	_Event.LLMMessageEventSchema = Type.Union([
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.start),
-			partial: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.textStart),
-			partIndex: Type.Number(),
-			partial: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.textDelta),
-			partIndex: Type.Number(),
-			delta: Type.String(),
-			partial: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.textEnd),
-			partIndex: Type.Number(),
-			content: Type.String(),
-			partial: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.thinkingStart),
-			partIndex: Type.Number(),
-			partial: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.thinkingDelta),
-			partIndex: Type.Number(),
-			delta: Type.String(),
-			partial: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.thinkingEnd),
-			partIndex: Type.Number(),
-			content: Type.String(),
-			partial: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.toolcallStart),
-			partIndex: Type.Number(),
-			partial: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.toolcallDelta),
-			partIndex: Type.Number(),
-			delta: Type.String(),
-			partial: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.toolcallEnd),
-			partIndex: Type.Number(),
-			toolCall: Message.ToolCallSchema,
-			partial: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.done),
-			reason: Type.Union([
-				Type.Literal("stop"),
-				Type.Literal("length"),
-				Type.Literal("toolUse")
-			]),
-			message: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(LLMMessageEventType.error),
-			reason: Type.Union([Type.Literal("aborted"), Type.Literal("error")]),
-			error: Message.AssistantMessageSchema
-		})
-	]);
-	const AgentEventType = _Event.AgentEventType = {
-		agentStart: "agent.start",
-		agentEnd: "agent.end",
-		turnStart: "turn.start",
-		turnEnd: "turn.end",
-		messageStart: "message.start",
-		messagePartStart: "message.part.start",
-		messagePartUpdate: "message.part.update",
-		messagePartEnd: "message.part.end",
-		messageUpdate: "message.update",
-		messageEnd: "message.end",
-		toolExecutionStart: "tool.execution.start",
-		toolExecutionUpdate: "tool.execution.update",
-		toolExecutionEnd: "tool.execution.end"
-	};
-	const UserMessagePart = Type.Union([Message.TextContentSchema, Message.ImageContentSchema]);
-	const AssistantMessagePart = Type.Union([
-		Message.TextContentSchema,
-		Message.ImageContentSchema,
-		Message.ThinkingContentSchema,
-		Message.ToolCallSchema
-	]);
-	const UserMessagePartStart = Type.Object({
-		type: Type.Literal(AgentEventType.messagePartStart),
-		message: Message.UserMessageSchema,
-		partIndex: Type.Number(),
-		part: UserMessagePart
-	});
-	const UserMessagePartEnd = Type.Object({
-		type: Type.Literal(AgentEventType.messagePartEnd),
-		message: Message.UserMessageSchema,
-		partIndex: Type.Number(),
-		part: UserMessagePart
-	});
-	const AssistantMessagePartStart = Type.Object({
-		type: Type.Literal(AgentEventType.messagePartStart),
-		message: Message.AssistantMessageSchema,
-		partIndex: Type.Number(),
-		part: AssistantMessagePart
-	});
-	const AssistantMessagePartLLMUpdate = Type.Object({
-		type: Type.Literal(AgentEventType.messagePartUpdate),
-		message: Message.AssistantMessageSchema,
-		partIndex: Type.Number(),
-		part: AssistantMessagePart,
-		source: Type.Literal("llm")
-	});
-	const AssistantMessagePartToolUpdate = Type.Object({
-		type: Type.Literal(AgentEventType.messagePartUpdate),
-		message: Message.AssistantMessageSchema,
-		partIndex: Type.Number(),
-		part: AssistantMessagePart,
-		source: Type.Literal("tool")
-	});
-	const AssistantMessagePartEnd = Type.Object({
-		type: Type.Literal(AgentEventType.messagePartEnd),
-		message: Message.AssistantMessageSchema,
-		partIndex: Type.Number(),
-		part: AssistantMessagePart
-	});
-	const ToolExecutionStartSchema = Type.Composite([Agent.ToolCallInFlightSchema, Type.Object({ type: Type.Literal(AgentEventType.toolExecutionStart) })]);
-	const ToolExecutionUpdateSchema = Type.Composite([
-		Agent.ToolCallInFlightSchema,
-		Type.Object({ type: Type.Literal(AgentEventType.toolExecutionUpdate) }),
-		Message.ToolRunningSchema
-	]);
-	const ToolExecutionEndSchema = Type.Composite([
-		Agent.ToolCallInFlightSchema,
-		Type.Object({ type: Type.Literal(AgentEventType.toolExecutionEnd) }),
-		Type.Union([Message.ToolCompletedSchema, Message.ToolErrorSchema])
-	]);
-	_Event.AgentEventSchema = Type.Union([
-		Type.Object({ type: Type.Literal(AgentEventType.agentStart) }),
-		Type.Object({
-			type: Type.Literal(AgentEventType.agentEnd),
-			messages: Type.Array(Message.MessageSchema)
-		}),
-		Type.Object({ type: Type.Literal(AgentEventType.turnStart) }),
-		Type.Object({
-			type: Type.Literal(AgentEventType.turnEnd),
-			message: Message.AssistantMessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(AgentEventType.messageStart),
-			message: Message.MessageSchema
-		}),
-		Type.Union([UserMessagePartStart, UserMessagePartEnd]),
-		Type.Union([
-			AssistantMessagePartStart,
-			AssistantMessagePartLLMUpdate,
-			AssistantMessagePartToolUpdate,
-			AssistantMessagePartEnd
-		]),
-		Type.Object({
-			type: Type.Literal(AgentEventType.messageUpdate),
-			message: Message.MessageSchema
-		}),
-		Type.Object({
-			type: Type.Literal(AgentEventType.messageEnd),
-			message: Message.MessageSchema
-		}),
-		ToolExecutionStartSchema,
-		ToolExecutionUpdateSchema,
-		ToolExecutionEndSchema
-	]);
-})(Event || (Event = {}));
-//#endregion
 //#region src/llm.ts
 const llmImpl = async (provider, model) => Model.getModel(provider, model);
 const llm = Object.assign(llmImpl, {
@@ -2503,13 +2333,10 @@ const stream = Object.assign(streamImpl, {
 	completeSimple: Stream.completeSimple,
 	resolveProtocolProvider: Stream.resolveProtocolProvider
 });
-/** Resolve a streamed response directly to the final assistant message. */
-const complete = Stream.complete;
-/** Start a stream using the simplified options surface. */
-const streamSimple = Stream.streamSimple;
-/** Resolve a simplified stream request directly to the final assistant message. */
-const completeSimple = Stream.completeSimple;
+Stream.complete;
+Stream.streamSimple;
+Stream.completeSimple;
 //#endregion
-export { Agent, Event, Message, Model, ModelCatalog, Stream, agent, complete, completeSimple, llm, stream, streamSimple, validateToolArguments, validateToolCall };
+export { Agent, Message, agent, llm, stream };
 
 //# sourceMappingURL=index.mjs.map