PanoEvJ commited on
Commit
7838852
1 Parent(s): f557bcb
Files changed (1) hide show
  1. app.py +5 -11
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 the Golden State Warriors to any other NBA player, one 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 of Golden State Warriors.
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 = [sql_nba_tool,
 
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,