@aigne/core 1.65.1-beta.2 → 1.66.0-beta.3

package/lib/dts/loader/schema.d.ts

@@ -22,42 +22,126 @@ export declare const inputOutputSchema: ({ path }: {
     required?: string[] | undefined;
     additionalProperties?: boolean | undefined;
 }>]>;
-export declare const defaultInputSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodObject<{
+export declare const defaultInputSchema: z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodUnknown, z.ZodObject<{
     $get: z.ZodString;
 }, "strip", z.ZodTypeAny, {
     $get: string;
 }, {
     $get: string;
-}>, z.ZodUnknown]>>;
+}>]>>;
 declare const chatModelObjectSchema: z.ZodObject<{
-    model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-    temperature: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    topP: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    frequencyPenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    presencePenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | undefined>;
+    model: ZodType<string | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, string | {
+        $get: string;
+    } | undefined>;
+    temperature: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    topP: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    frequencyPenalty: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    presencePenalty: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+        $get: string;
+    } | undefined>;
 }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
-    model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-    temperature: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    topP: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    frequencyPenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    presencePenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | undefined>;
+    model: ZodType<string | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, string | {
+        $get: string;
+    } | undefined>;
+    temperature: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    topP: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    frequencyPenalty: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    presencePenalty: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+        $get: string;
+    } | undefined>;
 }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
-    model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-    temperature: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    topP: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    frequencyPenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    presencePenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-    thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | undefined>;
+    model: ZodType<string | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, string | {
+        $get: string;
+    } | undefined>;
+    temperature: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    topP: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    frequencyPenalty: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    presencePenalty: ZodType<number | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | {
+        $get: string;
+    } | undefined>;
+    thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+        $get: string;
+    } | undefined>;
 }, z.ZodTypeAny, "passthrough">>;
 export declare const chatModelSchema: typeof chatModelObjectSchema;
 declare const imageModelObjectSchema: z.ZodObject<{
-    model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    model: ZodType<string | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, string | {
+        $get: string;
+    } | undefined>;
 }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
-    model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    model: ZodType<string | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, string | {
+        $get: string;
+    } | undefined>;
 }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
-    model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    model: ZodType<string | {
+        $get: string;
+    } | undefined, z.ZodTypeDef, string | {
+        $get: string;
+    } | undefined>;
 }, z.ZodTypeAny, "passthrough">>;
 export declare const imageModelSchema: typeof imageModelObjectSchema;
 export declare function optionalize<T>(schema: ZodType<T>): ZodType<T | undefined>;

package/lib/esm/agents/agent.d.ts

@@ -12,7 +12,7 @@ import { type Nullish, type PromiseOrValue, type XOr } from "../utils/type-utils
 import type { ChatModel } from "./chat-model.js";
 import type { GuideRailAgent, GuideRailAgentOutput } from "./guide-rail-agent.js";
 import type { ImageModel } from "./image-model.js";
