main
Kye 2 years ago
parent 5058acb814
commit 80750a32bd

@ -8,12 +8,14 @@ from langchain.callbacks.manager import (
class WorkerNode(BaseTool):
"""Useful for when you need to spawn an autonomous agent instance as a worker to accomplish complex tasks, it can search the internet or spawn child multi-modality models to process and generate images and text or audio and so on """
name = "WorkerNode"
description = "A worker node that can perform complex tasks"
llm: Any = Field(description="Language model")
tools: List[Tool] = Field(description="List of tools")
vectorstore: VectorStore = Field(description="Vector store")
def __init__(self, llm, tools, vectorstore):
super().__init__()
self.llm = llm
self.tools = tools
self.vectorstore = vectorstore
def __init__(self, **data):
super().__init__(**data)
def create_agent(self, ai_name, ai_role, human_in_the_loop, search_kwargs):
self.agent = AutoGPT.from_llm_and_tools(

Loading…
Cancel
Save