Jorge Manrubia
98 lines
4.0 KiB
Ruby
98 lines
4.0 KiB
Ruby
class Command::ChatQuery < Command
|
|
store_accessor :data, :query, :params
|
|
|
|
def title
|
|
"Chat query '#{query}'"
|
|
end
|
|
|
|
def execute
|
|
response = chat.ask query
|
|
Rails.logger.info "***\n#{response.content}\n***"
|
|
response = replace_names_with_ids(JSON.parse(response.content))
|
|
Command::Result::ChatResponse.new(JSON.pretty_generate(response))
|
|
end
|
|
|
|
private
|
|
def chat
|
|
chat = RubyLLM.chat
|
|
chat.with_instructions(prompt)
|
|
end
|
|
|
|
def prompt
|
|
<<~PROMPT
|
|
You are a helpful assistant that translates natural language into commands that Fizzy understand.
|
|
|
|
Fizzy supports the following commands:
|
|
|
|
- Assign users to cards: /assign [user]
|
|
- Close cards: /close [optional reason]
|
|
- Tag cards: /tag [tag-name]
|
|
- Search cards: /search. See how this works below:
|
|
|
|
asks for a certain set of cards, you can use the /search command to filter. The /search command (and only this
|
|
command) supports the following parameters:
|
|
|
|
- assignment_status: can be "unassigned". Only include if asking for unassigned cards explicitly
|
|
- indexed_by: can be "newest", "oldest", "latest", "stalled", "closed"
|
|
- engagement_status: can be "considering" or "doing"
|
|
- card_ids: a list of card ids
|
|
- assignee_ids: a list of assignee names
|
|
- creator_id: the name of a person
|
|
- collection_ids: a list of collection names. Cards are contained in collections. Don't use unless mentioning
|
|
specific collections.
|
|
- tag_ids: a list of tag names.
|
|
- terms: a list of terms to search for. Use this option to refine searches based on further keyword-based
|
|
queries.
|
|
|
|
The output will be in JSON. It will contain a list of commands. Each command will be a JSON object like:
|
|
|
|
{ command: "/close" }
|
|
|
|
For the case of the /search command, it can also contain additional params:
|
|
|
|
{ command: "/search", indexed_by: "closed", collection_ids: [ "Writebook", "Design" ] }
|
|
|
|
Notice that there are overlapping commands (filter by assignee or assign cards). Favor filtering/queries for
|
|
commands like "cards assigned to someone".
|
|
|
|
Notice that only /search commands carry additional JSON params. For /tag, /close and /assign just append the
|
|
param to the string command. This is important: notice that each of those commands receives a parameter (surrounded
|
|
by [] in the description above). Make sure if you invoke a given command you pass the params. Also, that you don't
|
|
' pass JSON params unless you are invoking a /search command.
|
|
|
|
For example, to assign a card, you invoke `assign kevin` instead of:
|
|
|
|
{
|
|
"command": "/assign",
|
|
"assignee_ids": [
|
|
"kevin"
|
|
]
|
|
}
|
|
|
|
Please combine commands to satisfy what the user needs. E.g: search with keywords and filters and then apply
|
|
as many commands as needed. Make sure you don't leave actions mentioned in the query needs unattended.'
|
|
|
|
Make sure to place into double quotes the strings in JSON values and that you generate valid JSON. I want a
|
|
JSON list like [{}, {}...]
|
|
PROMPT
|
|
end
|
|
|
|
def replace_names_with_ids(commands)
|
|
commands.each do |command|
|
|
if command["command"] == "/search"
|
|
command["assignee_ids"] = command["assignee_ids"]&.filter_map { |name| assignee_from(name)&.id }
|
|
command["creator_id"] = assignee_from(command["creator_id"])&.id if command["creator_id"]
|
|
command["collection_ids"] = command["collection_ids"]&.filter_map { |name| Collection.where("lower(name) = ?", name.downcase).first&.id }
|
|
command["tag_ids"] = command["tag_ids"]&.filter_map { |name| ::Tag.find_by_title(name)&.id }
|
|
command.compact!
|
|
end
|
|
end
|
|
end
|
|
|
|
def assignee_from(string)
|
|
string_without_at = string.delete_prefix("@")
|
|
User.all.find { |user| user.mentionable_handles.include?(string_without_at) }
|
|
end
|
|
|
|
end
|