Class responsible for managing tasks, including their creation, prioritization, and execution. It uses three chains for these operations: creationChain, prioritizationChain, and executionChain.

Example

const babyAGI = BabyAGI.fromLLM({
llm: new OpenAI({ temperature: 0 }),
vectorstore: new MemoryVectorStore(new OpenAIEmbeddings()),
maxIterations: 3,
});

const result = await babyAGI.call({
objective: "Write a weather report for SF today",
});

Hierarchy

Implements

Constructors

Properties

maxIterations: number
prioritizationChain: BaseChain<ChainValues, ChainValues>
taskIDCounter: number
taskList: Task[]
vectorstore: VectorStore
verbose: boolean

Whether to print out response text.

callbacks?: Callbacks
memory?: BaseMemory
metadata?: Record<string, unknown>
tags?: string[]

Accessors

  • get inputKeys(): string[]
  • Returns string[]

  • get outputKeys(): never[]
  • Returns never[]

Methods

  • Adds a task to the task list.

    Parameters

    • task: Task

      The task to be added.

    Returns Promise<void>

    Promise resolving to void.

  • Run the core logic of this chain and add to output if desired.

    Wraps _call and handles memory.

    Parameters

    Returns Promise<ChainValues>

  • Executes a task based on the objective and the task description.

    Parameters

    • objective: string

      The objective of the task.

    • task: string

      The task to be executed.

    • Optional runManager: CallbackManagerForChainRun

      Optional CallbackManagerForChainRun instance.

    Returns Promise<string>

    Promise resolving to the result of the task execution as a string.

  • Generates the next tasks based on the result of the previous task, the task description, and the objective.

    Parameters

    • result: string

      The result of the previous task.

    • task_description: string

      The description of the task.

    • objective: string

      The objective of the task.

    • Optional runManager: CallbackManagerForChainRun

      Optional CallbackManagerForChainRun instance.

    Returns Promise<Optional<Task, "taskID">[]>

    Promise resolving to an array of tasks without taskID.

  • Retrieves the top tasks that are most similar to the given query.

    Parameters

    • query: string

      The query to search for.

    • k: number = 5

      The number of top tasks to retrieve.

    Returns Promise<string[]>

    Promise resolving to an array of top tasks.

  • Invoke the chain with the provided input and returns the output.

    Parameters

    Returns Promise<ChainValues>

    Promise that resolves with the output of the chain run.

  • Create a new runnable sequence that runs each individual runnable in series, piping the output of one runnable into another runnable or runnable-like.

    Type Parameters

    • NewRunOutput

    Parameters

    • coerceable: RunnableLike<ChainValues, NewRunOutput>

      A runnable, function, or object whose values are functions or runnables.

    Returns RunnableSequence<ChainValues, Exclude<NewRunOutput, Error>>

    A new runnable sequence.

  • Parameters

    • inputs: Record<string, unknown>
    • outputs: Record<string, unknown>
    • returnOnlyOutputs: boolean = false

    Returns Promise<Record<string, unknown>>

  • Prints the next task to the console.

    Parameters

    • task: Task

      The next task to be printed.

    Returns void

    void

  • Prints the current task list to the console.

    Returns void

    void

  • Prints the result of a task to the console.

    Parameters

    • result: string

      The result of the task.

    Returns void

    void

  • Prioritizes the tasks based on the current task ID and the objective.

    Parameters

    • thisTaskID: number

      The ID of the current task.

    • objective: string

      The objective of the task.

    • Optional runManager: CallbackManagerForChainRun

      Optional CallbackManagerForChainRun instance.

    Returns Promise<{
        taskID: string;
        taskName: string;
    }[]>

    Promise resolving to an array of prioritized tasks.

  • Parameters

    Returns Promise<string>

  • Stream output in chunks.

    Parameters

    Returns Promise<IterableReadableStream<ChainValues>>

    A readable stream that is also an iterable.

  • Stream all output from a runnable, as reported to the callback system. This includes all inner runs of LLMs, Retrievers, Tools, etc. Output is streamed as Log objects, which include a list of jsonpatch ops that describe how the state of the run has changed in each step, and the final state of the run. The jsonpatch ops can be applied in order to construct state.

    Parameters

    Returns AsyncGenerator<RunLogPatch, any, unknown>

  • Default implementation of transform, which buffers input and then calls stream. Subclasses should override this method if they can start producing output while input is still being generated.

    Parameters

    Returns AsyncGenerator<ChainValues, any, unknown>

  • Bind lifecycle listeners to a Runnable, returning a new Runnable. The Run object contains information about the run, including its id, type, input, output, error, startTime, endTime, and any tags or metadata added to the run.

    Parameters

    • params: {
          onEnd?: ((run, config?) => void | Promise<void>);
          onError?: ((run, config?) => void | Promise<void>);
          onStart?: ((run, config?) => void | Promise<void>);
      }

      The object containing the callback functions.

      • Optional onEnd?: ((run, config?) => void | Promise<void>)
          • (run, config?): void | Promise<void>
          • Called after the runnable finishes running, with the Run object.

            Parameters

            Returns void | Promise<void>

      • Optional onError?: ((run, config?) => void | Promise<void>)
          • (run, config?): void | Promise<void>
          • Called if the runnable throws an error, with the Run object.

            Parameters

            Returns void | Promise<void>

      • Optional onStart?: ((run, config?) => void | Promise<void>)
          • (run, config?): void | Promise<void>
          • Called before the runnable starts running, with the Run object.

            Parameters

            Returns void | Promise<void>

    Returns Runnable<ChainValues, ChainValues, BaseCallbackConfig>

  • Static method to create a new BabyAGI instance from a BaseLanguageModel.

    Parameters

    Returns BabyAGI

    A new instance of BabyAGI.

Generated using TypeDoc