diff --git a/README.md b/README.md index 2c7b4c20..df32e4a9 100644 --- a/README.md +++ b/README.md @@ -53,10 +53,11 @@ swarm(objective) ## Usage ```python -from swarms import OpenAI -chat = OpenAI() -response = chat("Hello world!") +from swarms import HuggingFaceLLM + +hugging_face_model = HuggingFaceLLM(model_id="Voicelab/trurl-2-13b") +generated_text = hugging_face_model.generate("In a world where AI") ``` diff --git a/swarms/agents/base.py b/swarms/agents/base.py index 5a999416..ba0b26fa 100644 --- a/swarms/agents/base.py +++ b/swarms/agents/base.py @@ -76,10 +76,8 @@ class Agent: user_input = ( "Determine which next command to use, and respond using the format specified above:" ) - # Interaction Loop loop_count = 0 while True: - # Discontinue if continuous limit is reached loop_count += 1 # Send message to AI, get response @@ -90,7 +88,6 @@ class Agent: user_input=user_input, ) - # Print Assistant thoughts print(assistant_reply) self.chat_history_memory.add_message(HumanMessage(content=user_input)) self.chat_history_memory.add_message(AIMessage(content=assistant_reply))