rbs_goose 0.1.0

data/rbs_collection.lock.yaml

@@ -0,0 +1,188 @@
+---
+path: ".gem_rbs_collection"
+gems:
+- name: activesupport
+  version: '7.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: addressable
+  version: '2.8'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: ast
+  version: '2.4'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: baran
+  version: 0.1.10
+  source:
+    type: rubygems
+- name: base64
+  version: '0'
+  source:
+    type: stdlib
+- name: bigdecimal
+  version: '0'
+  source:
+    type: stdlib
+- name: concurrent-ruby
+  version: '1.1'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: connection_pool
+  version: '2.4'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: date
+  version: '0'
+  source:
+    type: stdlib
+- name: erb
+  version: '0'
+  source:
+    type: stdlib
+- name: faraday
+  version: '2.5'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: fileutils
+  version: '0'
+  source:
+    type: stdlib
+- name: forwardable
+  version: '0'
+  source:
+    type: stdlib
+- name: i18n
+  version: '1.10'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: json
+  version: '0'
+  source:
+    type: stdlib
+- name: logger
+  version: '0'
+  source:
+    type: stdlib
+- name: minitest
+  version: '0'
+  source:
+    type: stdlib
+- name: monitor
+  version: '0'
+  source:
+    type: stdlib
+- name: mutex_m
+  version: '0'
+  source:
+    type: stdlib
+- name: parallel
+  version: '1.20'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: parser
+  version: '3.2'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rainbow
+  version: '3.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rake
+  version: '13.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: regexp_parser
+  version: '2.8'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rubocop
+  version: '1.57'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rubocop-ast
+  version: '1.30'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: ec140aff951e3af39846eed4525ba5d040d294fa
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: securerandom
+  version: '0'
+  source:
+    type: stdlib
+- name: singleton
+  version: '0'
+  source:
+    type: stdlib
+- name: tiktoken_ruby
+  version: 0.0.6
+  source:
+    type: rubygems
+- name: time
+  version: '0'
+  source:
+    type: stdlib
+- name: timeout
+  version: '0'
+  source:
+    type: stdlib
+- name: uri
+  version: '0'
+  source:
+    type: stdlib
+gemfile_lock_path: Gemfile.lock

data/rbs_collection.yaml

@@ -0,0 +1,19 @@
+# Download sources
+sources:
+  - type: git
+    name: ruby/gem_rbs_collection
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    revision: main
+    repo_dir: gems
+
+# You can specify local directories as sources also.
+# - type: local
+#   path: path/to/your/local/repository
+
+# A directory to install the downloaded RBSs
+path: .gem_rbs_collection
+
+# gems:
+#   # If you want to avoid installing rbs files for gems, you can specify them here.
+#   - name: GEM_NAME
+#     ignore: true

data/renovate.json

@@ -0,0 +1,15 @@
+{
+  "$schema": "https://docs.renovatebot.com/renovate-schema.json",
+  "extends": [
+    "config:recommended"
+  ],
+  "labels": [
+    "dependencies"
+  ],
+  "packageRules": [
+    {
+      "groupName": "rubocop",
+      "matchPackagePatterns": [ "rubocop" ]
+    }
+  ]
+}

data/sig/rbs_goose/configuration.rbs

@@ -0,0 +1,14 @@
+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]
+end

data/sig/rbs_goose/error.rbs

@@ -0,0 +1,2 @@
+class RbsGoose::Error < ::StandardError
+end

data/sig/rbs_goose/io/example.rbs

@@ -0,0 +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
+end

data/sig/rbs_goose/io/example_group.rbs

@@ -0,0 +1,8 @@
+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]
+end

data/sig/rbs_goose/io/file.rbs

@@ -0,0 +1,19 @@
+class RbsGoose::IO::File
+  @path: String
+  @base_path: untyped
+  @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
+  def load_content: () -> String
+  def absolute_path: () -> String
+  def type: () -> (:ruby | :rbs)
+  def to_s: () -> String
+  def content=: (String) -> String
+  def write: () -> untyped
+  def ==: (untyped) -> untyped
+  attr_reader path: String
+  attr_reader content: String
+  attr_reader base_path: untyped
+end

data/sig/rbs_goose/io/target_group.rbs

@@ -0,0 +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
+end

data/sig/rbs_goose/io/typed_ruby.rbs

@@ -0,0 +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
+end

data/sig/rbs_goose/io.rbs

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

data/sig/rbs_goose/templates/default_template.rbs

@@ -0,0 +1,10 @@
+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

data/sig/rbs_goose/templates.rbs

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

data/sig/rbs_goose/type_inferrer.rbs

@@ -0,0 +1,3 @@
+class RbsGoose::TypeInferrer
+  def infer: (untyped) -> untyped
+end

data/sig/rbs_goose.rbs

@@ -0,0 +1,12 @@
+module RbsGoose
+  VERSION: "0.1.0"
+  self.@configuration: RbsGoose::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]
+end

data/sig_ext/class.rbs

@@ -0,0 +1,3 @@
+class Class
+  def self.def_delegators: (*untyped) -> untyped
+end

data/sig_ext/langchain/llm.rbs

@@ -0,0 +1,19 @@
+module Langchain
+  module LLM
+    class OpenAI
+      def initialize: (untyped) ->void
+      def complete: (untyped) -> untyped
+    end
+  end
+
+  module Prompt
+    class PromptTemplate
+      def initialize: (template: String, input_variables: Array[String]) -> void
+    end
+
+    class FewShotPromptTemplate
+      def initialize: (prefix: String, suffix: String, example_prompt: ::Langchain::Prompt::PromptTemplate, examples: Array[untyped], input_variables: Array[String]) -> void
+      def format: (untyped) -> untyped
+    end
+  end
+end

metadata

@@ -0,0 +1,128 @@
+--- !ruby/object:Gem::Specification
+name: rbs_goose
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- kokuyouwind
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-01-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: langchainrb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.8.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.8.2
+- !ruby/object:Gem::Dependency
+  name: ruby-openai
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 6.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 6.1.0
+description: RBS type inferrer with LLM
+email:
+- kokuyouwind@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".env.sample"
+- ".rspec"
+- ".rubocop.yml"
+- ".ruby-version"
+- CHANGELOG.md
+- CODE_OF_CONDUCT.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README-EN-US.md
+- README.md
+- Rakefile
+- Steepfile
+- assets/logo.svg
+- 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/templates.rb
+- lib/rbs_goose/templates/default_template.rb
+- lib/rbs_goose/type_inferrer.rb
+- lib/rbs_goose/version.rb
+- rbs_collection.lock.yaml
+- rbs_collection.yaml
+- renovate.json
+- sig/rbs_goose.rbs
+- sig/rbs_goose/configuration.rbs
+- sig/rbs_goose/error.rbs
+- sig/rbs_goose/io.rbs
+- sig/rbs_goose/io/example.rbs
+- sig/rbs_goose/io/example_group.rbs
+- sig/rbs_goose/io/file.rbs
+- sig/rbs_goose/io/target_group.rbs
+- sig/rbs_goose/io/typed_ruby.rbs
+- sig/rbs_goose/templates.rbs
+- sig/rbs_goose/templates/default_template.rbs
+- sig/rbs_goose/type_inferrer.rbs
+- sig_ext/class.rbs
+- sig_ext/langchain/llm.rbs
+homepage: https://github.com/kokuyouwind/rbs_goose
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/kokuyouwind/rbs_goose
+  source_code_uri: https://github.com/kokuyouwind/rbs_goose
+  changelog_uri: https://github.com/kokuyouwind/rbs_goose/CHANGELOG.md
+  rubygems_mfa_required: 'true'
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.0.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: RBS type inferrer with LLM
+test_files: []