Kunal Pai commited on
Commit
17c720f
·
1 Parent(s): d1ba4cb

Update GeminiManager model version and modify task prompt for stock investment inquiry

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -43,13 +43,13 @@ if __name__ == "__main__":
43
  # Load the tools using the ToolLoader class.
44
  tool_loader = ToolLoader()
45
 
46
- model_manager = GeminiManager(toolsLoader=tool_loader, gemini_model="gemini-2.5-pro-exp-03-25")
47
 
48
  # Example prompt instructing the CEO model to create a strategy for Ashton Hall.
49
  # The prompt explicitly mentions that it can use the web_search tool if needed,
50
  # and that it is allowed to choose the website for the search.
51
  task_prompt = (
52
- "Get me the current AAPL stock price and the latest news about it. "
53
  )
54
 
55
  # Request a CEO response with the prompt.
 
43
  # Load the tools using the ToolLoader class.
44
  tool_loader = ToolLoader()
45
 
46
+ model_manager = GeminiManager(toolsLoader=tool_loader, gemini_model="gemini-2.0-flash")
47
 
48
  # Example prompt instructing the CEO model to create a strategy for Ashton Hall.
49
  # The prompt explicitly mentions that it can use the web_search tool if needed,
50
  # and that it is allowed to choose the website for the search.
51
  task_prompt = (
52
+ "Should I invest in TSLA stocks? "
53
  )
54
 
55
  # Request a CEO response with the prompt.