@inkeep/agents-run-api 0.1.0

package/dist/a2a/transfer.js

@@ -0,0 +1,22 @@
+import { setActiveAgentForThread } from '@inkeep/agents-core';
+import { getLogger } from '../logger.js';
+import dbClient from '../data/db/dbClient.js';
+const logger = getLogger('Transfer');
+/**
+ * Executes a transfer by sending the original message to the target agent
+ */
+export async function executeTransfer({ tenantId, threadId, projectId, targetAgentId, }) {
+    logger.info({ targetAgent: targetAgentId }, 'Executing transfer to agent');
+    await setActiveAgentForThread(dbClient)({
+        scopes: { tenantId, projectId },
+        threadId,
+        agentId: targetAgentId,
+    });
+    return { success: true, targetAgentId };
+}
+/**
+ * Checks if a response is a transfer response
+ */
+export function isTransferResponse(result) {
+    return result?.artifacts.some((artifact) => artifact.parts.some((part) => part.kind === 'data' && part.data?.type === 'transfer'));
+}

package/dist/a2a/types.d.ts

@@ -0,0 +1,63 @@
+import { AgentCard, type Artifact, type TaskState } from '@inkeep/agents-core';
+export { AgentCard };
+export interface RegisteredAgent {
+    agentId: string;
+    tenantId: string;
+    projectId: string;
+    graphId: string;
+    agentCard: AgentCard;
+    taskHandler: (task: A2ATask) => Promise<A2ATaskResult>;
+}
+export interface A2ATask {
+    id: string;
+    input: {
+        parts: Array<{
+            kind: string;
+            text?: string;
+            data?: any;
+        }>;
+    };
+    context?: {
+        conversationId?: string;
+        userId?: string;
+        metadata?: Record<string, any>;
+    };
+}
+export interface A2ATaskResult {
+    status: {
+        state: TaskState;
+        message?: string;
+    };
+    artifacts?: Artifact[];
+}
+export interface TransferResponse {
+    type: 'transfer';
+    target: string;
+    task_id: string;
+    reason: string;
+    artifacts: Artifact[];
+    original_message: string;
+    context: {
+        conversationId?: string;
+        tenantId: string;
+        transfer_context?: Artifact[];
+    };
+}
+export interface JsonRpcRequest {
+    jsonrpc: '2.0';
+    method: string;
+    params?: any;
+    id?: string | number | null;
+}
+export interface JsonRpcResponse {
+    jsonrpc: '2.0';
+    result?: any;
+    error?: {
+        code: number;
+        message: string;
+        data?: any;
+    };
+    id?: string | number | null;
+}
+export type A2AMethod = 'agent.invoke' | 'agent.getCapabilities' | 'agent.getStatus';
+//# sourceMappingURL=types.d.ts.map

package/dist/a2a/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/a2a/types.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,KAAK,QAAQ,EAAE,KAAK,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAG/E,OAAO,EAAE,SAAS,EAAE,CAAC;AAErB,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,SAAS,CAAC;IACrB,WAAW,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,OAAO,CAAC,aAAa,CAAC,CAAC;CACxD;AAED,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE;QACL,KAAK,EAAE,KAAK,CAAC;YACX,IAAI,EAAE,MAAM,CAAC;YACb,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,IAAI,CAAC,EAAE,GAAG,CAAC;SACZ,CAAC,CAAC;KACJ,CAAC;IACF,OAAO,CAAC,EAAE;QACR,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAChC,CAAC;CACH;AAED,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE;QACN,KAAK,EAAE,SAAS,CAAC;QACjB,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;CACxB;AAGD,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,UAAU,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE;QACP,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,QAAQ,EAAE,MAAM,CAAC;QACjB,gBAAgB,CAAC,EAAE,QAAQ,EAAE,CAAC;KAC/B,CAAC;CACH;AAGD,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,KAAK,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;CAC7B;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,KAAK,CAAC;IACf,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,KAAK,CAAC,EAAE;QACN,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,GAAG,CAAC;KACZ,CAAC;IACF,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;CAC7B;AAGD,MAAM,MAAM,SAAS,GAAG,cAAc,GAAG,uBAAuB,GAAG,iBAAiB,CAAC"}

package/dist/a2a/types.js

@@ -0,0 +1 @@
+export {};

package/dist/agents/Agent.d.ts

