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.
|