Spaces:
Sleeping
Sleeping
Update agent prompt, model config
Browse files- agents/single_agent.py +1 -1
- app.py +2 -2
- config.py +1 -1
agents/single_agent.py
CHANGED
@@ -37,7 +37,7 @@ def create_single_agent(model):
|
|
37 |
additional_authorized_imports=['numpy', 'bs4', 'rich'],
|
38 |
max_steps=25,
|
39 |
verbosity_level=2,
|
40 |
-
description="Specialized agent for finding and extracting song lyrics specified by the user. Performs lyrics search and analysis using given tools. Provides detailed commentary with beautiful and human-readable format using rich library formatting. Format analysis results in rich colorful output.",
|
41 |
prompt_templates=prompt_templates
|
42 |
)
|
43 |
|
|
|
37 |
additional_authorized_imports=['numpy', 'bs4', 'rich'],
|
38 |
max_steps=25,
|
39 |
verbosity_level=2,
|
40 |
+
description="Specialized agent for finding and extracting song lyrics specified by the user. Performs lyrics search and after getting the lyrics full text from the web, it performs the analysis using given tools. Provides detailed commentary with beautiful and human-readable format using rich library formatting. Format analysis results in rich colorful output.",
|
41 |
prompt_templates=prompt_templates
|
42 |
)
|
43 |
|
app.py
CHANGED
@@ -36,8 +36,8 @@ def main():
|
|
36 |
|
37 |
# If using Ollama, we need to specify the API base URL
|
38 |
# Initialize the LLM model based on configuration
|
39 |
-
model_id = get_model_id(use_local=is_test
|
40 |
-
|
41 |
logger.info(f"Initializing with model: {model_id}")
|
42 |
if is_test:
|
43 |
api_base = get_ollama_api_base()
|
|
|
36 |
|
37 |
# If using Ollama, we need to specify the API base URL
|
38 |
# Initialize the LLM model based on configuration
|
39 |
+
model_id = get_model_id(use_local=is_test)
|
40 |
+
|
41 |
logger.info(f"Initializing with model: {model_id}")
|
42 |
if is_test:
|
43 |
api_base = get_ollama_api_base()
|
config.py
CHANGED
@@ -26,7 +26,7 @@ def load_api_keys():
|
|
26 |
os.environ["GEMINI_API_KEY"] = os.getenv("GEMINI_API_KEY")
|
27 |
|
28 |
|
29 |
-
def get_model_id(use_local=True, provider="
|
30 |
"""Get the appropriate model ID based on configuration.
|
31 |
|
32 |
Args:
|
|
|
26 |
os.environ["GEMINI_API_KEY"] = os.getenv("GEMINI_API_KEY")
|
27 |
|
28 |
|
29 |
+
def get_model_id(use_local=True, provider="gemini"):
|
30 |
"""Get the appropriate model ID based on configuration.
|
31 |
|
32 |
Args:
|