@takeshape/schema 11.63.7 → 11.66.0

package/dist/__tests__/validate.test.js

@@ -1951,4 +1951,30 @@ describe('latest', () => {
             }
         ]);
     });
+    test('validateSchema - invalid expressions', async () => {
+        const invalidSchema = agentSchema();
+        const { bbb, ccc, ddd } = invalidSchema['ai-experimental'].agents.findDogColor.states;
+        bbb.variables[0].steps[0].expression = ')';
+        ccc.transition[0].condition = '(';
+        ddd.variables[0].steps[0].condition = 'foo';
+        const { valid, errors } = await validateSchema(validateContext, invalidSchema);
+        expect(valid).toEqual(false);
+        expect(errors).toEqual([
+            {
+                message: 'Unexpected ")" at character 0',
+                path: ['ai-experimental', 'agents', 'findDogColor', 'states', 'bbb', 'variables', 0, 'steps', 0, 'expression'],
+                type: 'conflict'
+            },
+            {
+                message: 'Unclosed ( at character 1',
+                path: ['ai-experimental', 'agents', 'findDogColor', 'states', 'ccc', 'transition', 0, 'expression'],
+                type: 'conflict'
+            },
+            {
+                message: 'Unknown variable "foo"',
+                path: ['ai-experimental', 'agents', 'findDogColor', 'states', 'ddd', 'variables', 0, 'steps', 0, 'condition'],
+                type: 'conflict'
+            }
+        ]);
+    });
 });

package/dist/agents.d.ts

@@ -1,4 +1,4 @@
-import type { AgentAPIArgument, AgentJSON, Args, ProjectSchemaJSON } from './project-schema/index.ts';
+import type { AgentAPI, AgentAPIArgument, AgentJSON, Args, ProjectSchemaJSON } from './project-schema/index.ts';
 export declare const END_AGENT_EXECUTION = "endAgentExecution";
 export declare const BUILT_IN_CHAT_ARGS: AgentAPIArgument[];
 export declare const BUILT_IN_CHAT_ARG_NAMES: string[];
