kaba 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 7d0d15c1454667339192ecb68a40a0128221f0e2f176c9668fdddbaf6507a143
+  data.tar.gz: 63d4b6088c24453874a2acf9a1c183b6d7468a0297c43cf32feccd0e838f84c0
+SHA512:
+  metadata.gz: 0e937f2aea17053fd01b8a4b5f76f61d58d2817eed4e12942f4edaee0ef72b8cf2fc2c3febeb082d49131fad6aba495bfc7385d59bd13c127b8faaa0ad1a7719
+  data.tar.gz: 592a977e7ec2c0b80f76e9f4c9fa0875357466ab1e6cbb05e9dec0fdabe086d8f25fda39caad60d0334ed145a69483ef6ea091d25bc5c39299c9afc5380d6155

data/README.md

@@ -0,0 +1,9 @@
+# Kaba
+
+## 目录结构
+- data
+  - row
+  - schema
+
+## 命令
+`gem install kaba`

data/Rakefile

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+task default: %i[]

data/exe/kaba

@@ -0,0 +1,30 @@
+#!/usr/bin/env ruby
+require "bundler/setup"
+
+require 'async'
+require 'faraday'
+require 'colorize'
+require 'tty-progressbar'
+require 'async/http/faraday'
+
+require 'json'
+require "kaba"
+
+class Application
+  class << self
+    def connection
+      @connection ||= Faraday.new('https://lisa-typechat.listenai.com') do |faraday|
+        faraday.adapter :async_http, clients: Async::HTTP::Faraday::PersistentClients
+        faraday.request :json
+      end
+    end
+  end
+end
+
+def ddbug
+  require 'irb'
+  binding.irb
+end
+
+# 运行 DPodfile 文件，DPodfile 是一个 Ruby 文件
+load DatasetSource.podfile

data/kaba.gemspec

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require_relative "lib/kaba/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "kaba"
+  spec.version = Kaba::VERSION
+  spec.authors = ["MJ"]
+  spec.email = ["tywf91@gmail.com"]
+
+  spec.summary = "用来做数据集的工具"
+  spec.description = "用来做数据集的工具"
+  spec.homepage = "https://github.com/mjason/kaba.git"
+  spec.required_ruby_version = ">= 3.3.0"
+
+  # spec.metadata["allowed_push_host"] = "TODO: Set to your gem server 'https://example.com'"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "https://github.com/mjason/kaba.git"
+  spec.metadata["changelog_uri"] = "https://github.com/mjason/kaba.git"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(__dir__) do
+    `git ls-files -z`.split("\x0").reject do |f|
+      (File.expand_path(f) == __FILE__) ||
+        f.start_with?(*%w[bin/ test/ spec/ features/ .git .github appveyor Gemfile])
+    end
+  end
+  spec.bindir = "exe"
+  spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "async", "~> 2.20"
+  spec.add_dependency "faraday", "~> 2.12"
+  spec.add_dependency "async-http-faraday", "~> 0.19.0"
+  spec.add_dependency "colorize", "~> 1.1"
+  spec.add_dependency "tty-progressbar", "~> 0.18.3"
+
+  # For more information and examples about making a new gem, check out our
+  # guide at: https://bundler.io/guides/creating_gem.html
+end

data/lib/kaba/_DPodfile_

@@ -0,0 +1,51 @@
+## 使用 Ruby 语言编写的数据集校验脚本
+# ddbug 断点调试
+# 使用 colorize 来输出带颜色的信息，https://github.com/fazibear/colorize
+# 使用 progressbar 来显示进度条，https://github.com/piotrmurach/tty-progressbar
+# 设置数据集目录, 如果使用 Docker 方式运行，需要将数据集挂载到 /data 目录下，DatasetSource 会自动加载 /data 目录下的数据集
+source = DatasetSource.new(File.join(__dir__, 'data'))
+schema = source.schema.join('resume.ts').read
+type_name = 'Resume'
+prompt = Prompt.new(schema, type_name)
+validate = Validate.new(schema: schema, type_name: type_name)
+dataset = Dataset.new(source.row, prompt)
+
+# # 
+# # 可以使用 run_file 来验证单个文件
+# response = validate.run_file(source.row.join('1.target.json'))
+# puts response
+# # 
+
+## 一般来说直接 run_files 就可以了，支持 limit 来限制读取的文件数量
+validate.run_files(source.row)
+
+## 数据集处理部分
+# 支持 limit 来限制读取的文件数量
+dataset.scan()
+dataset.save(source.join('train.jsonl'))
+
+## 高级玩法，不要轻易尝试
+#
+### 可以加入 limit 来限制读取的文件数量，validate.run_files('./data/row', limit: 1) do |response, json, file| 
+# validate.run_files(source.row) do |response, progressbar| 
+#   progressbar.log response.to_s unless response.success?
+# end
+# 
+#
+## 也支持 limit 来进行限制
+# Async do
+#   dataset.each do |row, ds|
+#     Async do
+#       instruction = prompt.render(File.read row.input_file)
+#       target = <<~Markdown
+#       ```json
+#       #{JSON.pretty_generate(JSON.parse(File.read(row.target_path)))}
+#       ```
+#       Markdown
+#       ds.add({ instruction: instruction, output: target })
+
+#       instruction = prompt.render(File.read(row.input_file), export: true)
+#       ds.add({ instruction: instruction, output: target })
+#     end
+#   end
+# end.wait

