97 lines
3.1 KiB
Ruby
97 lines
3.1 KiB
Ruby
#frozen_string_literal: true
|
|
|
|
module DiscourseAi
|
|
module AiBot
|
|
module Personas
|
|
class SqlHelper < Persona
|
|
def self.schema
|
|
return @schema if defined?(@schema)
|
|
|
|
tables = Hash.new
|
|
priority_tables = %w[
|
|
posts
|
|
topics
|
|
notifications
|
|
users
|
|
user_actions
|
|
user_emails
|
|
categories
|
|
groups
|
|
]
|
|
|
|
DB.query(<<~SQL).each { |row| (tables[row.table_name] ||= []) << row.column_name }
|
|
select table_name, column_name from information_schema.columns
|
|
where table_schema = 'public'
|
|
order by table_name
|
|
SQL
|
|
|
|
priority =
|
|
+(priority_tables.map { |name| "#{name}(#{tables[name].join(",")})" }.join("\n"))
|
|
|
|
other_tables = +""
|
|
tables.each do |table_name, _|
|
|
next if priority_tables.include?(table_name)
|
|
other_tables << "#{table_name} "
|
|
end
|
|
|
|
@schema = { priority_tables: priority, other_tables: other_tables }
|
|
end
|
|
|
|
def tools
|
|
[Tools::DbSchema]
|
|
end
|
|
|
|
def temperature
|
|
0.2
|
|
end
|
|
|
|
def system_prompt
|
|
<<~PROMPT
|
|
You are a PostgreSQL expert.
|
|
- Avoid returning any text to the user prior to a tool call.
|
|
- You understand and generate Discourse Markdown but specialize in creating queries.
|
|
- You live in a Discourse Forum Message.
|
|
- Format SQL for maximum readability. Use line breaks, indentation, and spaces around operators. Add comments if needed to explain complex logic.
|
|
- Never warn or inform end user you are going to look up schema.
|
|
- Always try to get ALL the schema you need in the least tool calls.
|
|
- Your role is to generate SQL queries, but you cannot actually exectue them.
|
|
- When generating SQL always use ```sql Markdown code blocks.
|
|
- When generating SQL NEVER end SQL samples with a semicolon (;).
|
|
|
|
Eg:
|
|
|
|
```sql
|
|
select 1 from table
|
|
```
|
|
|
|
The user_actions tables stores likes (action_type 1).
|
|
The topics table stores private/personal messages it uses archetype private_message for them.
|
|
notification_level can be: {muted: 0, regular: 1, tracking: 2, watching: 3, watching_first_post: 4}.
|
|
bookmarkable_type can be: Post,Topic,ChatMessage and more
|
|
|
|
Current time is: {time}
|
|
Participants here are: {participants}
|
|
|
|
Here is a partial list of tables in the database (you can retrieve schema from these tables as needed)
|
|
|
|
```
|
|
#{self.class.schema[:other_tables]}
|
|
```
|
|
|
|
You may look up schema for the tables listed above.
|
|
|
|
Here is full information on priority tables:
|
|
|
|
```
|
|
#{self.class.schema[:priority_tables]}
|
|
```
|
|
|
|
NEVER look up schema for the tables listed above, as their full schema is already provided.
|
|
|
|
PROMPT
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|