From 05fbcf0d1e0a4dba2206d8079c1e9520acd405bd Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 27 Jul 2023 16:25:37 -0400 Subject: [PATCH] clean up Former-commit-id: d3ee2045ef25cc3cf4f6772af3bc736132ee84bc --- swarms/swarms.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/swarms/swarms.py b/swarms/swarms.py index 2be56179..c2c7f8b8 100644 --- a/swarms/swarms.py +++ b/swarms/swarms.py @@ -17,19 +17,21 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %( # TODO: BE MORE EXPLICIT ON TOOL USE, TASK DECOMPOSITION AND TASK COMPLETETION AND ALLOCATION # TODO: Add RLHF Data collection, ask user how the swarm is performing # TODO: Create an onboarding process if not settings are preconfigured like `from swarms import Swarm, Swarm()` => then initiate onboarding name your swarm + provide purpose + etc -# TODO: Off class HierarchicalSwarm: def __init__( self, model_id: Optional[str] = None, openai_api_key: Optional[str] = "", + use_vectorstore: Optional[bool] = True, embedding_size: Optional[int] = None, use_async: Optional[bool] = True, + human_in_the_loop: Optional[bool] = True, model_type: Optional[str] = None, boss_prompt: Optional[str] = None, + worker_prompt: Optional[str] = None, temperature: Optional[float] = None, max_iterations: Optional[int] = None, @@ -57,7 +59,7 @@ class HierarchicalSwarm: - def initialize_llm(self, llm_class: Optional = None): + def initialize_llm(self, llm_class): """ Init LLM