data/lib/kaba/dataset.rb

@@ -0,0 +1,68 @@
+class Dataset
+  attr_reader :lines
+
+  def initialize(data_dir, prompt)
+    @data_files = Dir.glob(File.join(File.expand_path(data_dir), '*.target.json'))
+    @lines = []
+    @prompt = prompt
+  end
+
+  ## 实现一个 each 方法，可以让用户通过 block 的方式遍历数据集，提供一个 add 方法，可以将数据添加到数据集中
+  def _each(limit: nil)
+    @data_files.first(limit || @data_files.size).each do |file|
+      yield(Row.new(file), self)
+    end
+  end
+
+  def each(limit: nil, &block)
+    puts "Waring: each is very dangerous".colorize(:red)
+    _each(limit: limit, &block)
+  end
+
+  def add(data)
+    @lines << data
+  end
+
+  def save(file_path)
+    File.open(File.expand_path(file_path), 'w') do |file|
+      @lines.each do |line|
+        file.puts(line.to_json)
+      end
+    end
+  end
+
+  def validate
+    @lines.size == (@data_files.size * 2)
+  end
+
+  def scan(limit: nil)
+    progressbar = TTY::ProgressBar.new("Dataset: [:bar] :percent :current/:total", total: @data_files.size)
+    Async do
+      _each(limit: limit) do |row, ds|
+        Async do
+          instruction = @prompt.render(File.read row.input_file)
+          target = <<~Markdown
+          ```json
+          #{JSON.pretty_generate(JSON.parse(File.read(row.target_path)))}
+          ```
+          Markdown
+          ds.add({ instruction: instruction, output: target })
+
+          instruction = @prompt.render(File.read(row.input_file), export: true)
+          ds.add({ instruction: instruction, output: target })
+
+          progressbar.advance
+        end
+      end
+    end.wait
+  end
+
+end
+
+class Row
+  attr_reader :target_path, :input_file
+  def initialize(file)
+    @target_path = File.expand_path(file)
+    @input_file = @target_path.sub(/\.target\.json$/, '.input.txt')
+  end
+end

data/lib/kaba/dataset_source.rb

@@ -0,0 +1,41 @@
+class DatasetSource
+  
+  attr_reader :path
+  def initialize(path)
+    @path = path
+  end
+
+  [:row, :schema].each do |method_name|
+    define_method(method_name) do
+      self.class.new(File.join(@path, method_name.to_s))
+    end
+  end
+
+  def read
+    File.read @path
+  end
+
+  def to_s
+    @path
+  end
+
+  def to_path
+    @path
+  end
+
+  def join(name)
+    self.class.new File.join(@path, name)
+  end
+
+  class << self
+    def podfile
+      d_podfile_path = File.join(Dir.pwd, 'DPodfile')
+      unless File.exist?(d_podfile_path)
+        FileUtils.cp(File.join(__dir__, '_DPodfile_'), d_podfile_path)
+      end
+      d_podfile_path
+    end
+  end
+
+end
+

data/lib/kaba/prompt.rb

@@ -0,0 +1,26 @@
+class Prompt
+  def initialize(schema, type_name)
+    @schema = schema
+    @type_name = type_name
+  end
+
+  def clear_export
+    @schema.gsub(/export\s+default\s+/, '').gsub(/export\s+/, '')
+  end
+
+  def render(input, export: false)
+    schema = export ? clear_export : @schema
+    request_body = {
+      schema: schema,
+      typeName: @type_name,
+      inpu: input
+    }
+    Application.connection.post('/prompt', request_body).body
+  end
+
+  class << self
+    def file(schema_path, type_name)
+      self.new File.read(File.expand_path schema_path), type_name
+    end
+  end
+end

