rbs_goose 0.1.1 → 0.2.0

data/sig/rbs_goose/configuration.rbs

@@ -1,14 +1,22 @@
 class RbsGoose::Configuration
-  @llm: Langchain::LLM::OpenAI
-  @template: Langchain::Prompt::FewShotPromptTemplate
-  def initialize: () ?{ (RbsGoose::Configuration)[self: RbsGoose::Configuration] -> void } -> void
-  attr_accessor llm: Langchain::LLM::OpenAI
-  attr_accessor instruction: String
-  attr_accessor example_groups: Array[untyped]
-  attr_accessor template_class: singleton(RbsGoose::Templates::DefaultTemplate)
-  def use_open_ai: (String open_ai_access_token, ?default_options: ::Hash[untyped, untyped]) -> untyped
-  def template: () -> RbsGoose::Templates::DefaultTemplate
-  private def default_template_class: () -> singleton(RbsGoose::Templates::DefaultTemplate)
-  private def default_instruction: () -> String
-  private def default_example_groups: () -> Array[untyped]
+  @llm: LLMConfig
+  extend Forwardable
+  class LLMConfig
+  end
+  class TemplateConfig
+    def build_template: () -> Templates::Base
+  end
+  def initialize: () ?{ () -> void } -> void
+  attr_accessor llm: LLMConfig
+  attr_accessor infer_template: TemplateConfig
+  attr_accessor fix_error_template: TemplateConfig
+  def use_open_ai: (String open_ai_access_token, ?model_name: String, ?mode: Symbol, ?llm_options: Hash[Symbol, untyped], ?default_options: Hash[Symbol, untyped]) -> void
+  def use_anthropic: (String access_token, ?model_name: String, ?mode: Symbol, ?default_options: Hash[Symbol, untyped]) -> void
+  def use_ollama: (?url: String, ?model_name: String, ?mode: Symbol, ?default_options: Hash[Symbol, untyped]) -> void
+  private def default_infer_template: () -> TemplateConfig
+  private def default_fix_error_template: () -> TemplateConfig
+  private def default_infer_instruction: () -> String
+  private def default_infer_example_groups: () -> Array[RbsGoose::IO::ExampleGroup]
+  private def default_fix_error_instruction: () -> String
+  private def default_fix_error_example_groups: () -> Array[RbsGoose::IO::ExampleGroup]
 end

data/sig/rbs_goose/io/example.rbs

@@ -1,9 +1,9 @@
 class RbsGoose::IO::Example
-  @typed_ruby: RbsGoose::IO::TypedRuby
-  @refined_rbs: RbsGoose::IO::File
-  def self.from_path: (ruby_path: String, rbs_path: String, refined_rbs_dir: String, base_path: untyped) -> RbsGoose::IO::Example
-  def initialize: (typed_ruby: RbsGoose::IO::TypedRuby, refined_rbs: RbsGoose::IO::File) -> void
-  def to_h: () -> { typed_ruby: RbsGoose::IO::TypedRuby, refined_rbs: RbsGoose::IO::File }
-  attr_reader typed_ruby: RbsGoose::IO::TypedRuby
-  attr_reader refined_rbs: RbsGoose::IO::File
+  @typed_ruby: TypedRuby
+  @refined_rbs: File
+  def self.from_path: (ruby_path: String, rbs_path: String, refined_rbs_dir: String, base_path: String) -> Example
+  def initialize: (typed_ruby: TypedRuby, refined_rbs: File) -> void
+  def to_h: () -> { typed_ruby: TypedRuby, refined_rbs: File }
+  attr_reader typed_ruby: TypedRuby
+  attr_reader refined_rbs: File
 end

data/sig/rbs_goose/io/example_group.rbs

@@ -1,8 +1,10 @@
-class RbsGoose::IO::ExampleGroup < ::Array[RbsGoose::IO::Example]
-  self.@default_examples: Hash[Symbol, Array[RbsGoose::IO::Example]]
-  def self.load_from: (untyped base_path, ?code_dir: ::String, ?sig_dir: ::String, ?refined_dir: ::String) -> RbsGoose::IO::ExampleGroup
-  def self.default_examples: () -> Hash[Symbol, Array[RbsGoose::IO::Example]]
-  private def self.to_rbs_path: (String, String) -> String
-  def to_target_group: () -> RbsGoose::IO::TargetGroup
-  def to_refined_rbs_list: () -> Array[RbsGoose::IO::File]
+class RbsGoose::IO::ExampleGroup < ::Array[Example]
+  self.@default_examples: Hash[Symbol, ExampleGroup]
+  attr_accessor error_messages: String?
+  def self.load_from: (String base_path, ?code_dir: String, ?sig_dir: String, ?refined_dir: String) -> ExampleGroup
+  def self.default_examples: () -> Hash[Symbol, ExampleGroup]
+  private def self.load_example: (String base_path, String code_dir, String path, String refined_dir, String sig_dir) -> Example
+  private def self.to_rbs_path: (String path, String sig_dir) -> String
+  def to_target_group: () -> TargetGroup
+  def to_refined_rbs_list: () -> Array[File]
 end

