@takeshape/schema 11.120.0 → 11.121.0

package/dist/index.d.ts

@@ -21,6 +21,8 @@ export * from './project-schema/index.ts';
 export * from './project-schema/migrate.ts';
 export * from './refs.ts';
 export * from './relationships.ts';
+export * from './resolvers/ai/agent-message-payload.ts';
+export type { AgentSessionPayload } from './resolvers/ai/agent-session-payload.ts';
 export * from './resolvers/takeshape/assets/constants.ts';
 export * from './runtime-schema.ts';
 export * from './scalars.ts';

package/dist/index.js

@@ -19,6 +19,7 @@ export * from "./project-schema/index.js";
 export * from "./project-schema/migrate.js";
 export * from "./refs.js";
 export * from "./relationships.js";
+export * from "./resolvers/ai/agent-message-payload.js";
 export * from "./resolvers/takeshape/assets/constants.js";
 export * from "./runtime-schema.js";
 export * from "./scalars.js";

package/dist/resolvers/ai/agent-chat-response.d.ts

@@ -98,6 +98,10 @@ export declare const schema: {
                     readonly type: "boolean";
                     readonly description: "Indicates whether the agent session has an associated history of agent responses.";
                 };
+                readonly hasFeedback: {
+                    readonly type: "boolean";
+                    readonly description: "Indicates whether the agent session has any messages with feedback.";
+                };
             };
             readonly required: ["id", "agentName", "createdAt", "updatedAt", "done", "currentStateId", "chatSessionIds", "sessionMemory", "stateOutputs", "projectId", "schemaId", "schemaHash"];
             readonly additionalProperties: false;
@@ -131,12 +135,13 @@ export declare const validate: import("ajv").ValidateFunction<{
             [x: string]: unknown;
         } | undefined;
         hasAgentHistory?: boolean | undefined;
-        id: string;
-        createdAt: string;
-        updatedAt: string;
-        agentName: string;
+        hasFeedback?: boolean | undefined;
         projectId: string;
         schemaId: string;
+        agentName: string;
+        createdAt: string;
+        id: string;
+        updatedAt: string;
         schemaHash: string;
         done: boolean;
         currentStateId: string;

package/dist/resolvers/ai/agent-message-payload.d.ts

@@ -1,4 +1,69 @@
+import type { AiUsage } from '@takeshape/util';
+import type { CoreMessage } from 'ai';
 import type { FromSchema } from 'json-schema-to-ts';
