Skip to content

Commit 9df1a40

Browse files
committed
fix sample script
1 parent 4ecd2cd commit 9df1a40

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

samples/agent/team_create.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,19 +16,18 @@
1616
select_ai.connect(user=user, password=password, dsn=dsn)
1717

1818
# Team
19-
team_attributes = TeamAttributes(
20-
agents=[{"name": "MOVIE_ANALYST", "task": "ANALYZE_MOVIE_TASK"}],
21-
process="sequential",
22-
)
2319
team = Team(
2420
team_name="MOVIE_AGENT_TEAM",
25-
attributes=team_attributes,
21+
attributes=TeamAttributes(
22+
agents=[{"name": "MOVIE_ANALYST", "task": "ANALYZE_MOVIE_TASK"}],
23+
process="sequential",
24+
),
2625
)
2726
team.create(enabled=True, replace=True)
2827

2928
print(
3029
team.run(
31-
prompt="Could list the movies in the database?",
30+
prompt="Could you list the movies in the database?",
3231
params={"conversation_id": conversation_id},
3332
)
3433
)

0 commit comments

Comments
 (0)