@makaio/framework 1.0.0-dev-1779046984397

package/dist/adapters/stream-session/index.mjs

@@ -0,0 +1 @@
+import{z as e}from"zod";import{MakaioBus as t,OnceAbortError as n}from"@makaio/framework/bus";import{AgentSubjects as r,McpSubjects as i,ToolSubjects as a}from"@makaio/framework/contracts";import{AIAgent as o,AgentCompleteEventSchema as s,AgentStartedEventSchema as c,BaseConnectorSession as l,ErrorEventSchema as u,ProceduralAgentConnector as d,ToolCompletedEventSchema as f,ToolStartedEventSchema as p,UserMessageQueue as m,extractMcpCallTarget as h,isMcpCallTool as g,processQueueMessages as _}from"@makaio/framework/adapters";const v=8e3;function y(e){if(e.length<=8e3)return e;let t=v,n=``;for(;;){let r=`\n...[truncated ${e.length-t} chars]`,i=v-r.length;if(i<0)return r.slice(0,v);if(i===t&&r===n)break;t=i,n=r}return`${e.slice(0,t)}${n}`}function b(e){return typeof e==`object`&&!!e&&!Array.isArray(e)}function x(e,t=`Adapter`){let n;try{n=JSON.parse(e.function.arguments)}catch{let n=e.function.arguments.length;throw console.error(`[${t}] Failed to parse tool arguments for "${e.function.name}" (argsLength=${n})`),Error(`Failed to parse tool arguments for "${e.function.name}"`)}if(!b(n)){let r=n===null?`null`:Array.isArray(n)?`array`:typeof n;throw console.error(`[${t}] Parsed tool arguments for "${e.function.name}" must be a JSON object (received ${r})`,n),Error(`Tool arguments for "${e.function.name}" must be a JSON object`)}return{toolName:e.function.name,args:n,toolCallId:e.id}}function S(e,t,n=`Adapter`){return{...x(e,n),agentId:t.agentId,adapterId:t.adapterId,adapterName:t.adapterName,adapterSessionId:t.adapterSessionId,sessionId:t.sessionId}}function C(e,t){t.updatedInput!==void 0&&(e.function.arguments=JSON.stringify(t.updatedInput))}async function w(e,n){try{return(await t.request(a.list,{adapterId:e,adapterName:n})).tools}catch(e){return console.warn(`[${n}] Failed to fetch tools from bus:`,e),[]}}function T(e){return e.filter(e=>e.inputSchema!==void 0)}async function E(e,n,r){await n({eventType:`tool_started`,toolName:e.function.name,toolCallId:e.id});try{let i=JSON.parse(e.function.arguments),o=await t.request(a.execute,{toolName:e.function.name,input:i,adapterId:r.adapterId,adapterName:r.adapterName,contextOverrides:{...r,toolCallId:e.id}});return n({eventType:`tool_completed`,toolName:e.function.name,toolCallId:e.id,result:o.success?JSON.stringify(o.data):JSON.stringify(o.error),success:o.success}).then(()=>o)}catch(t){let r=t instanceof Error?{message:t.message,name:t.name}:{message:String(t),name:`UnknownError`};throw await n({eventType:`tool_completed`,toolName:e.function.name,toolCallId:e.id,result:JSON.stringify(r),success:!1}),t}}async function D(e,t,n,r,i){let a=[];for(let o of e){let e=x(o,i),s=await t.requestToolApproval({...e,reasoning:n.reasoning});if(s.action===`deny`){if(s.shouldAbort)throw Error(`Tool use denied by approval handler: ${o.function.name}`);let e=y(JSON.stringify({error:s.message}));a.push(r(o.id,e,!0));continue}C(o,s);let c=await E(o,t.emitSdkEvent,n);if(c.success&&t.recordMcpCall&&g(e.toolName)){let n=h(s.updatedInput??e.args);n!==void 0&&t.recordMcpCall(n)}let l=c.success?typeof c.data==`string`?c.data:JSON.stringify(c.data??null):JSON.stringify({error:c.error.message,code:c.error.code});a.push(r(o.id,y(l),!c.success))}return a}const O=e.enum([`idle`,`turn_started`,`step_started`,`step_finished`,`turn_finished`]),k=e.object({adapterId:e.string(),agentId:e.string(),oldState:O,newState:O,timestamp:e.number()}),A=e.object({eventType:e.literal(`reasoning_delta`),content:e.string()}),j=e.object({eventType:e.literal(`reasoning_complete`),content:e.string()}),M=e.object({name:e.string(),arguments:e.string()}),N=e.object({id:e.string(),type:e.literal(`function`),function:M}),P=e.object({eventType:e.literal(`tool_calls`),toolCalls:e.array(N)}),F=e.object({id:e.string(),type:e.literal(`function`),function:M}),I=e.object({eventType:e.literal(`message_complete`),content:e.string().nullable(),reasoning:e.string().optional(),tool_calls:e.array(F).optional(),finish_reason:e.string().nullable()}),L=`stream_start_timeout`;var R=class extends l{static MAX_TOOL_CALL_ITERATIONS=8;lastAssistantMessage;currentModel;currentCwd;constructor(e){super(e),this.sessionId=crypto.randomUUID(),this.currentModel=e.model,this.currentCwd=e.cwd}async processQueue(e){await _(e,{getCurrentTurn:()=>this.currentTurn,startNewTurn:(e,t)=>this.startNewTurn(e,t)})}async startNewTurn(e,t){this.config.onTurnStart?.(e),this.sessionId=crypto.randomUUID(),e.adapterSessionId=this.sessionId,this.buildMessages(e,t),this.currentTurn=this.createTurn(e),this.lastAssistantMessage=void 0,e.markAcknowledged(),await this.config.emitSdkEvent({eventType:`agent_started`,model:this.currentModel});let n=this.currentTurn;queueMicrotask(async()=>{try{if(!n||(await this.runTurn(n,e),e.isProcessed||this.isTurnSuperseded(n)))return;let t={outcome:`completed`,result:{message:this.lastAssistantMessage||``}};e.markCompleted(t),this.config.onTurnComplete?.(e,t)}catch(t){this.handleTurnError(t,n,e)}finally{n&&(!n.isPaused()||e.deliveryMode!==`immediate`)&&(await n.markTurnFinished(),this.currentTurn===n&&(this.currentTurn=void 0))}})}handleTurnError(e,t,n){if(t?.isPaused()&&n.deliveryMode===`immediate`)return;let r=e instanceof Error?e:Error(String(e));if(t?.isPaused()){if(!n.isProcessed){let e={outcome:`error`,error:r};n.markCompleted(e),this.config.onTurnComplete?.(n,e)}return}if(console.error(`[${this.config.adapterName}] Turn error:`,e),!n.isProcessed){let e={outcome:`error`,error:r};n.markCompleted(e),this.config.onTurnComplete?.(n,e)}this.config.handleError(r,!1)}async runTurn(e,t){this.isTurnSuperseded(e)||(await e.start(),await this.runTurnIteration(e,t))}createMessageCompletePromise(e,t){return this.bus.once(this.getSdkEventSubject(),{timeoutMs:3e5,filter:{"event.eventType":`message_complete`,agentId:this.config.agentId,adapterId:this.config.adapterId,adapterSessionId:t},signal:e})}async executeStreamingStep(e,t,n){let r=this.config.streamStartTimeoutMs??3e4,i=new AbortController,a=setTimeout(()=>{i.abort()},r);try{await this.executeApiCall(e,AbortSignal.any([t,i.signal]),n)}catch(e){throw!t.aborted&&i.signal.aborted?Error(L,{cause:e}):e}finally{clearTimeout(a)}}async runTurnIteration(e,t,r=0){if(this.isTurnSuperseded(e))return;let i=e.getAbortSignal(),a=0,o;for(;;){if(this.shouldAbortTurnProcessing(e,t))return;try{let n=t.adapterSessionId??this.sessionId;if(!n)throw Error(`adapterSessionId must be set before waiting for message_complete`);if(o=this.createMessageCompletePromise(i,n),await this.executeStreamingStep(e,i,n),this.shouldAbortTurnProcessing(e,t)){o.catch(()=>{});return}let{payload:a}=await o,s=a.event;if(s.eventType!==`message_complete`)throw Error(`Expected message_complete event, got ${s.eventType}`);if(this.shouldAbortTurnProcessing(e,t))return;await this.applyMessageComplete(s,t,r,e);return}catch(e){if(o?.catch(()=>{}),e instanceof Error&&e.message===L&&!i.aborted&&a===0){a++;continue}if(i.aborted||e.name===`AbortError`||e instanceof n)return;let t=this.classifyError(e);throw r===0&&await this.config.emitSdkEvent({eventType:`error`,message:t.message,code:e?.code,type:e?.type}),t}}}buildToolConstraints(){if(this.config.allowedDirectories!==void 0)return{allowedDirectories:this.config.allowedDirectories}}updateModel(e){this.currentModel=e}updateCwd(e){this.currentCwd=e}getCurrentTurn(){return this.currentTurn}isTurnSuperseded(e){return this.currentTurn!==e}shouldAbortTurnProcessing(e,t){return t.isProcessed||e.isPaused()||this.isTurnSuperseded(e)}},z=class extends d{session;userMessageQueue=new m;mcpDirectTools=[];mcpSessionContextCache;hasPendingToolRefresh=!1;hasPendingLedgerInjection=!1;markToolRefreshPending(){this.hasPendingToolRefresh=!0}afterSessionCreated(e){}getSession(){return this.session}async ensureSession(){return this.session||await this.initializeSession(),this.session}getSessionQueue(){return this.userMessageQueue}async initializeSession(){await this.fetchToolsViaBus(),this.prepareMcpDirectTools(),this.session=this.createSession(),this.afterSessionCreated(this.session),this.wireSessionEvents(),this.mcpDirectTools.length>0&&this.session.updateTools?.(this.mcpDirectTools);let e=(this.pendingTurnNumber??this.currentTurnNumber)||1;this.config.toolLedger?.recordInjection(this.mcpDirectTools,e)}prepareMcpDirectTools(){let e=this.mcpSessionContextCache??this.config.mcpSessionContext;e&&(this.mcpDirectTools=e.directTools.map(e=>({name:e.fullName,description:e.description??``,toolsetName:e.serverName,inputSchema:e.inputSchema})))}async fetchToolsViaBus(){this.session||await this.fetchTools()}async refreshTools(){await this.resolveAndCacheMcpContext(),this.prepareMcpDirectTools(),this.hasPendingLedgerInjection=!0}async resolveAndCacheMcpContext(){let e=this.config.mcpSessionContext;if(!(!e||!(`sessionId`in e)))try{this.mcpSessionContextCache=await t.request(i.session.resolve,{sessionId:e.sessionId,profileId:e.profileId,projectId:e.projectId})}catch(e){console.warn(`[${this.config.adapterName}] MCP session re-resolve failed; using previous tool snapshot.`,e)}}getWireSessionConfig(){return{onTurnStarted:async()=>{if(this.consumeTurnNumber(),this.hasPendingToolRefresh)try{await this.refreshTools()}catch(e){console.warn(`[${this.config.adapterName}] Tool refresh failed at turn start; continuing with stale tools.`,e)}finally{this.hasPendingToolRefresh=!1}this.hasPendingLedgerInjection&&=(this.config.toolLedger?.recordInjection(this.mcpDirectTools,this.currentTurnNumber),!1)},onTurnFinished:async e=>{if(this.hasPendingToolRefresh)try{await this.refreshTools()}catch(e){console.warn(`[${this.config.adapterName}] Tool refresh failed; continuing with stale tools.`,e)}finally{this.hasPendingToolRefresh=!1}await e()}}}resolveSystemPrompt(){if(this.systemPrompt!==void 0)return typeof this.systemPrompt==`string`?this.systemPrompt:this.systemPrompt.content}async sendMessage(e,t){let n=this.createMessageHandle(e,t);return await this.processUserMessages([n]),n}abort(){this.session?.abort()}async close(){await this.session?.abort()}async changeModelInPlace(e){return this.session?.updateModel(e),!0}async changeCwdInPlace(e){return this.session?.updateCwd(e),!0}async changeReasoningInPlace(e){return this.session?.updateReasoning?.(e),!0}async interrupt(){await this.session?.abort()}async getAdapterSessionId(){if(this.adapterSessionId||this.initAdapterSessionId(),!this.adapterSessionId)throw Error(`Adapter session ID is missing in getAdapterSessionId(); ensure initAdapterSessionId() ran during connector initialization.`);return this.adapterSessionId}initAdapterSessionId(){this.adapterSessionId??=crypto.randomUUID()}};function B(e){try{let t=JSON.parse(e);return typeof t==`object`&&t&&!Array.isArray(t)?t:{}}catch{return{}}}var V=class extends o{toolBlockIndexMap=new Map;completedToolIds=new Set;toolCallArgsMap=new Map;resetPerTurnToolTracking(){this.toolBlockIndexMap.clear(),this.completedToolIds.clear(),this.toolCallArgsMap.clear()}wireEvents(e){this.wireSdkEvents(),this.wireSemanticEvents(e,this.getConnectorSubjects()),this.wireToolApprovalRpc(e)}wireSemanticEvents(e,t){this.wireMessageEvents(e,t),this.wireUsageEvents(e,t),this.wireToolEvents(e,t),this.wireReasoningEvents(e,t),this.wireLifecycleEvents(e,t)}wireMessageEvents(e,t){this.subscribeConnector(e,t.chunk,async e=>{let t=this.extractChunkText(e.payload);t&&await this.emitGlobal(r.message_delta,{text:t})}),this.subscribeConnector(e,t.messageComplete,async e=>{let{content:t}=e.payload;t&&(await this.emitStepStarted(`text`),await this.emitStepFinished(`text`,{type:`text`,content:t})),await this.emitGlobal(r.message,{content:t??``})})}wireUsageEvents(e,t){this.subscribeConnector(e,t.usage,async e=>{let t=this.extractUsagePayload(e.payload);await this.trackUsage(t),await this.emitUsageContextWindowUpdate(e.payload)})}wireToolEvents(e,t){this.subscribeConnector(e,t.toolCalls,async e=>{let{toolCalls:t}=e.payload;for(let e of t){if(this.completedToolIds.has(e.id)){console.warn(`[BaseStreamAgent] Ignoring late tool_calls for completed toolCallId ${e.id}`);continue}let t=B(e.function.arguments);this.toolCallArgsMap.set(e.id,t);let n=this.reserveToolCallBlockIndex(e);this.toolBlockIndexMap.set(e.id,n),await this.emitGlobal(r.step.started,{stepType:`tool_use`,blockIndex:n,blockData:{type:`tool_use`,toolName:e.function.name,toolCallId:e.id},content:{type:`tool_call`,toolCallId:e.id,name:e.function.name,args:t}}),this.afterToolCallStepEmitted(n),await this.emitGlobal(r.tool.use,{toolName:e.function.name,args:t,toolCallId:e.id})}}),this.subscribeConnector(e,t.toolStarted,async e=>{let{toolName:t,toolCallId:n}=e.payload;await this.emitGlobal(r.tool.started,{toolName:t,toolCallId:n})}),this.subscribeConnector(e,t.toolCompleted,async e=>{let{toolCallId:t,toolName:n,result:i,success:a}=e.payload;this.completedToolIds.add(t);let o=this.toolBlockIndexMap.get(t);o===void 0&&console.warn(`[BaseStreamAgent] toolCallId ${t} missing from toolBlockIndexMap; using fallback`);let s=o??this.getFallbackToolCompletedBlockIndex();this.toolBlockIndexMap.delete(t),await this.emitGlobal(r.step.finished,{stepType:`tool_use`,blockIndex:s,content:{type:`tool_output`,toolCallId:t,output:i,isError:!a}}),this.afterToolCompletedStepEmitted(s);let c=this.toolCallArgsMap.get(t);this.toolCallArgsMap.delete(t),c===void 0&&console.warn(`[BaseStreamAgent] toolCallId ${t} missing args for tool ${n}`),await this.emitGlobal(r.tool.completed,{toolName:n,args:c,result:{output:i},success:a,toolCallId:t})})}wireReasoningEvents(e,t){this.subscribeConnector(e,t.reasoningDelta,async e=>{let{content:t}=e.payload;await this.emitGlobal(r.reasoning_delta,{content:t})}),this.subscribeConnector(e,t.reasoningComplete,async e=>{let t=e.payload.content;if(t){let t=this.buildReasoningBlock(e.payload);await this.emitStepStarted(`reasoning`),await this.emitStepFinished(`reasoning`,t)}await this.emitGlobal(r.reasoning,{content:t??``})})}wireLifecycleEvents(e,t){this.subscribeConnector(e,t.agentStarted,async e=>{e.payload,this.resetPerTurnToolTracking(),await this.emitStart()}),this.subscribeConnector(e,t.agentComplete,async e=>{e.payload,this.resetPerTurnToolTracking()}),this.subscribeConnector(e,t.error,async e=>{let{message:t}=e.payload;this.emitError({error:t})})}};export{s as AgentCompleteEventSchema,c as AgentStartedEventSchema,V as BaseStreamAgent,z as BaseStreamConnector,R as BaseStreamSession,u as ErrorEventSchema,v as MAX_TOOL_RESULT_CONTENT_CHARS,I as MessageCompleteEventSchema,F as MessageToolCallSchema,j as ReasoningCompleteEventSchema,A as ReasoningDeltaEventSchema,O as StreamSessionTurnStateSchema,M as ToolCallFunctionSchema,N as ToolCallSchema,P as ToolCallsEventSchema,f as ToolCompletedEventSchema,p as ToolStartedEventSchema,k as TurnStateChangedSchema,C as applyApprovedArgs,y as boundToolResultContent,E as executeTool,x as extractToolCallPayload,T as filterToolsWithSchema,D as handleToolCalls,w as loadToolsFromRegistry,S as toGlobalToolApproval};

package/dist/adapters/stream-session/testing/index.d.mts

@@ -0,0 +1,29 @@
+import { AIReasoningLevel } from "@makaio/framework/contracts";
+
+//#region adapters/shared/stream-session/src/testing/index.d.ts
+/**
+ * Minimal public surface required from the connector under test.
+ *
+ * Intentionally narrow: the contract only exercises `changeReasoningInPlace`
+ * and the read-only `currentReasoningEffort` snapshot, keeping the helper
+ * decoupled from any concrete connector class.
+ */
+interface ChangeReasoningConnector {
+  /** Attempt an in-place reasoning level change; returns `true` on success. */
+  changeReasoningInPlace(level: AIReasoningLevel): Promise<boolean>;
+  /** Current reasoning effort level managed by the caller, never mutated in-place. */
+  readonly currentReasoningEffort: AIReasoningLevel | undefined;
+}
+/**
+ * Register the shared `changeReasoningInPlace` contract suite for a
+ * stream-based connector.
+ *
+ * Each test creates a fresh connector via `makeConnector()` to guarantee
+ * isolation — the factory must return a pre-start connector with no active
+ * session.
+ * @param connectorName - Display name used in the outer `describe` block
+ * @param makeConnector - Async factory that returns a fresh connector instance
+ */
+declare function describeChangeReasoningInPlaceContract(connectorName: string, makeConnector: () => Promise<ChangeReasoningConnector>): void;
+//#endregion
+export { ChangeReasoningConnector, describeChangeReasoningInPlaceContract };