+export declare const agentGenerateStateOutputSchema: {
+    readonly title: "AgentGenerateStateOutput";
+    readonly type: "object";
+    readonly properties: {
+        readonly type: {
+            readonly const: "generate";
+        };
+        readonly data: {};
+    };
+    readonly required: ["type", "data"];
+    readonly additionalProperties: false;
+};
+export declare const agentGraphQLStateOutputSchema: {
+    readonly title: "AgentGraphQLStateOutput";
+    readonly type: "object";
+    readonly properties: {
+        readonly type: {
+            readonly const: "graphql";
+        };
+        readonly data: {};
+    };
+    readonly required: ["type", "data"];
+    readonly additionalProperties: false;
+};
+export declare const agentChatStateOutputSchema: {
+    readonly title: "AgentChatStateOutput";
+    readonly type: "object";
+    readonly properties: {
+        readonly type: {
+            readonly const: "chat";
+        };
+        readonly data: {
+            readonly title: "AgentChatStateOutputData";
+            readonly type: "object";
+            readonly properties: {
+                readonly content: {
+                    readonly type: "string";
+                };
+                readonly artifact: {
+                    readonly type: "object";
+                };
+                readonly references: {
+                    readonly type: "array";
+                    readonly items: {
+                        readonly title: "AgentChatStateOutputReference";
+                        readonly type: "object";
+                        readonly properties: {
+                            readonly _tid: {
+                                readonly type: "string";
+                            };
+                        };
+                        readonly required: ["_tid"];
+                        readonly additionalProperties: false;
+                    };
+                };
+            };
+            readonly required: ["content", "references"];
+            readonly additionalProperties: false;
+        };
+    };
+    readonly required: ["type", "data"];
+    readonly additionalProperties: false;
+};
 export declare const schema: {
     readonly $id: "https://takeshape.io/schema/agent-message-payload";
     readonly type: "object";
@@ -23,11 +88,89 @@ export declare const schema: {
             readonly type: "string";
             readonly description: "Name of the agent this message was created by";
         };
-        readonly input: {
+        readonly args: {
             readonly type: "object";
         };
+        readonly input: {
+            readonly type: "string";
+        };
         readonly output: {
-            readonly type: "object";
+            readonly discriminator: {
+                readonly propertyName: "type";
+            };
+            readonly oneOf: [{
+                readonly title: "AgentGenerateStateOutput";
+                readonly type: "object";
+                readonly properties: {
+                    readonly type: {
+                        readonly const: "generate";
+                    };
+                    readonly data: {};
+                };
+                readonly required: ["type", "data"];
+                readonly additionalProperties: false;
+            }, {
+                readonly title: "AgentGraphQLStateOutput";
+                readonly type: "object";
+                readonly properties: {
+                    readonly type: {
+                        readonly const: "graphql";
+                    };
+                    readonly data: {};
+                };
+                readonly required: ["type", "data"];
+                readonly additionalProperties: false;
+            }, {
+                readonly title: "AgentChatStateOutput";
+                readonly type: "object";
+                readonly properties: {
+                    readonly type: {
+                        readonly const: "chat";
+                    };
+                    readonly data: {
+                        readonly title: "AgentChatStateOutputData";
+                        readonly type: "object";
+                        readonly properties: {
+                            readonly content: {
+                                readonly type: "string";
+                            };
+                            readonly artifact: {
+                                readonly type: "object";
+                            };
+                            readonly references: {
+                                readonly type: "array";
+                                readonly items: {
+                                    readonly title: "AgentChatStateOutputReference";
+                                    readonly type: "object";
+                                    readonly properties: {
+                                        readonly _tid: {
+                                            readonly type: "string";
+                                        };
+                                    };
+                                    readonly required: ["_tid"];
+                                    readonly additionalProperties: false;
+                                };
+                            };
+                        };
+                        readonly required: ["content", "references"];
+                        readonly additionalProperties: false;
+                    };
+                };
+                readonly required: ["type", "data"];
+                readonly additionalProperties: false;
+            }, {
+                readonly title: "AgentUnknownStateOutput";
+                readonly type: "object";
+                readonly properties: {
+                    readonly type: {
+                        readonly const: "unknown";
+                    };
+                    readonly data: {};
+                };
+                readonly required: ["type", "data"];
+                readonly additionalProperties: false;
+            }];
+            readonly required: ["type"];
         };
         readonly error: {
             readonly type: "object";
@@ -147,12 +290,44 @@ export declare const schema: {
     readonly additionalProperties: false;
 };
 export type AgentMessagePayload = FromSchema<typeof schema>;
+export type AgentStateOutput = NonNullable<AgentMessagePayload['output']>;
+export type AgentGenerateStateOutput = Extract<AgentMessagePayload['output'], {
+    type: 'generate';
+}>;
+export type AgentGraphQLStateOutput = Extract<AgentMessagePayload['output'], {
+    type: 'graphql';
+}>;
+export type AgentChatStateOutput = Extract<AgentMessagePayload['output'], {
+    type: 'chat';
+}>;
+export type AgentUnknownStateOutput = Extract<AgentMessagePayload['output'], {
+    type: 'unknown';
+}>;
 export declare const validate: import("ajv").ValidateFunction<{
-    input?: {
+    args?: {
         [x: string]: unknown;
     } | undefined;
+    input?: string | undefined;
     output?: {
-        [x: string]: unknown;
+        type: "generate";
+        data: unknown;
+    } | {
+        type: "graphql";
+        data: unknown;
+    } | {
+        type: "chat";
+        data: {
+            artifact?: {
+                [x: string]: unknown;
+            } | undefined;
+            content: string;
+            references: {
+                _tid: string;
+            }[];
+        };
+    } | {
+        type: "unknown";
+        data: unknown;
     } | undefined;
     error?: {
         message?: string | undefined;
@@ -163,10 +338,10 @@ export declare const validate: import("ajv").ValidateFunction<{
     startedAt?: number | undefined;
     finishedAt?: number | undefined;
     span?: {
-        createdAt: number;
         name: string;
         traceId: string;
         spanId: string;
+        createdAt: number;
     } | undefined;
     feedback?: {
         createdAt?: number | undefined;
@@ -185,10 +360,31 @@ export declare const validate: import("ajv").ValidateFunction<{
         guardrailUnits?: number | undefined;
     } | undefined;
     messageId: string;
-    agentName: string;
+    sessionId: string;
     projectId: string;
     schemaId: string;
-    sessionId: string;
+    agentName: string;
     runMode: "BACKGROUND" | "IMMEDIATE";
     sentAt: number;
 }>;
+export type AgentChatStateExecuteResult = {
+    input: string;
+    output: AgentChatStateOutput;
+    metadata: {
+        agentHistoryUpdate: CoreMessage[] | null;
+        usage: AiUsage;
+        blocked: boolean;
+    };
+};
+export type AgentGenerateStateExecuteResult = {
+    input: string;
+    output: AgentGenerateStateOutput;
+    metadata: {
+        usage: AiUsage;
+        blocked: boolean;
+    };
+};
+export type AgentGraphQLStateExecuteResult = {
+    output: AgentGraphQLStateOutput;
+};
+export type AgentStateExecuteResult = AgentChatStateExecuteResult | AgentGenerateStateExecuteResult | AgentGraphQLStateExecuteResult;

package/dist/resolvers/ai/agent-message-payload.js

@@ -1,4 +1,65 @@
 import { createTypedValidator } from '@takeshape/json-schema';
+export const agentGenerateStateOutputSchema = {
+    title: 'AgentGenerateStateOutput',
+    type: 'object',
+    properties: {
+        type: { const: 'generate' },
+        data: {}
+    },
+    required: ['type', 'data'],
+    additionalProperties: false
+};
+export const agentGraphQLStateOutputSchema = {
+    title: 'AgentGraphQLStateOutput',
+    type: 'object',
+    properties: {
+        type: { const: 'graphql' },
+        data: {}
+    },
+    required: ['type', 'data'],
+    additionalProperties: false
+};
+export const agentChatStateOutputSchema = {
+    title: 'AgentChatStateOutput',
+    type: 'object',
+    properties: {
+        type: { const: 'chat' },
+        data: {
+            title: 'AgentChatStateOutputData',
+            type: 'object',
+            properties: {
+                content: { type: 'string' },
+                artifact: { type: 'object' },
+                references: {
+                    type: 'array',
+                    items: {
+                        title: 'AgentChatStateOutputReference',
+                        type: 'object',
+                        properties: {
+                            _tid: { type: 'string' }
+                        },
+                        required: ['_tid'],
+                        additionalProperties: false
+                    }
+                }
+            },
+            required: ['content', 'references'],
+            additionalProperties: false
+        }
+    },
+    required: ['type', 'data'],
+    additionalProperties: false
+};
+const agentUnknownStateOutputSchema = {
+    title: 'AgentUnknownStateOutput',
+    type: 'object',
+    properties: {
+        type: { const: 'unknown' },
+        data: {}
+    },
+    required: ['type', 'data'],
+    additionalProperties: false
+};
 export const schema = {
     $id: 'https://takeshape.io/schema/agent-message-payload',
     type: 'object',
@@ -23,11 +84,23 @@ export const schema = {
             type: 'string',
             description: 'Name of the agent this message was created by'
         },
-        input: {
+        args: {
             type: 'object'
         },
+        input: {
+            type: 'string'
+        },
         output: {
-            type: 'object'
+            discriminator: {
+                propertyName: 'type'
+            },
+            oneOf: [
+                agentGenerateStateOutputSchema,
+                agentGraphQLStateOutputSchema,
+                agentChatStateOutputSchema,
+                agentUnknownStateOutputSchema
+            ],
+            required: ['type']
         },
         error: {
             type: 'object',

package/dist/resolvers/ai/agent-session-payload.d.ts

@@ -1,5 +1,5 @@
-import type { AgentExecutionOutputValue, AgentExecutionResultValue } from '@takeshape/util';
 import type { FromSchema } from 'json-schema-to-ts';
+import type { AgentStateExecuteResult, AgentStateOutput } from './agent-message-payload.ts';
 export declare const schema: {
     readonly type: "object";
     readonly properties: {
@@ -69,13 +69,17 @@ export declare const schema: {
             readonly type: "boolean";
             readonly description: "Indicates whether the agent session has an associated history of agent responses.";
         };
+        readonly hasFeedback: {
+            readonly type: "boolean";
+            readonly description: "Indicates whether the agent session has any messages with feedback.";
+        };
     };
     readonly required: ["id", "agentName", "createdAt", "updatedAt", "done", "currentStateId", "chatSessionIds", "sessionMemory", "stateOutputs", "projectId", "schemaId", "schemaHash"];
     readonly additionalProperties: false;
 };
 export type AgentSessionPayload = FromSchema<typeof schema> & {
-    stateOutputs: Record<string, AgentExecutionOutputValue | undefined>;
+    stateOutputs: Record<string, AgentStateOutput | undefined>;
     suspended?: boolean;
-    currentValue?: AgentExecutionResultValue;
+    currentValue?: Omit<AgentStateExecuteResult, 'metadata'>;
 };
 export declare const validate: import("ajv").ValidateFunction<AgentSessionPayload>;

package/dist/resolvers/ai/agent-session-payload.js

@@ -67,6 +67,10 @@ export const schema = {
         hasAgentHistory: {
             type: 'boolean',
             description: 'Indicates whether the agent session has an associated history of agent responses.'
+        },
+        hasFeedback: {
+            type: 'boolean',
+            description: 'Indicates whether the agent session has any messages with feedback.'
         }
     },
     required: [

package/dist/resolvers/ai/create-agent-session-response.d.ts

@@ -2,24 +2,24 @@ import type { FromSchema } from 'json-schema-to-ts';
 export declare const shapeName = "TSCreateAgentSessionResponse";
 export declare const schema: {
     readonly properties: {
-        readonly id: {
+        readonly agentName: {
             readonly type: "string";
-            readonly description: "The unique identifier for the agent session.";
+            readonly description: "The name of the agent associated with this session.";
         };
         readonly createdAt: {
             readonly type: "string";
             readonly description: "ISO8601 timestamp when the agent session was created";
             readonly format: "date-time";
         };
+        readonly id: {
+            readonly type: "string";
+            readonly description: "The unique identifier for the agent session.";
+        };
         readonly updatedAt: {
             readonly type: "string";
             readonly description: "ISO8601 timestamp when the agent session was updated";
             readonly format: "date-time";
         };
-        readonly agentName: {
-            readonly type: "string";
-            readonly description: "The name of the agent associated with this session.";
-        };
         readonly schemaHash: {
             readonly type: "string";
             readonly description: "The hash of the schema version associated with the agent session.";
@@ -60,6 +60,10 @@ export declare const schema: {
             readonly type: "boolean";
             readonly description: "Indicates whether the agent session has an associated history of agent responses.";
         };
+        readonly hasFeedback: {
+            readonly type: "boolean";
+            readonly description: "Indicates whether the agent session has any messages with feedback.";
+        };
     };
     readonly required: ["id", "agentName", "done", "currentStateId", "chatSessionIds", "sessionMemory", "stateOutputs"];
     readonly additionalProperties: false;
@@ -76,8 +80,9 @@ export declare const validate: import("ajv").ValidateFunction<{
         [x: string]: unknown;
     } | undefined;
     hasAgentHistory?: boolean | undefined;
-    id: string;
+    hasFeedback?: boolean | undefined;
     agentName: string;
+    id: string;
     done: boolean;
     currentStateId: string;
     chatSessionIds: {

package/dist/resolvers/ai/get-agent-message-response.d.ts

@@ -83,6 +83,10 @@ export declare const schema: {
                     readonly type: "boolean";
                     readonly description: "Indicates whether the agent session has an associated history of agent responses.";
                 };
+                readonly hasFeedback: {
+                    readonly type: "boolean";
+                    readonly description: "Indicates whether the agent session has any messages with feedback.";
+                };
             };
             readonly required: ["id", "agentName", "createdAt", "updatedAt", "done", "currentStateId", "chatSessionIds", "sessionMemory", "stateOutputs", "projectId", "schemaId", "schemaHash"];
             readonly additionalProperties: false;
@@ -95,9 +99,12 @@ export declare const schema: {
             readonly type: "string";
             readonly description: "Name of the agent this message was created by";
         };
-        readonly input: {
+        readonly args: {
             readonly type: "object";
         };
+        readonly input: {
+            readonly type: "string";
+        };
         readonly error: {
             readonly type: "object";
             readonly properties: {

package/dist/resolvers/ai/inspect-agent-response.d.ts

@@ -18,24 +18,24 @@ export declare const schema: {
             };
             readonly description: "A list of messages exchanged in the agent session, which may include user inputs and agent responses.";
         };
-        readonly id: {
+        readonly agentName: {
             readonly type: "string";
-            readonly description: "The unique identifier for the agent session.";
+            readonly description: "The name of the agent associated with this session.";
         };
         readonly createdAt: {
             readonly type: "string";
             readonly description: "ISO8601 timestamp when the agent session was created";
             readonly format: "date-time";
         };
+        readonly id: {
+            readonly type: "string";
+            readonly description: "The unique identifier for the agent session.";
+        };
         readonly updatedAt: {
             readonly type: "string";
             readonly description: "ISO8601 timestamp when the agent session was updated";
             readonly format: "date-time";
         };
-        readonly agentName: {
-            readonly type: "string";
-            readonly description: "The name of the agent associated with this session.";
-        };
         readonly schemaHash: {
             readonly type: "string";
             readonly description: "The hash of the schema version associated with the agent session.";
@@ -76,6 +76,10 @@ export declare const schema: {
             readonly type: "boolean";
             readonly description: "Indicates whether the agent session has an associated history of agent responses.";
         };
+        readonly hasFeedback: {
+            readonly type: "boolean";
+            readonly description: "Indicates whether the agent session has any messages with feedback.";
+        };
     };
     readonly required: ["id", "agentName", "done", "currentStateId", "chatSessionIds", "sessionMemory", "stateOutputs", "agentHistory", "messages"];
     readonly additionalProperties: false;

package/dist/resolvers/ai/send-agent-message-response.d.ts

@@ -75,6 +75,10 @@ export declare const schema: {
                     readonly type: "boolean";
                     readonly description: "Indicates whether the agent session has an associated history of agent responses.";
                 };
+                readonly hasFeedback: {
+                    readonly type: "boolean";
+                    readonly description: "Indicates whether the agent session has any messages with feedback.";
+                };
             };
             readonly required: ["id", "agentName", "createdAt", "updatedAt", "done", "currentStateId", "chatSessionIds", "sessionMemory", "stateOutputs", "projectId", "schemaId", "schemaHash"];
             readonly additionalProperties: false;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@takeshape/schema",
-  "version": "11.120.0",
+  "version": "11.121.0",
   "description": "TakeShape Schema",
   "homepage": "https://www.takeshape.io",
   "repository": {
@@ -56,9 +56,9 @@
     "p-reduce": "2.1.0",
     "semver": "7.7.2",
     "tiny-invariant": "1.3.3",
-    "@takeshape/errors": "11.120.0",
-    "@takeshape/util": "11.120.0",
-    "@takeshape/json-schema": "11.120.0"
+    "@takeshape/json-schema": "11.121.0",
+    "@takeshape/errors": "11.121.0",
+    "@takeshape/util": "11.121.0"
   },
   "devDependencies": {
     "@takeshape/json-schema-to-typescript": "11.0.0",
@@ -75,8 +75,8 @@
     "glob": "7.2.3",
     "json-schema-to-ts": "3.1.1",
     "shortid": "2.2.16",
-    "@takeshape/infra": "11.120.0",
-    "@takeshape/logger": "11.120.0"
+    "@takeshape/infra": "11.121.0",
+    "@takeshape/logger": "11.121.0"
   },
   "engines": {
     "node": ">=22"