@@ -10,5 +10,6 @@ export declare const getAgentEndTransitions: (agent: AgentJSON, includeSuspend?:
 export declare const getAgentEndStates: (agent: AgentJSON, includeSuspend?: boolean) => Set<string>;
 export declare const getInspectAgentSessionQueryName: (agentName: string) => string;
 export declare const removeBuiltInArgs: (args: AgentAPIArgument[]) => AgentAPIArgument[];
+export declare function getAgentApiArgs(api: AgentAPI): AgentAPIArgument[];
 export declare const createArgs: (agent: AgentJSON) => Args | undefined;
 export declare function addAiQueries(projectSchema: ProjectSchemaJSON): ProjectSchemaJSON;

package/dist/agents.js

@@ -27,7 +27,7 @@ const transitionCanEnd = (transition, includeSuspend = false) => {
         if (step.destination === END_AGENT_EXECUTION) {
             return true;
         }
-        if (step.condition.type === 'none') {
+        if (step.condition === undefined) {
             return false;
         }
     }
@@ -54,11 +54,15 @@ export const getInspectAgentSessionQueryName = (agentName) => {
 export const removeBuiltInArgs = (args) => {
     return args.filter((arg) => !BUILT_IN_CHAT_ARG_NAMES.includes(arg.argName));
 };
-export const createArgs = (agent) => {
-    let apiArguments = uniqBy(agent.api.arguments ?? [], (arg) => arg.argName);
-    if (agent.api.type === 'chat') {
+export function getAgentApiArgs(api) {
+    let apiArguments = uniqBy(api.arguments ?? [], (arg) => arg.argName);
+    if (api.type === 'chat') {
         apiArguments = removeBuiltInArgs(apiArguments).concat(BUILT_IN_CHAT_ARGS);
     }
+    return apiArguments;
+}
+export const createArgs = (agent) => {
+    const apiArguments = getAgentApiArgs(agent.api);
     return apiArguments.length > 0
         ? {
             type: 'object',

package/dist/index.d.ts

@@ -34,6 +34,7 @@ export * from './unions.ts';
 export * from './util/ai-tools.ts';
 export * from './util/api-indexing.ts';
 export * from './util/detect-cycles.ts';
+export * from './util/expressions.ts';
 export * from './util/find-shape-at-path.ts';
 export * from './util/form-config.ts';
 export * from './util/get-conflicting-properties.ts';

package/dist/index.js

@@ -32,6 +32,7 @@ export * from "./unions.js";
 export * from "./util/ai-tools.js";
 export * from "./util/api-indexing.js";
 export * from "./util/detect-cycles.js";
+export * from "./util/expressions.js";
 export * from "./util/find-shape-at-path.js";
 export * from "./util/form-config.js";
 export * from "./util/get-conflicting-properties.js";

package/dist/project-schema/latest.d.ts

@@ -285,16 +285,10 @@ export type CacheTriggerConfig = CacheScheduleTriggerConfig | CacheWebhookTrigge
 export type ShapeSchema = ShapeSchemaAllOf | ShapeSchemaExtends | ShapeSchemaOneOf | ShapeSchemaEnum | ObjectSchema | ShapeSchemaAny;
 export type AgentAPI = AgentAPIChat | AgentAPIGenerate;
 export type MemoryLocations = AgentSessionMemoryLocation[];
-export type AgentVariableStep = AgentVariableStepSessionMemory | AgentVariableStepGraphQLArg | AgentVariableStepStateOutput | AgentVariableStepPreviousStateOutput | AgentVariableStepStaticValue;
-export type AgentVariableStepCondition = AgentVariableStepConditionNone | AgentVariableStepConditionSourceState;
 export type Variables = AgentVariable[];
 export type AgentExecution = AgentExecutionGraphQL | AgentExecutionGenerate | AgentExecutionChat;
 export type AgentAIStateInput = AgentAIStateInputArgument | AgentAIStateInputTemplate;
-export type MinimatchPattern = string;
-/**
- * Remove properties from the schema sent to the model with keys matching these minimatch glob patterns.
- */
-export type RemovePropertyKeyPatterns = MinimatchPattern[];
+export type AgentToolConfigArg = AgentToolConfigArgAgent | AgentToolConfigArgVariable;
 export type History1 = boolean;
 /**
  * Maximum number of tokens to generate.
@@ -340,6 +334,11 @@ export type StructuredOutputs = boolean;
  * Enable OpenAI schema format. Must be enabled if `structuredOutputs` is enabled.
  */
 export type EnableOpenAISchemaFormat = boolean;
+export type MinimatchPattern = string;
+/**
+ * Remove properties from the schema sent to the model with keys matching these minimatch glob patterns.
+ */
+export type RemovePropertyKeyPatterns = MinimatchPattern[];
 export type AgentStateSessionMemoryLocation = AgentStateSessionMemoryLocationArtifact | AgentStateSessionMemoryLocationStaticValue;
 export type AgentStateSessionMemory = AgentStateSessionMemoryLocation[];
 export type GuardID = string;
@@ -1762,34 +1761,9 @@ export interface AgentStart {
 export interface AgentTransitionStep {
     destination: string;
     suspend?: boolean;
-    condition: AgentTransitionStepConditionNone | AgentTransitionStepConditionArtifact | AgentTransitionStepConditionStateOutput | AgentTransitionStepConditionStringContains | AgentTransitionStepConditionArgument;
+    condition?: string;
     limit?: number;
 }
-export interface AgentTransitionStepConditionNone {
-    type: 'none';
-}
-export interface AgentTransitionStepConditionArtifact {
-    type: 'artifact';
-    path?: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionStateOutput {
-    type: 'stateOutput';
-    stateId: string;
-    path?: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionStringContains {
-    type: 'stringContains';
-    path?: string;
-    string: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionArgument {
-    type: 'argument';
-    argument: string;
-    negated?: boolean;
-}
 /**
  * States that are traversed during the execution of an agent.
  */
@@ -1816,39 +1790,9 @@ export interface AgentVariable {
      */
     steps: AgentVariableStep[];
 }
-export interface AgentVariableStepSessionMemory {
-    type: 'sessionMemory';
-    memoryKey: string;
-    path?: string;
-}
-export interface AgentVariableStepGraphQLArg {
-    type: 'graphqlArg';
-    argName: string;
-}
-export interface AgentVariableStepStateOutput {
-    type: 'stateOutput';
-    stateId: string;
-    path?: string;
-}
-export interface AgentVariableStepPreviousStateOutput {
-    type: 'previousStateOutput';
-    path?: string;
-}
-/**
- * An Agent Variable that has a static value.
- */
-export interface AgentVariableStepStaticValue {
-    type: 'staticValue';
-    condition?: AgentVariableStepCondition;
-    value: string;
-}
-export interface AgentVariableStepConditionNone {
-    type: 'none';
-}
-export interface AgentVariableStepConditionSourceState {
-    type: 'sourceState';
-    stateId: string;
-    negated?: boolean;
+export interface AgentVariableStep {
+    condition?: string;
+    expression: string;
 }
 export interface AgentExecutionGraphQL {
     type: 'graphql';
@@ -1873,10 +1817,28 @@ export interface AgentAIStateInputTemplate {
     inputTemplate: string;
 }
 export interface AgentToolConfig {
-    description?: string;
     ref: string;
+    description?: string;
+    args?: AgentToolConfigArg[];
     selectionSet?: string;
-    removePropertyKeyPatterns?: RemovePropertyKeyPatterns;
+}
+export interface AgentToolConfigArgAgent {
+    type: 'agent';
+    /**
+     * The argument name that will be exposed by the tool for the LLM to use.
+     */
+    argName: string;
+}
+export interface AgentToolConfigArgVariable {
+    type: 'variable';
+    /**
+     * Name of argument to pass value to. This prevents the LLM from setting the argument.
+     */
+    argName: string;
+    /**
+     * Name of variable to pull value from.
+     */
+    variableName: string;
 }
 export interface AgentGenerateOptions {
     history?: History1;

package/dist/project-schema/v3.48.0.d.ts

@@ -264,16 +264,10 @@ export type CacheTriggerConfigV3_48_0 = CacheScheduleTriggerConfigV3_48_0 | Cach
 export type ShapeSchemaV3_48_0 = ShapeSchemaAllOfV3_48_0 | ShapeSchemaExtendsV3_48_0 | ShapeSchemaOneOfV3_48_0 | ShapeSchemaEnumV3_48_0 | ObjectSchemaV3_48_0 | ShapeSchemaAnyV3_48_0;
 export type AgentAPIV3_48_0 = AgentAPIChatV3_48_0 | AgentAPIGenerateV3_48_0;
 export type MemoryLocationsV3_48_0 = AgentSessionMemoryLocationV3_48_0[];
-export type AgentVariableStepV3_48_0 = AgentVariableStepSessionMemoryV3_48_0 | AgentVariableStepGraphQLArgV3_48_0 | AgentVariableStepStateOutputV3_48_0 | AgentVariableStepPreviousStateOutputV3_48_0 | AgentVariableStepStaticValueV3_48_0;
-export type AgentVariableStepConditionV3_48_0 = AgentVariableStepConditionNoneV3_48_0 | AgentVariableStepConditionSourceStateV3_48_0;
 export type VariablesV3_48_0 = AgentVariableV3_48_0[];
 export type AgentExecutionV3_48_0 = AgentExecutionGraphQLV3_48_0 | AgentExecutionGenerateV3_48_0 | AgentExecutionChatV3_48_0;
 export type AgentAIStateInputV3_48_0 = AgentAIStateInputArgumentV3_48_0 | AgentAIStateInputTemplateV3_48_0;
-export type MinimatchPatternV3_48_0 = string;
-/**
- * Remove properties from the schema sent to the model with keys matching these minimatch glob patterns.
- */
-export type RemovePropertyKeyPatternsV3_48_0 = MinimatchPatternV3_48_0[];
+export type AgentToolConfigArgV3_48_0 = AgentToolConfigArgAgentV3_48_0 | AgentToolConfigArgVariableV3_48_0;
 export type HistoryV3_48_01 = boolean;
 /**
  * Maximum number of tokens to generate.
@@ -319,6 +313,11 @@ export type StructuredOutputsV3_48_0 = boolean;
  * Enable OpenAI schema format. Must be enabled if `structuredOutputs` is enabled.
  */
 export type EnableOpenAISchemaFormatV3_48_0 = boolean;
+export type MinimatchPatternV3_48_0 = string;
+/**
+ * Remove properties from the schema sent to the model with keys matching these minimatch glob patterns.
+ */
+export type RemovePropertyKeyPatternsV3_48_0 = MinimatchPatternV3_48_0[];
 export type AgentStateSessionMemoryLocationV3_48_0 = AgentStateSessionMemoryLocationArtifactV3_48_0 | AgentStateSessionMemoryLocationStaticValueV3_48_0;
 export type AgentStateSessionMemoryV3_48_0 = AgentStateSessionMemoryLocationV3_48_0[];
 export type GuardIDV3_48_0 = string;
@@ -1629,34 +1628,9 @@ export interface AgentStartV3_48_0 {
 export interface AgentTransitionStepV3_48_0 {
     destination: string;
     suspend?: boolean;
-    condition: AgentTransitionStepConditionNoneV3_48_0 | AgentTransitionStepConditionArtifactV3_48_0 | AgentTransitionStepConditionStateOutputV3_48_0 | AgentTransitionStepConditionStringContainsV3_48_0 | AgentTransitionStepConditionArgumentV3_48_0;
+    condition?: string;
     limit?: number;
 }
-export interface AgentTransitionStepConditionNoneV3_48_0 {
-    type: 'none';
-}
-export interface AgentTransitionStepConditionArtifactV3_48_0 {
-    type: 'artifact';
-    path?: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionStateOutputV3_48_0 {
-    type: 'stateOutput';
-    stateId: string;
-    path?: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionStringContainsV3_48_0 {
-    type: 'stringContains';
-    path?: string;
-    string: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionArgumentV3_48_0 {
-    type: 'argument';
-    argument: string;
-    negated?: boolean;
-}
 /**
  * States that are traversed during the execution of an agent.
  */
@@ -1683,39 +1657,9 @@ export interface AgentVariableV3_48_0 {
      */
     steps: AgentVariableStepV3_48_0[];
 }
-export interface AgentVariableStepSessionMemoryV3_48_0 {
-    type: 'sessionMemory';
-    memoryKey: string;
-    path?: string;
-}
-export interface AgentVariableStepGraphQLArgV3_48_0 {
-    type: 'graphqlArg';
-    argName: string;
-}
-export interface AgentVariableStepStateOutputV3_48_0 {
-    type: 'stateOutput';
-    stateId: string;
-    path?: string;
-}
-export interface AgentVariableStepPreviousStateOutputV3_48_0 {
-    type: 'previousStateOutput';
-    path?: string;
-}
-/**
- * An Agent Variable that has a static value.
- */
-export interface AgentVariableStepStaticValueV3_48_0 {
-    type: 'staticValue';
-    condition?: AgentVariableStepConditionV3_48_0;
-    value: string;
-}
-export interface AgentVariableStepConditionNoneV3_48_0 {
-    type: 'none';
-}
-export interface AgentVariableStepConditionSourceStateV3_48_0 {
-    type: 'sourceState';
-    stateId: string;
-    negated?: boolean;
+export interface AgentVariableStepV3_48_0 {
+    condition?: string;
+    expression: string;
 }
 export interface AgentExecutionGraphQLV3_48_0 {
     type: 'graphql';
@@ -1740,10 +1684,28 @@ export interface AgentAIStateInputTemplateV3_48_0 {
     inputTemplate: string;
 }
 export interface AgentToolConfigV3_48_0 {
-    description?: string;
     ref: string;
+    description?: string;
+    args?: AgentToolConfigArgV3_48_0[];
     selectionSet?: string;
-    removePropertyKeyPatterns?: RemovePropertyKeyPatternsV3_48_0;
+}
+export interface AgentToolConfigArgAgentV3_48_0 {
+    type: 'agent';
+    /**
+     * The argument name that will be exposed by the tool for the LLM to use.
+     */
+    argName: string;
+}
+export interface AgentToolConfigArgVariableV3_48_0 {
+    type: 'variable';
+    /**
+     * Name of argument to pass value to. This prevents the LLM from setting the argument.
+     */
+    argName: string;
+    /**
+     * Name of variable to pull value from.
+     */
+    variableName: string;
 }
 export interface AgentGenerateOptionsV3_48_0 {
     history?: HistoryV3_48_01;

package/dist/project-schema/v3.49.0.d.ts

@@ -264,16 +264,10 @@ export type CacheTriggerConfigV3_49_0 = CacheScheduleTriggerConfigV3_49_0 | Cach
 export type ShapeSchemaV3_49_0 = ShapeSchemaAllOfV3_49_0 | ShapeSchemaExtendsV3_49_0 | ShapeSchemaOneOfV3_49_0 | ShapeSchemaEnumV3_49_0 | ObjectSchemaV3_49_0 | ShapeSchemaAnyV3_49_0;
 export type AgentAPIV3_49_0 = AgentAPIChatV3_49_0 | AgentAPIGenerateV3_49_0;
 export type MemoryLocationsV3_49_0 = AgentSessionMemoryLocationV3_49_0[];
-export type AgentVariableStepV3_49_0 = AgentVariableStepSessionMemoryV3_49_0 | AgentVariableStepGraphQLArgV3_49_0 | AgentVariableStepStateOutputV3_49_0 | AgentVariableStepPreviousStateOutputV3_49_0 | AgentVariableStepStaticValueV3_49_0;
-export type AgentVariableStepConditionV3_49_0 = AgentVariableStepConditionNoneV3_49_0 | AgentVariableStepConditionSourceStateV3_49_0;
 export type VariablesV3_49_0 = AgentVariableV3_49_0[];
 export type AgentExecutionV3_49_0 = AgentExecutionGraphQLV3_49_0 | AgentExecutionGenerateV3_49_0 | AgentExecutionChatV3_49_0;
 export type AgentAIStateInputV3_49_0 = AgentAIStateInputArgumentV3_49_0 | AgentAIStateInputTemplateV3_49_0;
-export type MinimatchPatternV3_49_0 = string;
-/**
- * Remove properties from the schema sent to the model with keys matching these minimatch glob patterns.
- */
-export type RemovePropertyKeyPatternsV3_49_0 = MinimatchPatternV3_49_0[];
+export type AgentToolConfigArgV3_49_0 = AgentToolConfigArgAgentV3_49_0 | AgentToolConfigArgVariableV3_49_0;
 export type HistoryV3_49_01 = boolean;
 /**
  * Maximum number of tokens to generate.
@@ -319,6 +313,11 @@ export type StructuredOutputsV3_49_0 = boolean;
  * Enable OpenAI schema format. Must be enabled if `structuredOutputs` is enabled.
  */
 export type EnableOpenAISchemaFormatV3_49_0 = boolean;
+export type MinimatchPatternV3_49_0 = string;
+/**
+ * Remove properties from the schema sent to the model with keys matching these minimatch glob patterns.
+ */
+export type RemovePropertyKeyPatternsV3_49_0 = MinimatchPatternV3_49_0[];
 export type AgentStateSessionMemoryLocationV3_49_0 = AgentStateSessionMemoryLocationArtifactV3_49_0 | AgentStateSessionMemoryLocationStaticValueV3_49_0;
 export type AgentStateSessionMemoryV3_49_0 = AgentStateSessionMemoryLocationV3_49_0[];
 export type GuardIDV3_49_0 = string;
@@ -1629,34 +1628,9 @@ export interface AgentStartV3_49_0 {
 export interface AgentTransitionStepV3_49_0 {
     destination: string;
     suspend?: boolean;
-    condition: AgentTransitionStepConditionNoneV3_49_0 | AgentTransitionStepConditionArtifactV3_49_0 | AgentTransitionStepConditionStateOutputV3_49_0 | AgentTransitionStepConditionStringContainsV3_49_0 | AgentTransitionStepConditionArgumentV3_49_0;
+    condition?: string;
     limit?: number;
 }
-export interface AgentTransitionStepConditionNoneV3_49_0 {
-    type: 'none';
-}
-export interface AgentTransitionStepConditionArtifactV3_49_0 {
-    type: 'artifact';
-    path?: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionStateOutputV3_49_0 {
-    type: 'stateOutput';
-    stateId: string;
-    path?: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionStringContainsV3_49_0 {
-    type: 'stringContains';
-    path?: string;
-    string: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionArgumentV3_49_0 {
-    type: 'argument';
-    argument: string;
-    negated?: boolean;
-}
 /**
  * States that are traversed during the execution of an agent.
  */
@@ -1683,39 +1657,9 @@ export interface AgentVariableV3_49_0 {
      */
     steps: AgentVariableStepV3_49_0[];
 }
-export interface AgentVariableStepSessionMemoryV3_49_0 {
-    type: 'sessionMemory';
-    memoryKey: string;
-    path?: string;
-}
-export interface AgentVariableStepGraphQLArgV3_49_0 {
-    type: 'graphqlArg';
-    argName: string;
-}
-export interface AgentVariableStepStateOutputV3_49_0 {
-    type: 'stateOutput';
-    stateId: string;
-    path?: string;
-}
-export interface AgentVariableStepPreviousStateOutputV3_49_0 {
-    type: 'previousStateOutput';
-    path?: string;
-}
-/**
- * An Agent Variable that has a static value.
- */
-export interface AgentVariableStepStaticValueV3_49_0 {
-    type: 'staticValue';
-    condition?: AgentVariableStepConditionV3_49_0;
-    value: string;
-}
-export interface AgentVariableStepConditionNoneV3_49_0 {
-    type: 'none';
-}
-export interface AgentVariableStepConditionSourceStateV3_49_0 {
-    type: 'sourceState';
-    stateId: string;
-    negated?: boolean;
+export interface AgentVariableStepV3_49_0 {
+    condition?: string;
+    expression: string;
 }
 export interface AgentExecutionGraphQLV3_49_0 {
     type: 'graphql';
@@ -1740,10 +1684,28 @@ export interface AgentAIStateInputTemplateV3_49_0 {
     inputTemplate: string;
 }
 export interface AgentToolConfigV3_49_0 {
-    description?: string;
     ref: string;
+    description?: string;
+    args?: AgentToolConfigArgV3_49_0[];
     selectionSet?: string;
-    removePropertyKeyPatterns?: RemovePropertyKeyPatternsV3_49_0;
+}
+export interface AgentToolConfigArgAgentV3_49_0 {
+    type: 'agent';
+    /**
+     * The argument name that will be exposed by the tool for the LLM to use.
+     */
+    argName: string;
+}
+export interface AgentToolConfigArgVariableV3_49_0 {
+    type: 'variable';
+    /**
+     * Name of argument to pass value to. This prevents the LLM from setting the argument.
+     */
+    argName: string;
+    /**
+     * Name of variable to pull value from.
+     */
+    variableName: string;
 }
 export interface AgentGenerateOptionsV3_49_0 {
     history?: HistoryV3_49_01;

package/dist/project-schema/v3.50.0.d.ts

@@ -276,16 +276,10 @@ export type CacheTriggerConfigV3_50_0 = CacheScheduleTriggerConfigV3_50_0 | Cach
 export type ShapeSchemaV3_50_0 = ShapeSchemaAllOfV3_50_0 | ShapeSchemaExtendsV3_50_0 | ShapeSchemaOneOfV3_50_0 | ShapeSchemaEnumV3_50_0 | ObjectSchemaV3_50_0 | ShapeSchemaAnyV3_50_0;
 export type AgentAPIV3_50_0 = AgentAPIChatV3_50_0 | AgentAPIGenerateV3_50_0;
 export type MemoryLocationsV3_50_0 = AgentSessionMemoryLocationV3_50_0[];
-export type AgentVariableStepV3_50_0 = AgentVariableStepSessionMemoryV3_50_0 | AgentVariableStepGraphQLArgV3_50_0 | AgentVariableStepStateOutputV3_50_0 | AgentVariableStepPreviousStateOutputV3_50_0 | AgentVariableStepStaticValueV3_50_0;
-export type AgentVariableStepConditionV3_50_0 = AgentVariableStepConditionNoneV3_50_0 | AgentVariableStepConditionSourceStateV3_50_0;
 export type VariablesV3_50_0 = AgentVariableV3_50_0[];
 export type AgentExecutionV3_50_0 = AgentExecutionGraphQLV3_50_0 | AgentExecutionGenerateV3_50_0 | AgentExecutionChatV3_50_0;
 export type AgentAIStateInputV3_50_0 = AgentAIStateInputArgumentV3_50_0 | AgentAIStateInputTemplateV3_50_0;
-export type MinimatchPatternV3_50_0 = string;
-/**
- * Remove properties from the schema sent to the model with keys matching these minimatch glob patterns.
- */
-export type RemovePropertyKeyPatternsV3_50_0 = MinimatchPatternV3_50_0[];
+export type AgentToolConfigArgV3_50_0 = AgentToolConfigArgAgentV3_50_0 | AgentToolConfigArgVariableV3_50_0;
 export type HistoryV3_50_01 = boolean;
 /**
  * Maximum number of tokens to generate.
@@ -331,6 +325,11 @@ export type StructuredOutputsV3_50_0 = boolean;
  * Enable OpenAI schema format. Must be enabled if `structuredOutputs` is enabled.
  */
 export type EnableOpenAISchemaFormatV3_50_0 = boolean;
+export type MinimatchPatternV3_50_0 = string;
+/**
+ * Remove properties from the schema sent to the model with keys matching these minimatch glob patterns.
+ */
+export type RemovePropertyKeyPatternsV3_50_0 = MinimatchPatternV3_50_0[];
 export type AgentStateSessionMemoryLocationV3_50_0 = AgentStateSessionMemoryLocationArtifactV3_50_0 | AgentStateSessionMemoryLocationStaticValueV3_50_0;
 export type AgentStateSessionMemoryV3_50_0 = AgentStateSessionMemoryLocationV3_50_0[];
 export type GuardIDV3_50_0 = string;
@@ -1681,34 +1680,9 @@ export interface AgentStartV3_50_0 {
 export interface AgentTransitionStepV3_50_0 {
     destination: string;
     suspend?: boolean;
-    condition: AgentTransitionStepConditionNoneV3_50_0 | AgentTransitionStepConditionArtifactV3_50_0 | AgentTransitionStepConditionStateOutputV3_50_0 | AgentTransitionStepConditionStringContainsV3_50_0 | AgentTransitionStepConditionArgumentV3_50_0;
+    condition?: string;
     limit?: number;
 }
-export interface AgentTransitionStepConditionNoneV3_50_0 {
-    type: 'none';
-}
-export interface AgentTransitionStepConditionArtifactV3_50_0 {
-    type: 'artifact';
-    path?: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionStateOutputV3_50_0 {
-    type: 'stateOutput';
-    stateId: string;
-    path?: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionStringContainsV3_50_0 {
-    type: 'stringContains';
-    path?: string;
-    string: string;
-    negated?: boolean;
-}
-export interface AgentTransitionStepConditionArgumentV3_50_0 {
-    type: 'argument';
-    argument: string;
-    negated?: boolean;
-}
 /**
  * States that are traversed during the execution of an agent.
  */
@@ -1735,39 +1709,9 @@ export interface AgentVariableV3_50_0 {
      */
     steps: AgentVariableStepV3_50_0[];
 }
-export interface AgentVariableStepSessionMemoryV3_50_0 {
-    type: 'sessionMemory';
-    memoryKey: string;
-    path?: string;
-}
-export interface AgentVariableStepGraphQLArgV3_50_0 {
-    type: 'graphqlArg';
-    argName: string;
-}
-export interface AgentVariableStepStateOutputV3_50_0 {
-    type: 'stateOutput';
-    stateId: string;
-    path?: string;
-}
-export interface AgentVariableStepPreviousStateOutputV3_50_0 {
-    type: 'previousStateOutput';
-    path?: string;
-}
-/**
- * An Agent Variable that has a static value.
- */
-export interface AgentVariableStepStaticValueV3_50_0 {
-    type: 'staticValue';
-    condition?: AgentVariableStepConditionV3_50_0;
-    value: string;
-}
-export interface AgentVariableStepConditionNoneV3_50_0 {
-    type: 'none';
-}
-export interface AgentVariableStepConditionSourceStateV3_50_0 {
-    type: 'sourceState';
-    stateId: string;
-    negated?: boolean;
+export interface AgentVariableStepV3_50_0 {
+    condition?: string;
+    expression: string;
 }
 export interface AgentExecutionGraphQLV3_50_0 {
     type: 'graphql';
@@ -1792,10 +1736,28 @@ export interface AgentAIStateInputTemplateV3_50_0 {
     inputTemplate: string;
 }
 export interface AgentToolConfigV3_50_0 {
-    description?: string;
     ref: string;
+    description?: string;
+    args?: AgentToolConfigArgV3_50_0[];
     selectionSet?: string;
-    removePropertyKeyPatterns?: RemovePropertyKeyPatternsV3_50_0;
+}
+export interface AgentToolConfigArgAgentV3_50_0 {
+    type: 'agent';
+    /**
+     * The argument name that will be exposed by the tool for the LLM to use.
+     */
+    argName: string;
+}
+export interface AgentToolConfigArgVariableV3_50_0 {
+    type: 'variable';
+    /**
+     * Name of argument to pass value to. This prevents the LLM from setting the argument.
+     */
+    argName: string;
+    /**
+     * Name of variable to pull value from.
+     */
+    variableName: string;
 }
 export interface AgentGenerateOptionsV3_50_0 {
     history?: HistoryV3_50_01;