pull/378/head
Kye 11 months ago
parent 070cccac92
commit a9f45f271f

@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
[tool.poetry]
name = "swarms"
version = "3.9.3"
version = "3.9.4"
description = "Swarms - Pytorch"
license = "MIT"
authors = ["Kye Gomez <kye@apac.ai>"]
@ -23,7 +23,7 @@ classifiers = [
[tool.poetry.dependencies]
python = "^3.6.1"
torch = "2.1.1"p
torch = "2.1.1"
transformers = "4.37.1"
openai = "0.28.0"
langchain = "0.0.333"

@ -83,7 +83,6 @@ class ModelParallelizer:
responses.append(llm(task))
return responses
# New Features
def save_responses_to_file(self, filename):
"""Save responses to file"""

@ -33,4 +33,17 @@ def disable_logging():
logger = logging.getLogger(logger_name)
logger.setLevel(
logging.WARNING
) # Supress DEBUG and info logs
) # Suppress DEBUG and info logs
# Create a file handler to log errors to the file
file_handler = logging.FileHandler("errors.txt")
file_handler.setLevel(logging.ERROR)
logging.getLogger().addHandler(file_handler)
# Create a stream handler to log errors to the terminal
stream_handler = logging.StreamHandler()
stream_handler.setLevel(logging.ERROR)
logging.getLogger().addHandler(stream_handler)
disable_logging()

@ -51,4 +51,3 @@ def test_swarm_network_remove_agent(swarm_network):
swarm_network.remove_agent(agent_to_remove)
assert len(swarm_network.agents) == 4
assert agent_to_remove not in swarm_network.agents

Loading…
Cancel
Save