mock_graphql_ai 1.0.4 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bc76561ab75b8fa4646d98737559285a790e3ee961643bdc44558092cbc885a1
-  data.tar.gz: 32a1e0f3fec394745dd10aa54df7e6fd160481e5b428347a0a689832dcebba19
+  metadata.gz: 9608b7928f7505d25c1d1b7c730eb6832777a392caa62b5f4fabee3aff322d30
+  data.tar.gz: 89bbced79385aff524126dd9b8dd09d54437070bef3424ec95f0db6d84ab4283
 SHA512:
-  metadata.gz: 74d93dbc1bf31910495a9baf022acd8ac356c496b840d6f098707335f8c8b55b3a863e5447dc29ac90f08970413394bdb74438e6666fc4ea3d3c67eb8eff9d7a
-  data.tar.gz: 064ed835690f84f6824dc1590e1f3bc37c86ca8e63a8a8d90add371bfb4cbb2ebcd7c705ba7067d79f37502003a8c8ec93b3169d70dea8e869250b989cc06d0d
+  metadata.gz: 2200ec1dd098149f2383a626531041d67814f07bb1b6a83154b30d6373177af14df93360a59d9b1f5ad8762f480fe4785a857483e2909adfbc54a1647c18e8f7
+  data.tar.gz: 893ccf6715ec28d3917b1fe1e3baf413f2a02bb1aa3834dbf5a2d68e7815f74f1c83b1f5a2ba346df2d1708f9a762191471a86e348dbc733d71662476bb0f280

data/lib/generators/mock_graphql_ai/install_generator.rb

@@ -3,6 +3,8 @@ require 'rails/generators'
 module MockGraphqlAi
   module Generators
     class InstallGenerator < Rails::Generators::Base
+      include Rails::Generators::Migration
+
       source_root(File.expand_path('../templates', __dir__))
 
       def copy_module_file
@@ -12,10 +14,32 @@ module MockGraphqlAi
       def copy_controller_file
         copy_file('graphql_mocks_controller.rb', 'app/controllers/graphql_mocks_controller.rb')
       end
+      
+      def copy_model_file
+        copy_file('mock_graphql_response.rb', 'app/models/mock_graphql_response.rb')
+      end
+
+      def copy_migration
+        migration_template(
+          'migration.rb', 
+          'db/migrate/create_mock_graphql_responses.rb', 
+          migration_version: migration_version
+        )
+      end
 
       def register_controller_main_route
         route "get '/imitate_graphql', to: 'graphql_mocks#imitate', as: :imitate_graphql"
       end
+
+      private
+
+      def self.next_migration_number(dirname)
+        Time.now.utc.strftime('%Y%m%d%H%M%S')
+      end      
+      
+      def migration_version
+        "[#{Rails::VERSION::MAJOR}.#{Rails::VERSION::MINOR}]"
+      end
     end
   end
 end

data/lib/generators/templates/graphql_mocks_controller.rb

@@ -1,5 +1,10 @@
 class GraphqlMocksController < ApplicationController
   def imitate
-    render json: { "data" => { "query" => params[:query], "say" => "Here we'll send Mock API response!" } }    
+    result = MockGraphqlAi::Resource.new(
+      query: params[:query],
+      token: "add-openai-access-token-here" # Rails.application.credentials.dig(:openai, :access_token) 
+    ).respond!
+
+    render json: result
   end 
 end

data/lib/generators/templates/migration.rb

@@ -0,0 +1,13 @@
+class CreateMockGraphqlResponses < ActiveRecord::Migration<%= migration_version %>
+  def change
+    create_table :mock_graphql_responses do |t|
+      t.string :name
+      t.integer :req_type, null: false, default: 0
+      t.integer :output_mode, null: false, default: 0
+      t.jsonb :data, null: false, default: {}
+
+      t.timestamps
+    end
+  end
+end
+

data/lib/generators/templates/mock_graphql_response.rb

@@ -0,0 +1,6 @@
+class MockGraphqlResponse < ApplicationRecord
+  enum req_type: [:query, :mutation]
+  
+  # All output modes except :void map to the OpenAI response attribute 'finish_reason' to determine model output
+  enum output_mode: [:void, :token_limit, :content_filter, :incomplete, :stop]  
+end

data/lib/mock_graphql_ai/version.rb

@@ -1,3 +1,3 @@
 module MockGraphqlAi
-  VERSION = '1.0.4'
+  VERSION = '1.1.0'
 end

