You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
842 B
35 lines
842 B
import os
|
|
|
|
from dotenv import load_dotenv
|
|
from loguru import logger
|
|
from swarm_models import OpenAIChat
|
|
|
|
from swarms.agents.create_agents_from_yaml import (
|
|
create_agents_from_yaml,
|
|
)
|
|
|
|
# Load environment variables
|
|
load_dotenv()
|
|
|
|
# Path to your YAML file
|
|
yaml_file = "agents.yaml"
|
|
|
|
# Get the OpenAI API key from the environment variable
|
|
api_key = os.getenv("OPENAI_API_KEY")
|
|
|
|
# Create an instance of the OpenAIChat class
|
|
model = OpenAIChat(
|
|
openai_api_key=api_key, model_name="gpt-4o-mini", temperature=0.1
|
|
)
|
|
|
|
|
|
try:
|
|
# Create agents and run tasks (using 'both' to return agents and task results)
|
|
task_results = create_agents_from_yaml(
|
|
model=model, yaml_file=yaml_file, return_type="tasks"
|
|
)
|
|
|
|
logger.info(f"Results from agents: {task_results}")
|
|
except Exception as e:
|
|
logger.error(f"An error occurred: {e}")
|