-import { type TransferAgentOutput } from "./types.js";
+import { type GetterSchema, type TransferAgentOutput } from "./types.js";
 export * from "./types.js";
 export declare const DEFAULT_INPUT_ACTION_GET = "$get";
 /**
@@ -293,11 +293,7 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
     renderTaskTitle(input: I): Promise<string | undefined>;
     taskRenderMode?: TaskRenderMode;
     private readonly _inputSchema?;
-    defaultInput?: Partial<{
-        [key in keyof I]: {
-            [DEFAULT_INPUT_ACTION_GET]: string;
-        } | I[key];
-    }>;
+    defaultInput?: GetterSchema<I>;
     private readonly _outputSchema?;
     /**
      * Get the input data schema for this agent

package/lib/esm/agents/agent.js

@@ -363,7 +363,11 @@ export class Agent {
         input = this.mergeDefaultInput(input);
         logger.debug("Invoke agent %s started with input: %O", this.name, input);
         if (!this.disableEvents)
-            opts.context.emit("agentStarted", { agent: this, input });
+            opts.context.emit("agentStarted", {
+                agent: this,
+                input,
+                taskTitle: await this.renderTaskTitle(input),
+            });
         try {
             const s = await this.callHooks("onStart", { input }, opts);
             if (s?.input)

package/lib/esm/agents/ai-agent.js

@@ -273,6 +273,7 @@ export class AIAgent extends Agent {
             input,
             model,
         });
+        modelInput.modelOptions = await model.getModelOptions(input, options);
         const toolsMap = new Map(toolAgents?.map((i) => [i.name, i]));
         if (this.toolChoice === "router") {
             return yield* this._processRouter(input, model, modelInput, options, toolsMap);

package/lib/esm/agents/chat-model.d.ts

@@ -1,12 +1,12 @@
 import { z } from "zod";
 import { type PromiseOrValue } from "../utils/type-utils.js";
-import { type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type AgentResponse, type AgentResponseStream, type Message } from "./agent.js";
+import { type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type AgentResponse, type AgentResponseStream, type GetterSchema, type Message } from "./agent.js";
 import { type FileType, type FileUnionContent, Model } from "./model.js";
 export declare class StructuredOutputError extends Error {
 }
 export interface ChatModelOptions extends Omit<AgentOptions<ChatModelInput, ChatModelOutput>, "model" | "inputSchema" | "outputSchema"> {
     model?: string;
-    modelOptions?: Omit<ChatModelInputOptions, "model">;
+    modelOptions?: ChatModelInputOptionsWithGetter;
 }
 /**
  * ChatModel is an abstract base class for interacting with Large Language Models (LLMs).
@@ -40,6 +40,7 @@ export declare abstract class ChatModel extends Model<ChatModelInput, ChatModelO
         apiKey?: string;
         model?: string;
     }>;
+    getModelOptions(input: Message, options: AgentInvokeOptions): Promise<ChatModelInputOptions>;
     /**
      * Indicates whether the model supports parallel tool calls
      *
@@ -156,7 +157,7 @@ export interface ChatModelInput extends Message {
     /**
      * Model-specific configuration options
      */
-    modelOptions?: ChatModelInputOptions;
+    modelOptions?: ChatModelInputOptionsWithGetter;
 }
 /**
  * Message role types
@@ -361,7 +362,7 @@ export type Modality = "text" | "image" | "audio";
  *
  * Contains various parameters for controlling model behavior, such as model name, temperature, etc.
  */
