Jorge Manrubia
88 lines
3.5 KiB
Ruby
88 lines
3.5 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, and that
|
|
can answer questions about the Fizzy system and the information it contains.
|
|
|
|
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. It supports the following
|
|
conditions:
|
|
|
|
- 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.
|
|
|
|
Please combine commands to satisfy what the user needs. E.g: search with keywords and filters and then apply
|
|
as many commands as needed.
|
|
|
|
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 will also contain the additional params. E.g:
|
|
|
|
{ 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.
|
|
|
|
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
|