beeai-framework 0.1.2

package/dist/agents/experimental/replan/prompts.d.cts

@@ -0,0 +1,153 @@
+import * as ajv from 'ajv';
+import { z } from 'zod';
+import { PromptTemplate } from '../../../template.cjs';
+import { AnyTool } from '../../../tools/base.cjs';
+import '../../../errors.cjs';
+import '../../../internals/types.cjs';
+import '../../../internals/helpers/guards.cjs';
+import '../../../internals/serializable.cjs';
+import '../../../context.cjs';
+import '../../../emitter-BWtGHYn0.cjs';
+import '../../../internals/helpers/promise.cjs';
+import 'promise-based-task';
+import '../../../cache/base.cjs';
+import '../../../internals/helpers/schema.cjs';
+import 'zod-to-json-schema';
+
+declare function createRePlanOutputSchema<T extends AnyTool>(tools: T[]): Promise<{
+    definition: z.ZodObject<{
+        information: z.ZodRecord<z.ZodString, z.ZodString>;
+        lookback: z.ZodString;
+        plan: z.ZodArray<z.ZodObject<{
+            title: z.ZodString;
+            decision: z.ZodString;
+            research: z.ZodBoolean;
+            computation: z.ZodBoolean;
+        }, "strip", z.ZodTypeAny, {
+            title: string;
+            decision: string;
+            research: boolean;
+            computation: boolean;
+        }, {
+            title: string;
+            decision: string;
+            research: boolean;
+            computation: boolean;
+        }>, "many">;
+        nextStep: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
+            type: z.ZodLiteral<"message">;
+            message: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            message: string;
+            type: "message";
+        }, {
+            message: string;
+            type: "message";
+        }>, ...z.ZodObject<{
+            type: z.ZodLiteral<"tool">;
+            calls: z.ZodArray<z.ZodDiscriminatedUnion<"name", [z.ZodObject<{
+                name: z.ZodLiteral<string>;
+                input: z.ZodObject<{}, "passthrough", z.ZodTypeAny, z.objectOutputType<{}, z.ZodTypeAny, "passthrough">, z.objectInputType<{}, z.ZodTypeAny, "passthrough">>;
+            }, "strip", z.ZodTypeAny, {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }, {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }>, ...z.ZodObject<{
+                name: z.ZodLiteral<string>;
+                input: z.ZodObject<{}, "passthrough", z.ZodTypeAny, z.objectOutputType<{}, z.ZodTypeAny, "passthrough">, z.objectInputType<{}, z.ZodTypeAny, "passthrough">>;
+            }, "strip", z.ZodTypeAny, {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }, {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }>[]]>, "many">;
+        }, "strip", z.ZodTypeAny, {
+            type: "tool";
+            calls: {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }[];
+        }, {
+            type: "tool";
+            calls: {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }[];
+        }>[]]>;
+    }, "strip", z.ZodTypeAny, {
+        information: Record<string, string>;
+        lookback: string;
+        plan: {
+            title: string;
+            decision: string;
+            research: boolean;
+            computation: boolean;
+        }[];
+        nextStep: {
+            message: string;
+            type: "message";
+        } | {
+            type: "tool";
+            calls: {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }[];
+        };
+    }, {
+        information: Record<string, string>;
+        lookback: string;
+        plan: {
+            title: string;
+            decision: string;
+            research: boolean;
+            computation: boolean;
+        }[];
+        nextStep: {
+            message: string;
+            type: "message";
+        } | {
+            type: "tool";
+            calls: {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }[];
+        };
+    }>;
+    json: ajv.SchemaObject;
+}>;
+type RePlanState = z.output<Awaited<ReturnType<typeof createRePlanOutputSchema>>["definition"]>;
+declare const RePlanSystemPrompt: PromptTemplate<z.ZodObject<{
+    schema: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    schema: string;
+}, {
+    schema: string;
+}>>;
+declare const RePlanAssistantPrompt: PromptTemplate<z.ZodObject<{
+    results: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    results: string;
+}, {
+    results: string;
+}>>;
+
+export { RePlanAssistantPrompt, type RePlanState, RePlanSystemPrompt, createRePlanOutputSchema };

package/dist/agents/experimental/replan/prompts.d.ts

@@ -0,0 +1,153 @@
+import * as ajv from 'ajv';
+import { z } from 'zod';
+import { PromptTemplate } from '../../../template.js';
+import { AnyTool } from '../../../tools/base.js';
+import '../../../errors.js';
+import '../../../internals/types.js';
+import '../../../internals/helpers/guards.js';
+import '../../../internals/serializable.js';
+import '../../../context.js';
+import '../../../emitter-l0W9gC1A.js';
+import '../../../internals/helpers/promise.js';
+import 'promise-based-task';
+import '../../../cache/base.js';
+import '../../../internals/helpers/schema.js';
+import 'zod-to-json-schema';
+
+declare function createRePlanOutputSchema<T extends AnyTool>(tools: T[]): Promise<{
+    definition: z.ZodObject<{
+        information: z.ZodRecord<z.ZodString, z.ZodString>;
+        lookback: z.ZodString;
+        plan: z.ZodArray<z.ZodObject<{
+            title: z.ZodString;
+            decision: z.ZodString;
+            research: z.ZodBoolean;
+            computation: z.ZodBoolean;
+        }, "strip", z.ZodTypeAny, {
+            title: string;
+            decision: string;
+            research: boolean;
+            computation: boolean;
+        }, {
+            title: string;
+            decision: string;
+            research: boolean;
+            computation: boolean;
+        }>, "many">;
+        nextStep: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
+            type: z.ZodLiteral<"message">;
+            message: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            message: string;
+            type: "message";
+        }, {
+            message: string;
+            type: "message";
+        }>, ...z.ZodObject<{
+            type: z.ZodLiteral<"tool">;
+            calls: z.ZodArray<z.ZodDiscriminatedUnion<"name", [z.ZodObject<{
+                name: z.ZodLiteral<string>;
+                input: z.ZodObject<{}, "passthrough", z.ZodTypeAny, z.objectOutputType<{}, z.ZodTypeAny, "passthrough">, z.objectInputType<{}, z.ZodTypeAny, "passthrough">>;
+            }, "strip", z.ZodTypeAny, {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }, {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }>, ...z.ZodObject<{
+                name: z.ZodLiteral<string>;
+                input: z.ZodObject<{}, "passthrough", z.ZodTypeAny, z.objectOutputType<{}, z.ZodTypeAny, "passthrough">, z.objectInputType<{}, z.ZodTypeAny, "passthrough">>;
+            }, "strip", z.ZodTypeAny, {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }, {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }>[]]>, "many">;
+        }, "strip", z.ZodTypeAny, {
+            type: "tool";
+            calls: {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }[];
+        }, {
+            type: "tool";
+            calls: {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }[];
+        }>[]]>;
+    }, "strip", z.ZodTypeAny, {
+        information: Record<string, string>;
+        lookback: string;
+        plan: {
+            title: string;
+            decision: string;
+            research: boolean;
+            computation: boolean;
+        }[];
+        nextStep: {
+            message: string;
+            type: "message";
+        } | {
+            type: "tool";
+            calls: {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }[];
+        };
+    }, {
+        information: Record<string, string>;
+        lookback: string;
+        plan: {
+            title: string;
+            decision: string;
+            research: boolean;
+            computation: boolean;
+        }[];
+        nextStep: {
+            message: string;
+            type: "message";
+        } | {
+            type: "tool";
+            calls: {
+                name: string;
+                input: {} & {
+                    [k: string]: unknown;
+                };
+            }[];
+        };
+    }>;
+    json: ajv.SchemaObject;
+}>;
+type RePlanState = z.output<Awaited<ReturnType<typeof createRePlanOutputSchema>>["definition"]>;
+declare const RePlanSystemPrompt: PromptTemplate<z.ZodObject<{
+    schema: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    schema: string;
+}, {
+    schema: string;
+}>>;
+declare const RePlanAssistantPrompt: PromptTemplate<z.ZodObject<{
+    results: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    results: string;
+}, {
+    results: string;
+}>>;
+
+export { RePlanAssistantPrompt, type RePlanState, RePlanSystemPrompt, createRePlanOutputSchema };

package/dist/agents/experimental/replan/prompts.js

@@ -0,0 +1,85 @@
+import { z, ZodFirstPartyTypeKind } from 'zod';
+import { PromptTemplate } from '../../../template.js';
+import { hasAtLeast, map } from 'remeda';
+import { toJsonSchema } from '../../../internals/helpers/schema.js';
+import { ignoreOverride } from 'zod-to-json-schema';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+async function createRePlanOutputSchema(tools) {
+  const toolSchemas = await Promise.all(tools.map(async (tool) => ({
+    name: tool.name,
+    description: tool.description,
+    input: z.object({}).passthrough(),
+    inputSchema: await tool.getInputJsonSchema()
+  })));
+  const zodSchemaToJsonSchema = new WeakMap(toolSchemas.map((tool) => [
+    tool.input.shape,
+    tool.inputSchema
+  ]));
+  const definition = z.object({
+    information: z.record(z.string()).describe("Summary of the factual information that was collected so far. Eg. 'height of the Eiffel tower': '300m'. Only information that was provided by tools or the user in this very conversation is allowed to be included here. Other information must not be included."),
+    lookback: z.string().describe("A full summary of what has happened so far, focusing on what the assistant tried, what went well and what failed. This repeats in every message, but always concerns the full history up to that point."),
+    plan: z.array(z.object({
+      title: z.string().describe("Title of this step, shortly describing what needs to be done."),
+      decision: z.string().describe("Assistant's decision of how to tackle this step."),
+      research: z.boolean().describe("Does this step involve looking up factual information through tools?"),
+      computation: z.boolean().describe("Does this step involve calculating or computing information through tools?")
+    })).describe("Detailed step-by-step plan of what steps will the assistant take from start to finish to fulfill the user's request. Includes concrete facts and numbers wherever possible. This repeats in every message, but always contains all the future steps from this point on."),
+    nextStep: z.discriminatedUnion("type", [
+      z.object({
+        type: z.literal("message"),
+        message: z.string().describe("Message with the response, that is sent back to the user. Always include a bit of info on how you arrived at the answer. Be friendly and helpful.")
+      }).describe("Message the user -- either to give the answer, or to ask for more information."),
+      ...hasAtLeast(toolSchemas, 1) ? [
+        z.object({
+          type: z.literal("tool"),
+          calls: z.array(z.discriminatedUnion("name", map(toolSchemas, (tool) => z.object({
+            name: z.literal(tool.name),
+            input: tool.input
+          }).describe(tool.description))))
+        }).describe("Obtain more information using tools.")
+      ] : []
+    ])
+  });
+  return {
+    definition,
+    json: toJsonSchema(definition, {
+      override: /* @__PURE__ */ __name((_def) => {
+        const def = _def;
+        if (def.typeName === ZodFirstPartyTypeKind.ZodObject) {
+          const shape = def.shape();
+          const schema = zodSchemaToJsonSchema.get(shape);
+          if (schema) {
+            zodSchemaToJsonSchema.delete(shape);
+            return schema;
+          }
+        }
+        return ignoreOverride;
+      }, "override")
+    })
+  };
+}
+__name(createRePlanOutputSchema, "createRePlanOutputSchema");
+const RePlanSystemPrompt = new PromptTemplate({
+  schema: z.object({
+    schema: z.string()
+  }),
+  template: `The assistant is created by IBM and refers to itself as Bee. It's named after the IBM logo.
+
+The assistant is very intelligent and helpful. It always thinks ahead, and uses smart approaches to solve the user's problems. The assistant is an expert-level user of the provided tools, and can utilize them to their maximum potential.
+
+The assistant is forbidden from using factual information that was not provided by the user or tools in this very conversation. All information about places, people, events, etc. is unknown to the assistant, and the assistant must use tools to obtain it.
+
+Output Schema: {{schema}}`
+});
+const RePlanAssistantPrompt = new PromptTemplate({
+  schema: z.object({
+    results: z.string()
+  }),
+  template: `{{results}}`
+});
+
+export { RePlanAssistantPrompt, RePlanSystemPrompt, createRePlanOutputSchema };
+//# sourceMappingURL=prompts.js.map
+//# sourceMappingURL=prompts.js.map

