clean up agents

Former-commit-id: ec22711d09
group-chat
Kye 1 year ago
parent b229f41053
commit 0bc1dae750

@ -1,25 +1,21 @@
import argparse
import asyncio
import enum import enum
import logging
import os import os
from pathlib import Path import re
import shutil
import sys import sys
import time import time
import shutil from pathlib import Path
import argparse from typing import Any, Callable, List, Optional
import asyncio
import re
from typing import List, Optional, Callable, Any
import openai import openai
from agent_protocol import Agent, Step, Task
from openai_function_call import openai_function from openai_function_call import openai_function
from tenacity import retry, stop_after_attempt, wait_random_exponential from smol_dev.prompts import generate_code_sync, plan, specify_file_paths
import logging
from smol_dev.prompts import plan, specify_file_paths, generate_code_sync
from smol_dev.utils import generate_folder, write_file from smol_dev.utils import generate_folder, write_file
from tenacity import retry, stop_after_attempt, wait_random_exponential
from agent_protocol import Agent, Step, Task
class DeveloperAgent: class DeveloperAgent:

@ -1,11 +1,7 @@
# from swarms.workers.multi_modal_workers.multi_modal_agent import MultiModalVisualAgent # from swarms.workers.multi_modal_workers.multi_modal_agent import MultiModalVisualAgent
from swarms.workers.multi_modal_workers.multi_modal_agent import MultiModalVisualAgent from swarms.workers.multi_modal_workers.multi_modal_agent import MultiModalVisualAgent
from langchain.tools import BaseTool
class MultiModalVisualAgentTool(BaseTool):
name = "multi_visual_agent"
description = "Multi-Modal Visual agent tool"
class MultiModalVisualAgentTool:
def __init__(self, agent: MultiModalVisualAgent): def __init__(self, agent: MultiModalVisualAgent):
self.agent = agent self.agent = agent
@ -13,6 +9,3 @@ class MultiModalVisualAgentTool(BaseTool):
#run the multi-modal visual agent with the give task #run the multi-modal visual agent with the give task
return self.agent.run_text(text) return self.agent.run_text(text)
multimodal_agent = MultiModalVisualAgent()
multimodal_agent_tool = MultiModalVisualAgentTool(multimodal_agent)
Loading…
Cancel
Save