mj 0.6.2 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4f73fffd1a785c75120ffed30d970ffbc9d64f525a5a5729c814a785f7c0868e
-  data.tar.gz: 8a09ecd20bb4bae4e54f07d30f09c9fa534b4d2c6090d87353c3297cb1bb2210
+  metadata.gz: 8b15f4c4efd81a03bf545e20a2bebb6beff7a12612c470173073f35c5b67cab2
+  data.tar.gz: 6c61d4acd6cc0d8590835bfd038fe0e291df05fd3100db7fff41f189eeade3fe
 SHA512:
-  metadata.gz: 90f2712ab6a67d83f45ce62f4a4213e0b85404ead8f218263771095ef09121dcaf29732e81a17e459be7d2788bb836f172ec34bb6d90e3e697f49599075378eb
-  data.tar.gz: d05317d65dd3fb88eb5379f3e0e4439a2264c5b40d6a2276fec219058668748f852ea4474468fc7a56f527357ea5b8808d15a541bbb9558203ca6602edaec50e
+  metadata.gz: 7d1017ad9f4b623c86d864e16c5110b9ecc1e131bbe96e27f23e0782b79ce209257972837a857a6557877a484dd0de834c2bf0c5be6c38d2ff5c67866243872c
+  data.tar.gz: 77da1cdfb825c43e50d8ae4cf7b203f585fc7c4be33c1121ad47e46c30befe773ec9cfe345a980112e02b625e0f3a82bd4158ef5d15438e8f46f46816b416529

data/Gemfile

@@ -16,3 +16,4 @@ gem "rubocop-rake"
 gem "rubocop-rspec"
 gem "simplecov", require: false
 gem "simplecov-lcov", require: false
+gem "uri", "~> 0.13"

data/Gemfile.lock

@@ -1,62 +1,103 @@
 PATH
   remote: .
   specs:
-    mj (0.6.2)
+    mj (0.8.0)
+      koine-rest_client
+      mj-hash_utils
+      ruby-openai (~> 6.2.0)
       thor (~> 1.2.1)
 
 GEM
   remote: https://rubygems.org/
   specs:
+    addressable (2.8.6)
+      public_suffix (>= 2.0.2, < 6.0)
     ast (2.4.2)
     awesome_print (1.9.2)
     coderay (1.1.3)
-    diff-lcs (1.5.0)
+    diff-lcs (1.5.1)
     docile (1.4.0)
+    event_stream_parser (0.3.0)
+    faraday (2.9.0)
+      faraday-net_http (>= 2.0, < 3.2)
+    faraday-multipart (1.0.4)
+      multipart-post (~> 2)
+    faraday-net_http (3.1.0)
+      net-http
+    httparty (0.21.0)
+      mini_mime (>= 1.0.0)
+      multi_xml (>= 0.5.2)
+    json (2.7.1)
+    koine-rest_client (1.1.1)
+      addressable (>= 2.3)
+      httparty (>= 0.17.0)
     koine-test_runner (0.4.0)
+    language_server-protocol (3.17.0.3)
     method_source (1.0.0)
-    parallel (1.21.0)
-    parser (3.1.1.0)
+    mini_mime (1.1.5)
+    mj-hash_utils (0.1.0)
+    multi_xml (0.6.0)
+    multipart-post (2.4.0)
+    net-http (0.4.1)
+      uri
+    parallel (1.24.0)
+    parser (3.3.0.5)
       ast (~> 2.4.1)
+      racc
     pry (0.14.2)
       coderay (~> 1.1)
       method_source (~> 1.0)
+    public_suffix (5.0.4)
+    racc (1.7.3)
     rainbow (3.1.1)
