tonko22 commited on
Commit
92894aa
·
1 Parent(s): e018c03

Update agent prompt, model config

Browse files
Files changed (3) hide show
  1. agents/single_agent.py +1 -1
  2. app.py +2 -2
  3. 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, provider="gemini")
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="ollama"):
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: