bstraehle commited on
Commit
24cc8ee
·
verified ·
1 Parent(s): 86b200e

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +10 -0
tools.py CHANGED
@@ -6,6 +6,16 @@ from crewai.tools import tool
6
  from google import genai
7
  from google.genai import types
8
 
 
 
 
 
 
 
 
 
 
 
9
  @tool("Web Search Tool")
10
  def web_search_tool(question: str) -> str:
11
  """Given a question only, search the web to answer the question.
 
6
  from google import genai
7
  from google.genai import types
8
 
9
+ WEB_SEARCH_MODEL = "gemini-2.5-flash-preview-04-17"
10
+ IMAGE_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
11
+ AUDIO_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
12
+ VIDEO_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
13
+ YOUTUBE_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
14
+ DOCUMENT_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
15
+ ARITHMETIC_MODEL = "gemini-2.5-flash-preview-04-17"
16
+ CODE_GENERATION_MODEL = "gemini-2.5-flash-preview-04-17"
17
+ CODE_EXECUTION_MODEL = "gemini-2.5-flash-preview-04-17"
18
+
19
  @tool("Web Search Tool")
20
  def web_search_tool(question: str) -> str:
21
  """Given a question only, search the web to answer the question.