@@ -0,0 +1,154 @@
+import { type ToolSet } from 'ai';
+import type { AgentExecutionServer } from '../AgentExecutionServer.js';
+import { type Artifact, type DataComponentApiInsert, type McpTool, type AgentConversationHistoryConfig, type ArtifactComponentApiInsert, type Models } from '@inkeep/agents-core';
+import type { StreamHelper } from '../utils/stream-helpers.js';
+/**
+ * Creates a stopWhen condition that stops when any tool call name starts with the given prefix
+ * @param prefix - The prefix to check for in tool call names
+ * @returns A function that can be used as a stopWhen condition
+ */
+export declare function hasToolCallWithPrefix(prefix: string): ({ steps }: {
+    steps: Array<any>;
+}) => any;
+export type AgentConfig = {
+    id: string;
+    tenantId: string;
+    projectId: string;
+    graphId: string;
+    baseUrl: string;
+    apiKey?: string;
+    apiKeyId?: string;
+    name: string;
+    description: string;
+    agentPrompt: string;
+    agentRelations: AgentConfig[];
+    transferRelations: AgentConfig[];
+    delegateRelations: DelegateRelation[];
+    tools?: McpTool[];
+    artifacts?: Record<string, Artifact>;
+    functionTools?: Array<{
+        name: string;
+        description: string;
+        execute: (params: any) => Promise<any>;
+        parameters?: Record<string, any>;
+        schema?: any;
+    }>;
+    contextConfigId?: string;
+    dataComponents?: DataComponentApiInsert[];
+    artifactComponents?: ArtifactComponentApiInsert[];
+    conversationHistoryConfig?: AgentConversationHistoryConfig;
+    models?: Models;
+    stopWhen?: {
+        stepCountIs?: number;
+    };
+};
+export type ExternalAgentConfig = {
+    id: string;
+    name: string;
+    description: string;
+    baseUrl: string;
+};
+export type DelegateRelation = {
+    type: 'internal';
+    config: AgentConfig;
+} | {
+    type: 'external';
+    config: ExternalAgentConfig;
+};
+export type ToolType = 'transfer' | 'delegation' | 'mcp' | 'tool';
+export declare class Agent {
+    private config;
+    private systemPromptBuilder;
+    private responseFormatter;
+    private credentialStuffer?;
+    private streamHelper?;
+    private conversationId?;
+    private artifactComponents;
+    private isDelegatedAgent;
+    private contextResolver?;
+    constructor(config: AgentConfig, framework?: AgentExecutionServer);
+    /**
+     * Get the maximum number of generation steps for this agent
+     * Uses agent's stopWhen.stepCountIs config or defaults to CONSTANTS.MAX_GENERATION_STEPS
+     */
+    private getMaxGenerationSteps;
+    /**
+     * Get the primary model settings for text generation and thinking
+     * Defaults to claude-4-sonnet if not specified
+     */
+    private getPrimaryModel;
+    /**
+     * Get the model settings for structured output generation
+     * Defaults to GPT-4.1-mini for structured outputs if not specified
+     */
+    private getStructuredOutputModel;
+    /**
+     * Get the model settings for summarization tasks
+     * Defaults to gpt-4.1-nano if not specified
+     */
+    private getSummarizerModel;
+    setConversationId(conversationId: string): void;
+    /**
+     * Set delegation status for this agent instance
+     */
+    setDelegationStatus(isDelegated: boolean): void;
+    /**
+     * Get streaming helper if this agent should stream to user
+     * Returns undefined for delegated agents to prevent streaming data operations to user
+     */
+    getStreamingHelper(): StreamHelper | undefined;
+    /**
+     * Wraps a tool with streaming lifecycle tracking (start, complete, error) and GraphSession recording
+     */
+    private wrapToolWithStreaming;
+    getRelationTools(runtimeContext?: {
+        contextId: string;
+        metadata: {
+            conversationId: string;
+            threadId: string;
+            streamRequestId?: string;
+            streamBaseUrl?: string;
+            apiKey?: string;
+            baseUrl?: string;
+        };
+    }, sessionId?: string): any;
+    getMcpTools(sessionId?: string, streamRequestId?: string): Promise<ToolSet>;
+    /**
+     * Convert database McpTool to builder MCPToolConfig format
+     */
+    private convertToMCPToolConfig;
+    getMcpTool(tool: McpTool): Promise<Record<string, any>>;
+    getFunctionTools(streamRequestId?: string): ToolSet;
+    /**
+     * Get resolved context using ContextResolver - will return cached data or fetch fresh data as needed
+     */
+    getResolvedContext(conversationId: string, requestContext?: Record<string, unknown>): Promise<Record<string, unknown> | null>;
+    /**
+     * Get the graph prompt for this agent's graph
+     */
+    private getGraphPrompt;
+    /**
+     * Check if any agent in the graph has artifact components configured
+     */
+    private hasGraphArtifactComponents;
+    /**
+     * Build adaptive system prompt for Phase 2 structured output generation
+     * based on configured data components and artifact components across the graph
+     */
+    private buildPhase2SystemPrompt;
+    private buildSystemPrompt;
+    private getArtifactTools;
+    private createThinkingCompleteTool;
+    private getDefaultTools;
+    private graphHasArtifactComponents;
+    generate(userMessage: string, runtimeContext?: {
+        contextId: string;
+        metadata: {
+            conversationId: string;
+            threadId: string;
+            taskId: string;
+            streamRequestId: string;
+        };
+    }): Promise<any>;
+}
+//# sourceMappingURL=Agent.d.ts.map

package/dist/agents/Agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"Agent.d.ts","sourceRoot":"","sources":["../../src/agents/Agent.ts"],"names":[],"mappings":"AACA,OAAO,EAA4C,KAAK,OAAO,EAAQ,MAAM,IAAI,CAAC;AAElF,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AACvE,OAAO,EAYL,KAAK,QAAQ,EACb,KAAK,sBAAsB,EAC3B,KAAK,OAAO,EAEZ,KAAK,8BAA8B,EACnC,KAAK,0BAA0B,EAE/B,KAAK,MAAM,EAEZ,MAAM,qBAAqB,CAAC;AAgB7B,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAU/D;;;;GAIG;AACH,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,MAAM,IAC1C,WAAW;IAAE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAA;CAAE,SAOzC;AAuBD,MAAM,MAAM,WAAW,GAAG;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,WAAW,EAAE,CAAC;IAC9B,iBAAiB,EAAE,WAAW,EAAE,CAAC;IACjC,iBAAiB,EAAE,gBAAgB,EAAE,CAAC;IACtC,KAAK,CAAC,EAAE,OAAO,EAAE,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACrC,aAAa,CAAC,EAAE,KAAK,CAAC;QACpB,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,OAAO,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;QACvC,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACjC,MAAM,CAAC,EAAE,GAAG,CAAC;KACd,CAAC,CAAC;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,cAAc,CAAC,EAAE,sBAAsB,EAAE,CAAC;IAC1C,kBAAkB,CAAC,EAAE,0BAA0B,EAAE,CAAC;IAClD,yBAAyB,CAAC,EAAE,8BAA8B,CAAC;IAC3D,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE;QACT,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,gBAAgB,GACxB;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,MAAM,EAAE,WAAW,CAAA;CAAE,GACzC;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,MAAM,EAAE,mBAAmB,CAAA;CAAE,CAAC;AAEtD,MAAM,MAAM,QAAQ,GAAG,UAAU,GAAG,YAAY,GAAG,KAAK,GAAG,MAAM,CAAC;AAIlE,qBAAa,KAAK;IAChB,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,mBAAmB,CAAiD;IAC5E,OAAO,CAAC,iBAAiB,CAAoB;IAC7C,OAAO,CAAC,iBAAiB,CAAC,CAAoB;IAC9C,OAAO,CAAC,YAAY,CAAC,CAAe;IACpC,OAAO,CAAC,cAAc,CAAC,CAAS;IAChC,OAAO,CAAC,kBAAkB,CAAoC;IAC9D,OAAO,CAAC,gBAAgB,CAAkB;IAC1C,OAAO,CAAC,eAAe,CAAC,CAAkB;gBAE9B,MAAM,EAAE,WAAW,EAAE,SAAS,CAAC,EAAE,oBAAoB;IAyCjE;;;OAGG;IACH,OAAO,CAAC,qBAAqB;IAI7B;;;OAGG;IACH,OAAO,CAAC,eAAe;IAUvB;;;OAGG;IACH,OAAO,CAAC,wBAAwB;IAwBhC;;;OAGG;IACH,OAAO,CAAC,kBAAkB;IAwB1B,iBAAiB,CAAC,cAAc,EAAE,MAAM;IAIxC;;OAEG;IACH,mBAAmB,CAAC,WAAW,EAAE,OAAO;IAIxC;;;OAGG;IACH,kBAAkB,IAAI,YAAY,GAAG,SAAS;IAI9C;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAyE7B,gBAAgB,CACd,cAAc,CAAC,EAAE;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,EAAE;YACR,cAAc,EAAE,MAAM,CAAC;YACvB,QAAQ,EAAE,MAAM,CAAC;YACjB,eAAe,CAAC,EAAE,MAAM,CAAC;YACzB,aAAa,CAAC,EAAE,MAAM,CAAC;YACvB,MAAM,CAAC,EAAE,MAAM,CAAC;YAChB,OAAO,CAAC,EAAE,MAAM,CAAC;SAClB,CAAC;KACH,EACD,SAAS,CAAC,EAAE,MAAM;IAqDd,WAAW,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,eAAe,CAAC,EAAE,MAAM;IA+D9D;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAaxB,UAAU,CAAC,IAAI,EAAE,OAAO;IA0E9B,gBAAgB,CAAC,eAAe,CAAC,EAAE,MAAM;IAwBzC;;OAEG;IACG,kBAAkB,CACtB,cAAc,EAAE,MAAM,EACtB,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACvC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IA8D1C;;OAEG;YACW,cAAc;IAuB5B;;OAEG;YACW,0BAA0B;IAkCxC;;;OAGG;YACW,uBAAuB;YAoCvB,iBAAiB;IAwH/B,OAAO,CAAC,gBAAgB;IAyBxB,OAAO,CAAC,0BAA0B;YAYpB,eAAe;YAsCf,0BAA0B;IAkBlC,QAAQ,CACZ,WAAW,EAAE,MAAM,EACnB,cAAc,CAAC,EAAE;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,EAAE;YACR,cAAc,EAAE,MAAM,CAAC;YACvB,QAAQ,EAAE,MAAM,CAAC;YACjB,MAAM,EAAE,MAAM,CAAC;YACf,eAAe,EAAE,MAAM,CAAC;SACzB,CAAC;KACH;CA+dJ"}