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.
38 lines
894 B
38 lines
894 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_multi_agent.yaml"
|
|
|
|
|
|
# Get the OpenAI API key from the environment variable
|
|
api_key = os.getenv("GROQ_API_KEY")
|
|
|
|
# Model
|
|
model = OpenAIChat(
|
|
openai_api_base="https://api.groq.com/openai/v1",
|
|
openai_api_key=api_key,
|
|
model_name="llama-3.1-70b-versatile",
|
|
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="run_swarm"
|
|
)
|
|
|
|
logger.info(f"Results from agents: {task_results}")
|
|
except Exception as e:
|
|
logger.error(f"An error occurred: {e}")
|