kaba 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0add8ecf9ac4613decb084b678c044809896db95136b59e24602ae47a426d448
-  data.tar.gz: 2dda3c5fa0152b4928c4a2fadc11b18ed429cc724e87c5b38c794083cb731c69
+  metadata.gz: 986fbab4a45443b0ffc9676cbe6ea2857a7eaec7ba74294561c4d8abe8bf8869
+  data.tar.gz: e5a2254a5f2dc1778083df00141c59654749e2538eb695384c3528c49a8820dd
 SHA512:
-  metadata.gz: 622be2e55740919481236e525f2c2045f9f633eccfa36363a55d959c8d25cd859cfc349e6668df429571052d9501077f8068c35e1a72bac25ced650188c3407e
-  data.tar.gz: 56c1a740604cb48c6bfb14f5c93854597fa53922c2befd03e58cf192d29e2b2b1ac87d9480255ae6c6ef175d2652299b67b0cba8cb1e0deccf405adce9efa22b
+  metadata.gz: 42b2a10981e7133b1b34b5a9e38ea5c12a0b7965ce4b12b689a8840b2d45b672ba1b04d6a6462b2d4f93ce8fa2d34fbd77769ee09725801f217f3f25145a5a8c
+  data.tar.gz: ba4cfed522723fca973d6c265eeb5e9d97edad4326bbbd36086cf385bae31ed8d37b7607e368dd40129311d37f43711e5749fd88cbd79e29e3c0f99a74592b97

data/exe/kaba

@@ -6,14 +6,42 @@ require 'dotenv'
 Dotenv.load
 
 require "kaba"
+require 'thor'
 
-if ARGV[0] == 'test'
-  # 运行测试程序
-  load DatasetSource.testfile
-elsif ARGV[0] == 'version'
-  # 显示版本号
-  puts Kaba::VERSION
-else
-  # 运行 DPodfile 文件，DPodfile 是一个 Ruby 文件
-  load DatasetSource.podfile
-end
+class KabaCommand < Thor
+  desc "test", "Run the test program"
+  def test
+    load DatasetSource.testfile
+  end
+
+  desc "version", "Show the version"
+  def version
+    puts Kaba::VERSION
+  end
+
+  desc "exec FILE", "Run a specified Ruby file"
+  def exec(file)
+    load file
+  end
+
+  desc "create", "Run the DPodfile (default behavior)"
+  def create
+    load DatasetSource.podfile
+  end
+
+  desc "validate", "Validate the environment"
+  def validate
+    Boot.validate
+  end
+
+  # 默认行为处理（兼容现有行为）
+  def self.start(args = ARGV, config = {})
+    if args.empty?
+      new.create
+    else
+      super
+    end
+  end
+end
+
+KabaCommand.start

data/kaba.gemspec

@@ -40,6 +40,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency "dotenv", "~> 3.1"
   spec.add_dependency "ruby-openai", "~> 7.3"
   spec.add_dependency "json-repair", "~> 0.2.0"
+  spec.add_dependency "thor", '~> 1.3', '>= 1.3.2'
 
   # For more information and examples about making a new gem, check out our
   # guide at: https://bundler.io/guides/creating_gem.html

data/lib/kaba/boot.rb

@@ -0,0 +1,32 @@
+module Boot
+  module_function
+  def init
+    # 判断 .env 文件是否存在, 不存在则创建
+    unless File.exist?(".env")
+      File.open(".env", "w") do |f|
+        f.write("LISA_ACCESS_TOKEN=<Lisa access token, visit: https://platform.listenai.com/>\n")
+        f.write("JUDGE_ACCCESS_TOKEN=<Judge access token, visit: https://platform.listenai.com/>\n")
+
+        f.write("; if you are using a different typechat endpoint, please fill in the following line\n")
+        f.write("; LISA_TYPECHAT_ENDPOINT=https://lisa-typechat.listenai.com\n")
+
+        f.write("; if you are using a different LLM endpoint, please fill in the following line\n")
+        f.write("; LISA_LLM_URI_BASE=https://api.listenai.com\n")
+
+        f.write("; if you are using a different JUDGE LLM endpoint, please fill in the following line\n")
+        f.write("; JUDGE_LLM_URI_BASE=https://api.listenai.com\n")
+      end
+
+      puts "Created .env file, please fill in the necessary information."
+    end
+  end
+
+  def validate
+    puts "Validating..."
+    Application.env!('LISA_ACCESS_TOKEN')
+    Application.env!('JUDGE_ACCCESS_TOKEN')
+    puts "Validation passed."
+  rescue => e
+    puts "Validation failed: #{e.message}"
+  end
+end

data/lib/kaba/version.rb

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

data/lib/kaba.rb

@@ -19,6 +19,7 @@ require_relative "kaba/prompt"
 require_relative "kaba/validate"
 require_relative "kaba/judge"
 require_relative "kaba/test_runner"
+require_relative "kaba/boot"
 
 module Kaba
   class Error < StandardError; end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kaba
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - MJ
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-11-18 00:00:00.000000000 Z
+date: 2024-11-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: async
@@ -122,6 +122,26 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.2.0
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.3.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.3.2
 description: 用来做数据集的工具
 email:
 - tywf91@gmail.com
@@ -140,6 +160,7 @@ files:
 - lib/kaba/_DPodfile_
 - lib/kaba/_DTestfile_
 - lib/kaba/application.rb
+- lib/kaba/boot.rb
 - lib/kaba/dataset.rb
 - lib/kaba/dataset_source.rb
 - lib/kaba/json.rb