Class BabyAGI

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 (view full)

Implements

Constructors

constructor

Properties

creationChain executionChain maxIterations prioritizationChain taskIDCounter taskList vectorstore memory?

Accessors

inputKeys outputKeys

Methods

addTask apply call executeTask getNextTasks getTopTasks invoke prepOutputs printNextTask printTaskList printTaskResult prioritizeTasks run serialize deserialize fromLLM

Constructors

constructor

Properties

creationChain

creationChain: BaseChain<ChainValues, ChainValues>

executionChain

executionChain: BaseChain<ChainValues, ChainValues>

maxIterations

maxIterations: number

prioritizationChain

prioritizationChain: BaseChain<ChainValues, ChainValues>

taskIDCounter

taskIDCounter: number

taskList

taskList: Task[]

vectorstore

vectorstore: VectorStoreInterface

Optionalmemory

memory?: any

Accessors

inputKeys

outputKeys

Methods

addTask

apply

  • apply(inputs, config?): Promise<ChainValues[]>

  • Parameters

    • inputs: ChainValues[]
    • Optionalconfig: any[]

    Returns Promise<ChainValues[]>

    Deprecated

    Use .batch() instead. Will be removed in 0.2.0.

    Call the chain on all inputs in the list

call

  • call(values, config?, tags?): Promise<ChainValues>

  • Parameters

    • values: any
    • Optionalconfig: any
    • Optionaltags: string[]

      Deprecated

    Returns Promise<ChainValues>

    Deprecated

    Use .invoke() instead. Will be removed in 0.2.0.

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

    Wraps _call and handles memory.

executeTask

  • executeTask(objective, task, runManager?): Promise<string>

  • 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.

    • OptionalrunManager: any

      Optional CallbackManagerForChainRun instance.

    Returns Promise<string>

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

getNextTasks

  • getNextTasks(result, task_description, objective, runManager?): Promise<Optional<Task, "taskID">[]>

  • 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.

    • OptionalrunManager: any

      Optional CallbackManagerForChainRun instance.

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

    Promise resolving to an array of tasks without taskID.

getTopTasks

  • getTopTasks(query, k?): Promise<any>

  • 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<any>

    Promise resolving to an array of top tasks.

invoke

  • invoke(input, options?): Promise<ChainValues>

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

    Parameters

    • input: ChainValues

      Input values for the chain run.

    • Optionaloptions: any

    Returns Promise<ChainValues>

    Promise that resolves with the output of the chain run.

prepOutputs

  • prepOutputs(inputs, outputs, returnOnlyOutputs?): Promise<Record<string, unknown>>

  • Parameters

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

    Returns Promise<Record<string, unknown>>

printNextTask

printTaskList

printTaskResult

prioritizeTasks

  • prioritizeTasks(thisTaskID, objective, runManager?): Promise<{
        taskID: string;
        taskName: string;
    }[]>

  • 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.

    • OptionalrunManager: any

      Optional CallbackManagerForChainRun instance.

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

    Promise resolving to an array of prioritized tasks.

run

  • run(input, config?): Promise<string>

  • Parameters

    • input: any
    • Optionalconfig: any

    Returns Promise<string>

    Deprecated

    Use .invoke() instead. Will be removed in 0.2.0.

serialize

Staticdeserialize

StaticfromLLM

  • fromLLM(__namedParameters): BabyAGI

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

    Parameters

    • __namedParameters: Omit<BabyAGIInputs, "executionChain" | "creationChain" | "prioritizationChain"> & Partial<Pick<BabyAGIInputs, "executionChain" | "creationChain" | "prioritizationChain">> & {
          llm: BaseLanguageModelInterface;
      }

    Returns BabyAGI

    A new instance of BabyAGI.

Settings

Member Visibility

On This Page

Constructors
Properties
Accessors
Methods