package/dist/agents/experimental/replan/prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/experimental/replan/prompts.ts"],"names":["createRePlanOutputSchema","tools","toolSchemas","Promise","all","map","tool","name","description","input","z","object","passthrough","inputSchema","getInputJsonSchema","zodSchemaToJsonSchema","WeakMap","shape","definition","information","record","string","describe","lookback","plan","array","title","decision","research","boolean","computation","nextStep","discriminatedUnion","type","literal","message","hasAtLeast","calls","json","toJsonSchema","override","_def","def","typeName","ZodFirstPartyTypeKind","ZodObject","schema","get","delete","ignoreOverride","RePlanSystemPrompt","PromptTemplate","template","RePlanAssistantPrompt","results"],"mappings":";;;;;;;;AAuBA,eAAsBA,yBAA4CC,KAAU,EAAA;AAC1E,EAAA,MAAMC,cAAc,MAAMC,OAAAA,CAAQC,IAChCH,KAAMI,CAAAA,GAAAA,CAAI,OAAOC,IAAU,MAAA;AACzBC,IAAAA,IAAAA,EAAMD,IAAKC,CAAAA,IAAAA;AACXC,IAAAA,WAAAA,EAAaF,IAAKE,CAAAA,WAAAA;AAClBC,IAAAA,KAAAA,EAAOC,CAAEC,CAAAA,MAAAA,CAAO,EAAC,EAAGC,WAAW,EAAA;IAC/BC,WAAa,EAAA,MAAMP,KAAKQ,kBAAkB;AAC5C,GAAA,CAAA,CAAA,CAAA;AAEF,EAAA,MAAMC,wBAAwB,IAAIC,OAAAA,CAChCd,WAAYG,CAAAA,GAAAA,CAAI,CAACC,IAAS,KAAA;AAACA,IAAAA,IAAAA,CAAKG,KAAMQ,CAAAA,KAAAA;IAAOX,IAAKO,CAAAA;GAAY,CAAA,CAAA;AAGhE,EAAMK,MAAAA,UAAAA,GAAaR,EAAEC,MAAO,CAAA;AAC1BQ,IAAAA,WAAAA,EAAaT,EACVU,MAAOV,CAAAA,CAAAA,CAAEW,QAAM,CAAA,CACfC,SACC,mQAAA,CAAA;AAEJC,IAAAA,QAAAA,EAAUb,CACPW,CAAAA,MAAAA,EACAC,CAAAA,QAAAA,CACC,yMAAA,CAAA;IAEJE,IAAMd,EAAAA,CAAAA,CACHe,KACCf,CAAAA,CAAAA,CAAEC,MAAO,CAAA;AACPe,MAAAA,KAAAA,EAAOhB,CACJW,CAAAA,MAAAA,EACAC,CAAAA,QAAAA,CAAS,+DAAA,CAAA;AACZK,MAAAA,QAAAA,EAAUjB,CAAEW,CAAAA,MAAAA,EAASC,CAAAA,QAAAA,CAAS,kDAAA,CAAA;AAC9BM,MAAAA,QAAAA,EAAUlB,CACPmB,CAAAA,OAAAA,EACAP,CAAAA,QAAAA,CAAS,sEAAA,CAAA;AACZQ,MAAAA,WAAAA,EAAapB,CACVmB,CAAAA,OAAAA,EACAP,CAAAA,QAAAA,CAAS,4EAAA;KACd,CAAA,CAEDA,CAAAA,QAAAA,CACC,yQAAA,CAAA;IAEJS,QAAUrB,EAAAA,CAAAA,CAAEsB,mBAAmB,MAAQ,EAAA;AACrCtB,MAAAA,CAAAA,CACGC,MAAO,CAAA;QACNsB,IAAMvB,EAAAA,CAAAA,CAAEwB,QAAQ,SAAA,CAAA;AAChBC,QAAAA,OAAAA,EAASzB,CACNW,CAAAA,MAAAA,EACAC,CAAAA,QAAAA,CACC,mJAAA;OAEN,CAAA,CACCA,SAAS,gFAAA,CAAA;SACRc,UAAWlC,CAAAA,WAAAA,EAAa,CAAA,CACxB,GAAA;AACEQ,QAAAA,CAAAA,CACGC,MAAO,CAAA;UACNsB,IAAMvB,EAAAA,CAAAA,CAAEwB,QAAQ,MAAA,CAAA;UAChBG,KAAO3B,EAAAA,CAAAA,CAAEe,KACPf,CAAAA,CAAAA,CAAEsB,kBACA,CAAA,MAAA,EACA3B,IAAIH,WAAa,EAAA,CAACI,IAChBI,KAAAA,CAAAA,CACGC,MAAO,CAAA;YACNJ,IAAMG,EAAAA,CAAAA,CAAEwB,OAAQ5B,CAAAA,IAAAA,CAAKC,IAAI,CAAA;AACzBE,YAAAA,KAAAA,EAAOH,IAAKG,CAAAA;AACd,WAAA,EACCa,QAAShB,CAAAA,IAAAA,CAAKE,WAAW,CAAA,CAAA,CAAA;SAIpC,CAAA,CACCc,SAAS,sCAAA;UAEd;AACL,KAAA;GACH,CAAA;AAEA,EAAO,OAAA;AACLJ,IAAAA,UAAAA;AACAoB,IAAAA,IAAAA,EAAMC,aAAarB,UAAY,EAAA;AAC7BsB,MAAAA,QAAAA,0BAAWC,IAAAA,KAAAA;AACT,QAAA,MAAMC,GAAMD,GAAAA,IAAAA;AACZ,QAAIC,IAAAA,GAAAA,CAAIC,QAAaC,KAAAA,qBAAAA,CAAsBC,SAAW,EAAA;AACpD,UAAM5B,MAAAA,KAAAA,GAAQyB,IAAIzB,KAAK,EAAA;AACvB,UAAM6B,MAAAA,MAAAA,GAAS/B,qBAAsBgC,CAAAA,GAAAA,CAAI9B,KAAAA,CAAAA;AACzC,UAAA,IAAI6B,MAAQ,EAAA;AACV/B,YAAAA,qBAAAA,CAAsBiC,OAAO/B,KAAAA,CAAAA;AAC7B,YAAO6B,OAAAA,MAAAA;AACT;AACF;AACA,QAAOG,OAAAA,cAAAA;OAVC,EAAA,UAAA;KAYZ;AACF,GAAA;AACF;AA/FsBjD,MAAAA,CAAAA,wBAAAA,EAAAA,0BAAAA,CAAAA;AAqGTkD,MAAAA,kBAAAA,GAAqB,IAAIC,cAAe,CAAA;AACnDL,EAAAA,MAAAA,EAAQpC,EAAEC,MAAO,CAAA;AACfmC,IAAAA,MAAAA,EAAQpC,EAAEW,MAAM;GAClB,CAAA;EACA+B,QAAU,EAAA,CAAA;;;;;;;AAOZ,CAAA;AAEaC,MAAAA,qBAAAA,GAAwB,IAAIF,cAAe,CAAA;AACtDL,EAAAA,MAAAA,EAAQpC,EAAEC,MAAO,CAAA;AAAE2C,IAAAA,OAAAA,EAAS5C,EAAEW,MAAM;GAAG,CAAA;EACvC+B,QAAU,EAAA,CAAA,WAAA;AACZ,CAAA","file":"prompts.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { z, ZodFirstPartyTypeKind, ZodObjectDef, ZodTypeDef } from \"zod\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { hasAtLeast, map } from \"remeda\";\nimport { toJsonSchema } from \"@/internals/helpers/schema.js\";\nimport { ignoreOverride, JsonSchema7Type } from \"zod-to-json-schema\";\n\nexport async function createRePlanOutputSchema<T extends AnyTool>(tools: T[]) {\n  const toolSchemas = await Promise.all(\n    tools.map(async (tool) => ({\n      name: tool.name,\n      description: tool.description,\n      input: z.object({}).passthrough(),\n      inputSchema: await tool.getInputJsonSchema(),\n    })),\n  );\n  const zodSchemaToJsonSchema = new WeakMap(\n    toolSchemas.map((tool) => [tool.input.shape, tool.inputSchema]),\n  );\n\n  const definition = z.object({\n    information: z\n      .record(z.string())\n      .describe(\n        \"Summary of the factual information that was collected so far. Eg. 'height of the Eiffel tower': '300m'. Only information that was provided by tools or the user in this very conversation is allowed to be included here. Other information must not be included.\",\n      ),\n    lookback: z\n      .string()\n      .describe(\n        \"A full summary of what has happened so far, focusing on what the assistant tried, what went well and what failed. This repeats in every message, but always concerns the full history up to that point.\",\n      ),\n    plan: z\n      .array(\n        z.object({\n          title: z\n            .string()\n            .describe(\"Title of this step, shortly describing what needs to be done.\"),\n          decision: z.string().describe(\"Assistant's decision of how to tackle this step.\"),\n          research: z\n            .boolean()\n            .describe(\"Does this step involve looking up factual information through tools?\"),\n          computation: z\n            .boolean()\n            .describe(\"Does this step involve calculating or computing information through tools?\"),\n        }),\n      )\n      .describe(\n        \"Detailed step-by-step plan of what steps will the assistant take from start to finish to fulfill the user's request. Includes concrete facts and numbers wherever possible. This repeats in every message, but always contains all the future steps from this point on.\",\n      ),\n    nextStep: z.discriminatedUnion(\"type\", [\n      z\n        .object({\n          type: z.literal(\"message\"),\n          message: z\n            .string()\n            .describe(\n              \"Message with the response, that is sent back to the user. Always include a bit of info on how you arrived at the answer. Be friendly and helpful.\",\n            ),\n        })\n        .describe(\"Message the user -- either to give the answer, or to ask for more information.\"),\n      ...(hasAtLeast(toolSchemas, 1)\n        ? [\n            z\n              .object({\n                type: z.literal(\"tool\"),\n                calls: z.array(\n                  z.discriminatedUnion(\n                    \"name\",\n                    map(toolSchemas, (tool) =>\n                      z\n                        .object({\n                          name: z.literal(tool.name),\n                          input: tool.input,\n                        })\n                        .describe(tool.description),\n                    ),\n                  ),\n                ),\n              })\n              .describe(\"Obtain more information using tools.\"),\n          ]\n        : []),\n    ]),\n  });\n\n  return {\n    definition,\n    json: toJsonSchema(definition, {\n      override: (_def: ZodTypeDef) => {\n        const def = _def as ZodObjectDef;\n        if (def.typeName === ZodFirstPartyTypeKind.ZodObject) {\n          const shape = def.shape();\n          const schema = zodSchemaToJsonSchema.get(shape) as JsonSchema7Type;\n          if (schema) {\n            zodSchemaToJsonSchema.delete(shape);\n            return schema;\n          }\n        }\n        return ignoreOverride;\n      },\n    }),\n  };\n}\n\nexport type RePlanState = z.output<\n  Awaited<ReturnType<typeof createRePlanOutputSchema>>[\"definition\"]\n>;\n\nexport const RePlanSystemPrompt = new PromptTemplate({\n  schema: z.object({\n    schema: z.string(),\n  }),\n  template: `The assistant is created by IBM and refers to itself as Bee. It's named after the IBM logo.\n\nThe assistant is very intelligent and helpful. It always thinks ahead, and uses smart approaches to solve the user's problems. The assistant is an expert-level user of the provided tools, and can utilize them to their maximum potential.\n\nThe assistant is forbidden from using factual information that was not provided by the user or tools in this very conversation. All information about places, people, events, etc. is unknown to the assistant, and the assistant must use tools to obtain it.\n\nOutput Schema: {{schema}}`,\n});\n\nexport const RePlanAssistantPrompt = new PromptTemplate({\n  schema: z.object({ results: z.string() }),\n  template: `{{results}}`,\n});\n"]}