-    rake (13.0.6)
-    regexp_parser (2.2.1)
-    rexml (3.2.5)
-    rspec (3.11.0)
-      rspec-core (~> 3.11.0)
-      rspec-expectations (~> 3.11.0)
-      rspec-mocks (~> 3.11.0)
-    rspec-core (3.11.0)
-      rspec-support (~> 3.11.0)
-    rspec-expectations (3.11.0)
+    rake (13.1.0)
+    regexp_parser (2.9.0)
+    rexml (3.2.6)
+    rspec (3.13.0)
+      rspec-core (~> 3.13.0)
+      rspec-expectations (~> 3.13.0)
+      rspec-mocks (~> 3.13.0)
+    rspec-core (3.13.0)
+      rspec-support (~> 3.13.0)
+    rspec-expectations (3.13.0)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.11.0)
-    rspec-mocks (3.11.0)
+      rspec-support (~> 3.13.0)
+    rspec-mocks (3.13.0)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.11.0)
-    rspec-support (3.11.0)
-    rubocop (1.25.1)
+      rspec-support (~> 3.13.0)
+    rspec-support (3.13.1)
+    rubocop (1.62.0)
+      json (~> 2.3)
+      language_server-protocol (>= 3.17.0)
       parallel (~> 1.10)
-      parser (>= 3.1.0.0)
+      parser (>= 3.3.0.2)
       rainbow (>= 2.2.2, < 4.0)
       regexp_parser (>= 1.8, < 3.0)
-      rexml
-      rubocop-ast (>= 1.15.1, < 2.0)
+      rexml (>= 3.2.5, < 4.0)
+      rubocop-ast (>= 1.31.1, < 2.0)
       ruby-progressbar (~> 1.7)
-      unicode-display_width (>= 1.4.0, < 3.0)
-    rubocop-ast (1.16.0)
-      parser (>= 3.1.1.0)
-    rubocop-performance (1.13.2)
-      rubocop (>= 1.7.0, < 2.0)
-      rubocop-ast (>= 0.4.0)
+      unicode-display_width (>= 2.4.0, < 3.0)
+    rubocop-ast (1.31.2)
+      parser (>= 3.3.0.4)
+    rubocop-capybara (2.20.0)
+      rubocop (~> 1.41)
+    rubocop-factory_bot (2.25.1)
+      rubocop (~> 1.41)
+    rubocop-performance (1.20.2)
+      rubocop (>= 1.48.1, < 2.0)
+      rubocop-ast (>= 1.30.0, < 2.0)
     rubocop-rake (0.6.0)
       rubocop (~> 1.0)
-    rubocop-rspec (2.9.0)
-      rubocop (~> 1.19)
-    ruby-progressbar (1.11.0)
-    simplecov (0.21.2)
+    rubocop-rspec (2.27.1)
+      rubocop (~> 1.40)
+      rubocop-capybara (~> 2.17)
+      rubocop-factory_bot (~> 2.22)
+    ruby-openai (6.2.0)
+      event_stream_parser (>= 0.3.0, < 1.0.0)
+      faraday (>= 1)
+      faraday-multipart (>= 1)
+    ruby-progressbar (1.13.0)
+    simplecov (0.22.0)
       docile (~> 1.1)
       simplecov-html (~> 0.11)
       simplecov_json_formatter (~> 0.1)
@@ -64,7 +105,8 @@ GEM
     simplecov-lcov (0.8.0)
     simplecov_json_formatter (0.1.4)
     thor (1.2.2)
-    unicode-display_width (2.1.0)
+    unicode-display_width (2.5.0)
+    uri (0.13.0)
 
 PLATFORMS
   ruby
@@ -82,6 +124,7 @@ DEPENDENCIES
   rubocop-rspec
   simplecov
   simplecov-lcov
+  uri (~> 0.13)
 
 BUNDLED WITH
    2.3.16

data/README.md

@@ -28,6 +28,92 @@ Or install it yourself as:
 mj help
 ```
 
+### ChatGPT
+
+```
+bundle exec mj chatgpt ask "Who won the World Cup in 1994?" --request-file samples/chatgpt/football-team.yml | jq
+```
+
+Where the config file looks like:
+
+```yaml
+request:
+  parameters:
+    model: gpt-3.5-turbo-16k
+    temperature: 0.1
+    messages:
+    - role: system
+      content: You re an expert in futebool history.
+```
+
+Response:
+
+```json
+{
+  "id": "chatcmpl-90a0HhlOIqYkMPRbuICPy5m798yjQ",
+  "object": "chat.completion",
+  "created": 1709925425,
+  "model": "gpt-3.5-turbo-16k-0613",
+  "choices": [
+    {
+      "index": 0,
+      "message": {
+        "role": "assistant",
+        "content": "The World Cup in 1994 was won by the Brazilian national team. They defeated Italy in the final match, which was held at the Rose Bowl stadium in Pasadena, California, United States. The match ended in a 0-0 draw after extra time, and Brazil won the penalty shootout 3-2 to claim their fourth World Cup title."
+      },
+      "logprobs": null,
+      "finish_reason": "stop"
+    }
+  ],
+  "usage": {
+    "prompt_tokens": 31,
+    "completion_tokens": 71,
+    "total_tokens": 102
+  },
+  "system_fingerprint": null
+}
+```
+
+You can also do `jq '.choices[0].message.content'`:
+
+```sh
+mj chatgpt ask "Who won the World Cup in 1994?" \
+  --config-file samples/chatgpt/football-team.yml | jq '.choices[0].message.content'
+
+# "The World Cup in 1994 was won by the Brazilian national team. They defeated Italy in the final match, which was held at the Rose Bowl stadium in Pasadena, California, United States. The match ended in a 0-0 draw after extra time, and Brazil won the penalty shootout 3-2 to claim their fourth World Cup title."
+```
+
+### GraphQl
+
+```
+bundle exec mj graphql query_file AccountUsers.graphql | jq
+```
+
+Your graqphql fiel is a regular file with variables at the top:
+
+```graphql
+# gql-endpoint: http://api.myapi.me/graphql
+# gql-header: Authorization: Bearer cb923f33617877578961a19cf4566ec2
+# gql-header: Content-Type: application/json
+# gql-variables: { "input": { "id": "some-id" } }
+
+query QueryAccountUsers(
+  $input: AccountInput!
+) {
+  account(input: $input) {
+    users {
+      id
+      name
+    }
+    errors {
+      path
+      message
+    }
+  }
+}
+```
+
+
 ### Alternative file
 
 Examples:

data/lib/mj/chatgpt/commands/ask_command.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Mj
+  module ChatGpt
+    module Commands
+      class AskCommand
+        attr_reader :question
+        attr_reader :options
+
+        def initialize(question, options: {})
+          @question = question
+          @options = options
+        end
+      end
+    end
+  end
+end

data/lib/mj/chatgpt/commands/ask_command_handler.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require "openai"
+require_relative "../request"
+
+module Mj
+  module ChatGpt
+    module Commands
+      class AskCommandHandler
+        def initialize(client:)
+          @client = client
+        end
+
+        def handle(command)
+          request = Request.from_config_file(command.options.fetch("config_file"))
+          @client.chat(**request.asking(command.question).to_h)
+        rescue Faraday::Error => exception
+          {
+            error: exception.message,
+            body: exception.response&.dig(:body, "error") || exception.response
+          }
+        end
+      end
+    end
+  end
+end

data/lib/mj/chatgpt/request.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Mj
+  module ChatGpt
+    class Request
+      def initialize(data = {})
+        @data = HashUtils.new.deep_symbolize_keys(data)
+      end
+
+      def asking(question)
+        data = @data.dup
+        data[:parameters] ||= {}
+        data[:parameters][:messages] ||= []
+        data[:parameters][:messages] << {
+          role: :user,
+          content: question
+        }
+        self.class.new(data)
+      end
+
+      def to_h
+        @data
+      end
+
+      def self.from_config_file(file)
+        data = YAML.load_file(file)
+        new(data.fetch("request"))
+      end
+    end
+  end
+end

data/lib/mj/chatgpt/thor_command.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require_relative "commands/ask_command_handler"
+require_relative "commands/ask_command"
+
+module Mj
+  module ChatGpt
+    class ThorCommand < Thor
+      desc "ask <question>", "Ask ChatGPT a question"
+      option :config_file, type: :string, desc: "Config file"
+      def ask(question)
+        client_config = {
+          access_token: ENV.fetch("OPENAI_ACCESS_TOKEN")
+        }
+        client = OpenAI::Client.new(**client_config.compact)
+        handler = Commands::AskCommandHandler.new(client: client)
+        data = handler.handle(Commands::AskCommand.new(question, options: options))
+        $stdout.puts JSON.generate(data)
+      end
+    end
+  end
+end

data/lib/mj/cli.rb

@@ -3,6 +3,8 @@
 require "thor"
 require_relative "version"
 require_relative "alternative_file/thor_command"
+require_relative "graphql/thor_command"
+require_relative "chatgpt/thor_command"
 
 module Mj
   class Cli < Thor
@@ -17,5 +19,11 @@ module Mj
 
     desc "alternative_file", "Lists alternative files"
     subcommand "alternative_file", AlternativeFile::ThorCommand
+
+    desc "graphql", "CLI client for GraphQL"
+    subcommand "graphql", GraphQL::ThorCommand
+
+    desc "chatgpt", "CLI client for ChatGPT"
+    subcommand "chatgpt", ChatGpt::ThorCommand
   end
 end

data/lib/mj/graphql/client.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require "koine/rest_client"
+
+module Mj
+  module GraphQL
+    class Client
+      def send(request)
+        base_url = request.endpoint.split("/graphql").first
+        base_request = Koine::RestClient::Request.new(base_url: base_url)
+        http_client = Koine::RestClient::Client.new(base_request: base_request)
+        http_client.post("/graphql", request.to_h, headers: request.headers)
+      end
+    end
+  end
+end

data/lib/mj/graphql/commands/query_file_command.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Mj
+  module GraphQL
+    module Commands
+      class QueryFileCommand
+        attr_reader :file
+        attr_reader :options
+
+        def initialize(file, options)
+          @file = file
+          @options = options
+        end
+      end
+    end
+  end
+end

data/lib/mj/graphql/commands/query_file_command_handler.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require_relative "../request_factory"
+require_relative "../client"
+
+module Mj
+  module GraphQL
+    module Commands
+      class QueryFileCommandHandler
+        def handle(command)
+          request = RequestFactory.new.from_file(command.file)
+          Client.new.send(request)
+        end
+      end
+    end
+  end
+end

data/lib/mj/graphql/request.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+module Mj
+  module GraphQL
+    class Request
+      attr_reader :endpoint, :headers, :query, :variables
+
+      def initialize(endpoint: nil, query: nil, variables: {}, headers: {})
+        @endpoint = endpoint
+        @query = query
+        @variables = variables
+        @headers = headers
+      end
+
+      def with_header(name, value)
+        with(:headers, headers.merge(name => value))
+      end
+
+      def with_variables(variables)
+        with(:variables, variables)
+      end
+
+      def to_h
+        {
+          query: query,
+          variables: variables
+        }
+      end
+
+      private
+
+      def with(variable, value)
+        config = {
+          endpoint: endpoint,
+          query: query,
+          variables: variables,
+          headers: headers
+        }.merge(variable => value)
+
+        self.class.new(**config)
+      end
+    end
+  end
+end

data/lib/mj/graphql/request_factory.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require_relative "request"
+
+module Mj
+  module GraphQL
+    class RequestFactory
+      def from_file(file) # rubocop:disable Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity, Metrics/AbcSize, Metrics/MethodLength
+        contents = File.read(file)
+        comments = contents.split("\n").take_while { |line| line.match(/^#/) }.join("\n")
+        query = contents.split("\n").drop_while { |line| line.match(/^#/) }.join("\n")
+        endpoint = comments.match(/# gql-endpoint: (.+)/)&.captures&.first
+
+        request = Request.new(query: query, endpoint: endpoint)
+
+        headers = comments.split("\n")
+          .select { |line| line.match(/^# gql-header:/) }
+          .map { |line| line.split("gql-header:")[1..].join.strip }
+
+        headers.map do |value|
+          values = value.split(":").map(&:strip)
+          request = request.with_header(*values)
+        end
+
+        variables_string = comments.match(/# gql-variables:\s?(.+)/)&.captures&.first
+
+        if variables_string
+          variables = JSON.parse(variables_string)
+          request = request.with_variables(variables)
+        end
+
+        request
+      end
+    end
+  end
+end

data/lib/mj/graphql/thor_command.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+require_relative "commands/query_file_command_handler"
+require_relative "commands/query_file_command"
+require "json"
+
+module Mj
+  module GraphQL
+    class ThorCommand < Thor
+      desc "query_file <file>", "Execute a query based on a .graqphql file"
+      def query_file(file)
+        handler = Commands::QueryFileCommandHandler.new
+        data = handler.handle(Commands::QueryFileCommand.new(file, options))
+        $stdout.puts JSON.generate(data)
+      end
+    end
+  end
+end

data/lib/mj/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Mj
-  VERSION = "0.6.2"
+  VERSION = "0.8.0"
 end

data/lib/mj.rb

@@ -2,6 +2,7 @@
 
 require_relative "mj/version"
 require_relative "mj/cli"
+require "mj/hash_utils"
 
 module Mj
   class Error < StandardError; end

data/mj.gemspec

@@ -31,6 +31,9 @@ Gem::Specification.new do |spec|
   spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
+  spec.add_dependency "koine-rest_client"
+  spec.add_dependency "mj-hash_utils"
+  spec.add_dependency "ruby-openai", "~> 6.2.0"
   spec.add_dependency "thor", "~> 1.2.1"
 
   # For more information and examples about making a new gem, check out our

data/samples/chatgpt/football-team.yml

@@ -0,0 +1,10 @@
+# samples/chatgpt/football-team.yml
+# usage: bundle exec mj chatgpt ask "Who won the World Cup in 1994?" --config-file samples/chatgpt/football-team.yml | jq
+request:
+  parameters:
+    model: gpt-3.5-turbo-16k
+    temperature: 0.1
+    max_tokens: 8192
+    messages:
+    - role: system
+      content: You re an expert in futebool history.

metadata

@@ -1,15 +1,57 @@
 --- !ruby/object:Gem::Specification
 name: mj
 version: !ruby/object:Gem::Version
-  version: 0.6.2
+  version: 0.8.0
 platform: ruby
 authors:
 - Marcelo Jacobus
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-02-05 00:00:00.000000000 Z
+date: 2024-03-08 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: koine-rest_client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mj-hash_utils
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: ruby-openai
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 6.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 6.2.0
 - !ruby/object:Gem::Dependency
   name: thor
   requirement: !ruby/object:Gem::Requirement
@@ -58,9 +100,20 @@ files:
 - lib/mj/alternative_file/resolvers/ruby/vendored_gems_resolver.rb
 - lib/mj/alternative_file/resolvers/ruby/view_component_resolver.rb
 - lib/mj/alternative_file/thor_command.rb
+- lib/mj/chatgpt/commands/ask_command.rb
+- lib/mj/chatgpt/commands/ask_command_handler.rb
+- lib/mj/chatgpt/request.rb
+- lib/mj/chatgpt/thor_command.rb
 - lib/mj/cli.rb
+- lib/mj/graphql/client.rb
+- lib/mj/graphql/commands/query_file_command.rb
+- lib/mj/graphql/commands/query_file_command_handler.rb
+- lib/mj/graphql/request.rb
+- lib/mj/graphql/request_factory.rb
+- lib/mj/graphql/thor_command.rb
 - lib/mj/version.rb
 - mj.gemspec
+- samples/chatgpt/football-team.yml
 - sig/mj.rbs
 homepage: https://github.com/mjacobus/mj
 licenses: