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

@ -26,7 +26,7 @@ print(task)
########## V2
# ######### V2
# from swarms.agents.models.openai import OpenAI
# chat = OpenAI()
@ -35,7 +35,7 @@ print(task)
# ############# v3
# # ############# v3
# import os
# from swarms.workers.vortex_worker import VortexWorkerAgent

@ -3,6 +3,7 @@ from __future__ import annotations
from typing import List, Optional
from langchain.chains.llm import LLMChain
from pydantic import ValidationError
from swarms.agents.memory.base import VectorStoreRetriever
@ -26,7 +27,6 @@ from swarms.agents.utils.human_input import HumanInputRun
class Agent:
"""Base Agent class"""
def __init__(
self,
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')
# ---------- Constants ----------
ROOT_DIR = "./data/"
class HierarchicalSwarm:
def __init__(
self,
@ -50,7 +47,7 @@ class HierarchicalSwarm:
self.boss_node = BossNode(
api_key=self.openai_api_key,
worker_node=self.worker_node,
# worker_node=self.worker_node,
llm_class=OpenAI,
max_iterations=self.max_iterations,
verbose=self.verbose

Loading…
Cancel
Save