Spaces:
Running
Running
Kunal Pai
commited on
Commit
·
17c720f
1
Parent(s):
d1ba4cb
Update GeminiManager model version and modify task prompt for stock investment inquiry
Browse files
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.
|
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 |
-
"
|
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.
|