Former-commit-id: f9b900e21f
group-chat
Kye 1 year ago
parent e521cb499a
commit a11e02e60a

@ -35,7 +35,7 @@ print(task)
# ############# v3 # # ############# v3
# import os # import os
# from swarms.workers.vortex_worker import VortexWorkerAgent # from swarms.workers.vortex_worker import VortexWorkerAgent

@ -3,6 +3,7 @@ from __future__ import annotations
from typing import List, Optional from typing import List, Optional
from langchain.chains.llm import LLMChain from langchain.chains.llm import LLMChain
from pydantic import ValidationError from pydantic import ValidationError
from swarms.agents.memory.base import VectorStoreRetriever from swarms.agents.memory.base import VectorStoreRetriever
@ -26,7 +27,6 @@ from swarms.agents.utils.human_input import HumanInputRun
class Agent: class Agent:
"""Base Agent class""" """Base Agent class"""
def __init__( def __init__(
self, self,
ai_name: str, ai_name: str,

@ -9,12 +9,9 @@ from swarms.workers.worker_node import WorkerNode
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s') logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
# ---------- Constants ---------- # ---------- Constants ----------
ROOT_DIR = "./data/" ROOT_DIR = "./data/"
class HierarchicalSwarm: class HierarchicalSwarm:
def __init__( def __init__(
self, self,
@ -50,7 +47,7 @@ class HierarchicalSwarm:
self.boss_node = BossNode( self.boss_node = BossNode(
api_key=self.openai_api_key, api_key=self.openai_api_key,
worker_node=self.worker_node, # worker_node=self.worker_node,
llm_class=OpenAI, llm_class=OpenAI,
max_iterations=self.max_iterations, max_iterations=self.max_iterations,
verbose=self.verbose verbose=self.verbose

Loading…
Cancel
Save