gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

wip

Browse source
This commit is contained in:
paulpaliychuk 2024-08-26 21:19:31 -04:00
parent c304758791
commit d972e25547
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
examples/nba/nba_agent.py
View file

@ -36,9 +36,7 @@ async def invoke_tool(tool_name: str, **kwargs):
@tool @tool
async def get_team_roster(team_name: str): async def get_team_roster(team_name: str):
"""Get the current roster for a specific team.""" """Get the current roster for a specific team."""
search_result = await graphiti_client.search(f'{team_name.lower()}', num_results=1) search_result = await graphiti_client.search(f'{team_name.lower()}', num_results=10)
print(search_result)
print(team_name.lower())
roster = [] roster = []
for fact in search_result: for fact in search_result:
roster.append(fact) roster.append(fact)