beeai-framework 0.1.6 → 0.1.8

package/dist/workflows/agent.d.cts

@@ -4,10 +4,10 @@ import { Message, MessageContentPart } from '../backend/message.cjs';
 import { AnyTool } from '../tools/base.cjs';
 import { ReadOnlyMemory } from '../memory/base.cjs';
 import { z } from 'zod';
-import { a as BaseAgent } from '../base-daM6fHUP.cjs';
-import { BeeRunInput, BeeRunOutput, BeeRunOptions, BeeAgentExecutionConfig } from '../agents/bee/types.cjs';
+import { a as BaseAgent } from '../base-Csr9guq2.cjs';
 import { ChatModel } from '../backend/chat.cjs';
-import '../emitter-C9EN5B0j.cjs';
+import { ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentExecutionConfig } from '../agents/toolCalling/types.cjs';
+import '../emitter-BWtGHYn0.cjs';
 import '../internals/types.cjs';
 import '../internals/helpers/guards.cjs';
 import '../internals/serializable.cjs';
@@ -19,144 +19,324 @@ import 'promise-based-task';
 import '../cache/base.cjs';
 import '../internals/helpers/schema.cjs';
 import 'zod-to-json-schema';
-import '../agents/bee/prompts.cjs';
-import '../template.cjs';
-import '../parsers/linePrefix.cjs';
-import '../parsers/field.cjs';
-import '@streamparser/json';
-import 'jsonrepair/stream';
-import '../parsers/errors.cjs';
 import '../backend/utils.cjs';
 import '../backend/constants.cjs';
 import '@ai-sdk/provider';
+import '../agents/toolCalling/prompts.cjs';
+import '../template.cjs';
 
-type AgentInstance = BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions>;
+type AgentInstance = BaseAgent<ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions>;
 type AgentFactory = (memory: ReadOnlyMemory) => AgentInstance | Promise<AgentInstance>;
 interface AgentFactoryInput {
-    name: string;
+    name?: string;
+    role?: string;
     llm: ChatModel;
     instructions?: string;
     tools?: AnyTool[];
-    execution?: BeeAgentExecutionConfig;
+    execution?: ToolCallingAgentExecutionConfig;
 }
 declare class AgentWorkflow {
     protected readonly workflow: Workflow<z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
         finalAnswer: string;
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
     }, {
-        messages: Message<MessageContentPart, string>[];
         finalAnswer: string;
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, string>;
-    static readonly schema: z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+    readonly schema: z.ZodObject<{
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>;
     constructor(name?: string);
-    run(messages: Message[], options?: WorkflowRunOptions<string>): Run<WorkflowRun<z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+    run(inputs: (ToolCallingAgentRunInput | Message)[], options?: WorkflowRunOptions<string>): Run<WorkflowRun<z.ZodObject<{
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
         finalAnswer: string;
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
     }, {
-        messages: Message<MessageContentPart, string>[];
         finalAnswer: string;
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, string>, Workflow<z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
         finalAnswer: string;
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
     }, {
-        messages: Message<MessageContentPart, string>[];
         finalAnswer: string;
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, string>, readonly [{
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }, WorkflowRunOptions<string>]>;
     addAgent(agent: AgentFactory | AgentFactoryInput): this;
     addAgent(agent: AgentInstance): Promise<this>;
     delAgent(name: string): Workflow<z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
         finalAnswer: string;
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
     }, {
-        messages: Message<MessageContentPart, string>[];
         finalAnswer: string;
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, never>;
     protected _createFactory(input: AgentFactoryInput): AgentFactory;