@caretakerai/agent 0.0.13 → 0.0.15

package/README.md

@@ -8,8 +8,6 @@ The Agent Framework provides a flexible and extensible environment for building
 
 - `Activity`: Represents a single unit of work or thought that the agent performs. Activities can be of different kinds, such as `Observation`, `Thought`, or `Action`.
 
-- `Action`: A subclass of `Activity` that represents a concrete action the agent can take.
-
 - `Optimizer`: A component used to improve the agent's performance by optimizing its activities and decisions.
 
 ## Initialization Parameters
@@ -17,21 +15,23 @@ The Agent Framework provides a flexible and extensible environment for building
 When creating an `Agent`, you must provide an `AgentPrams` object with the following properties:
 
 - `name`: The name of the agent.
+- `description`: A description of the agent's purpose and capabilities.
 - `llm`: The language model the agent will use for understanding and generating text.
-- `actions`: A list of `Action` objects that the agent can perform.
-- `description`: (Optional) A description of the agent's purpose and capabilities.
+- `isChatModel`: (Optional) A flag to indicate if a chat model is used, affecting prompt formatting.
+- `typeDefs`: GraphQL type definitions for the agent's actions.
+- `resolvers`: (Optional) GraphQL resolvers for implementing the actions.
+- `executor`: (Optional) A custom GraphQL executor to handle agent actions.
 - `history`: (Optional) A list of `Activity` objects representing the agent's past experiences.
 - `examples`: (Optional) Examples of activities to guide the agent's behavior.
-- `constraints`: (Optional) A list of constraints that the agent must adhere to.
 - `objective`: (Optional) The goal the agent is trying to achieve.
-- `thoughtSuffix`: (Optional) A suffix to append to thoughts in the agent's output.
-- `actionSuffix`: (Optional) A suffix to append to actions in the agent's output.
-- `maxIterations`: (Optional) The maximum number of iterations the agent can perform in a single run.
+- `instruction`: (Optional) Completion instruction for the language model.
+- `maxIterations`: (Optional) The maximum number of iterations the agent can perform.
 - `maxRetries`: (Optional) The maximum number of retries for actions.
 - `optimizer`: The optimizer used to improve the agent's performance.
+- `signal`: (Optional) An abort signal to stop the agent's operation.
 - `template`: (Optional) The template for generating prompts for the agent.
+- `stop`: (Optional) A list of strings that, if generated by the agent, should cause it to stop.
 - `logger`: (Optional) The logger the agent will use for outputting information.
-
 ## Objective
 The `objective` parameter defines the goal or purpose that the Agent is trying to achieve. It is a guiding force for the Agent's behavior and decision-making process. When initializing an Agent, you can optionally provide an objective to inform the Agent's actions and ensure they align with the desired outcomes. Here is an example of how an objective might be set:
 
@@ -45,122 +45,76 @@ const agent = new Agent({
 The objective will be included in the Agent's prompt template to provide context to the language model, ensuring that the Agent's interactions are focused and relevant to the goal.
 
 
-## Action
-Actions are the mechanisms through which an AI agent interacts with the application and its environment. They are the operational components that translate the agent's decisions into executable tasks. Each action extends from an abstract `Action` class, which provides a common structure and ensures compatibility with the agent framework.
-
-Here's an example of an `Action` subclass:
-
-```typescript
-import { Action, ActionInput } from '@caretaker/agent';
-
-export class Say extends Action {
-  async call(input: ActionInput): Promise<void> {
-    // Logic to perform the action, e.g., output a message
-    console.log(input.params.message);
-  }
-}
-```
-
-In this `Say` action example, the `call` method is where the action's specific behavior is implemented. When an agent decides to use the `Say` action, it will call this method, passing in the necessary parameters.
-
-Actions are defined and added to the agent's configuration, allowing the agent to use them during its processing loop. They serve as the bridge between the AI's decision-making capabilities and the application's functional operations, enabling the agent to perform tasks such as communicating with the user or calculating results.
-
-The parameters for actions are defined using JSON Schema, which provides a clear and enforceable structure for the data that an action can accept. This ensures that the agent and the actions it performs can interact with consistent and validated data formats.
-
-For example, in the `Say` action from `@say.ts`, the parameters are defined as follows:
-
-```typescript
-const SayParamsSchema = z.object({
-  message: z.string().describe('message to say to the user'),
-}).describe('Parameters for Say action');
-```
-
-This schema is then converted to a JSON Schema, which is used by the `Say` action to validate the parameters before the action is executed:
-
-```typescript
-const SayParamsJsonSchema = zodToJsonSchema(SayParamsSchema, 'SayParamsSchema')
-  .definitions!.SayParamsSchema as JSONSchema;
-```
-
-The use of JSON Schema in action parameters allows for the automatic validation of inputs, ensuring that only the correct data types and structures are passed to the action's `call` method. This is crucial for maintaining the integrity of the agent's operations and preventing runtime errors due to invalid data.
-
-
-The `Multiply` action example above demonstrates how to define an action with input parameters and a return type using Zod schemas and TypeScript. The `params` property is assigned a JSON Schema that is generated from the Zod schema, which is used for runtime validation of the action's input parameters. The `call` method implements the logic to multiply an array of numbers, which are provided as input parameters, and returns the multiplication result. This action can be used by the agent to perform multiplication operations as part of its task execution.
-
+## GraphQL Actions
+In the Agent Framework, actions are defined within a GraphQL schema, which provides a structured and enforceable method for specifying the data the agent can act upon. This schema-driven approach facilitates the declaration of queries and mutations that correspond to the actions available to the agent.
 
-```typescript
-import { z } from 'zod';
-import { zodToJsonSchema } from 'zod-to-json-schema';
-import { Action, ActionInput } from '@caretaker/agent';
-import { JSONSchema } from 'json-schema-to-typescript';
-
-// Define the parameters schema using Zod for the Multiply action
-const MultiplyParamsSchema = z.array(z.number()).describe('Array of numbers to multiply');
-// Infer the TypeScript type from the Zod schema
-type MultiplyParams = z.infer<typeof MultiplyParamsSchema>;
-// Convert the Zod schema to a JSON Schema for runtime validation
-const MultiplyParamsJsonSchema = zodToJsonSchema(MultiplyParamsSchema, 'MultiplyParamsSchema')
-  .definitions!.MultiplyParamsSchema as JSONSchema;
-
-// Define the result schema using Zod for the Multiply action
-const MultiplyResultSchema = z.number().describe('The result of the multiplication');
-// Infer the TypeScript type from the Zod schema
-type MultiplyResult = z.infer<typeof MultiplyResultSchema>;
-// Convert the Zod schema to a JSON Schema for runtime validation
-const MultiplyResultJsonSchema = zodToJsonSchema(MultiplyResultSchema, 'MultiplyResultSchema')
-  .definitions!.MultiplyResultSchema as JSONSchema;
-
-// Multiply action class definition
-export class Multiply extends Action<MultiplyParams, MultiplyResult> {
-  // Assign the JSON Schema to the params and result properties
-  readonly params = MultiplyParamsJsonSchema;
-  readonly result = MultiplyResultJsonSchema;
-  // Define whether this action should signal the agent to exit
-  readonly exit = false;
-  // Define the kind of action, which is the class name
-  readonly kind = Multiply.name;
-  // Provide a description for the action
-  readonly description = 'Multiply the numbers and provide you with the result.';
-  // Provide examples of usage (optional)
-  readonly examples = [];
-
-  // The call method is invoked by the agent when this action is performed
-  async call({ params }: ActionInput<MultiplyParams>): Promise<MultiplyResult> {
-    // Implement the action logic: multiply the numbers provided in params
-    return params.reduce((acc, n) => acc * n, 1);
-  }
-}
-```
+Actions are expressed as mutations within the GraphQL schema, and their execution is handled through resolvers. These resolvers implement the logic that the agent performs when an action is invoked. The parameters for these actions are defined using GraphQL's strong typing system, ensuring that the agent receives and acts upon well-structured and validated data.
 
 ## Usage
 
-To use the Agent Framework, create an instance of the `Agent` class with the necessary parameters, including the language model, actions, and any optional configurations such as history, objectives, and optimizers. Then, invoke the `agent.invoke()` method to start the agent's processing loop. The agent will utilize the provided actions, such as `Say` to communicate with the user or `Multiply` to perform multiplication tasks, to fulfill its objectives within the given constraints.
+To use the Agent Framework, instantiate the `Agent` class with the necessary initialization parameters, including the language model, GraphQL type definitions, and resolvers. You can also include optional configurations such as history, objectives, and optimizers. Once the agent is configured, call the `agent.invoke()` method to begin the agent's processing loop. The agent will execute actions defined in the GraphQL schema to interact with the user and perform tasks in line with its objectives.
 
-```typescript
-import { Agent, Activity, ActivityKind } from '@caretaker/agent';
-import { OpenAI } from 'langchain/llms/openai';
-import { Say } from './actions/say';
-
-const llm = new OpenAI({
-  modelName: 'gpt-3.5-turbo',
-  temperature: 0.7,
-  maxTokens: 256,
-});
+Here is an example of how to set up and use the Agent Framework:
+```ts
 
 const agent = new Agent({
   name: 'CalculatorAI',
-  description: 'An agent that performs arithmetic operations',
-  llm,
-  actions: [
-    new Say(),
-    // ... other actions
-  ],
+  llm: new OpenAI(), // Define LLM
+  objective: 'Help the user with math.', // Describe the objective for the agent.
+  typeDefs: dedent`
+    schema {
+      query: Query
+      mutation: Mutation
+    }
+
+    type Query {
+      # No queries for this case
+    }
+
+    type Mutation {
+      """
+      Relay information to the user and wait for the reply. Note that this is only way of communicating information to the user.
+      """
+      say(input: SayInput!): SayResult
+      add(input: MathInput!): MathResult
+      # ... Rest of mutations
+    }
+
+    # ... Rest of schema
+  `.trim(),
+  resolvers: {
+    Mutation: {
+      say: async (_, { input: { message } }) => {
+        console.log(`${chalk.bold(`${agent.name}:`)} ${message}`);
+
+        const reply = await inputPrompt({
+          message: 'Human:'
+        });
+
+        return { reply };
+      },
+      add: (_, { input: { left, right } }) => {
+        try {
+          return { result: left + right };
+        } catch (error) {
+          return { error };
+        }
+      },
+      // ... rest of resolvers
+    }
+  },
   history: [
-    new Activity({ kind: ActivityKind.Observation, input: 'The user says: How can you help me?' })
+    // Add some agent primer
+    new Activity({
+      kind: ActivityKind.Observation, input: JSON.stringify({
+        data: { say: { reply: 'Hi!, how can you help me?' } }
+      }, null, 2)
+    })
   ],
-  // ... other parameters
+  optimizer: // Add history optimizer
 });
 
+// Invoke the agent
 await agent.invoke();
-```
 
+```

package/dist/activity.d.ts

@@ -6,7 +6,6 @@ export declare enum ActivityKind {
 export type ActivityParams = {
     kind: ActivityKind;
     input: string;
-    attributes?: Record<string, string>;
 };
 export declare class Activity implements ActivityParams {
     kind: ActivityKind;
@@ -15,6 +14,6 @@ export declare class Activity implements ActivityParams {
     constructor(params: ActivityParams);
     prompt(): string;
     toObject(): this;
-    static fromObject({ kind, attributes, input }: Record<string, any>): Activity;
+    static fromObject({ kind, input }: Record<string, any>): Activity;
     static parse(text: string): Activity[];
 }

package/dist/activity.js

@@ -23,8 +23,8 @@ class Activity {
     toObject() {
         return { ...this };
     }
-    static fromObject({ kind, attributes, input }) {
-        return new Activity({ kind, attributes, input });
+    static fromObject({ kind, input }) {
+        return new Activity({ kind, input });
     }
     static parse(text) {
         const { elements: [root] } = (0, xml_js_1.xml2js)(`<root>${text}</root>`, { trim: true });

package/dist/activity.js.map

@@ -1 +1 @@
-{"version":3,"file":"activity.js","sourceRoot":"","sources":["../src/activity.ts"],"names":[],"mappings":";;;AAAA,mCAAiD;AAKjD,IAAY,YAOX;AAPD,WAAY,YAAY;IAEtB,2CAA2B,CAAA;IAE3B,mCAAmB,CAAA;IAEnB,iCAAiB,CAAA;AACnB,CAAC,EAPW,YAAY,4BAAZ,YAAY,QAOvB;AAcD,MAAa,QAAQ;IACnB,IAAI,CAAgB;IACpB,UAAU,CAA0B;IACpC,KAAK,CAAU;IAEf,YAAY,MAAsB;QAChC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC9B,CAAC;IAED,MAAM;QACJ,OAAO,IAAA,eAAM,EACX,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,IAAI,CAAC,KAAK,IAAI,GAAG,EAAE,EACpF,EAAE,OAAO,EAAE,IAAI,EAAE,CAClB;aACE,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC;aACvB,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,QAAQ;QACN,OAAO,EAAE,GAAG,IAAI,EAAE,CAAC;IACrB,CAAC;IAED,MAAM,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAuB;QAChE,OAAO,IAAI,QAAQ,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,IAAY;QACvB,MAAM,EAAE,QAAQ,EAAE,CAAC,IAAI,CAAC,EAAE,GAAG,IAAA,eAAM,EAAC,SAAS,IAAI,SAAS,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAE5E,OAAQ,IAAI,CAAC,QAAsB;aAChC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAW,EAAE,EAAE;YAC/C,MAAM,KAAK,GAAG,IAAA,eAAM,EAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,KAAK,MAAM,CAAE,EAAE,CAAC;iBAChF,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC;iBACvB,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAE3B,OAAO,QAAQ,CAAC,UAAU,CAAC;gBACzB,IAAI,EAAE,IAAI;gBACV,KAAK,EAAE,KAAK;gBACZ,UAAU,EAAE,UAAU;aACvB,CAAC,CAAC;QACL,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC,WAAW,EAAE,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;IAC1G,CAAC;CACF;AA3CD,4BA2CC"}
+{"version":3,"file":"activity.js","sourceRoot":"","sources":["../src/activity.ts"],"names":[],"mappings":";;;AAAA,mCAAiD;AAKjD,IAAY,YAOX;AAPD,WAAY,YAAY;IAEtB,2CAA2B,CAAA;IAE3B,mCAAmB,CAAA;IAEnB,iCAAiB,CAAA;AACnB,CAAC,EAPW,YAAY,4BAAZ,YAAY,QAOvB;AAYD,MAAa,QAAQ;IACnB,IAAI,CAAgB;IACpB,UAAU,CAA0B;IACpC,KAAK,CAAU;IAEf,YAAY,MAAsB;QAChC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC9B,CAAC;IAED,MAAM;QACJ,OAAO,IAAA,eAAM,EACX,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,IAAI,CAAC,KAAK,IAAI,GAAG,EAAE,EACpF,EAAE,OAAO,EAAE,IAAI,EAAE,CAClB;aACE,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC;aACvB,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,QAAQ;QACN,OAAO,EAAE,GAAG,IAAI,EAAE,CAAC;IACrB,CAAC;IAED,MAAM,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,KAAK,EAAuB;QACpD,OAAO,IAAI,QAAQ,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,IAAY;QACvB,MAAM,EAAE,QAAQ,EAAE,CAAC,IAAI,CAAC,EAAE,GAAG,IAAA,eAAM,EAAC,SAAS,IAAI,SAAS,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAE5E,OAAQ,IAAI,CAAC,QAAsB;aAChC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAW,EAAE,EAAE;YAC/C,MAAM,KAAK,GAAG,IAAA,eAAM,EAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,KAAK,MAAM,CAAE,EAAE,CAAC;iBAChF,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC;iBACvB,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAE3B,OAAO,QAAQ,CAAC,UAAU,CAAC;gBACzB,IAAI,EAAE,IAAI;gBACV,KAAK,EAAE,KAAK;gBACZ,UAAU,EAAE,UAAU;aACvB,CAAC,CAAC;QACL,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC,WAAW,EAAE,YAAY,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;IAC1G,CAAC;CACF;AA3CD,4BA2CC"}

package/dist/agent.d.ts

@@ -1,15 +1,19 @@
-import { Logger } from 'winston';
+import { Logger } from 'pino';
 import { BasePromptTemplate } from '@langchain/core/prompts';
 import { BaseLanguageModel } from '@langchain/core/language_models/base';
-import { Action } from './action';
+import type { TypeSource, IResolvers } from '@graphql-tools/utils';
+import { GraphQLSchema } from 'graphql';
 import { Activity } from './activity';
 import { Optimizer } from './types';
+type GraphQLExecutor = (query: string) => Promise<Record<string, unknown>>;
 interface AgentPrams {
     name: string;
     description: string;
     llm: BaseLanguageModel;
     isChatModel?: boolean;
-    actions: Action[];
+    typeDefs: TypeSource;
+    resolvers?: IResolvers;
+    executor?: GraphQLExecutor;
     history?: Activity[];
     examples?: Activity[];
     objective?: string;
@@ -17,6 +21,7 @@ interface AgentPrams {
     maxIterations?: number;
     maxRetries?: number;
     optimizer: Optimizer;
+    signal?: AbortSignal;
     template?: BasePromptTemplate;
     stop?: string[];
     logger?: Logger;
@@ -25,23 +30,25 @@ export declare class Agent implements AgentPrams {
     name: string;
     description: string;
     llm: BaseLanguageModel;
-    actions: Action[];
+    typeDefs: TypeSource;
+    resolvers: IResolvers;
     history: Activity[];
     examples: Activity[];
     objective: string;
     instruction: string;
-    actionSuffix: string;
     maxIterations: number;
     maxRetries: number;
     isChatModel: boolean;
+    signal: AbortSignal;
     optimizer: Optimizer;
     logger: Logger;
     template?: BasePromptTemplate;
+    executor?: GraphQLExecutor;
+    readonly schema: GraphQLSchema;
     static defaults: Partial<AgentPrams>;
     constructor(params: AgentPrams);
     addActivities(...activities: Activity[]): void;
     prompt(params?: Record<string, string>): Promise<void>;
-    execute({ attributes, input }: Activity): Promise<string>;
-    invoke(params?: Record<string, any>): Promise<string>;
+    invoke(params?: Record<string, any>): Promise<void>;
 }
 export {};

package/dist/agent.js

@@ -5,37 +5,45 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Agent = void 0;
 const dedent_1 = __importDefault(require("dedent"));
-const winston_1 = require("winston");
+const yaml_1 = require("yaml");
+const pino_1 = __importDefault(require("pino"));
 const prompts_1 = require("@langchain/core/prompts");
 const output_parsers_1 = require("@langchain/core/output_parsers");
+const schema_1 = require("@graphql-tools/schema");
+const graphql_1 = require("graphql");
 const constants_1 = require("./constants");
 const activity_1 = require("./activity");
 class Agent {
     name;
     description;
     llm;
-    actions;
+    typeDefs;
+    resolvers;
     history;
     examples;
     objective;
     instruction;
-    actionSuffix;
     maxIterations;
     maxRetries;
     isChatModel;
+    signal;
     optimizer;
     logger;
     template;
+    executor;
+    schema;
     static defaults = {
         template: prompts_1.PromptTemplate.fromTemplate((0, dedent_1.default) `
       # Objective
       {objective}
 
-      # Actions
+      # GraphQL Schema
       The only permissible actions you may take are listed below:
-      {actions}
+      \`\`\`graphql
+      {schema}
+      \`\`\`
 
-      **Continue the History with the following format in your response:**
+      **Continue the History with your thoughts and actions following format in your response:**
       {examples}
 
       # History:
@@ -45,43 +53,52 @@ class Agent {
     `),
         objective: 'You are helpful assistant.',
         instruction: (0, dedent_1.default) `
-      Think your further actions step by step before taking any.
-      Your must always explain your choice in your thoughts.
-      Use only actions listed in Actions section.
-      Reject any request that are not related to your objective and cannot be fulfilled within the given list of actions.
-      Provide your Thought and Action here.
+      Plan your further actions step by step before taking any.
+      Always explain your choice in your thoughts.
+      Use only actions listed in the Actions section.
+      Do multiple queries or mutations in a single request if possible.
+      Reject any requests that are not related to your objective and cannot be fulfilled within the given list of actions.
+      Thought and Action goes here.
     `,
         maxRetries: 7,
         isChatModel: false,
         maxIterations: Number.MAX_SAFE_INTEGER,
-        logger: (0, winston_1.createLogger)({ transports: [new winston_1.transports.Console()] }),
+        logger: (0, pino_1.default)(),
         examples: [
             new activity_1.Activity({
                 kind: activity_1.ActivityKind.Observation,
-                input: 'The result of previously taken action',
+                input: (0, dedent_1.default) `
+          <!-- The result of the previous action e.g. -->
+          data
+            theBestNumber:
+              result: 73
+        `.trim(),
             }),
             new activity_1.Activity({
                 kind: activity_1.ActivityKind.Thought,
-                input: 'You must always think before taking the action',
+                input: 'Now I know that the best number is 73. I should share this information with the user immediately.',
             }),
             new activity_1.Activity({
                 kind: activity_1.ActivityKind.Action,
-                attributes: { kind: 'the action kind to take, should be one of listed in Actions section' },
                 input: (0, dedent_1.default) `
-          <!-- The action input as valid JSON e.g. -->
-          {
-            "message": "hello!!"
+          <!-- The action must be a single executable GraphQL request e.g. -->
+          \`\`\`graphql
+          mutation {
+            say(input: { message: "The best number is 73!" }) {
+              reply
+            }
           }
+          \`\`\`
         `.trim(),
             }),
         ],
     };
     constructor(params) {
-        const { actions } = params;
-        if (!actions.length) {
-            throw new Error('Actions list must be non empty');
-        }
         Object.assign(this, Agent.defaults, params);
+        const { typeDefs, resolvers, executor } = params;
+        if (!executor) {
+            this.schema = (0, schema_1.makeExecutableSchema)({ typeDefs, resolvers });
+        }
     }
     addActivities(...activities) {
         activities.forEach(a => this.logger.debug(a));
@@ -101,10 +118,6 @@ class Agent {
             const activitiesStrings = activities.map(a => a.prompt()).join(constants_1.ACTIVITY_SEP);
             return activitiesStrings;
         };
-        const actions = async () => {
-            const actionsStrings = await Promise.all(this.actions.map(a => a._prompt()));
-            return actionsStrings.join(constants_1.ACTION_SEP);
-        };
         const instruction = () => this.instruction
             .split('\n')
             .filter(i => i)
@@ -116,15 +129,15 @@ class Agent {
         };
         const template = await this.template.partial({
             objective: this.objective,
+            schema: this.typeDefs.toString(),
             history,
-            actions,
             examples,
             instruction,
             completions,
         });
         const chain = template
             .pipe((prompt) => prompt.toString().trim())
-            .pipe(this.llm.bind({ stop: [`<${activity_1.ActivityKind.Observation}`] }))
+            .pipe(this.llm.bind({ stop: [`<${activity_1.ActivityKind.Observation}>`] }))
             .pipe(new output_parsers_1.StringOutputParser());
         for (let i = 0; i < this.maxRetries; ++i) {
             let completion = await chain.invoke(params ?? {});
@@ -155,11 +168,16 @@ class Agent {
                 continue;
             }
             try {
-                const observation = await this.execute(activity);
+                const source = activity.input
+                    .replace(/.*?```graphql\s*/, '')
+                    .replace(/```$/, '')
+                    .trim();
+                const result = this.executor
+                    ? await this.executor(source)
+                    : await (0, graphql_1.graphql)({ schema: this.schema, source });
                 this.addActivities(...activities, new activity_1.Activity({
                     kind: activity_1.ActivityKind.Observation,
-                    input: observation,
-                    attributes: { of: activity.attributes.kind }
+                    input: (0, yaml_1.stringify)(result),
                 }));
                 return;
             }
@@ -168,7 +186,6 @@ class Agent {
                 activities.push(new activity_1.Activity({
                     kind: activity_1.ActivityKind.Observation,
                     input: err.toString(),
-                    attributes: { of: activity.attributes.kind }
                 }));
                 this.addActivities(...activities);
                 activities = [];
@@ -178,29 +195,16 @@ class Agent {
         }
         throw new Error('Max number of retries reached.');
     }
-    async execute({ attributes, input }) {
-        const { kind } = attributes;
-        const action = this.actions.find(a => a.kind === kind);
-        if (!action) {
-            throw new Error(`Action "${kind}" is not allowed. Correct your spelling.`);
-        }
-        const observation = await action._call(input, this);
-        return observation;
-    }
     async invoke(params) {
         if (!this.history.length) {
-            throw new Error('Activity list must not be empty.');
+            throw new Error('History must not be empty.');
         }
         if (this.history.at(-1)?.kind !== activity_1.ActivityKind.Observation) {
             throw new Error('Latest experience must be of Observation kind');
         }
         for (let i = 0; i < this.maxIterations; ++i) {
             await this.prompt(params);
-            const activity = this.history.at(-2);
-            const action = this.actions.find(a => a.kind === activity.attributes.kind);
-            if (action.exit) {
-                return this.history.at(-1).input;
-            }
+            this.signal?.throwIfAborted();
         }
         throw new Error('Max number of iterations reached.');
     }

package/dist/agent.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAC5B,qCAA2D;AAC3D,qDAA6E;AAC7E,mEAAoE;AAGpE,2CAAuD;AACvD,yCAAoD;AAuCpD,MAAa,KAAK;IAChB,IAAI,CAAU;IACd,WAAW,CAAU;IACrB,GAAG,CAAqB;IACxB,OAAO,CAAY;IACnB,OAAO,CAAc;IACrB,QAAQ,CAAc;IACtB,SAAS,CAAU;IACnB,WAAW,CAAU;IACrB,YAAY,CAAU;IACtB,aAAa,CAAU;IACvB,UAAU,CAAU;IACpB,WAAW,CAAW;IACtB,SAAS,CAAa;IACtB,MAAM,CAAU;IAChB,QAAQ,CAAsB;IAE9B,MAAM,CAAC,QAAQ,GAAwB;QACrC,QAAQ,EAAE,wBAAc,CAAC,YAAY,CAAC,IAAA,gBAAM,EAAA;;;;;;;;;;;;;;;KAe3C,CAAC;QACF,SAAS,EAAE,4BAA4B;QACvC,WAAW,EAAE,IAAA,gBAAM,EAAA;;;;;;KAMlB;QACD,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,KAAK;QAClB,aAAa,EAAE,MAAM,CAAC,gBAAgB;QACtC,MAAM,EAAE,IAAA,sBAAY,EAAC,EAAE,UAAU,EAAE,CAAC,IAAI,oBAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;QAChE,QAAQ,EAAE;YACR,IAAI,mBAAQ,CAAC;gBACX,IAAI,EAAE,uBAAY,CAAC,WAAW;gBAC9B,KAAK,EAAE,uCAAuC;aAC/C,CAAC;YACF,IAAI,mBAAQ,CAAC;gBACX,IAAI,EAAE,uBAAY,CAAC,OAAO;gBAC1B,KAAK,EAAE,gDAAgD;aACxD,CAAC;YACF,IAAI,mBAAQ,CAAC;gBACX,IAAI,EAAE,uBAAY,CAAC,MAAM;gBACzB,UAAU,EAAE,EAAE,IAAI,EAAE,qEAAqE,EAAE;gBAC3F,KAAK,EAAE,IAAA,gBAAM,EAAA;;;;;SAKZ,CAAC,IAAI,EAAE;aACT,CAAC;SACH;KACF,CAAA;IAED,YAAY,MAAkB;QAC5B,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC;QAE3B,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;YACnB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;SACnD;QAED,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC9C,CAAC;IAED,aAAa,CAAC,GAAG,UAAsB;QACrC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9C,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAA;IAClC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAA+B;QAC1C,IAAI,UAAU,GAAe,EAAE,CAAC;QAEhC,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;YACzB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC5D,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAY,CAAC,CAAC;YACvE,OAAO,cAAc,CAAC;QACxB,CAAC,CAAC;QACF,MAAM,WAAW,GAAG,KAAK,IAAI,EAAE;YAE7B,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;gBAC3C,OAAO,IAAI,uBAAY,CAAC,OAAO,GAAG,CAAC;aACpC;YAED,MAAM,iBAAiB,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAY,CAAC,CAAC;YAC7E,OAAO,iBAAiB,CAAC;QAC3B,CAAC,CAAC;QAEF,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;YACzB,MAAM,cAAc,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC7E,OAAO,cAAc,CAAC,IAAI,CAAC,sBAAU,CAAC,CAAC;QACzC,CAAC,CAAC;QAEF,MAAM,WAAW,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW;aACvC,KAAK,CAAC,IAAI,CAAC;aACX,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;aACd,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;aACzB,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE;YAC1B,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAY,CAAC,CAAC;YAC9E,OAAO,eAAe,CAAC;QACzB,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;YAC3C,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO;YACP,OAAO;YACP,QAAQ;YACR,WAAW;YACX,WAAW;SACZ,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,QAAQ;aACnB,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;aAC1C,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,IAAI,uBAAY,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;aAC/D,IAAI,CAAC,IAAI,mCAAkB,EAAE,CAAC,CAAC;QAElC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,EAAE,CAAC,EAAE;YACxC,IAAI,UAAU,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;YAGlD,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,GAAG,uBAAY,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;gBAC1E,UAAU,GAAG,IAAI,uBAAY,CAAC,OAAO,MAAM,UAAU,EAAE,CAAC;aACzD;YAED,IAAI;gBACF,IAAI,aAAa,GAAG,mBAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBAE3D,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE;oBACzB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;iBAC7C;gBAGD,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,uBAAY,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE;oBACzE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;iBACjD;gBAGD,MAAM,aAAa,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,uBAAY,CAAC,MAAM,CAAC,CAAC;gBACnF,aAAa,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,GAAG,CAAC,CAAC,CAAA;gBAEzD,UAAU,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;aACnC;YAAC,OAAO,CAAC,EAAE;gBACV,MAAM,GAAG,GAAG,CAAU,CAAC;gBACvB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC9B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,6BAA6B,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC5E,SAAS;aACV;YAED,MAAM,QAAQ,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAGnC,IAAI,QAAQ,CAAC,IAAI,KAAK,uBAAY,CAAC,OAAO,EAAE;gBAC1C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;gBAC1D,SAAS;aACV;YAED,IAAI;gBACF,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACjD,IAAI,CAAC,aAAa,CAChB,GAAG,UAAU,EACb,IAAI,mBAAQ,CAAC;oBACX,IAAI,EAAE,uBAAY,CAAC,WAAW;oBAC9B,KAAK,EAAE,WAAW;oBAClB,UAAU,EAAE,EAAE,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE;iBAC7C,CAAC,CACH,CAAC;gBACF,OAAO;aACR;YAAC,OAAM,CAAC,EAAE;gBACT,MAAM,GAAG,GAAG,CAAU,CAAC;gBACvB,UAAU,CAAC,IAAI,CAAC,IAAI,mBAAQ,CAAC;oBAC3B,IAAI,EAAE,uBAAY,CAAC,WAAW;oBAC9B,KAAK,EAAE,GAAG,CAAC,QAAQ,EAAE;oBACrB,UAAU,EAAE,EAAE,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE;iBAC7C,CAAC,CAAC,CAAA;gBACH,IAAI,CAAC,aAAa,CAAC,GAAG,UAAU,CAAC,CAAC;gBAClC,UAAU,GAAG,EAAE,CAAC;gBAChB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,yBAAyB,GAAG,EAAE,CAAC,CAAC;gBAChE,SAAS;aACV;SACF;QAED,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,KAAK,EAAY;QAC3C,MAAM,EAAE,IAAI,EAAE,GAAG,UAAU,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QAEvD,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CAAC,WAAW,IAAI,0CAA0C,CAAC,CAAC;SAC5E;QAED,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACpD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAA4B;QACvC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;YACxB,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;SACrD;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,uBAAY,CAAC,WAAW,EAAE;YAC1D,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;SAClE;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC,EAAE;YAC3C,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC1B,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;YACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAE3E,IAAI,MAAM,CAAC,IAAI,EAAE;gBACf,OAAO,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;aAClC;SACF;QAED,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACvD,CAAC;;AAvOH,sBAwOC"}
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAC5B,+BAAiC;AACjC,gDAAoC;AAEpC,qDAA6E;AAC7E,mEAAoE;AAGpE,kDAA6D;AAE7D,qCAAiD;AAEjD,2CAA2C;AAC3C,yCAAoD;AA+CpD,MAAa,KAAK;IAChB,IAAI,CAAU;IACd,WAAW,CAAU;IACrB,GAAG,CAAqB;IACxB,QAAQ,CAAc;IACtB,SAAS,CAAa;IACtB,OAAO,CAAc;IACrB,QAAQ,CAAa;IACrB,SAAS,CAAS;IAClB,WAAW,CAAS;IACpB,aAAa,CAAS;IACtB,UAAU,CAAS;IACnB,WAAW,CAAU;IACrB,MAAM,CAAc;IACpB,SAAS,CAAa;IACtB,MAAM,CAAU;IAChB,QAAQ,CAAsB;IAC9B,QAAQ,CAAmB;IAElB,MAAM,CAAgB;IAE/B,MAAM,CAAC,QAAQ,GAAwB;QACrC,QAAQ,EAAE,wBAAc,CAAC,YAAY,CAAC,IAAA,gBAAM,EAAA;;;;;;;;;;;;;;;;;KAiB3C,CAAC;QACF,SAAS,EAAE,4BAA4B;QACvC,WAAW,EAAE,IAAA,gBAAM,EAAA;;;;;;;KAOlB;QACD,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,KAAK;QAClB,aAAa,EAAE,MAAM,CAAC,gBAAgB;QACtC,MAAM,EAAE,IAAA,cAAI,GAAE;QACd,QAAQ,EAAE;YACR,IAAI,mBAAQ,CAAC;gBACX,IAAI,EAAE,uBAAY,CAAC,WAAW;gBAC9B,KAAK,EAAE,IAAA,gBAAM,EAAA;;;;;SAKZ,CAAC,IAAI,EAAE;aACT,CAAC;YACF,IAAI,mBAAQ,CAAC;gBACX,IAAI,EAAE,uBAAY,CAAC,OAAO;gBAC1B,KAAK,EAAE,mGAAmG;aAC3G,CAAC;YACF,IAAI,mBAAQ,CAAC;gBACX,IAAI,EAAE,uBAAY,CAAC,MAAM;gBACzB,KAAK,EAAE,IAAA,gBAAM,EAAA;;;;;;;;;SASZ,CAAC,IAAI,EAAE;aACT,CAAC;SACH;KACF,CAAA;IAED,YAAY,MAAkB;QAC5B,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC5C,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;QAGjD,IAAI,CAAC,QAAQ,EAAE;YACb,IAAI,CAAC,MAAM,GAAG,IAAA,6BAAoB,EAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;SAC7D;IACH,CAAC;IAED,aAAa,CAAC,GAAG,UAAsB;QACrC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9C,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAA;IAClC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAA+B;QAC1C,IAAI,UAAU,GAAe,EAAE,CAAC;QAEhC,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;YACzB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC5D,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAY,CAAC,CAAC;YACvE,OAAO,cAAc,CAAC;QACxB,CAAC,CAAC;QAEF,MAAM,WAAW,GAAG,KAAK,IAAI,EAAE;YAE7B,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;gBAC3C,OAAO,IAAI,uBAAY,CAAC,OAAO,GAAG,CAAC;aACpC;YAED,MAAM,iBAAiB,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAY,CAAC,CAAC;YAC7E,OAAO,iBAAiB,CAAC;QAC3B,CAAC,CAAC;QAEF,MAAM,WAAW,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW;aACvC,KAAK,CAAC,IAAI,CAAC;aACX,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;aACd,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;aACzB,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE;YAC1B,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAY,CAAC,CAAC;YAC9E,OAAO,eAAe,CAAC;QACzB,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;YAC3C,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE;YAChC,OAAO;YACP,QAAQ;YACR,WAAW;YACX,WAAW;SACZ,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,QAAQ;aACnB,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;aAC1C,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,IAAI,uBAAY,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC,CAAC;aAChE,IAAI,CAAC,IAAI,mCAAkB,EAAE,CAAC,CAAC;QAElC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,EAAE,CAAC,EAAE;YACxC,IAAI,UAAU,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;YAGlD,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,GAAG,uBAAY,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;gBAC1E,UAAU,GAAG,IAAI,uBAAY,CAAC,OAAO,MAAM,UAAU,EAAE,CAAC;aACzD;YAED,IAAI;gBACF,IAAI,aAAa,GAAG,mBAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBAE3D,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE;oBACzB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;iBAC7C;gBAGD,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,uBAAY,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE;oBACzE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;iBACjD;gBAGD,MAAM,aAAa,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,uBAAY,CAAC,MAAM,CAAC,CAAC;gBACnF,aAAa,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,GAAG,CAAC,CAAC,CAAA;gBAEzD,UAAU,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;aACnC;YAAC,OAAO,CAAC,EAAE;gBACV,MAAM,GAAG,GAAG,CAAU,CAAC;gBACvB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC9B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,6BAA6B,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC5E,SAAS;aACV;YAED,MAAM,QAAQ,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAGnC,IAAI,QAAQ,CAAC,IAAI,KAAK,uBAAY,CAAC,OAAO,EAAE;gBAC1C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;gBAC1D,SAAS;aACV;YAGD,IAAI;gBACF,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK;qBAC1B,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC;qBAC/B,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;qBACnB,IAAI,EAAE,CAAC;gBAIV,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ;oBAC1B,CAAC,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;oBAC7B,CAAC,CAAC,MAAM,IAAA,iBAAO,EAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;gBAGnD,IAAI,CAAC,aAAa,CAChB,GAAG,UAAU,EACb,IAAI,mBAAQ,CAAC;oBACX,IAAI,EAAE,uBAAY,CAAC,WAAW;oBAC9B,KAAK,EAAE,IAAA,gBAAS,EAAC,MAAM,CAAC;iBACzB,CAAC,CACH,CAAC;gBAEF,OAAO;aACR;YAAC,OAAM,CAAC,EAAE;gBACT,MAAM,GAAG,GAAG,CAAU,CAAC;gBAEvB,UAAU,CAAC,IAAI,CAAC,IAAI,mBAAQ,CAAC;oBAC3B,IAAI,EAAE,uBAAY,CAAC,WAAW;oBAC9B,KAAK,EAAE,GAAG,CAAC,QAAQ,EAAE;iBACtB,CAAC,CAAC,CAAC;gBAEJ,IAAI,CAAC,aAAa,CAAC,GAAG,UAAU,CAAC,CAAC;gBAClC,UAAU,GAAG,EAAE,CAAC;gBAChB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,yBAAyB,GAAG,EAAE,CAAC,CAAC;gBAChE,SAAS;aACV;SACF;QAED,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAA4B;QACvC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;YACxB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;SAC/C;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,uBAAY,CAAC,WAAW,EAAE;YAC1D,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;SAClE;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC,EAAE;YAC3C,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC1B,IAAI,CAAC,MAAM,EAAE,cAAc,EAAE,CAAC;SAC/B;QAED,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACvD,CAAC;;AA9OH,sBA+OC"}

package/dist/constants.d.ts

@@ -1,2 +1 @@
-export declare const ACTION_SEP = "\n\n";
 export declare const ACTIVITY_SEP = "\n";

package/dist/constants.js

@@ -1,6 +1,5 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ACTIVITY_SEP = exports.ACTION_SEP = void 0;
-exports.ACTION_SEP = '\n\n';
+exports.ACTIVITY_SEP = void 0;
 exports.ACTIVITY_SEP = '\n';
 //# sourceMappingURL=constants.js.map

package/dist/constants.js.map

@@ -1 +1 @@
-{"version":3,"file":"constants.js","sourceRoot":"","sources":["../src/constants.ts"],"names":[],"mappings":";;;AAAa,QAAA,UAAU,GAAG,MAAM,CAAA;AACnB,QAAA,YAAY,GAAG,IAAI,CAAC"}
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../src/constants.ts"],"names":[],"mappings":";;;AAAa,QAAA,YAAY,GAAG,IAAI,CAAC"}

package/dist/index.d.ts

@@ -1,4 +1,3 @@
-export * from './action';
 export * from './agent';
 export * from './constants';
 export * from './activity';

package/dist/index.js

@@ -14,7 +14,6 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./action"), exports);
 __exportStar(require("./agent"), exports);
 __exportStar(require("./constants"), exports);
 __exportStar(require("./activity"), exports);

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAyB;AACzB,0CAAwB;AACxB,8CAA4B;AAC5B,6CAA2B;AAC3B,0CAAwB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,0CAAwB;AACxB,8CAA4B;AAC5B,6CAA2B;AAC3B,0CAAwB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@caretakerai/agent",
-  "version": "0.0.13",
+  "version": "0.0.15",
   "description": "Single framework for building text-agents",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -21,11 +21,14 @@
     "ajv-errors": "^3.0.0",
     "debug": "^4.3.4",
     "json-schema-to-typescript": "^13.1.1",
-    "winston": "^3.11.0",
-    "xml-js": "^1.6.11"
+    "pino": "^9.0.0",
+    "xml-js": "^1.6.11",
+    "yaml": "^2.4.0"
   },
   "devDependencies": {
+    "@graphql-tools/schema": "^10.0.2",
     "@types/node": "^20.4.1",
+    "graphql": "^16.8.1",
     "rimraf": "^5.0.5",
     "tsx": "^3.12.7",
     "typescript": "^5.2.2"

package/dist/action.d.ts

@@ -1,23 +0,0 @@
-import type { Agent } from './agent';
-import { type JSONSchema } from 'json-schema-to-typescript';
-import { Activity } from './activity';
-export interface ActionInput<T> {
-    params: T;
-    agent: Agent;
-}
-export interface ActionExample {
-    description?: string;
-    activities: Activity[];
-}
-export declare abstract class Action<P = any, R = any> {
-    abstract get exit(): boolean;
-    abstract get kind(): string;
-    abstract get description(): string;
-    abstract get params(): JSONSchema;
-    abstract get result(): JSONSchema;
-    abstract get examples(): ActionExample[];
-    abstract call(input: ActionInput<P>): Promise<R>;
-    private _examplesPrompt;
-    _prompt(template?: string): Promise<string>;
-    _call(input: string, agent: Agent): Promise<string>;
-}

package/dist/action.js

@@ -1,86 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Action = void 0;
-const dedent_1 = __importDefault(require("dedent"));
-const ajv_1 = __importDefault(require("ajv"));
-const ajv_errors_1 = __importDefault(require("ajv-errors"));
-const prompts_1 = require("@langchain/core/prompts");
-const json_schema_to_typescript_1 = require("json-schema-to-typescript");
-const constants_1 = require("./constants");
-const ajv = (0, ajv_errors_1.default)(new ajv_1.default({
-    useDefaults: true,
-    removeAdditional: true,
-    allErrors: true,
-}));
-const ACTION_TEMPLATE = (`
-\`\`\`ts
-/**
- * @kind {kind}
-{description}
- * /
-
-{params}
-
-{result}
-{examples}\`\`\`
-`).trim();
-class Action {
-    _examplesPrompt() {
-        return this.examples
-            .map(({ activities, description }) => [
-            (0, dedent_1.default) `
-        /**
-         * @example ${description}`,
-            activities.map(a => a.prompt())
-                .join(constants_1.ACTIVITY_SEP)
-                .split('\n')
-                .map(s => ` * ${s}`)
-                .join('\n'),
-            ' */'
-        ].join('\n'))
-            .join('\n\n') + '\n';
-    }
-    async _prompt(template = ACTION_TEMPLATE) {
-        const paramsType = `${this.kind}Params`;
-        const resultType = `${this.kind}Result`;
-        const partial = await prompts_1.PromptTemplate.fromTemplate(template).partial({
-            params: async () => {
-                const ts = await (0, json_schema_to_typescript_1.compile)(this.params, paramsType, { bannerComment: '', additionalProperties: false });
-                return ts.replace(/^(export\s*)/gm, '').trim();
-            },
-            result: async () => {
-                const ts = await (0, json_schema_to_typescript_1.compile)(this.result, resultType, { bannerComment: '', additionalProperties: false });
-                return ts.replace(/^(export\s*)/gm, '').trim();
-            },
-            examples: () => this._examplesPrompt(),
-            kind: this.kind,
-            description: this.description
-                .split('\n')
-                .map(s => ` * ${s}`)
-                .join('\n'),
-            paramsType,
-            resultType,
-        });
-        return partial
-            .pipe(prompt => prompt.toString().trim())
-            .invoke({});
-    }
-    async _call(input, agent) {
-        const params = JSON.parse(input);
-        const validator = ajv.compile(this.params);
-        const isValid = validator(params);
-        if (!isValid) {
-            throw new Error(`Action "${this.kind}" params are not valid: ${ajv.errorsText(validator.errors)} `);
-        }
-        const result = await this.call({ params, agent });
-        if (typeof result === 'string') {
-            return result;
-        }
-        return JSON.stringify(result);
-    }
-}
-exports.Action = Action;
-//# sourceMappingURL=action.js.map

package/dist/action.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"action.js","sourceRoot":"","sources":["../src/action.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAC5B,8CAAsB;AACtB,4DAAyC;AACzC,qDAAyD;AAEzD,yEAAqE;AAErE,2CAA2C;AAE3C,MAAM,GAAG,GAAG,IAAA,oBAAe,EACzB,IAAI,aAAG,CAAC;IACN,WAAW,EAAE,IAAI;IACjB,gBAAgB,EAAE,IAAI;IACtB,SAAS,EAAE,IAAI;CAChB,CAAC,CACH,CAAC;AAEF,MAAM,eAAe,GAAG,CAAC;;;;;;;;;;;CAWxB,CAAC,CAAC,IAAI,EAAE,CAAC;AAaV,MAAsB,MAAM;IAUlB,eAAe;QACrB,OAAO,IAAI,CAAC,QAAQ;aACjB,GAAG,CAAC,CAAC,EAAE,UAAU,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;YACpC,IAAA,gBAAM,EAAA;;sBAEQ,WAAW,EAAE;YAC3B,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;iBAC5B,IAAI,CAAC,wBAAY,CAAC;iBAClB,KAAK,CAAC,IAAI,CAAC;iBACX,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;iBACnB,IAAI,CAAC,IAAI,CAAC;YACb,KAAK;SACN,CAAC,IAAI,CAAC,IAAI,CAAC,CACX;aACA,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAA;IACxB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAQ,GAAG,eAAe;QACtC,MAAM,UAAU,GAAG,GAAG,IAAI,CAAC,IAAI,QAAQ,CAAC;QACxC,MAAM,UAAU,GAAG,GAAG,IAAI,CAAC,IAAI,QAAQ,CAAC;QAExC,MAAM,OAAO,GAAG,MAAM,wBAAc,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YAClE,MAAM,EAAE,KAAK,IAAI,EAAE;gBACjB,MAAM,EAAE,GAAG,MAAM,IAAA,mCAAO,EAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAAC,CAAC;gBACtG,OAAO,EAAE,CAAC,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAA;YAChD,CAAC;YACD,MAAM,EAAE,KAAK,IAAI,EAAE;gBACjB,MAAM,EAAE,GAAG,MAAM,IAAA,mCAAO,EAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAAC,CAAC;gBACtG,OAAO,EAAE,CAAC,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAA;YAChD,CAAC;YACD,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE;YACtC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;iBAC1B,KAAK,CAAC,IAAI,CAAC;iBACX,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;iBACnB,IAAI,CAAC,IAAI,CAAC;YACb,UAAU;YACV,UAAU;SACX,CAAC,CAAC;QAEH,OAAO,OAAO;aACX,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;aACxC,MAAM,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAa,EAAE,KAAY;QACrC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAM,CAAC;QACtC,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QAC1C,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;QAElC,IAAI,CAAC,OAAO,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,WAAW,IAAI,CAAC,IAAI,2BAA2B,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;SACrG;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAElD,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE;YAC9B,OAAO,MAAM,CAAC;SACf;QAED,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;CACF;AAxED,wBAwEC"}