package/dist/agents/experimental/streamlit/agent.cjs

@@ -0,0 +1,161 @@
+'use strict';
+
+var base_cjs = require('../../base.cjs');
+var emitter_cjs = require('../../../emitter/emitter.cjs');
+var remeda = require('remeda');
+var message_cjs = require('../../../backend/message.cjs');
+var prompts_cjs = require('./prompts.cjs');
+var tokenMemory_cjs = require('../../../memory/tokenMemory.cjs');
+var string_cjs = require('../../../internals/helpers/string.cjs');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class StreamlitAgent extends base_cjs.BaseAgent {
+  static {
+    __name(this, "StreamlitAgent");
+  }
+  input;
+  emitter;
+  constructor(input) {
+    super(), this.input = input, this.emitter = new emitter_cjs.Emitter({
+      namespace: [
+        "agent",
+        "experimental",
+        "streamlit"
+      ],
+      creator: this
+    });
+  }
+  get meta() {
+    return {
+      name: `Streamlit`,
+      tools: [],
+      description: `StreamlitAgent is an experimental meta-app agent that uses \`Meta LLaMa 3.1 70B\` to build \`IBM Granite 3 8B\`-powered apps using Streamlit -- a simple UI framework for Python.`
+    };
+  }
+  set memory(memory) {
+    this.input.memory = memory;
+  }
+  get memory() {
+    return this.input.memory;
+  }
+  async _run(input, _options, run) {
+    const { userMessage, runMemory } = await this.prepare(input);
+    let content = "";
+    const raw = await this.input.llm.create({
+      stream: true,
+      messages: runMemory.messages,
+      abortSignal: run.signal
+    }).observe((emitter) => {
+      emitter.on("newToken", async ({ value: chunk }) => {
+        const delta = chunk.getTextContent();
+        if (delta) {
+          content += delta;
+          await run.emitter.emit("newToken", {
+            delta,
+            state: {
+              content
+            },
+            chunk
+          });
+        }
+      });
+    });
+    const result = this.parse(content || raw.getTextContent());
+    const assistantMessage = new message_cjs.AssistantMessage(content);
+    await this.memory.addMany([
+      userMessage,
+      assistantMessage
+    ].filter(remeda.isTruthy));
+    return {
+      result,
+      message: assistantMessage,
+      memory: runMemory
+    };
+  }
+  async prepare(input) {
+    const systemMessage = new message_cjs.SystemMessage((this.input.templates?.system ?? prompts_cjs.StreamlitAgentSystemPrompt).render({}));
+    const userMessage = input.prompt !== null || this.memory.isEmpty() ? new message_cjs.UserMessage(input.prompt ?? "No message.", {
+      createdAt: /* @__PURE__ */ new Date()
+    }) : null;
+    const inputMessages = [
+      ...this.memory.messages,
+      userMessage
+    ].filter(remeda.isTruthy);
+    const runMemory = new tokenMemory_cjs.TokenMemory({
+      capacityThreshold: 0.85,
+      syncThreshold: 0.6,
+      handlers: {
+        removalSelector(msgs) {
+          const prevConversationMsg = msgs.find((msg) => inputMessages.includes(msg));
+          if (prevConversationMsg && prevConversationMsg !== remeda.last(inputMessages)) {
+            return prevConversationMsg;
+          }
+          const lastMsg = msgs.length > 3 && msgs.find((m) => m.role === message_cjs.Role.ASSISTANT);
+          if (!lastMsg) {
+            throw new base_cjs.AgentError("Cannot fit the current conversation into the context window!");
+          }
+          return lastMsg;
+        }
+      }
+    });
+    await runMemory.addMany([
+      systemMessage,
+      ...inputMessages
+    ].filter(remeda.isTruthy));
+    return {
+      runMemory,
+      userMessage
+    };
+  }
+  parse(raw) {
+    const blocks = [];
+    for (let i = 0; i < raw.length; ) {
+      const text = raw.substring(i);
+      const code = string_cjs.findFirstPair(text, [
+        "```python-app\n",
+        "\n```"
+      ], {
+        allowOverlap: true
+      });
+      if (!code) {
+        blocks.push({
+          start: i,
+          end: i + text.length,
+          content: text,
+          name: "text"
+        });
+        break;
+      }
+      if (code.start > 0) {
+        blocks.push({
+          name: "text",
+          start: i,
+          end: i + code.start,
+          content: text.substring(0, code.start)
+        });
+      }
+      blocks.push({
+        name: "app",
+        content: code.inner,
+        start: i + code.start,
+        end: i + code.end
+      });
+      i += code.end;
+    }
+    return {
+      raw,
+      blocks
+    };
+  }
+  createSnapshot() {
+    return {
+      ...super.createSnapshot(),
+      input: this.input
+    };
+  }
+}
+
+exports.StreamlitAgent = StreamlitAgent;
+//# sourceMappingURL=agent.cjs.map
+//# sourceMappingURL=agent.cjs.map