data/lib/mock_graphql_ai.rb

@@ -1,4 +1,95 @@
 require 'mock_graphql_ai/version'
+require 'openai'
+require 'graphql'
 
 module MockGraphqlAi
+  class Resource
+    attr_accessor :query, :type, :name, :client, :openai_access_token, :response
+
+    def initialize(options={})
+      @query = options[:query]
+      @openai_access_token = options[:token]
+      @temperature = 0.3
+      @model = "gpt-3.5-turbo"
+    end
+  
+    def respond!
+      @type, @name = grab_query_details
+      initiate_openai_client
+      get_response_from_ai
+    end
+
+    private
+      
+      def grab_query_details
+        query_definition = GraphQL.parse(query).definitions[0]
+        [query_definition.operation_type, query_definition.selections[0].name]
+      end
+
+      def initiate_openai_client
+        @client = OpenAI::Client.new(access_token: openai_access_token)        
+      end
+
+      def get_response_from_ai
+        return mock_response.data if mock_response
+        
+        @response = client.chat(
+          parameters: {
+            model: @model,
+            temperature: @temperature,
+            messages: [
+              { role: "system", content: instructions_to_model },
+              { role: "user", content: "#{query}" }
+            ],
+            max_tokens: 1000
+          }
+        )
+
+        persist_response! if complete_model_output?
+        response.dig("choices", 0, "message", "content")
+      end
+
+      def mock_response
+        MockGraphqlResponse.find_by(name: name, req_type: type)
+      end
+
+      def instructions_to_model
+        "You are a helpful coding assistant. Below are the points you must consider while helping me
+  
+        - I will give you a GraphQL query or mutation
+        - You have to return me the only response back
+        - Don't append any conversational or explanation text around it
+        - Don't give me code implementations you used to achieve this response
+        - Use double quotes around key value pairs
+        - Give me data objects to consume within
+        - I want real data instead of empty collection back to consume within my React Frontend app
+        - Avoid errors because I have to parse data back to JSON"
+      end
+
+      def persist_response!
+        MockGraphqlResponse.create(
+          data: @response.dig("choices", 0, "message", "content"),
+          name: name,
+          req_type: type,
+          output_mode: figure_output_mode 
+        )
+      end
+      
+      def complete_model_output?
+        @response.dig("choices", 0, "finish_reason") == "stop"
+      end
+
+      def figure_output_mode
+        finish_reason = response.dig("choices", 0, "finish_reason")
+
+        reasons_mapper = {
+          "length" => :token_limit,
+          "content_filter" => :content_filter,
+          "null" => :incomplete,
+          "stop" => :stop 
+        }
+
+        reasons_mapper[finish_reason]
+      end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mock_graphql_ai
 version: !ruby/object:Gem::Version
-  version: 1.0.4
+  version: 1.1.0
 platform: ruby
 authors:
 - Moiz Ali
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-03-29 00:00:00.000000000 Z
+date: 2023-04-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -181,10 +181,10 @@ files:
 - LICENSE
 - README.md
 - Rakefile
-- lib/generators/mock_controller.rb
 - lib/generators/mock_graphql_ai/install_generator.rb
-- lib/generators/mock_module.rb
 - lib/generators/templates/graphql_mocks_controller.rb
+- lib/generators/templates/migration.rb
+- lib/generators/templates/mock_graphql_response.rb
 - lib/generators/templates/mockable.rb
 - lib/mock_graphql_ai.rb
 - lib/mock_graphql_ai/version.rb

data/lib/generators/mock_controller.rb

@@ -1,15 +0,0 @@
-require "rails/generators"
-
-module Generators
-  class MockController < Rails::Generators::Base
-    source_root(File.expand_path("../generators/templates/controllers", __dir__))
-
-    def copy_controller_file
-      copy_file("graphql_mocks_controller.rb", "app/controllers/graphql_mocks_controller.rb")
-    end
-
-    def register_controller_main_route
-      route "get '/imitate_graphql', to: 'graphql_mocks#imitate', as: :imitate_graphql"
-    end
-  end
-end

data/lib/generators/mock_module.rb

@@ -1,11 +0,0 @@
-require 'rails/generators'
-
-module Generators
-  class MockModule < Rails::Generators::Base
-    source_root(File.expand_path('../generators/templates', __dir__))
-  
-    def copy_module_file
-      copy_file("mockable.rb", "app/controllers/concerns/mockable.rb")    
-    end  
-  end
-end