data/sig/rbs_goose/io/file.rbs

@@ -1,19 +1,19 @@
 class RbsGoose::IO::File
   @path: String
-  @base_path: untyped
+  @base_path: String?
   @content: String
-  @type: :ruby | :rbs
-  MARKDOWN_REGEXP: ::Regexp
-  def self.from_markdown: (String) -> RbsGoose::IO::File
-  def initialize: (path: String, ?content: String, ?base_path: untyped) -> void
+  @type: Symbol
+  MARKDOWN_REGEXP: Regexp
+  def self.from_markdown: (String markdown) -> File
+  def initialize: (path: String, ?content: String?, ?base_path: String?) -> void
   def load_content: () -> String
   def absolute_path: () -> String
-  def type: () -> (:ruby | :rbs)
+  def type: () -> Symbol
   def to_s: () -> String
-  def content=: (String) -> String
-  def write: () -> untyped
-  def ==: (untyped) -> untyped
+  def content=: (String content) -> String
+  def write: () -> void
+  def ==: (Object other) -> bool
   attr_reader path: String
   attr_reader content: String
-  attr_reader base_path: untyped
+  attr_reader base_path: String?
 end

data/sig/rbs_goose/io/target_group.rbs

@@ -1,3 +1,3 @@
-class RbsGoose::IO::TargetGroup < ::Array[RbsGoose::IO::TypedRuby]
-  def self.load_from: (untyped base_path, ?code_dir: ::String, ?sig_dir: ::String) -> RbsGoose::IO::TargetGroup
+class RbsGoose::IO::TargetGroup < ::Array[TypedRuby]
+  def self.load_from: (String base_path, ?code_dir: String, ?sig_dir: String) -> TargetGroup
 end

data/sig/rbs_goose/io/typed_ruby.rbs

@@ -1,9 +1,9 @@
 class RbsGoose::IO::TypedRuby
-  @ruby: RbsGoose::IO::File
-  @rbs: RbsGoose::IO::File | nil
-  def self.from_path: (ruby_path: String, rbs_path: String, base_path: untyped) -> RbsGoose::IO::TypedRuby
-  def initialize: (ruby: RbsGoose::IO::File, rbs: RbsGoose::IO::File | nil) -> void
-  def to_s: () -> (untyped | String)
-  attr_reader ruby: RbsGoose::IO::File
-  attr_reader rbs: RbsGoose::IO::File | nil
+  @ruby: File
+  @rbs: File?
+  def self.from_path: (ruby_path: String, rbs_path: String, base_path: String) -> TypedRuby
+  def initialize: (ruby: File, rbs: File?) -> void
+  def to_s: () -> String
+  attr_reader ruby: File
+  attr_reader rbs: File?
 end

data/sig/rbs_goose/io.rbs

@@ -1,2 +1,2 @@
 module RbsGoose::IO
-end
+end

data/sig/rbs_goose/orthoses/fix_error.rbs

@@ -0,0 +1,11 @@
+# Call RbsGoose::TypeInferrer to fix steep type errors.
+#   use TbsGoose::Orthoses::FixError, code_dir: 'lib', sig_dir: 'sig' do |config|
+#     config.use_open_ai('open_ai_access_token')
+#   end
+class RbsGoose::Orthoses::FixError
+  @loader: Proc
+  @code_dir: String
+  @sig_dir: String
+  def initialize: (loader: Proc, ?code_dir: String, ?sig_dir: String) ?{ () -> void } -> void
+  def call: () -> untyped
+end

data/sig/rbs_goose/orthoses/infer.rbs

@@ -0,0 +1,11 @@
+# Call RbsGoose::TypeInferrer to infer RBS type signatures.
+#   use TbsGoose::Orthoses::Infer, code_dir: 'lib', sig_dir: 'sig' do |config|
+#     config.use_open_ai('open_ai_access_token')
+#   end
+class RbsGoose::Orthoses::Infer
+  @loader: Proc
+  @code_dir: String
+  @sig_dir: String
+  def initialize: (loader: Proc, ?code_dir: String, ?sig_dir: String) ?{ () -> void } -> void
+  def call: () -> untyped
+end

data/sig/rbs_goose/orthoses.rbs

@@ -0,0 +1,2 @@
+module RbsGoose::Orthoses
+end

data/sig/rbs_goose/templates/base.rbs

@@ -0,0 +1,18 @@
+class RbsGoose::Templates::Base
+  @instruction: String
+  @example_groups: Array[RbsGoose::IO::ExampleGroup]
+  def initialize: (instruction: String, example_groups: Array[RbsGoose::IO::ExampleGroup]) -> void
+  def format: (**untyped args) -> String
+  def format_system_prompt: () -> String
+  def format_user_prompt: (**untyped args) -> String
+  def parse_result: (String result) -> Array[RbsGoose::IO::File]
+  private def input_template_string: () -> String
+  private def input_variables: () -> Array[String]
+  private def transform_example_group: (RbsGoose::IO::ExampleGroup example_group) -> Hash[String, String]
+  private def format_args: (untyped args) -> Hash[String, String]
+  private def prefix: () -> String
+  private def suffix: () -> String
+  private def example_prompt: () -> Langchain::Prompt::PromptTemplate
+  private attr_reader instruction: String
+  private attr_reader example_groups: Array[RbsGoose::IO::ExampleGroup]
+end

data/sig/rbs_goose/templates/fix_error_template.rbs

@@ -0,0 +1,6 @@
+class RbsGoose::Templates::FixErrorTemplate < ::RbsGoose::Templates::Base
+  private def input_variables: () -> Array["typed_ruby_list", "error_messages"]
+  private def input_template_string: () -> String
+  private def format_args: (untyped args) -> { typed_ruby_list: String, error_messages: String }
+  private def transform_example_group: (RbsGoose::IO::ExampleGroup example_group) -> { typed_ruby_list: String, error_messages: String?, refined_rbs_list: String }
+end

data/sig/rbs_goose/templates/infer_template.rbs

@@ -0,0 +1,6 @@
+class RbsGoose::Templates::InferTemplate < ::RbsGoose::Templates::Base
+  private def input_variables: () -> Array["typed_ruby_list"]
+  private def input_template_string: () -> String
+  private def format_args: (untyped args) -> { typed_ruby_list: String }
+  private def transform_example_group: (RbsGoose::IO::ExampleGroup example_group) -> { typed_ruby_list: String, refined_rbs_list: String }
+end

data/sig/rbs_goose/templates.rbs

@@ -1,2 +1,2 @@
 module RbsGoose::Templates
-end
+end

data/sig/rbs_goose/type_inferrer.rbs

@@ -1,3 +1,11 @@
 class RbsGoose::TypeInferrer
-  def infer: (untyped) -> untyped
+  def infer: (RbsGoose::IO::TargetGroup target_group) -> Array[RbsGoose::IO::File]
+  def fix_error: (RbsGoose::IO::TargetGroup target_group) -> Array[RbsGoose::IO::File]
+  private def call_llm: (RbsGoose::Templates::Base template, Hash[String, untyped] format_args) -> Array[RbsGoose::IO::File]
+  private def call_llm_complete: (Hash[String, untyped] format_args, RbsGoose::Templates::Base template) -> String
+  private def call_llm_chat: (Hash[String, untyped] format_args, RbsGoose::Templates::Base template) -> String
+  private def llm_debug: (String prompt) { () -> Langchain::LLM::Result } -> Langchain::LLM::Result
+  private def steep_check: () -> String
+  private def io_stubs: () -> Array[StringIO]
+  private def disable_rainbow: () { () -> void } -> void
 end

data/sig/rbs_goose.rbs

@@ -1,12 +1,21 @@
 module RbsGoose
-  VERSION: "0.1.0"
-  self.@configuration: RbsGoose::Configuration
+  VERSION: String
+
+  self.@configuration: Configuration?
+
   extend Forwardable
-  def self.configure: () { (RbsGoose::Configuration)[self: RbsGoose::Configuration] -> void } -> void
-  def self.reset_configuration: () -> untyped
-  def self.run: (?code_dir: ::String, ?sig_dir: ::String, ?base_path: untyped) -> untyped
-  attr_reader self.configuration: RbsGoose::Configuration
-  def self.llm: () -> Langchain::LLM::OpenAI
-  def self.instruction: () -> String
-  def self.example_groups: () -> Array[untyped]
+
+  def self.configure: () { () -> void } -> Configuration
+
+  def self.reset_configuration: () -> nil
+
+  def self.run: (?code_dir: String, ?sig_dir: String, ?base_path: String) -> void
+
+  def self.fix_error: (?code_dir: String, ?sig_dir: String, ?base_path: String) -> void
+
+  def self.infer_template: () -> RbsGoose::Templates::InferTemplate
+
+  def self.fix_error_template: () -> RbsGoose::Templates::FixErrorTemplate
+
+  attr_reader self.configuration: Configuration?
 end

metadata

@@ -1,22 +1,82 @@
 --- !ruby/object:Gem::Specification
 name: rbs_goose
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - kokuyouwind
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-01-31 00:00:00.000000000 Z
+date: 2024-05-09 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.9.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.9.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 3.0.0
 - !ruby/object:Gem::Dependency
   name: langchainrb
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.8.2
+        version: 0.12.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.12.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+- !ruby/object:Gem::Dependency
+  name: steep
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.6.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.6.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+- !ruby/object:Gem::Dependency
+  name: anthropic
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.2.0
     - - "<"
       - !ruby/object:Gem::Version
         version: 1.0.0
@@ -26,10 +86,30 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.8.2
+        version: 0.2.0
     - - "<"
       - !ruby/object:Gem::Version
         version: 1.0.0
+- !ruby/object:Gem::Dependency
+  name: orthoses
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.13.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.13.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 2.0.0
 - !ruby/object:Gem::Dependency
   name: ruby-openai
   requirement: !ruby/object:Gem::Requirement
@@ -39,7 +119,7 @@ dependencies:
         version: 6.1.0
     - - "<"
       - !ruby/object:Gem::Version
-        version: 7.0.0
+        version: 8.0.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -49,7 +129,7 @@ dependencies:
         version: 6.1.0
     - - "<"
       - !ruby/object:Gem::Version
-        version: 7.0.0
+        version: 8.0.0
 description: RBS type inferrer with LLM
 email:
 - kokuyouwind@gmail.com
@@ -66,31 +146,62 @@ files:
 - Gemfile
 - Gemfile.lock
 - LICENSE.txt
-- README-EN-US.md
+- README-ja.md
 - README.md
 - Rakefile
 - Steepfile
 - assets/logo.svg
+- evaluations/01_rbs_goose/claude_haiku.log
+- evaluations/01_rbs_goose/claude_opus.log
+- evaluations/01_rbs_goose/claude_sonnet.log
+- evaluations/01_rbs_goose/gpt-3-5-turbo.log
+- evaluations/01_rbs_goose/gpt-4-turbo.log
+- examples/fix_errors/Steepfile
+- examples/fix_errors/error_messages.txt
+- examples/fix_errors/lib/email.rb
+- examples/fix_errors/lib/person.rb
+- examples/fix_errors/lib/phone.rb
+- examples/fix_errors/refined/sig/email.rbs
+- examples/fix_errors/refined/sig/person.rbs
+- examples/fix_errors/refined/sig/phone.rbs
+- examples/fix_errors/sig/email.rbs
+- examples/fix_errors/sig/person.rbs
+- examples/fix_errors/sig/phone.rbs
+- examples/rbs_samples/Steepfile
+- examples/rbs_samples/lib/email.rb
+- examples/rbs_samples/lib/person.rb
+- examples/rbs_samples/lib/phone.rb
+- examples/rbs_samples/refined/sig/email.rbs
+- examples/rbs_samples/refined/sig/person.rbs
+- examples/rbs_samples/refined/sig/phone.rbs
+- examples/rbs_samples/sig/email.rbs
+- examples/rbs_samples/sig/person.rbs
+- examples/rbs_samples/sig/phone.rbs
+- examples/rbs_syntax/Steepfile
+- examples/rbs_syntax/lib/a.rb
+- examples/rbs_syntax/lib/list.rb
+- examples/rbs_syntax/lib/main.rb
+- examples/rbs_syntax/refined/sig/a.rbs
+- examples/rbs_syntax/refined/sig/list.rbs
+- examples/rbs_syntax/refined/sig/main.rbs
+- examples/rbs_syntax/sig/a.rbs
+- examples/rbs_syntax/sig/list.rbs
+- examples/rbs_syntax/sig/main.rbs
 - lib/rbs_goose.rb
 - lib/rbs_goose/configuration.rb
-- lib/rbs_goose/examples/rbs_samples/Steepfile
-- lib/rbs_goose/examples/rbs_samples/lib/email.rb
-- lib/rbs_goose/examples/rbs_samples/lib/person.rb
-- lib/rbs_goose/examples/rbs_samples/lib/phone.rb
-- lib/rbs_goose/examples/rbs_samples/refined/sig/email.rbs
-- lib/rbs_goose/examples/rbs_samples/refined/sig/person.rbs
-- lib/rbs_goose/examples/rbs_samples/refined/sig/phone.rbs
-- lib/rbs_goose/examples/rbs_samples/sig/email.rbs
-- lib/rbs_goose/examples/rbs_samples/sig/person.rbs
-- lib/rbs_goose/examples/rbs_samples/sig/phone.rbs
 - lib/rbs_goose/io.rb
 - lib/rbs_goose/io/example.rb
 - lib/rbs_goose/io/example_group.rb
 - lib/rbs_goose/io/file.rb
 - lib/rbs_goose/io/target_group.rb
 - lib/rbs_goose/io/typed_ruby.rb
+- lib/rbs_goose/orthoses.rb
+- lib/rbs_goose/orthoses/fix_error.rb
+- lib/rbs_goose/orthoses/infer.rb
 - lib/rbs_goose/templates.rb
-- lib/rbs_goose/templates/default_template.rb
+- lib/rbs_goose/templates/base.rb
+- lib/rbs_goose/templates/fix_error_template.rb
+- lib/rbs_goose/templates/infer_template.rb
 - lib/rbs_goose/type_inferrer.rb
 - lib/rbs_goose/version.rb
 - rbs_collection.lock.yaml
@@ -105,8 +216,13 @@ files:
 - sig/rbs_goose/io/file.rbs
 - sig/rbs_goose/io/target_group.rbs
 - sig/rbs_goose/io/typed_ruby.rbs
+- sig/rbs_goose/orthoses.rbs
+- sig/rbs_goose/orthoses/fix_error.rbs
+- sig/rbs_goose/orthoses/infer.rbs
 - sig/rbs_goose/templates.rbs
-- sig/rbs_goose/templates/default_template.rbs
+- sig/rbs_goose/templates/base.rbs
+- sig/rbs_goose/templates/fix_error_template.rbs
+- sig/rbs_goose/templates/infer_template.rbs
 - sig/rbs_goose/type_inferrer.rbs
 - sig_ext/class.rbs
 - sig_ext/langchain/llm.rbs
@@ -126,14 +242,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.0.0
+      version: 3.3.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.3
+rubygems_version: 3.5.9
 signing_key:
 specification_version: 4
 summary: RBS type inferrer with LLM

data/README-EN-US.md