data/lib/kaba/validate.rb

@@ -0,0 +1,84 @@
+## 用于将所有数据进行验证
+class Validate
+  def initialize(schema:, type_name:)
+    @schema = schema
+    @type_name = type_name
+  end
+
+  def run(input)
+    request_body = {
+      schema: @schema,
+      typeName: @type_name,
+      jsonData: input
+    }
+
+    Application.connection.post('/validate', request_body)
+  end
+
+  # 读取某个文件然后运行
+  def run_file(file)
+    input = JSON.parse File.read(File.expand_path file)
+    ValidateReponse.new run(input)
+  end
+
+  # 读取某个文件夹下的然后运行，运行有结果了 block 会被调用
+  # limit 用于限制读取的文件数量, 为 nil 时读取所有文件
+  def run_files(dir, limit: nil, &block)
+    files = Dir[File.join(File.expand_path(dir), '*.target.json')]
+    files = files.first(limit) if limit
+
+    progressbar = TTY::ProgressBar.new("Validate: [:bar] :percent :current/:total", total: files.size)
+
+    Async do
+      files.each do |file|
+        Async do
+          input = JSON.parse(File.read(file))
+          response = ValidateReponse.new run(input), file: file
+
+          if block
+            block.call(response, progressbar)
+          else
+            unless response.success?
+              progressbar.log "validate failed".colorize(:red)
+              response.to_s.split("\n").each do |line|
+                progressbar.log line
+              end
+              progressbar.log "\n"
+            end
+          end
+          progressbar.advance
+        end
+      end
+    end.wait
+  end
+
+  class ValidateReponse
+    attr_reader :response, :body, :file
+
+    def initialize(response, file: nil)
+      @response = response
+      @body = JSON.parse(response.body)
+      @file = file
+    end
+
+    def success?
+      @response.status == 200 && @body["success"]
+    end
+
+    def message
+      @body["message"]
+    end
+
+    def data
+      @body["data"]
+    end
+
+    def to_s
+      s = "#{'success:'.colorize(:bold_blue)} #{success? ? 'true'.colorize(:green) : 'false'.colorize(:red)}"
+      s += "\n#{'file:'.colorize(:bold_blue)} #{file.colorize(:yellow)}"
+      s += "\n#{'message:'.colorize(:bold_blue)} #{message.colorize(:yellow)}" unless success?
+      s += "\n\n"
+    end
+  end
+
+end

data/lib/kaba/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Kaba
+  VERSION = "0.1.0"
+end

data/lib/kaba.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require_relative "kaba/version"
+require_relative "kaba/dataset"
+require_relative "kaba/dataset_source"
+require_relative "kaba/prompt"
+require_relative "kaba/validate"
+
+module Kaba
+  class Error < StandardError; end
+  # Your code goes here...
+end

data/sig/kaba.rbs

@@ -0,0 +1,4 @@
+module Kaba
+  VERSION: String
+  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+end

metadata

@@ -0,0 +1,128 @@
+--- !ruby/object:Gem::Specification
+name: kaba
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- MJ
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2024-11-13 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: async
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.20'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.20'
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.12'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.12'
+- !ruby/object:Gem::Dependency
+  name: async-http-faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.19.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.19.0
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: tty-progressbar
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.18.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.18.3
+description: 用来做数据集的工具
+email:
+- tywf91@gmail.com
+executables:
+- kaba
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- Rakefile
+- exe/kaba
+- kaba.gemspec
+- lib/kaba.rb
+- lib/kaba/_DPodfile_
+- lib/kaba/dataset.rb
+- lib/kaba/dataset_source.rb
+- lib/kaba/prompt.rb
+- lib/kaba/validate.rb
+- lib/kaba/version.rb
+- sig/kaba.rbs
+homepage: https://github.com/mjason/kaba.git
+licenses: []
+metadata:
+  homepage_uri: https://github.com/mjason/kaba.git
+  source_code_uri: https://github.com/mjason/kaba.git
+  changelog_uri: https://github.com/mjason/kaba.git
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.3.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.3
+signing_key: 
+specification_version: 4
+summary: 用来做数据集的工具
+test_files: []