requirements.tx

Former-commit-id: 6c93628499
group-chat
Kye 1 year ago
parent a111b99c0c
commit ea9fbfe9c6

@ -1,4 +1,7 @@
faiss-gpu
transformers transformers
pandas
numpy
langchain==0.0.240 langchain==0.0.240
nest_asyncio nest_asyncio
pegasusx pegasusx
@ -23,10 +26,6 @@ exxa
open-interpreter open-interpreter
tabulate tabulate
colored colored
# git+https://github.com/IDEA-Research/GroundingDINO.git
# git+https://github.com/facebookresearch/segment-anything.git
addict addict
albumentations albumentations

@ -101,15 +101,10 @@ class OmniModalAgent:
def __init__( def __init__(
self, self,
llm: BaseLanguageModel, llm: BaseLanguageModel,
tools: List[BaseTool] # tools: List[BaseTool]
): ):
self.llm = llm self.llm = llm
self.tools = tools
self.chat_planner = load_chat_planner(llm)
self.response_generator = load_response_generator(llm)
self.task_executor: TaskExecutor
def tools_init(self):
self.tools = [ self.tools = [
load_tool(tool_name) load_tool(tool_name)
for tool_name in [ for tool_name in [
@ -130,6 +125,10 @@ class OmniModalAgent:
] ]
] ]
self.chat_planner = load_chat_planner(llm)
self.response_generator = load_response_generator(llm)
self.task_executor: TaskExecutor
def run(self, input: str) -> str: def run(self, input: str) -> str:
plan = self.chat_planner.plan( plan = self.chat_planner.plan(

Loading…
Cancel
Save