@aigne/core 0.4.201-0 → 0.4.201-1

package/lib/types/data-type.d.ts

@@ -30,6 +30,9 @@ export interface DataTypeArray extends DataTypeBase {
     defaultValue?: object[];
     items?: OmitPropsFromUnion<DataType, 'id'>;
 }
+export type DataTypes = {
+    [name: string]: OmitPropsFromUnion<DataType, 'id' | 'name'>;
+};
 type SchemaTypeInner<T extends Record<string, OmitPropsFromUnion<DataType, 'id' | 'name'>>> = {
     [K in keyof T]: T[K]['type'] extends 'string' ? string : T[K]['type'] extends 'number' ? number : T[K]['type'] extends 'boolean' ? boolean : T[K]['type'] extends 'object' ? object : T[K]['type'] extends 'array' ? object[] : never;
 };

package/lib/types/function-agent.d.ts

@@ -1,15 +1,11 @@
-import { DataType, SchemaType } from './data-type';
+import { DataType, DataTypes, SchemaType } from './data-type';
 import { FunctionRunner } from './function-runner';
 import { RunOptions, Runnable, RunnableDefinition, RunnableResponseStream } from './runnable';
 import { OmitPropsFromUnion } from './utils/omit';
 export declare class FunctionAgent<I extends {} = {}, O extends {} = {}> extends Runnable<I, O> {
     definition: FunctionAgentDefinition;
     runner?: FunctionRunner | undefined;
-    static create<I extends {
-        [name: string]: OmitPropsFromUnion<DataType, 'id' | 'name'>;
-    }, O extends {
-        [name: string]: OmitPropsFromUnion<DataType, 'id' | 'name'>;
-    }>(options: Parameters<typeof createFunctionAgentDefinition<I, O>>[0]): FunctionAgent<SchemaType<I>, SchemaType<O>>;
+    static create<I extends {} = {}, O extends {} = {}, Inputs extends DataTypes = {}, Outputs extends DataTypes = {}>(options: Parameters<typeof createFunctionAgentDefinition<Inputs, Outputs>>[0]): FunctionAgent<SchemaType<Inputs> & I, SchemaType<Outputs> & O>;
     constructor(definition: FunctionAgentDefinition, runner?: FunctionRunner | undefined);
     run(input: I, options: RunOptions & {
         stream: true;

package/lib/types/llm-agent.d.ts

@@ -1,4 +1,4 @@
-import { DataType, SchemaType } from './data-type';
+import { DataType, DataTypes, SchemaType } from './data-type';
 import { LLMModel, LLMModelInputs, Role } from './llm-model';
 import { RunOptions, Runnable, RunnableDefinition, RunnableResponseStream } from './runnable';
 import { OmitPropsFromUnion } from './utils/omit';
@@ -6,11 +6,7 @@ import { OrderedRecord } from './utils/ordered-map';
 export declare class LLMAgent<I extends {} = {}, O extends {} = {}> extends Runnable<I, O> {
     definition: LLMAgentDefinition;
     model?: LLMModel | undefined;
-    static create<I extends {
-        [name: string]: OmitPropsFromUnion<DataType, 'id' | 'name'>;
-    }, O extends {
-        [name: string]: OmitPropsFromUnion<DataType, 'id' | 'name'>;
-    }>(options: Parameters<typeof createLLMAgentDefinition<I, O>>[0]): LLMAgent<SchemaType<I>, SchemaType<O>>;
+    static create<I extends {} = {}, O extends {} = {}, Inputs extends DataTypes = {}, Outputs extends DataTypes = {}>(options: Parameters<typeof createLLMAgentDefinition<I, O>>[0]): LLMAgent<SchemaType<Inputs> & I, SchemaType<Outputs> & O>;
     constructor(definition: LLMAgentDefinition, model?: LLMModel | undefined);
     run(input: I, options: RunOptions & {
         stream: true;

package/lib/types/llm-decision-agent.d.ts

@@ -1,5 +1,5 @@
 import type { Context } from './context';
-import { DataType, SchemaType } from './data-type';
+import { DataType, DataTypes, SchemaType } from './data-type';
 import { LLMModel, LLMModelInputMessage, LLMModelInputs, LLMModelOptions } from './llm-model';
 import { RunOptions, Runnable, RunnableDefinition, RunnableResponseStream } from './runnable';
 import { OrderedRecord } from './utils';
@@ -10,11 +10,7 @@ export declare class LLMDecisionAgent<I extends {
     definition: LLMDecisionAgentDefinition;
     model?: LLMModel | undefined;
     context?: Context | undefined;
-    static create<I extends {
-        [name: string]: OmitPropsFromUnion<DataType, 'id' | 'name'>;
-    }, O extends {
-        [name: string]: OmitPropsFromUnion<DataType, 'id' | 'name'>;
-    }>(options: Parameters<typeof createLLMDecisionAgentDefinition<I>>[0]): LLMDecisionAgent<SchemaType<I>, SchemaType<O>>;
+    static create<I extends {} = {}, O extends {} = {}, Inputs extends DataTypes = {}, Outputs extends DataTypes = {}>(options: Parameters<typeof createLLMDecisionAgentDefinition<Inputs>>[0]): LLMDecisionAgent<SchemaType<Inputs> & I, SchemaType<Outputs> & O>;
     constructor(definition: LLMDecisionAgentDefinition, model?: LLMModel | undefined, context?: Context | undefined);
     run(input: I, options: RunOptions & {
         stream: true;

package/lib/types/local-function-agent.d.ts

@@ -1,15 +1,11 @@
 import type { Context } from './context';
-import { DataType, SchemaType } from './data-type';
+import { DataType, DataTypes, SchemaType } from './data-type';
 import { RunOptions, Runnable, RunnableDefinition, RunnableResponse, RunnableResponseStream } from './runnable';
 import { OmitPropsFromUnion } from './utils/omit';
 export declare class LocalFunctionAgent<I extends {} = {}, O extends {} = {}, State = {}> extends Runnable<I, O> {
     definition: LocalFunctionAgentDefinition<I, O, State>;
     context?: Context<State> | undefined;
-    static create<I extends {
-        [name: string]: OmitPropsFromUnion<DataType, 'id' | 'name'>;
-    }, O extends {
-        [name: string]: OmitPropsFromUnion<DataType, 'id' | 'name'>;
-    }, State = {}>(options: Parameters<typeof createLocalFunctionAgentDefinition<I, O, State>>[0]): LocalFunctionAgent<SchemaType<I>, SchemaType<O>, State>;
+    static create<I extends {} = {}, O extends {} = {}, State = {}, Inputs extends DataTypes = {}, Outputs extends DataTypes = {}>(options: Parameters<typeof createLocalFunctionAgentDefinition<Inputs, Outputs, State>>[0]): LocalFunctionAgent<SchemaType<Inputs> & I, SchemaType<Outputs> & O, State>;
     constructor(definition: LocalFunctionAgentDefinition<I, O, State>, context?: Context<State> | undefined);
     run(input: I, options: RunOptions & {
         stream: true;

package/lib/types/pipeline-agent.d.ts

@@ -1,5 +1,5 @@
 import type { Context } from './context';
-import { DataType, SchemaType } from './data-type';
+import { DataType, DataTypes, SchemaType } from './data-type';
 import { RunOptions, Runnable, RunnableDefinition, RunnableOutput, RunnableResponseStream } from './runnable';
 import { OmitPropsFromUnion } from './utils/omit';
 import { OrderedRecord } from './utils/ordered-map';
@@ -8,14 +8,12 @@ export declare class PipelineAgent<I extends {
 } = {}, O extends {} = {}> extends Runnable<I, O> {
     definition: PipelineAgentDefinition;
     context?: Context | undefined;
-    static create<I extends {
-        [name: string]: OmitPropsFromUnion<DataType, 'id' | 'name'>;
-    }, O extends {
-        [name: string]: OmitPropsFromUnion<DataType, 'id' | 'name'> & {
+    static create<I extends {} = {}, O extends {} = {}, Inputs extends DataTypes = {}, Outputs extends {
+        [name: string]: DataTypes[string] & {
             fromVariable: string;
             fromVariablePropPath?: string[];
         };
-    }>(options: Parameters<typeof createPipelineAgentDefinition<I, O>>[0]): PipelineAgent<SchemaType<I>, SchemaType<O>>;
+    } = {}>(options: Parameters<typeof createPipelineAgentDefinition<Inputs, Outputs>>[0]): PipelineAgent<SchemaType<Inputs> & I, SchemaType<Outputs> & O>;
     constructor(definition: PipelineAgentDefinition, context?: Context | undefined);
     run(input: I, options: RunOptions & {
         stream: true;