Spaces:
Running
Running
File size: 2,091 Bytes
ce0ec3b 7c390d2 ce0ec3b 775ea0b ce0ec3b 0e9bb01 ce0ec3b 775ea0b ce0ec3b 775ea0b ce0ec3b 7c390d2 ce0ec3b 0e9bb01 ce0ec3b 7c390d2 775ea0b 3726684 ce0ec3b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
"""
Analysis agent for interpreting song lyrics and providing deeper context.
This agent is responsible for analyzing song lyrics and formatting the analysis
in a user-friendly way, showing lyrics with comments underneath them.
"""
from smolagents import CodeAgent, VisitWebpageTool
from loguru import logger
from config import load_prompt_templates
from tools.search_tools import ThrottledDuckDuckGoSearchTool
from tools.analysis_tools import AnalyzeLyricsTool
from tools.formatting_tools import FormatAnalysisResultsTool
def create_analysis_agent(model):
"""
Create an agent specialized in analyzing and interpreting song lyrics.
Args:
model: The LLM model to use with this agent
Returns:
A configured CodeAgent for lyrics analysis
"""
# Load prompt templates
prompt_templates = load_prompt_templates()
# Create the throttled search tool with hardcoded values
throttled_search_tool = ThrottledDuckDuckGoSearchTool(
min_delay=3.0,
max_delay=7.0
)
# Create and return the agent with specialized lyrics analysis instructions
# Customize the prompts to use our specialized lyrics_analysis_agent prompt
custom_prompt_templates = prompt_templates.copy()
# Set our special prompt as the system prompt for better instruction
if 'lyrics_analysis_agent' in custom_prompt_templates:
custom_prompt_templates['system_prompt'] = custom_prompt_templates['lyrics_analysis_agent']
agent = CodeAgent(
model=model,
tools=[throttled_search_tool, VisitWebpageTool(), AnalyzeLyricsTool(), FormatAnalysisResultsTool()],
name="lyrics_analysis_agent",
description="Specialized agent for analyzing song lyrics and providing detailed commentary",
additional_authorized_imports=["numpy", "bs4", "json", "re"],
max_steps=5, # Hardcoded from config
verbosity_level=2, # Hardcoded from config
prompt_templates=prompt_templates
)
logger.info("Analysis agent created successfully")
return agent
|