package/dist/agents/experimental/streamlit/agent.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/experimental/streamlit/agent.ts"],"names":["StreamlitAgent","BaseAgent","emitter","constructor","input","Emitter","namespace","creator","meta","name","tools","description","memory","_run","_options","run","userMessage","runMemory","prepare","content","raw","llm","create","stream","messages","abortSignal","signal","observe","on","value","chunk","delta","getTextContent","emit","state","result","parse","assistantMessage","AssistantMessage","addMany","filter","isTruthy","message","systemMessage","SystemMessage","templates","system","StreamlitAgentSystemPrompt","render","prompt","isEmpty","UserMessage","createdAt","Date","inputMessages","TokenMemory","capacityThreshold","syncThreshold","handlers","removalSelector","msgs","prevConversationMsg","find","msg","includes","last","lastMsg","length","m","role","Role","ASSISTANT","AgentError","blocks","i","text","substring","code","findFirstPair","allowOverlap","push","start","end","inner","createSnapshot"],"mappings":";;;;;;;;;;;;AAqEO,MAAMA,uBAAuBC,kBAAAA,CAAAA;EArEpC;;;;AAsESC,EAAAA,OAAAA;AAKPC,EAAAA,WAAAA,CAA+BC,KAA4B,EAAA;AACzD,IAAA,KAAA,IAAK,IADwBA,CAAAA,KAAAA,GAAAA,OAAAA,IALxBF,CAAAA,OAAAA,GAAU,IAAIG,mBAAyB,CAAA;MAC5CC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA,cAAA;AAAgB,QAAA;;MACrCC,OAAS,EAAA;KACX,CAAA;AAIA;AAEA,EAAA,IAAWC,IAAkB,GAAA;AAC3B,IAAO,OAAA;MACLC,IAAM,EAAA,CAAA,SAAA,CAAA;AACNC,MAAAA,KAAAA,EAAO,EAAA;MACPC,WAAa,EAAA,CAAA,iLAAA;AACf,KAAA;AACF;AAEA,EAAA,IAAIC,OAAOA,MAAoB,EAAA;AAC7B,IAAA,IAAA,CAAKR,MAAMQ,MAASA,GAAAA,MAAAA;AACtB;AAEA,EAAA,IAAWA,MAAS,GAAA;AAClB,IAAA,OAAO,KAAKR,KAAMQ,CAAAA,MAAAA;AACpB;EAEA,MAAgBC,IAAAA,CACdT,KACAU,EAAAA,QAAAA,EACAC,GAC6B,EAAA;AAC7B,IAAA,MAAM,EAAEC,WAAaC,EAAAA,SAAAA,KAAc,MAAM,IAAA,CAAKC,QAAQd,KAAAA,CAAAA;AAEtD,IAAA,IAAIe,OAAU,GAAA,EAAA;AACd,IAAA,MAAMC,GAAM,GAAA,MAAM,IAAKhB,CAAAA,KAAAA,CAAMiB,IAC1BC,MAAO,CAAA;MACNC,MAAQ,EAAA,IAAA;AACRC,MAAAA,QAAAA,EAAUP,SAAUO,CAAAA,QAAAA;AACpBC,MAAAA,WAAAA,EAAaV,GAAIW,CAAAA;KACnB,CAAA,CACCC,OAAQ,CAAA,CAACzB,OAAAA,KAAAA;AACRA,MAAAA,OAAAA,CAAQ0B,GAAG,UAAY,EAAA,OAAO,EAAEC,KAAAA,EAAOC,OAAO,KAAA;AAC5C,QAAMC,MAAAA,KAAAA,GAAQD,MAAME,cAAc,EAAA;AAClC,QAAA,IAAID,KAAO,EAAA;AACTZ,UAAWY,OAAAA,IAAAA,KAAAA;AACX,UAAMhB,MAAAA,GAAAA,CAAIb,OAAQ+B,CAAAA,IAAAA,CAAK,UAAY,EAAA;AAAEF,YAAAA,KAAAA;YAAOG,KAAO,EAAA;AAAEf,cAAAA;AAAQ,aAAA;AAAGW,YAAAA;WAAM,CAAA;AACxE;OACF,CAAA;KACF,CAAA;AACF,IAAA,MAAMK,SAAS,IAAKC,CAAAA,KAAAA,CAAMjB,OAAWC,IAAAA,GAAAA,CAAIY,gBAAc,CAAA;AAEvD,IAAMK,MAAAA,gBAAAA,GAAmB,IAAIC,4BAAAA,CAAiBnB,OAAAA,CAAAA;AAC9C,IAAM,MAAA,IAAA,CAAKP,OAAO2B,OAAQ,CAAA;AAACvB,MAAAA,WAAAA;AAAaqB,MAAAA;AAAkBG,KAAAA,CAAAA,MAAAA,CAAOC,eAAAA,CAAAA,CAAAA;AAEjE,IAAO,OAAA;AACLN,MAAAA,MAAAA;MACAO,OAASL,EAAAA,gBAAAA;MACTzB,MAAQK,EAAAA;AACV,KAAA;AACF;AAEA,EAAA,MAAgBC,QAAQd,KAA0B,EAAA;AAChD,IAAMuC,MAAAA,aAAAA,GAAgB,IAAIC,yBAAAA,CAAAA,CACvB,IAAKxC,CAAAA,KAAAA,CAAMyC,SAAWC,EAAAA,MAAAA,IAAUC,sCAA4BC,EAAAA,MAAAA,CAAO,EAAC,CAAA,CAAA;AAGvE,IAAA,MAAMhC,WACJZ,GAAAA,KAAAA,CAAM6C,MAAW,KAAA,IAAA,IAAQ,IAAKrC,CAAAA,MAAAA,CAAOsC,OAAO,EAAA,GACxC,IAAIC,uBAAAA,CAAY/C,KAAM6C,CAAAA,MAAAA,IAAU,aAAe,EAAA;AAAEG,MAAAA,SAAAA,sBAAeC,IAAAA;AAAO,KAAA,CACvE,GAAA,IAAA;AAEN,IAAA,MAAMC,aAAgB,GAAA;AAAI,MAAA,GAAA,IAAA,CAAK1C,MAAOY,CAAAA,QAAAA;AAAUR,MAAAA;AAAawB,KAAAA,CAAAA,MAAAA,CAAOC,eAAAA,CAAAA;AAEpE,IAAMxB,MAAAA,SAAAA,GAAY,IAAIsC,2BAAY,CAAA;MAChCC,iBAAmB,EAAA,IAAA;MACnBC,aAAe,EAAA,GAAA;MACfC,QAAU,EAAA;AACRC,QAAAA,eAAAA,CAAgBC,IAAI,EAAA;AAElB,UAAMC,MAAAA,mBAAAA,GAAsBD,KAAKE,IAAK,CAAA,CAACC,QAAQT,aAAcU,CAAAA,QAAAA,CAASD,GAAAA,CAAAA,CAAAA;AACtE,UAAA,IAAIF,mBAAuBA,IAAAA,mBAAAA,KAAwBI,WAAKX,CAAAA,aAAAA,CAAgB,EAAA;AACtE,YAAOO,OAAAA,mBAAAA;AACT;AAEA,UAAMK,MAAAA,OAAAA,GAAUN,IAAKO,CAAAA,MAAAA,GAAS,CAAKP,IAAAA,IAAAA,CAAKE,IAAK,CAAA,CAACM,CAAMA,KAAAA,CAAAA,CAAEC,IAASC,KAAAA,gBAAAA,CAAKC,SAAS,CAAA;AAC7E,UAAA,IAAI,CAACL,OAAS,EAAA;AACZ,YAAM,MAAA,IAAIM,oBAAW,8DAAA,CAAA;AACvB;AACA,UAAON,OAAAA,OAAAA;AACT;AACF;KACF,CAAA;AACA,IAAA,MAAMjD,UAAUsB,OAAQ,CAAA;AAACI,MAAAA,aAAAA;AAAkBW,MAAAA,GAAAA;AAAed,KAAAA,CAAAA,MAAAA,CAAOC,eAAAA,CAAAA,CAAAA;AAEjE,IAAO,OAAA;AAAExB,MAAAA,SAAAA;AAAWD,MAAAA;AAAY,KAAA;AAClC;AAEUoB,EAAAA,KAAAA,CAAMhB,GAAqB,EAAA;AACnC,IAAA,MAAMqD,SAAkB,EAAA;AAExB,IAAA,KAAA,IAASC,CAAI,GAAA,CAAA,EAAGA,CAAItD,GAAAA,GAAAA,CAAI+C,MAAU,IAAA;AAChC,MAAMQ,MAAAA,IAAAA,GAAOvD,GAAIwD,CAAAA,SAAAA,CAAUF,CAAAA,CAAAA;AAE3B,MAAMG,MAAAA,IAAAA,GAAOC,yBAAcH,IAAM,EAAA;AAAC,QAAA,iBAAA;AAAmB,QAAA;AAAU,OAAA,EAAA;QAAEI,YAAc,EAAA;OAAK,CAAA;AACpF,MAAA,IAAI,CAACF,IAAM,EAAA;AACTJ,QAAAA,MAAAA,CAAOO,IAAK,CAAA;UAAEC,KAAOP,EAAAA,CAAAA;AAAGQ,UAAAA,GAAAA,EAAKR,IAAIC,IAAKR,CAAAA,MAAAA;UAAQhD,OAASwD,EAAAA,IAAAA;UAAMlE,IAAM,EAAA;SAAO,CAAA;AAC1E,QAAA;AACF;AAEA,MAAIoE,IAAAA,IAAAA,CAAKI,QAAQ,CAAG,EAAA;AAClBR,QAAAA,MAAAA,CAAOO,IAAK,CAAA;UACVvE,IAAM,EAAA,MAAA;UACNwE,KAAOP,EAAAA,CAAAA;AACPQ,UAAAA,GAAAA,EAAKR,IAAIG,IAAKI,CAAAA,KAAAA;AACd9D,UAAAA,OAAAA,EAASwD,IAAKC,CAAAA,SAAAA,CAAU,CAAGC,EAAAA,IAAAA,CAAKI,KAAK;SACvC,CAAA;AACF;AACAR,MAAAA,MAAAA,CAAOO,IAAK,CAAA;QACVvE,IAAM,EAAA,KAAA;AACNU,QAAAA,OAAAA,EAAS0D,IAAKM,CAAAA,KAAAA;AACdF,QAAAA,KAAAA,EAAOP,IAAIG,IAAKI,CAAAA,KAAAA;AAChBC,QAAAA,GAAAA,EAAKR,IAAIG,IAAKK,CAAAA;OAChB,CAAA;AACAR,MAAAA,CAAAA,IAAKG,IAAKK,CAAAA,GAAAA;AACZ;AAEA,IAAO,OAAA;AACL9D,MAAAA,GAAAA;AACAqD,MAAAA;AACF,KAAA;AACF;EAEAW,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;AACThF,MAAAA,KAAAA,EAAO,IAAKA,CAAAA;AACd,KAAA;AACF;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { isTruthy, last } from \"remeda\";\nimport { AssistantMessage, Message, Role, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport {\n  StreamlitAgentSystemPrompt,\n  StreamlitAgentTemplates,\n} from \"@/agents/experimental/streamlit/prompts.js\";\nimport { TokenMemory } from \"@/memory/tokenMemory.js\";\nimport { findFirstPair } from \"@/internals/helpers/string.js\";\nimport { ChatModel, ChatModelOutput } from \"@/backend/chat.js\";\n\nexport interface StreamlitAgentInput {\n  llm: ChatModel;\n  memory: BaseMemory;\n  templates?: Partial<StreamlitAgentTemplates>;\n}\n\nexport interface StreamlitRunInput {\n  prompt: string | null;\n}\n\ninterface Block {\n  name: \"text\" | \"app\";\n  content: string;\n  start: number;\n  end: number;\n}\n\ninterface Result {\n  raw: string;\n  blocks: Block[];\n}\n\nexport interface StreamlitRunOutput {\n  result: Result;\n  message: Message;\n  memory: BaseMemory;\n}\n\nexport interface StreamlitEvents {\n  newToken: Callback<{\n    delta: string;\n    state: Readonly<{\n      content: string;\n    }>;\n    chunk: ChatModelOutput;\n  }>;\n}\n\nexport class StreamlitAgent extends BaseAgent<StreamlitRunInput, StreamlitRunOutput> {\n  public emitter = new Emitter<StreamlitEvents>({\n    namespace: [\"agent\", \"experimental\", \"streamlit\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: StreamlitAgentInput) {\n    super();\n  }\n\n  public get meta(): AgentMeta {\n    return {\n      name: `Streamlit`,\n      tools: [],\n      description: `StreamlitAgent is an experimental meta-app agent that uses \\`Meta LLaMa 3.1 70B\\` to build \\`IBM Granite 3 8B\\`-powered apps using Streamlit -- a simple UI framework for Python.`,\n    };\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  public get memory() {\n    return this.input.memory;\n  }\n\n  protected async _run(\n    input: StreamlitRunInput,\n    _options: BaseAgentRunOptions,\n    run: GetRunContext<typeof this>,\n  ): Promise<StreamlitRunOutput> {\n    const { userMessage, runMemory } = await this.prepare(input);\n\n    let content = \"\";\n    const raw = await this.input.llm\n      .create({\n        stream: true,\n        messages: runMemory.messages,\n        abortSignal: run.signal,\n      })\n      .observe((emitter) => {\n        emitter.on(\"newToken\", async ({ value: chunk }) => {\n          const delta = chunk.getTextContent();\n          if (delta) {\n            content += delta;\n            await run.emitter.emit(\"newToken\", { delta, state: { content }, chunk });\n          }\n        });\n      });\n    const result = this.parse(content || raw.getTextContent());\n\n    const assistantMessage = new AssistantMessage(content);\n    await this.memory.addMany([userMessage, assistantMessage].filter(isTruthy));\n\n    return {\n      result,\n      message: assistantMessage,\n      memory: runMemory,\n    };\n  }\n\n  protected async prepare(input: StreamlitRunInput) {\n    const systemMessage = new SystemMessage(\n      (this.input.templates?.system ?? StreamlitAgentSystemPrompt).render({}),\n    );\n\n    const userMessage =\n      input.prompt !== null || this.memory.isEmpty()\n        ? new UserMessage(input.prompt ?? \"No message.\", { createdAt: new Date() })\n        : null;\n\n    const inputMessages = [...this.memory.messages, userMessage].filter(isTruthy);\n\n    const runMemory = new TokenMemory({\n      capacityThreshold: 0.85,\n      syncThreshold: 0.6,\n      handlers: {\n        removalSelector(msgs) {\n          // First we remove messages from the past conversations\n          const prevConversationMsg = msgs.find((msg) => inputMessages.includes(msg));\n          if (prevConversationMsg && prevConversationMsg !== last(inputMessages)) {\n            return prevConversationMsg;\n          }\n\n          const lastMsg = msgs.length > 3 && msgs.find((m) => m.role === Role.ASSISTANT);\n          if (!lastMsg) {\n            throw new AgentError(\"Cannot fit the current conversation into the context window!\");\n          }\n          return lastMsg;\n        },\n      },\n    });\n    await runMemory.addMany([systemMessage, ...inputMessages].filter(isTruthy));\n\n    return { runMemory, userMessage };\n  }\n\n  protected parse(raw: string): Result {\n    const blocks: Block[] = [];\n\n    for (let i = 0; i < raw.length; ) {\n      const text = raw.substring(i);\n\n      const code = findFirstPair(text, [\"```python-app\\n\", \"\\n```\"], { allowOverlap: true });\n      if (!code) {\n        blocks.push({ start: i, end: i + text.length, content: text, name: \"text\" });\n        break;\n      }\n\n      if (code.start > 0) {\n        blocks.push({\n          name: \"text\",\n          start: i,\n          end: i + code.start,\n          content: text.substring(0, code.start),\n        });\n      }\n      blocks.push({\n        name: \"app\",\n        content: code.inner,\n        start: i + code.start,\n        end: i + code.end,\n      });\n      i += code.end;\n    }\n\n    return {\n      raw,\n      blocks,\n    };\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: this.input,\n    };\n  }\n}\n"]}