@@ -1,85 +0,0 @@
-<p align="center">
-  <img src="https://raw.githubusercontent.com/kokuyouwind/rbs_goose/main/assets/logo.svg" alt="RuboCop Logo"/>
-</p>
-
-[![en-US README](https://img.shields.io/badge/Multilingual_README-en--US-blue.svg)](/README-EN-US.md)
-[![en-JP README](https://img.shields.io/badge/Multilingual_README-ja--JP-orangered.svg)](/README.md)
-
-[![Ruby](https://github.com/kokuyouwind/rbs_goose/actions/workflows/main.yml/badge.svg)](https://github.com/kokuyouwind/rbs_goose/actions/workflows/main.yml)
-
-RBS Goose is a tool that uses ChatGPT and other large language models to infer the RBS signature of Ruby code.
-
-> [!CAUTION]
-> Currently undergoing technical validation, so you may get little or no output of the appropriate types.
-> Also, the tool uses ChatGPT API and other tools to infer RBS signatures, so depending on the size of your code, the usage fee may be expensive.
-
-## Installation
-
-```bash
-$ gem install rbs_goose
-## Depending on the LangChain LLM you are using, install the corresponding gem
-$ gem install ruby-openai
-```
-
-If you use `bundler`, add the following to your `Gemfile` instead.
-
-```ruby
-gem 'rbs_goose'.
-# Depending on your LangChain LLM, install the corresponding gem
-gem 'ruby-openai'
-```
-
-## Usage
-
-Currently, command line tools are not yet available, so please call `RbsGoose.run` directly from `Rakefile` or other sources.
-
-If you use [OpenAI API](https://openai.com/blog/openai-api), do the following.
-
-```ruby
-require 'rbs_goose'
-require 'openai'
-
-desc 'refine RBS files in sig directory' ```ruby require 'rbs_goose' require 'openai'
-task :refine do
-  RbsGoose.configure do |c|
-    c.use_open_ai(ENV.fetch('OPENAI_ACCESS_TOKEN'))
-  end
-  RbsGoose.run
-end
-```
-
-Running this task will reference the Ruby code under `lib` and the RBS signature under `sig` and override the guessed signature.
-
-To change the target directory, specify the following arguments to `RbsGoose.run`.
-
-```ruby
-RbsGoose.run(code_dir: 'app', sig_dir: 'types', base_path: Rails.root)
-```
-
-[Langchain.rb](https://github.com/andreibondarev/langchainrb) is used to invoke the large language model.
-
-To use other large-scale language models, set `llm` directly as follows.
-
-```ruby
-RbsGoose.configure do |c|
-  c.llm = Langchain::LLM::GooglePalm.new(api_key: ENV["GOOGLE_PALM_API_KEY"])
-end
-```
-
-## Development
-
-After checking out the repository, run bin/setup to install the dependencies. Next, run `rake spec` to run the tests. You can also try running `bin/console` as an interactive prompt.
-
-To install the gem on your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb` and run `bundle exec rake release`. This will create a Git tag for the version, push the Git commit and the created tag, and push the .gem file to rubygems.org.
-
-## Contributing
-
-Bug reports and pull requests are accepted on GitHub at https://github.com/kokuyouwind/rbs_goose. The project aims to be a safe and welcoming space for collaborative work, and contributors are expected to follow the [CODE OF CONDUCT](/CODE_OF_CONDUCT.md).
-
-## License
-
-This gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
-
-## Code of Conduct
-
-Everyone involved with the RbsGoose project codebase, issue tracker, chat room, and mailing lists is expected to follow the [CODE OF CONDUCT](/CODE_OF_CONDUCT.md).

data/lib/rbs_goose/templates/default_template.rb

@@ -1,49 +0,0 @@
-# frozen_string_literal: true
-
-require 'langchain'
-
-module RbsGoose
-  module Templates
-    class DefaultTemplate
-      def initialize(instruction:, example_groups:)
-        @template = Langchain::Prompt::FewShotPromptTemplate.new(
-          prefix: instruction,
-          suffix: "#{input_template_string}\n",
-          example_prompt: example_prompt,
-          examples: example_groups.map { transform_example_group(_1) },
-          input_variables: %w[typed_ruby_list]
-        )
-      end
-
-      def format(typed_ruby_list)
-        template.format(typed_ruby_list: typed_ruby_list.join("\n"))
-      end
-
-      def parse_result(result)
-        result.scan(/```.+?```/m).map { IO::File.from_markdown(_1) }
-      end
-
-      private
-
-      attr_reader :template
-
-      def example_prompt
-        Langchain::Prompt::PromptTemplate.new(
-          template: "#{input_template_string}\n{refined_rbs_list}",
-          input_variables: %w[typed_ruby_list refined_rbs_list]
-        )
-      end
-
-      def input_template_string
-        "========Input========\n{typed_ruby_list}\n\n========Output========"
-      end
-
-      def transform_example_group(example_group)
-        {
-          typed_ruby_list: example_group.map(&:typed_ruby).join("\n"),
-          refined_rbs_list: example_group.map(&:refined_rbs).join("\n")
-        }
-      end
-    end
-  end
-end

data/sig/rbs_goose/templates/default_template.rbs

@@ -1,10 +0,0 @@
-class RbsGoose::Templates::DefaultTemplate
-  @template: Langchain::Prompt::FewShotPromptTemplate
-  def initialize: (instruction: String, example_groups: Array[untyped]) -> void
-  def format: (Array[untyped]) -> untyped
-  def parse_result: (untyped) -> untyped
-  private attr_reader template: Langchain::Prompt::FewShotPromptTemplate
-  private def example_prompt: () -> Langchain::Prompt::PromptTemplate
-  private def input_template_string: () -> String
-  private def transform_example_group: (untyped) -> { typed_ruby_list: untyped, refined_rbs_list: untyped }
-end