-export interface ChatModelInputOptions {
+export interface ChatModelInputOptions extends Record<string, unknown> {
     /**
      * Model name or version
      */
@@ -390,6 +391,7 @@ export interface ChatModelInputOptions {
     preferInputFileType?: "file" | "url";
     reasoningEffort?: number | "minimal" | "low" | "medium" | "high";
 }
+export type ChatModelInputOptionsWithGetter = GetterSchema<ChatModelInputOptions>;
 /**
  * Output message format for ChatModel
  *

package/lib/esm/agents/chat-model.js

@@ -4,7 +4,7 @@ import { convertJsonSchemaToZod } from "zod-from-json-schema";
 import { wrapAutoParseJsonSchema } from "../utils/json-schema.js";
 import { logger } from "../utils/logger.js";
 import { checkArguments, isNil, omitByDeep } from "../utils/type-utils.js";
-import { agentOptionsSchema, } from "./agent.js";
+import { agentOptionsSchema, DEFAULT_INPUT_ACTION_GET, getterSchema, } from "./agent.js";
 import { fileContentSchema, fileUnionContentSchema, localContentSchema, Model, urlContentSchema, } from "./model.js";
 const CHAT_MODEL_DEFAULT_RETRY_OPTIONS = {
     retries: 3,
@@ -61,6 +61,27 @@ export class ChatModel extends Model {
     get credential() {
         return {};
     }
+    async getModelOptions(input, options) {
+        const result = {};
+        for (const [key, val] of Object.entries({
+            ...this.options?.modelOptions,
+            ...("modelOptions" in input ? input.modelOptions : {}),
+        })) {
+            if (val &&
+                typeof val === "object" &&
+                DEFAULT_INPUT_ACTION_GET in val &&
+                typeof val[DEFAULT_INPUT_ACTION_GET] === "string") {
+                const getterPath = val[DEFAULT_INPUT_ACTION_GET];
+                const value = input[getterPath] ?? options.context.userContext[getterPath];
+                if (!isNil(value))
+                    Object.assign(result, { [key]: value });
+            }
+            else {
+                Object.assign(result, { [key]: val });
+            }
+        }
+        return result;
+    }
     /**
      * Indicates whether the model supports parallel tool calls
      *
@@ -292,22 +313,20 @@ const chatModelInputToolChoiceSchema = z.union([
     chatModelInputToolSchema,
 ]);
 const modelOptionsSchema = z.object({
-    model: z.string().optional(),
-    temperature: z.number().optional(),
-    topP: z.number().optional(),
-    frequencyPenalty: z.number().optional(),
-    presencePenalty: z.number().optional(),
-    parallelToolCalls: z.boolean().optional().default(true),
-    modalities: z.array(z.enum(["text", "image", "audio"])).optional(),
-    reasoningEffort: z
-        .union([
+    model: getterSchema(z.string()).optional(),
+    temperature: getterSchema(z.number()).optional(),
+    topP: getterSchema(z.number()).optional(),
+    frequencyPenalty: getterSchema(z.number()).optional(),
+    presencePenalty: getterSchema(z.number()).optional(),
+    parallelToolCalls: getterSchema(z.boolean()).optional().default(true),
+    modalities: getterSchema(z.array(z.enum(["text", "image", "audio"]))).optional(),
+    reasoningEffort: getterSchema(z.union([
         z.number(),
         z.literal("minimal"),
         z.literal("low"),
         z.literal("medium"),
         z.literal("high"),
-    ])
-        .optional(),
+    ])).optional(),
 });
 const chatModelOptionsSchema = agentOptionsSchema.extend({
     model: z.string().optional(),

package/lib/esm/agents/image-model.d.ts

@@ -1,6 +1,6 @@
 import { type ZodType, z } from "zod";
 import type { PromiseOrValue } from "../utils/type-utils.js";
-import type { AgentInvokeOptions, AgentOptions, AgentProcessResult, AgentResponse, AgentResponseStream, Message } from "./agent.js";
+import type { AgentInvokeOptions, AgentOptions, AgentProcessResult, AgentResponse, AgentResponseStream, GetterSchema, Message } from "./agent.js";
 import { type ChatModelOutputUsage } from "./chat-model.js";
 import { type FileType, type FileUnionContent, Model } from "./model.js";
 export interface ImageModelOptions<I extends ImageModelInput = ImageModelInput, O extends ImageModelOutput = ImageModelOutput> extends Omit<AgentOptions<I, O>, "model"> {
@@ -33,6 +33,7 @@ export interface ImageModelInputOptions extends Record<string, unknown> {
     model?: string;
     preferInputFileType?: "file" | "url";
 }
+export type ImageModelInputOptionsWithGetter = GetterSchema<ImageModelInputOptions>;
 export declare const imageModelInputSchema: z.ZodObject<{
     prompt: z.ZodString;
     image: z.ZodOptional<z.ZodArray<z.ZodDiscriminatedUnion<"type", [z.ZodObject<{

package/lib/esm/agents/types.d.ts

@@ -1,4 +1,5 @@
-import type { Agent, Message } from "./agent.js";
+import z, { type ZodType } from "zod";
+import { type Agent, DEFAULT_INPUT_ACTION_GET, type Message } from "./agent.js";
 export declare const transferAgentOutputKey = "$transferAgentTo";
 export interface TransferAgentOutput extends Message {
     [transferAgentOutputKey]: {
@@ -8,3 +9,15 @@ export interface TransferAgentOutput extends Message {
 export declare function transferToAgentOutput(agent: Agent): TransferAgentOutput;
 export declare function isTransferAgentOutput(output: Message): output is TransferAgentOutput;
 export declare function replaceTransferAgentToName(output: Message): Message;
+export type GetterSchema<I extends Record<string, unknown>> = Partial<{
+    [key in keyof I]: {
+        [DEFAULT_INPUT_ACTION_GET]: string;
+    } | I[key];
+}>;
+export declare function getterSchema<T extends ZodType>(schema: T): z.ZodUnion<[T, z.ZodObject<{
+    $get: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    $get: string;
+}, {
+    $get: string;
+}>]>;

package/lib/esm/agents/types.js

@@ -1,3 +1,5 @@
+import z from "zod";
+import { DEFAULT_INPUT_ACTION_GET } from "./agent.js";
 export const transferAgentOutputKey = "$transferAgentTo";
 export function transferToAgentOutput(agent) {
     return {
@@ -18,3 +20,11 @@ export function replaceTransferAgentToName(output) {
     }
     return output;
 }
+export function getterSchema(schema) {
+    return z.union([
+        schema,
+        z.object({
+            [DEFAULT_INPUT_ACTION_GET]: z.string(),
+        }),
+    ]);
+}

package/lib/esm/agents/video-model.d.ts

@@ -1,6 +1,6 @@
 import { type ZodType, z } from "zod";
 import type { PromiseOrValue } from "../utils/type-utils.js";
-import type { AgentInvokeOptions, AgentOptions, AgentProcessResult, AgentResponse, AgentResponseStream, Message } from "./agent.js";
+import type { AgentInvokeOptions, AgentOptions, AgentProcessResult, AgentResponse, AgentResponseStream, GetterSchema, Message } from "./agent.js";
 import { type ChatModelOutputUsage } from "./chat-model.js";
 import { type FileType, type FileUnionContent, Model } from "./model.js";
 export interface VideoModelOptions<I extends VideoModelInput = VideoModelInput, O extends VideoModelOutput = VideoModelOutput> extends Omit<AgentOptions<I, O>, "model"> {
@@ -33,6 +33,7 @@ export interface VideoModelInput extends Message {
 export interface VideoModelInputOptions extends Record<string, unknown> {
     model?: string;
 }
+export type VideoModelInputOptionsWithGetter = GetterSchema<VideoModelInputOptions>;
 export declare const videoModelInputSchema: z.ZodObject<{
     prompt: z.ZodString;
     model: z.ZodOptional<z.ZodString>;

package/lib/esm/aigne/context.js

@@ -204,7 +204,7 @@ export class AIGNEContext {
         try {
             switch (eventName) {
                 case "agentStarted": {
-                    const { agent, input } = args[0];
+                    const { agent, input, taskTitle } = args[0];
                     span.updateName(agent.name);
                     span.setAttribute("custom.trace_id", this.rootId);
                     span.setAttribute("custom.span_id", this.id);
@@ -215,6 +215,9 @@ export class AIGNEContext {
                     span.setAttribute("custom.started_at", b.timestamp);
                     span.setAttribute("input", JSON.stringify(input));
                     span.setAttribute("agentTag", agent.tag ?? "UnknownAgent");
+                    if (taskTitle) {
+                        span.setAttribute("taskTitle", taskTitle);
+                    }
                     try {
                         span.setAttribute("userContext", JSON.stringify(this.userContext));
                     }

package/lib/esm/loader/agent-js.js

@@ -2,11 +2,12 @@ import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
 import { Agent } from "../agents/agent.js";
 import { tryOrThrow } from "../utils/type-utils.js";
 import { parseAgentFile } from "./agent-yaml.js";
+import { LoadJsAgentError } from "./error.js";
 const importFn = new Function("path", "return import(path)");
 export async function loadAgentFromJsFile(path) {
     if (nodejs.path.isAbsolute(path))
         path = nodejs.url.pathToFileURL(path).toString();
-    const { default: agent } = await tryOrThrow(() => importFn(path), (error) => new Error(`Failed to load agent definition from ${path}: ${error.message}`));
+    const { default: agent } = await tryOrThrow(() => importFn(path), (error) => new LoadJsAgentError(`Failed to load agent definition from ${path}: ${error.message}`));
     if (agent instanceof Agent)
         return agent;
     if (typeof agent !== "function") {

package/lib/esm/loader/error.d.ts

@@ -0,0 +1,2 @@
+export declare class LoadJsAgentError extends Error {
+}

package/lib/esm/loader/error.js

@@ -0,0 +1,2 @@
+export class LoadJsAgentError extends Error {
+}

package/lib/esm/loader/index.d.ts

@@ -33,24 +33,80 @@ declare const aigneFileSchema: z.ZodObject<{
     name: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
     description: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
     model: ZodType<z.objectInputType<{
-        model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        temperature: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        topP: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        frequencyPenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        presencePenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | undefined>;
+        model: ZodType<string | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, string | {
+            $get: string;
+        } | undefined>;
+        temperature: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        topP: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        frequencyPenalty: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        presencePenalty: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+            $get: string;
+        } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined, z.ZodTypeDef, z.objectInputType<{
-        model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        temperature: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        topP: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        frequencyPenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        presencePenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | undefined>;
+        model: ZodType<string | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, string | {
+            $get: string;
+        } | undefined>;
+        temperature: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        topP: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        frequencyPenalty: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        presencePenalty: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+            $get: string;
+        } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined>;
     imageModel: ZodType<z.objectInputType<{
-        model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+        model: ZodType<string | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, string | {
+            $get: string;
+        } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined, z.ZodTypeDef, z.objectInputType<{
-        model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+        model: ZodType<string | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, string | {
+            $get: string;
+        } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined>;
     agents: ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
     skills: ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
@@ -68,17 +124,45 @@ declare const aigneFileSchema: z.ZodObject<{
     } | undefined>;
 }, "strip", z.ZodTypeAny, {
     model?: z.objectInputType<{
-        model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        temperature: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        topP: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        frequencyPenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        presencePenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | undefined>;
+        model: ZodType<string | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, string | {
+            $get: string;
+        } | undefined>;
+        temperature: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        topP: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        frequencyPenalty: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        presencePenalty: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+            $get: string;
+        } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined;
     name?: string | undefined;
     description?: string | undefined;
     imageModel?: z.objectInputType<{
-        model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+        model: ZodType<string | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, string | {
+            $get: string;
+        } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined;
     skills?: string[] | undefined;
     agents?: string[] | undefined;
@@ -91,17 +175,45 @@ declare const aigneFileSchema: z.ZodObject<{
     } | undefined;
 }, {
     model?: z.objectInputType<{
-        model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        temperature: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        topP: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        frequencyPenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        presencePenalty: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | undefined>;
+        model: ZodType<string | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, string | {
+            $get: string;
+        } | undefined>;
+        temperature: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        topP: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        frequencyPenalty: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        presencePenalty: ZodType<number | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | {
+            $get: string;
+        } | undefined>;
+        thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+            $get: string;
+        } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined;
     name?: string | undefined;
     description?: string | undefined;
     imageModel?: z.objectInputType<{
-        model: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+        model: ZodType<string | {
+            $get: string;
+        } | undefined, z.ZodTypeDef, string | {
+            $get: string;
+        } | undefined>;
     }, z.ZodTypeAny, "passthrough"> | undefined;
     skills?: string[] | undefined;
     agents?: string[] | undefined;

package/lib/esm/loader/index.js

@@ -100,11 +100,15 @@ async function parseHooks(path, hooks, options) {
         onHandoff: hook.onHandoff ? await loadNestAgent(path, hook.onHandoff, options) : undefined,
     })));
 }
+async function loadSkills(path, skills, options) {
+    const loadedSkills = [];
+    for (const skill of skills) {
+        loadedSkills.push(await loadNestAgent(path, skill, options));
+    }
+    return loadedSkills;
+}
 async function parseAgent(path, agent, options, agentOptions) {
-    const skills = "skills" in agent
-        ? agent.skills &&
-            (await Promise.all(agent.skills.map((skill) => loadNestAgent(path, skill, options))))
-        : undefined;
+    const skills = "skills" in agent && agent.skills ? await loadSkills(path, agent.skills, options) : undefined;
     const memory = "memory" in agent && options?.memories?.length
         ? await loadMemory(options.memories, typeof agent.memory === "object" ? agent.memory.provider : undefined, typeof agent.memory === "object" ? agent.memory : {})
         : undefined;