better_prompt 0.2.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 1b0275e59070ddcb50314443ed257f616af962ab8e7e328ee78ff337ef106b37
+  data.tar.gz: 6e2a503dfa2399a1ddf25276647edf7a85422b723aeaf444c01832869d504e8b
+SHA512:
+  metadata.gz: e14e38212b0ac9b0840647f37acac1f61602e9a5152bce1dff23be3c2836bb658fb5a4c50434be1d1d9a22c9aaecfe0e16ab4f71467ebc35772996679c852117
+  data.tar.gz: 2614387d896abbfe25104ed50e38c2752fc363d3a876d47368c6f7ea4afb55bc5677f5f56a7a819df9721f61a52db453e83346a887d096648bc4965ec7e6a858

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 庄表伟
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/README.cn.md

@@ -0,0 +1,107 @@
+# BetterPrompt: 命令行中的 AI 提示词管理利器
+
+BetterPrompt 是一个功能强大的命令行工具，旨在帮助开发者和内容创作者高效地管理、测试和调用 AI 模型的提示词（Prompts）。
+
+## 🚀 功能特性
+
+*   **提示词管理**: 轻松实现提示词的增、删、改、查。
+*   **模板支持**: 创建和管理提示词模板，提高复用性。
+*   **模型调用**: 直接在命令行中调用 AI 模型并查看结果。
+*   **本地存储**: 使用 SQLite 数据库在本地安全地存储您的提示词和历史记录。
+*   **灵活配置**: 支持自定义 AI 模型的 API 地址和密钥。
+
+## 📦 安装
+
+通过 RubyGems 安装 BetterPrompt:
+
+```bash
+gem install better_prompt
+```
+
+## ⚙️ 配置
+
+首次运行时，BetterPrompt 会引导您完成初始化。您也可以手动执行：
+
+```bash
+bp --init
+```
+
+该命令会在您的用户主目录下创建 `.better_prompt` 文件夹，并包含 `config.yml` 配置文件和 `prompt.db` 数据库文件。
+
+请修改 `config.yml` 文件，填入您的 AI 模型提供商的 `api_key` 和 `api_url`。
+
+```yaml
+# ~/.better_prompt/config.yml
+api_key: "YOUR_API_KEY"
+api_url: "https://api.openai.com/v1/chat/completions"
+model: "gpt-3.5-turbo"
+```
+
+## 🛠️ 使用方法
+
+BetterPrompt 提供了简洁的命令行接口来管理您的提示词。
+
+### 主要命令
+
+*   **列出所有提示词**
+    ```bash
+    bp --list
+    # or
+    bp -l
+    ```
+
+*   **显示一个提示词的详细内容**
+    ```bash
+    bp --show <ID>
+    # or
+    bp -s <ID>
+    ```
+
+*   **添加一个新的提示词**
+    *BetterPrompt 会启动您默认的文本编辑器来编写提示词内容。*
+    ```bash
+    bp --add
+    # or
+    bp -a
+    ```
+
+*   **编辑一个已存在的提示词**
+    *BetterPrompt 会启动您默认的文本编辑器来修改提示词内容。*
+    ```bash
+    bp --edit <ID>
+    # or
+    bp -e <ID>
+    ```
+
+*   **删除一个提示词**
+    ```bash
+    bp --delete <ID>
+    # or
+    bp -d <ID>
+    ```
+
+*   **调用 AI 模型**
+    使用指定的提示词 ID 来调用在 `config.yml` 中配置的 AI 模型。
+    ```bash
+    bp --call <ID>
+    ```
+
+*   **提供反馈**
+    打开一个链接，为项目提供反馈。
+    ```bash
+    bp --feedback
+    ```
+
+## 👨‍💻 开发
+
+如果您想为 BetterPrompt 贡献代码，请遵循以下步骤：
+
+1.  **Fork** 本项目。
+2.  创建您的功能分支 (`git checkout -b feature/AmazingFeature`)。
+3.  提交您的更改 (`git commit -m 'Add some AmazingFeature'`)。
+4.  将代码推送到分支 (`git push origin feature/AmazingFeature`)。
+5.  开启一个 **Pull Request**。
+
+## 📄 许可证
+
+本项目采用 MIT 许可证。详情请见 `LICENSE` 文件。

data/README.md

@@ -0,0 +1,100 @@
+# Better Prompt
+
+A Ruby Gem for tracking and analyzing LLM prompt interactions, designed to work with [smart_prompt](https://github.com/zhuangbiaowei/smart_prompt).
+
+## Features
+
+- Records complete prompt/response history including:
+  - Prompt content and length
+  - Response content, length and timing
+  - Model information (name, provider, size)
+  - Call parameters (temperature, max_tokens, etc.)
+  - Stream vs non-stream calls
+- Stores user feedback on response quality:
+  - Star ratings (1-5)
+  - Detailed scores (accuracy, relevance, etc.)
+  - Free-form comments
+- Organizes prompts with:
+  - Tags and categories
+  - Projects grouping
+- SQLite backend for easy analysis
+- Character interface for analyzing logs and comparing responses:
+  - Execute `better_prompt ./path/database.db` to launch
+  - View interaction logs and history
+  - Compare different model responses side-by-side
+  - Evaluate and rate response quality
+  - Built with [ruby_rich](https://github.com/zhuangbiaowei/ruby_rich) for rich terminal UI
+
+## Database Schema
+
+The database contains these main tables:
+
+1. `users` - User accounts
+2. `models` - LLM model information
+3. `prompts` - Prompt content and metadata
+4. `model_calls` - Individual API calls
+5. `responses` - Model responses
+6. `feedback` - User ratings and comments
+7. `tags`/`prompt_tags` - Prompt categorization
+8. `projects`/`project_prompts` - Prompt organization
+
+See [db/init.sql](db/init.sql) for complete schema.
+
+## Installation
+
+Add to your Gemfile:
+
+```ruby
+gem 'better_prompt'
+```
+
+Then execute:
+
+```bash
+bundle install
+```
+
+## Usage
+
+Basic setup:
+
+```ruby
+require 'better_prompt'
+
+# Initialize with database path
+BetterPrompt.setup(db_path: 'path/to/database.db')
+
+# Record a prompt and response
+BetterPrompt.record(
+  user_id: 1,
+  prompt: "Explain quantum computing",
+  response: "Quantum computing uses qubits...", 
+  model_name: "gpt-4",
+  response_time_ms: 1250,
+  is_stream: false
+)
+
+# Add user feedback
+BetterPrompt.add_feedback(
+  response_id: 123,
+  rating: 4,
+  comment: "Helpful but could be more detailed"
+)
+```
+
+## Development
+
+After checking out the repo:
+
+```bash
+bin/setup
+bundle exec rake test
+```
+
+## Contributing
+
+Bug reports and pull requests are welcome.
+
+## License
+
+The gem is available as open source under MIT License.

data/bin/bp

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require 'better_prompt'
+
+if ARGV[0]
+  db_path = ARGV[0]
+else
+  db_path = "./db/prompt.db"
+end
+db_path = File.expand_path(db_path)
+BetterPrompt.setup(db_path: db_path)
+
+BetterPrompt::CLI.start(db_path)
+

data/db/init.sql

@@ -0,0 +1,83 @@
+-- 创建模型表
+CREATE TABLE models (
+    model_id INTEGER PRIMARY KEY AUTOINCREMENT,
+    model_provider TEXT NOT NULL,
+    model_name TEXT NOT NULL,
+    model_version TEXT,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+-- 创建提示词表
+CREATE TABLE prompts (
+    prompt_id INTEGER PRIMARY KEY AUTOINCREMENT,
+    role TEXT NOT NULL,
+    prompt_template_name TEXT NOT NULL,
+    prompt_title TEXT,
+    prompt_content TEXT NOT NULL,
+    prompt_length INTEGER NOT NULL,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+-- 创建调用记录表
+CREATE TABLE model_calls (
+    call_id INTEGER PRIMARY KEY AUTOINCREMENT,
+    prompt_list TEXT NOT NULL,
+    model_id INTEGER,
+    is_streaming INTEGER NOT NULL, -- 在SQLite中使用0/1代替布尔值
+    temperature REAL,
+    max_tokens INTEGER,
+    top_p REAL,
+    top_k INTEGER,
+    additional_parameters TEXT, -- 使用JSON字符串存储
+    call_timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
+    FOREIGN KEY (model_id) REFERENCES models(model_id)
+);
+
+-- 创建响应表
+CREATE TABLE responses (
+    response_id INTEGER PRIMARY KEY AUTOINCREMENT,
+    call_id INTEGER,
+    response_content TEXT NOT NULL,
+    response_length INTEGER NOT NULL,
+    response_time_ms INTEGER NOT NULL, -- 响应时间(毫秒)
+    is_streaming INTEGER NOT NULL, -- 在SQLite中使用0/1代替布尔值
+    token_count INTEGER,               -- 令牌数量
+    cost REAL,                         -- 调用成本
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    FOREIGN KEY (call_id) REFERENCES model_calls(call_id)
+);
+
+-- 创建用户评价表
+CREATE TABLE feedback (
+    feedback_id INTEGER PRIMARY KEY AUTOINCREMENT,
+    response_id INTEGER,
+    rating INTEGER CHECK (rating BETWEEN 1 AND 5), -- 1-5星评价
+    feedback_comment TEXT,
+    accuracy_score INTEGER CHECK (accuracy_score BETWEEN 1 AND 10),
+    relevance_score INTEGER CHECK (relevance_score BETWEEN 1 AND 10),
+    creativity_score INTEGER CHECK (creativity_score BETWEEN 1 AND 10),
+    helpfulness_score INTEGER CHECK (helpfulness_score BETWEEN 1 AND 10),
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+    FOREIGN KEY (response_id) REFERENCES responses(response_id)
+);
+
+-- 创建标签表，用于更好地分类提示词
+CREATE TABLE tags (
+    tag_id INTEGER PRIMARY KEY AUTOINCREMENT,
+    tag_name TEXT NOT NULL UNIQUE,
+    created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+
+-- 创建提示词-标签关联表
+CREATE TABLE prompt_tags (
+    prompt_id INTEGER,
+    tag_id INTEGER,
+    PRIMARY KEY (prompt_id, tag_id),
+    FOREIGN KEY (prompt_id) REFERENCES prompts(prompt_id),
+    FOREIGN KEY (tag_id) REFERENCES tags(tag_id)
+);
+
+-- 创建索引以提高查询性能
+CREATE INDEX idx_responses_call_id ON responses(call_id);
+CREATE INDEX idx_feedback_response_id ON feedback(response_id);
+CREATE INDEX idx_prompt_tags_prompt_id ON prompt_tags(prompt_id);

data/lib/better_prompt/cli.rb

@@ -0,0 +1,27 @@
+require "ruby_rich"
+
+module BetterPrompt
+  class CLI
+    class << self
+      def start(db_path)
+        @layout = Component::Root.build
+        @layout.split_row(
+          Component::Sidebar.build,
+          Component::Main.build
+        )
+        @layout["sidebar"].split_column(
+          Component::TemplateList.build,
+          Component::PromptList.build,
+        )
+        @layout["main"].split_column(
+          Component::ModelCall.build,
+          Component::Response.build
+        )
+        ORM.setup("sqlite://"+db_path)
+        RubyRich::Live.start(@layout, refresh_rate: 24) do |live|
+          live.listening = true
+        end
+      end
+    end
+  end
+end

data/lib/better_prompt/components/feedback.rb

@@ -0,0 +1,33 @@
+module BetterPrompt
+  module Component
+    class Feedback
+      def self.build(live)
+        dialog = RubyRich::Dialog.new(title: "加载对话", content: "请问这个回答的内容打分，1~5 之间。", width: 80, height: 20, buttons: [:cancel])
+        dialog.live = live
+        register_event_listener(dialog)
+        return dialog
+      end
+      def self.register_event_listener(dialog)
+        dialog.key(:escape){|event, live|
+          live.layout.hide_dialog
+        }
+        dialog.key(:string){|event, live|
+          key_value = event[:value].to_i
+          if key_value >=1 && key_value <=5
+            if feedback=live.params[:feedback]
+              feedback.rating = key_value
+              feedback.save
+            else
+              feedback = ORM::Feedback.new(response_id: live.params[:response_id], rating: key_value)
+              feedback.save
+              live.params[:feedback] = feedback
+            end
+            response = live.find_panel("response")
+            response.title = "回答  (Ctrl+r/Ctrl+a/Ctrl+t)  评分： #{key_value}"
+            live.layout.hide_dialog
+          end
+        }
+      end
+    end
+  end
+end

data/lib/better_prompt/components/main.rb

@@ -0,0 +1,10 @@
+module BetterPrompt
+  module Component
+    class Main
+      def self.build
+        layout = RubyRich::Layout.new(name: "main", ratio: 3) 
+        return layout
+      end
+    end
+  end
+end  

data/lib/better_prompt/components/model_call.rb

@@ -0,0 +1,159 @@
+module BetterPrompt
+  module Component
+    class ModelCall
+      def self.build
+        @current_pos = 0
+        layout = RubyRich::Layout.new(name: "model_call", size: 14)
+        draw_view(layout)
+        register_event_listener(layout)
+        return layout
+      end
+
+      def self.draw_view(layout)
+        layout.update_content(RubyRich::Panel.new(
+          "",
+          title: "模型调用 (Shift+3) 打分 (Ctrl+f)"
+        ))
+      end
+
+      def self.update_model_call_list(live, prompt_id)
+        model_call = live.find_panel("model_call")
+        @call_list = ORM::ModelCall.where_all(Sequel.like(:prompt_list, "%,#{prompt_id}]"))
+        @page_count = (@call_list.size / 9.0).ceil
+        model_call.content = generate_content()
+      end
+
+      def self.generate_content(pos=0, page_num=1)
+        i = 0
+        @current_page = page_num
+        start_pos = (page_num - 1) * 9
+        end_pos = start_pos + 9
+        call_list_str = ""
+        @last_pos = 0
+        @call_list.each do |call|
+          i += 1
+          model = ORM::Model[call.model_id]
+          row_str = "id: #{call.call_id}"
+          row_str = row_str + " "*(10-row_str.length) + "model:" + model.model_provider + "/" + model.model_name
+          if call.prompt_list.length < 42
+            row_str = row_str + " "*(60-row_str.length) + "prompts: "+ call.prompt_list
+          else
+            row_str = row_str + " "*(60-row_str.length) + "prompts: ..."+ call.prompt_list[-42..-1]
+          end
+          if i > start_pos && i <= end_pos
+            if i-start_pos==pos
+              call_list_str += RubyRich::AnsiCode.color(:blue)+ "#{i-start_pos} #{row_str}\n" + RubyRich::AnsiCode.reset
+            else
+              call_list_str += RubyRich::AnsiCode.color(:blue)+ "#{i-start_pos}" + RubyRich::AnsiCode.reset + " #{row_str}\n"
+            end
+            @last_pos += 1
+          end
+        end
+        @max_pos = i
+        if start_pos>0 or i > end_pos
+          call_list_str += " \n" * (11 - call_list_str.split("\n").size)
+        end
+        if start_pos>0
+          call_list_str += " " * 11 + "←" + " " * 2
+        else
+          call_list_str += " " * 14          
+        end
+        if i > end_pos
+          call_list_str += " " * 2 + "→"
+        end
+        return call_list_str
+      end
+
+      def self.try_parse(string)
+        return JSON.parse(string)
+      rescue JSON::ParserError
+        begin
+          return JSON.parse(string.gsub("=>",":").gsub("nil","\"nil\""))
+        rescue JSON::ParserError
+          return false
+        end
+      end
+
+      def self.show_call(model_call)
+        model_call.content = generate_content(@current_pos, @current_page)
+        call = @call_list[(@current_page - 1)*9+@current_pos-1]
+        response = ORM::Response.where(call_id: call.call_id).first
+        if response
+          response_content = response.response_content
+          if json = try_parse(response_content)
+            Response.set_content(json.dig("choices", 0, "message", "content"))
+            Response.set_reason_content(json.dig("choices", 0, "message", "reasoning_content"))
+            Response.set_tool_content(json.dig("choices",0,"message", "tool_calls"))
+          else
+            Response.set_content(response_content)
+            Response.set_reason_content("")
+            Response.set_tool_content("")
+          end
+          response_id = response.response_id
+          feedback = ORM::Feedback.where(response_id: response_id).first
+        else
+            Response.set_content("响应不存在")
+            Response.set_reason_content("响应不存在")
+            Response.set_tool_content("响应不存在")
+        end
+        Response.show_content(:content, feedback)
+        return feedback, response_id
+      end
+
+      def self.register_event_listener(layout)        
+        layout.key(:string) do |event, live|
+          model_call = live.find_panel("model_call")
+          if model_call.border_style == :green && @call_list
+            if event[:value].to_i >= 1 && event[:value].to_i <= @last_pos
+              @current_pos = event[:value].to_i
+              live.params[:feedback], live.params[:response_id] = show_call(model_call)
+            end
+          end
+        end
+        layout.key(:right) do |event, live|
+          model_call = live.find_panel("model_call")
+          if model_call.border_style == :green && @call_list
+            if @current_page<@page_count
+              model_call.content = generate_content(0, @current_page+1)
+              @current_pos = 0
+            end
+          end
+        end
+        layout.key(:left) do |event, live|
+          model_call = live.find_panel("model_call")
+          if model_call.border_style == :green && @call_list
+            if @current_page>1              
+              model_call.content = generate_content(0, @current_page-1)
+              @current_pos = 0
+            end
+          end
+        end
+        layout.key(:up) do |event, live|
+          model_call = live.find_panel("model_call")
+          if model_call.border_style == :green && @call_list
+            if @current_pos > 1
+              @current_pos -= 1
+              show_call(model_call)
+            end
+          end
+        end
+        layout.key(:down) do |event, live|
+          model_call = live.find_panel("model_call")
+          if model_call.border_style == :green && @call_list
+            if @current_pos < @max_pos
+              @current_pos += 1
+              show_call(model_call)
+            end
+          end
+        end
+        layout.key(:ctrl_f) do |event, live|
+          model_call = live.find_panel("model_call")
+          if model_call.border_style == :green && @current_pos>0 && @call_list
+            dialog = Feedback.build(live)
+            live.layout.show_dialog(dialog)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/better_prompt/components/prompt_list.rb

@@ -0,0 +1,112 @@
+module BetterPrompt
+  module Component
+    class PromptList
+      def self.build
+        layout = RubyRich::Layout.new(name: "prompt_list",  ratio: 4)
+        draw_view(layout)
+        register_event_listener(layout)
+        return layout
+      end
+
+      def self.draw_view(layout)
+        layout.update_content(RubyRich::Panel.new(
+          "",
+          title: "提示词列表  (Shift+2)"
+        ))
+      end
+
+      def self.update_title(prompt)
+        result = BetterPrompt.engine.call_worker(:summarize, {text: prompt[:prompt_content]})
+        prompt[:prompt_title] = result.strip
+        prompt.save
+        return result.strip
+      end
+
+      def self.update_prompt_list(live, template_name)
+        prompt_list = live.find_panel("prompt_list")
+        @prompt_list = BetterPrompt::ORM::Prompt.where(:prompt_template_name=>template_name).all
+        @page_count = (@prompt_list.size / 9.0).ceil
+        prompt_list.content = generate_content()
+      end
+
+      def self.generate_content(pos=0, page_num=1)
+        i = 0
+        @current_page = page_num
+        start_pos = (page_num - 1) * 9
+        end_pos = start_pos + 9
+        prompt_list_str = ""
+        @prompt_list.each do |prompt|
+          i += 1
+          title = prompt.prompt_title
+          if title==nil
+            title = update_title(prompt)
+          end
+          title_width = title.chars.sum { |c| Unicode::DisplayWidth.of(c) }
+          if title_width>22
+            new_title = ""
+            title.chars.each do |c|
+              if (new_title+c).chars.sum { |c| Unicode::DisplayWidth.of(c) }>22
+                new_title += ".."
+                break
+              else
+                new_title += c
+              end
+            end
+            title = new_title
+          end
+          if i > start_pos && i <= end_pos            
+            if i-start_pos==pos
+              prompt_list_str += RubyRich::AnsiCode.color(:blue)+ "#{i-start_pos} #{title}\n" + RubyRich::AnsiCode.reset
+            else
+              prompt_list_str += RubyRich::AnsiCode.color(:blue)+ "#{i-start_pos}" + RubyRich::AnsiCode.reset + " #{title}\n"
+            end
+          end
+        end
+        if start_pos>0 or i > end_pos
+          prompt_list_str += " \n" * (25 - prompt_list_str.split("\n").size)
+        end
+        if start_pos>0
+          prompt_list_str += " " * 11 + "←" + " " * 2
+        else
+          prompt_list_str += " " * 14          
+        end
+        if i > end_pos
+          prompt_list_str += " " * 2 + "→"
+        end
+        return prompt_list_str
+      end
+
+      def self.register_event_listener(layout)
+        layout.key(:string) do |event, live|
+          prompt_list = live.find_panel("prompt_list")
+          if prompt_list.border_style == :green
+            if event[:value].to_i >= 1 && event[:value].to_i <= @prompt_list.count
+              prompt_list.content = generate_content(event[:value].to_i, @current_page)
+              prompt = @prompt_list[(@current_page - 1)*9+event[:value].to_i-1]
+              dialog = ShowPrompt.build(prompt.prompt_content, live)
+              live.layout.show_dialog(dialog)
+              Response.clean
+              ModelCall.update_model_call_list(live, prompt.prompt_id)
+            end
+          end
+        end
+        layout.key(:right) do |event, live|
+          prompt_list = live.find_panel("prompt_list")
+          if prompt_list.border_style == :green
+            if @current_page<@page_count
+              prompt_list.content = generate_content(0, @current_page+1)
+            end
+          end
+        end
+        layout.key(:left) do |event, live|
+          prompt_list = live.find_panel("prompt_list")
+          if prompt_list.border_style == :green
+            if @current_page>1              
+              prompt_list.content = generate_content(0, @current_page-1)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/better_prompt/components/response.rb

@@ -0,0 +1,74 @@
+module BetterPrompt
+  module Component
+    class Response
+      def self.build
+        layout = RubyRich::Layout.new(name: "response",  ratio: 1)
+        draw_view(layout)
+        register_event_listener(layout)
+        return layout
+      end
+
+      def self.draw_view(layout)
+        @panel = RubyRich::Panel.new(
+          "",
+          title: "回答  (Ctrl+r/Ctrl+a/Ctrl+t)  评分： 1~5"
+        )
+        layout.update_content(@panel)
+      end
+
+      def self.clean
+        @panel.content = ""
+      end
+
+      def self.update_content(content)
+        if content.strip.empty?          
+          @panel.content = "<Empty>"
+        else
+          @panel.content = ""
+          @panel.content = content
+        end
+        @panel.home
+      end
+
+      def self.set_content(content)
+        @content = content
+      end
+
+      def self.set_reason_content(content)
+        @reason_content = content
+      end
+
+      def self.set_tool_content(content)
+        @tool_content = content
+      end
+
+      def self.show_content(type, feedback)
+        rating = if feedback
+          feedback.rating
+        else
+          "1~5"
+        end
+        if type==:content
+          @panel.title = "回答  (Ctrl+r/Ctrl+a/Ctrl+t)  评分： #{rating}"
+          update_content(@content.to_s)
+        end
+        if type==:reason
+          @panel.title = "推理  (Ctrl+r/Ctrl+a/Ctrl+t)  评分： #{rating}"
+          update_content(@reason_content.to_s)
+        end
+        if type==:tool
+          @panel.title = "工具调用  (Ctrl+r/Ctrl+a/Ctrl+t)  评分： #{rating}"
+          update_content(@tool_content.to_s)
+        end
+      end
+
+      def self.register_event_listener(layout)
+        layout.key(:ctrl_r) { |event, live|  show_content(:reason, live.params[:feedback])}
+        layout.key(:ctrl_a) { |event, live|  show_content(:content, live.params[:feedback])}
+        layout.key(:ctrl_t) { |event, live|  show_content(:tool, live.params[:feedback])}
+        layout.key(:page_up) { |event, live| @panel.page_up}
+        layout.key(:page_down) { |event, live| @panel.page_down}
+      end
+    end
+  end
+end

data/lib/better_prompt/components/root.rb

@@ -0,0 +1,38 @@
+module BetterPrompt
+  module Component
+    class Root
+      def self.build
+        layout = RubyRich::Layout.new
+        register_event_listener(layout)
+        return layout
+      end
+      def self.register_event_listener(layout)
+        layout.key(:string) do |event, live|
+          key = event[:value]
+          if key=="!"
+            template_list = live.find_panel("template_list")
+            template_list.border_style = :green
+            prompt_list = live.find_panel("prompt_list")
+            prompt_list.border_style = :white
+            model_call = live.find_panel("model_call")
+            model_call.border_style = :white
+          elsif key=="@"
+            template_list = live.find_panel("template_list")
+            template_list.border_style = :white
+            prompt_list = live.find_panel("prompt_list")
+            prompt_list.border_style = :green
+            model_call = live.find_panel("model_call")
+            model_call.border_style = :white
+          elsif key=="#"
+            template_list = live.find_panel("template_list")
+            template_list.border_style = :white
+            prompt_list = live.find_panel("prompt_list")
+            prompt_list.border_style = :white
+            model_call = live.find_panel("model_call")
+            model_call.border_style = :green
+          end
+        end
+      end
+    end
+  end
+end

data/lib/better_prompt/components/show_prompt.rb

@@ -0,0 +1,19 @@
+module BetterPrompt
+  module Component
+    class ShowPrompt
+      def self.build(content, live)
+        width = 60
+        dialog = RubyRich::Dialog.new(title: "提示词内容", content: content, width: width, height: 20, buttons: [:ok])
+        dialog.live = live
+        register_event_listener(dialog)
+        return dialog
+      end
+
+      def self.register_event_listener(dialog)
+        dialog.key(:enter){|event, live|
+          live.layout.hide_dialog
+        }
+      end
+    end
+  end
+end

data/lib/better_prompt/components/sidebar.rb

@@ -0,0 +1,15 @@
+module BetterPrompt
+  module Component
+    class Sidebar
+      def self.build
+        layout = RubyRich::Layout.new(name: "sidebar", size: 30)
+        register_event_listener(layout)
+        return layout
+      end
+      
+      def self.register_event_listener(layout)        
+        layout.key(:ctrl_c) { |event, live| live.stop }        
+      end
+    end
+  end
+end

data/lib/better_prompt/components/template_list.rb

@@ -0,0 +1,90 @@
+module BetterPrompt
+  module Component
+    class TemplateList
+      def self.build
+        layout = RubyRich::Layout.new(name: "template_list", size: 14)
+        draw_view(layout)
+        register_event_listener(layout)
+        return layout
+      end
+
+      def self.get_list_content
+        @template_list = ORM::Prompt
+          .where(prompt_template_name: 'NULL')
+          .invert
+          .select(:prompt_template_name)
+          .distinct
+          .map(&:prompt_template_name)
+        @page_count = (@template_list.size / 9.0).ceil
+        generate_content
+      end
+
+      def self.generate_content(pos=0, page_num=1)
+        i = 0
+        @current_page = page_num
+        start_pos = (page_num - 1) * 9
+        end_pos = start_pos + 9
+        template_list_str = ""
+        @template_list.each do |template|
+          i += 1
+          if i > start_pos && i <= end_pos            
+            if i-start_pos==pos
+              template_list_str += RubyRich::AnsiCode.color(:blue)+ "#{i-start_pos} #{template}\n" + RubyRich::AnsiCode.reset
+            else
+              template_list_str += RubyRich::AnsiCode.color(:blue)+ "#{i-start_pos}" + RubyRich::AnsiCode.reset + " #{template}\n"
+            end
+          end          
+        end
+        if start_pos>0 or i > end_pos
+          template_list_str += " \n" * (11 - template_list_str.split("\n").size)
+        end
+        if start_pos>0
+          template_list_str += " " * 11 + "←" + " " * 2
+        else
+          template_list_str += " " * 14          
+        end
+        if i > end_pos
+          template_list_str += " " * 2 + "→"
+        end
+        return template_list_str
+      end
+
+      def self.draw_view(layout)
+        layout.update_content(RubyRich::Panel.new(
+          get_list_content(),
+          title: "模板列表 (Shift+1)",
+          border_style: :green
+        ))
+      end
+
+      def self.register_event_listener(layout)
+        layout.key(:string) do |event, live|
+          template_list = live.find_panel("template_list")
+          if template_list.border_style == :green
+            if event[:value].to_i >= 1 && event[:value].to_i <= @template_list.count
+              template_list.content = generate_content(event[:value].to_i, @current_page)
+              template_name = @template_list[(@current_page - 1)*9+event[:value].to_i-1]
+              PromptList.update_prompt_list(live, template_name)
+            end
+          end
+        end
+        layout.key(:right) do |event, live|
+          template_list = live.find_panel("template_list")
+          if template_list.border_style == :green
+            if @current_page<@page_count
+              template_list.content = generate_content(0, @current_page+1)
+            end
+          end
+        end
+        layout.key(:left) do |event, live|
+          template_list = live.find_panel("template_list")
+          if template_list.border_style == :green
+            if @current_page>1
+              template_list.content = generate_content(0, @current_page-1)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/better_prompt/models.rb

@@ -0,0 +1,31 @@
+require 'sequel'
+module BetterPrompt
+  module ORM
+    class Model < Sequel::Model(ORM.db[:models])
+      plugin :timestamps, update_on_create: true
+    end
+
+    class Prompt < Sequel::Model(ORM.db[:prompts])
+      plugin :timestamps, update_on_create: true
+    end
+
+    class ModelCall < Sequel::Model(ORM.db[:model_calls])
+      plugin :timestamps, update_on_create: true
+    end
+
+    class Response < Sequel::Model(ORM.db[:responses])
+      plugin :timestamps, update_on_create: true
+    end
+
+    class Feedback < Sequel::Model(ORM.db[:feedback])
+      plugin :timestamps, update_on_create: true
+    end
+
+    class Tag < Sequel::Model(ORM.db[:tags])
+      plugin :timestamps, update_on_create: true
+    end
+
+    class PromptTag < Sequel::Model(ORM.db[:prompt_tags])
+    end
+  end
+end

data/lib/better_prompt/orm.rb

@@ -0,0 +1,16 @@
+require 'sequel'
+module BetterPrompt
+  module ORM
+    class << self
+      attr_reader :db
+
+      def setup(database_url)
+        raise "DATABASE_URL environment variable not set" unless database_url
+        @db = Sequel.connect(database_url)
+        @db.test_connection
+      rescue => e
+        raise "Failed to connect to database: #{e.message}"
+      end
+    end
+  end
+end

data/lib/better_prompt/version.rb

@@ -0,0 +1,3 @@
+module BetterPrompt
+  VERSION = "0.2.0"
+end

data/lib/better_prompt.rb

@@ -0,0 +1,158 @@
+require "sqlite3"
+require "fileutils"
+require "time"
+require "smart_prompt"
+
+require_relative 'better_prompt/cli'
+require_relative 'better_prompt/orm'
+Dir["#{__dir__}/better_prompt/components/*.rb"].each { |f| require_relative f }
+
+module BetterPrompt
+  class << self
+    def setup(db_path:)
+      BetterPrompt.logger = Logger.new("./log/better.log")
+      @db_path = db_path
+      # Return true if database already exists
+      if File.exist?(db_path)
+        ORM.setup("sqlite://"+db_path)
+        require_relative 'better_prompt/models'
+        return true 
+      end
+
+      # Ensure directory exists
+      FileUtils.mkdir_p(File.dirname(db_path))
+
+      # Initialize database
+      SQLite3::Database.new(db_path) do |db|
+        # Execute each statement from init.sql
+        File.read(File.join(__dir__, "../db/init.sql")).split(/;\s*\n/).each do |statement|
+          db.execute(statement.strip) unless statement.strip.empty?
+        end
+      end
+      ORM.setup("sqlite://"+db_path)
+      require_relative 'better_prompt/models'
+      true
+    rescue => e
+      raise "Failed to initialize database: #{e.message}"
+    end
+
+    def add_model(provider, model_name)
+      model_version = if model_name.include?(':')
+                        model_name.split(':', 2).last
+                      else
+                        'latest'
+                      end
+      model_name = model_name.split(":")[0]
+      
+      ORM::Model.find_or_create(
+        model_provider: provider,
+        model_name: model_name,
+        model_version: model_version
+      )
+    rescue => e
+      raise "Failed to add model: #{e.message}"
+    end
+
+    def add_prompt(template_name, role, prompt_content)
+      prompt_length = prompt_content.length
+      
+      ORM::Prompt.find_or_create(
+        prompt_template_name: template_name,
+        role: role,
+        prompt_content: prompt_content
+      ) do |prompt|
+        prompt.prompt_length = prompt_length
+      end
+    rescue => e
+      raise "Failed to add prompt: #{e.message}"
+    end
+
+    def add_model_call(provider, model_name, messages, streaming=false, temperature=0.7, max_tokens=0, top_p=0.0, top_k=0, params={})
+      model_version = if model_name.include?(':')
+                        model_name.split(':', 2).last
+                      else
+                        'latest'
+                      end
+      model_base_name = model_name.split(":")[0]
+      
+      # 查找或创建model记录
+      model = ORM::Model.first(
+        model_provider: provider,
+        model_name: model_base_name,
+        model_version: model_version
+      )
+      raise "Model not found: #{provider}/#{model_name}" unless model
+
+      # 从messages构建prompt_list
+      prompt_list = messages.map do |msg|
+        role = msg[:role] || msg["role"]
+        content = msg[:content] || msg["content"]
+        if role == "assistant" || role == "tool"
+          add_prompt("NULL", role, content)
+        end
+        # 查找prompt_id
+        prompt = ORM::Prompt.first(role: role, prompt_content: content)
+        prompt&.prompt_id || 0
+      end.to_json
+
+      # 创建model_call记录
+      model_call = ORM::ModelCall.create(
+        prompt_list: prompt_list,
+        model_id: model.model_id,
+        is_streaming: streaming,
+        temperature: temperature,
+        max_tokens: max_tokens,
+        top_p: top_p,
+        top_k: top_k,
+        additional_parameters: params.to_json
+      )
+
+      model_call.call_id
+    rescue => e
+      raise "Failed to add model call: #{e.message}"
+    end
+
+    def add_response(call_id, response_content, is_streaming)
+      response_content = response_content.to_s
+      response_length = response_content.length
+      current_time = Time.now
+      
+      # 查找model_call记录
+      model_call = ORM::ModelCall[call_id]
+      raise "Model call not found: #{call_id}" unless model_call
+
+      # 计算响应时间(毫秒)
+      response_time_ms = ((current_time - model_call.call_timestamp) * 1000).to_i
+
+      # 创建response记录
+      response = ORM::Response.create(
+        call_id: call_id,
+        response_content: response_content,
+        response_length: response_length,
+        response_time_ms: response_time_ms,
+        is_streaming: is_streaming
+      )
+
+      response.response_id
+    rescue => e
+      raise "Failed to add response: #{e.message}"
+    end
+    def engine
+      if @engine == nil
+        file_path = "./config/config.yml"
+        file_path = "./config/llm_config.yml" unless File.exist?(file_path)
+        BetterPrompt.logger.info(file_path)
+        @engine = SmartPrompt::Engine.new(file_path)
+      end
+      return @engine
+    end
+    def logger=(logger)
+      @logger = logger
+    end
+    def logger
+      @logger ||= Logger.new($stdout).tap do |log|
+        log.progname = "Better Prompt"
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,130 @@
+--- !ruby/object:Gem::Specification
+name: better_prompt
+version: !ruby/object:Gem::Version
+  version: 0.2.0
+platform: ruby
+authors:
+- zhuang biaowei
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ruby_rich
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.1
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.4'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: Provides enhanced command line prompt functionality with database storage
+email:
+- zbw@kaiyuanshe.org
+executables:
+- bp
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.cn.md
+- README.md
+- bin/bp
+- db/init.sql
+- lib/better_prompt.rb
+- lib/better_prompt/cli.rb
+- lib/better_prompt/components/feedback.rb
+- lib/better_prompt/components/main.rb
+- lib/better_prompt/components/model_call.rb
+- lib/better_prompt/components/prompt_list.rb
+- lib/better_prompt/components/response.rb
+- lib/better_prompt/components/root.rb
+- lib/better_prompt/components/show_prompt.rb
+- lib/better_prompt/components/sidebar.rb
+- lib/better_prompt/components/template_list.rb
+- lib/better_prompt/models.rb
+- lib/better_prompt/orm.rb
+- lib/better_prompt/version.rb
+homepage: https://github.com/zhuangbiaowei/better_prompt
+licenses:
+- MIT
+metadata: {}
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.6.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: A better command line prompt utility
+test_files: []