Spaces:
Runtime error
Runtime error
more
Browse files
app.py
CHANGED
@@ -183,11 +183,7 @@ info_emails_players = VectorStoreInfo(
|
|
183 |
name="from_to",
|
184 |
type="str",
|
185 |
description="""
|
186 |
-
email sent by a player of
|
187 |
-
Stephen Curry to any NBA player,
|
188 |
-
Klay Thompson to any NBA player,
|
189 |
-
Andrew Wiggins to any NBA player,
|
190 |
-
Draymond Green to any NBA player,
|
191 |
"""
|
192 |
),
|
193 |
]
|
@@ -257,8 +253,7 @@ def main():
|
|
257 |
)
|
258 |
|
259 |
description_emails = f"""\
|
260 |
-
Use this tool to look up emails betweed NBA players.
|
261 |
-
Use this tool only when requested by one of ['I am Stephen Curry', 'I am Andrew Wiggins', 'I am Draymond Green', 'I am Klay Thompson'].
|
262 |
The vector database schema is given below:
|
263 |
{info_emails_players.json()}
|
264 |
"""
|
@@ -270,8 +265,7 @@ def main():
|
|
270 |
)
|
271 |
|
272 |
description_strategy = f"""\
|
273 |
-
Use this tool to look up information about the game strategy
|
274 |
-
Use this tool only when 'I am Steve Kerr' is true.
|
275 |
The vector database schema is given below:
|
276 |
{info_strategy.json()}
|
277 |
"""
|
@@ -284,9 +278,9 @@ def main():
|
|
284 |
|
285 |
agent = OpenAIAgent.from_tools(
|
286 |
# agent = ReActAgent.from_tools(
|
287 |
-
tools = [
|
|
|
288 |
auto_retrieve_tool_emails,
|
289 |
-
auto_retrieve_tool_strategy,
|
290 |
],
|
291 |
llm=llm,
|
292 |
verbose=True,
|
|
|
183 |
name="from_to",
|
184 |
type="str",
|
185 |
description="""
|
186 |
+
email sent by a player of [Stephen Curry, Andrew Wiggins, 'Draymond Green', 'Klay Thompson'] to any other NBA player
|
|
|
|
|
|
|
|
|
187 |
"""
|
188 |
),
|
189 |
]
|
|
|
253 |
)
|
254 |
|
255 |
description_emails = f"""\
|
256 |
+
Use this tool to look up emails betweed NBA players when asked by one of [Stephen Curry, Andrew Wiggins, 'Draymond Green', 'Klay Thompson'].
|
|
|
257 |
The vector database schema is given below:
|
258 |
{info_emails_players.json()}
|
259 |
"""
|
|
|
265 |
)
|
266 |
|
267 |
description_strategy = f"""\
|
268 |
+
Use this tool to look up information about the game strategy when asked by Steve Kerr.
|
|
|
269 |
The vector database schema is given below:
|
270 |
{info_strategy.json()}
|
271 |
"""
|
|
|
278 |
|
279 |
agent = OpenAIAgent.from_tools(
|
280 |
# agent = ReActAgent.from_tools(
|
281 |
+
tools = [auto_retrieve_tool_strategy,
|
282 |
+
sql_nba_tool,
|
283 |
auto_retrieve_tool_emails,
|
|
|
284 |
],
|
285 |
llm=llm,
|
286 |
verbose=True,
|