package/dist/adapters/stream-session/testing/index.mjs

@@ -0,0 +1 @@
+import{describe as e,expect as t,it as n,vi as r}from"vitest";function i(i,a){e(`${i}.changeReasoningInPlace()`,()=>{n(`always returns true (stream-based adapters apply reasoning per-request)`,async()=>{t(await(await a()).changeReasoningInPlace(`medium`)).toBe(!0)}),n(`returns true for every canonical reasoning level`,async()=>{let e=await a();for(let n of[`none`,`low`,`medium`,`high`,`extra-high`])t(await e.changeReasoningInPlace(n)).toBe(!0)}),n(`does NOT mutate connector.currentReasoningEffort (mutation contract — caller is responsible)`,async()=>{let e=await a(),n=e.currentReasoningEffort;await e.changeReasoningInPlace(`high`),t(e.currentReasoningEffort).toBe(n)}),n(`calls session.updateReasoning with the requested level when a session is active`,async()=>{let e=await a(),n=r.fn();Reflect.set(e,`session`,{updateReasoning:n}),await e.changeReasoningInPlace(`high`),t(n).toHaveBeenCalledOnce(),t(n).toHaveBeenCalledWith(`high`)}),n(`does not throw and returns true when no session exists (pre-start state)`,async()=>{t(await(await a()).changeReasoningInPlace(`medium`)).toBe(!0)}),n(`returns true and does not throw when session is null (optional-chain guard)`,async()=>{let e=await a();Reflect.set(e,`session`,null),t(await e.changeReasoningInPlace(`medium`)).toBe(!0)})})}export{i as describeChangeReasoningInPlaceContract};