Skip to content

Commit

Permalink
Tweak purpose prompts
Browse files Browse the repository at this point in the history
  • Loading branch information
philippjfr committed Dec 16, 2024
1 parent 20e00cf commit 71c24a4
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions lumen/ai/agents.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,11 +178,9 @@ class SourceAgent(Agent):
"""

purpose = param.String(default="""
The SourceAgent allows a user to upload unavailable,
new datasets, tables, or documents.
The SourceAgent allows a user to upload unavailable, new datasets, tables, or documents.
Only use this if the user is requesting to add a completely new table
or you think additional information is required to solve the user query.
Only use this if the user is requesting to add a completely new table.
Not useful for answering what's available or loading existing datasets.
""")

Expand Down Expand Up @@ -285,9 +283,9 @@ class AnalystAgent(ChatAgent):
class TableListAgent(Agent):

purpose = param.String(default="""
Renders a list of all availables tables to the user.
Renders a list of all availables tables to the user and lets the user
pick one, do not use if user has already picked a table.
Not useful for gathering information about the tables.
Use with ChatAgent to provide info about the tables.
""")

prompts = param.Dict(
Expand Down

0 comments on commit 71c24a4

Please sign in to comment.