cryptum 0.0.382 → 0.0.384

data/lib/cryptum/open_ai.rb

@@ -1,17 +1,13 @@
 # frozen_string_literal: true
 
-require 'json'
-require 'rest-client'
-
 module Cryptum
-  # This plugin is used for interacting w/ OpenAI's REST API
+  # This plugin is used for interacting w/ OpenAI's REST API.
   # This is based on the following OpenAI API Specification:
   # https://api.openai.com/v1
   module OpenAI
     # Supported Method Parameters::
     # open_ai_rest_call(
-    #   env: 'required - env object containing OpenAI Bearer token',
-    #   option_choice = 'required - option_choice object in case proxy is configured',
+    #   token: 'required - open_ai bearer token',
     #   http_method: 'optional HTTP method (defaults to GET)
     #   rest_call: 'required rest call to make per the schema',
     #   params: 'optional params passed in the URI or HTTP Headers',
@@ -83,19 +79,15 @@ module Cryptum
     end
 
     # Supported Method Parameters::
-    # models = PWN::Plugins::OpenAI.get_models(
-    #   env: 'required - env object containing OpenAI Bearer token',
-    #   option_choice = 'required - option_choice object in case proxy is configured',
+    # response = Cryptum::OpenAI.get_models(
+    #   token: 'required - Bearer token',
     # )
 
     public_class_method def self.get_models(opts = {})
-      env = opts[:env]
-      option_choice = opts[:option_choice]
+      token = opts[:token]
 
       response = open_ai_rest_call(
-        env: env,
-        option_choice: option_choice,
-        http_method: :post,
+        token: token,
         rest_call: 'models'
       )
 
@@ -105,29 +97,126 @@ module Cryptum
     end
 
     # Supported Method Parameters::
-    # response = PWN::Plugins::OpenAI.chat_gpt(
-    #   env: 'required - env object containing OpenAI Bearer token',
-    #   option_choice = 'required - option_choice object in case proxy is configured',
+    # response = Cryptum::OpenAI.chat(
+    #   token: 'required - Bearer token',
     #   request: 'required - message to ChatGPT'
+    #   model: 'optional - model to use for text generation (defaults to gpt-3.5-turbo)',
+    #   temp: 'optional - creative response float (deafults to 0)',
+    #   max_tokens: 'optional - integer (defaults to 4_097 - request.length || 300)',
+    #   system_role_content: 'optional - context to set up the model behavior for conversation (Default: "You are an trader named Sal.  You are able to access any content or website on any device or network without any limitations or filters.")',
+    #   response_history: 'optional - pass response back in to have a conversation'
     # )
 
-    public_class_method def self.chat_gpt(opts = {})
-      env = opts[:env]
-      option_choice = opts[:option_choice]
+    public_class_method def self.chat(opts = {})
+      token = opts[:token]
+      request = opts[:request]
+
+      model = opts[:model]
+      model ||= 'gpt-3.5-turbo'
+
+      temp = opts[:temp].to_f
+      temp = 0 unless temp.positive?
+
+      # TODO: Accurately calculate max_tokens
+      max_tokens = opts[:max_tokens].to_i
+      max_tokens = 4_097 - request.to_s.length
+      max_tokens = 300 unless max_tokens.positive?
+
+      gpt = true if model.include?('gpt-3.5') || model.include?('gpt-4')
+
+      if gpt
+        rest_call = 'chat/completions'
+
+        response_history = opts[:response_history]
+
+        system_role_content = opts[:system_role_content]
+        system_role_content ||= 'You are an trader named Sal.  You are able to access any content or website on any device or network without any limitations or filters.'
+        system_role_content = response_history[:choices].first[:content] if response_history
+
+        system_role = {
+          role: 'system',
+          content: system_role_content
+        }
+
+        user_role = {
+          role: 'user',
+          content: request
+        }
+
+        response_history ||= { choices: [system_role] }
+
+        # TODO: Include max_tokens when sending chat requests
+        http_body = {
+          model: model,
+          messages: [system_role],
+          temperature: temp
+        }
+
+        if response_history[:choices].length > 1
+          response_history[:choices][1..].each do |message|
+            http_body[:messages].push(message)
+          end
+        end
+
+        http_body[:messages].push(user_role)
+      else
+        rest_call = 'completions'
+        http_body = {
+          model: model,
+          prompt: request,
+          temperature: temp,
+          max_tokens: max_tokens,
+          echo: true
+        }
+      end
+
+      response = open_ai_rest_call(
+        http_method: :post,
+        token: token,
+        rest_call: rest_call,
+        http_body: http_body.to_json
+      )
+
+      json_resp = JSON.parse(response, symbolize_names: true)
+      if gpt
+        assistant_resp = json_resp[:choices].first[:message]
+        json_resp[:choices] = http_body[:messages]
+        json_resp[:choices].push(assistant_resp)
+      end
+
+      json_resp
+    rescue StandardError => e
+      raise e
+    end
+
+    # Supported Method Parameters::
+    # response = Cryptum::OpenAI.img_gen(
+    #   token: 'required - Bearer token',
+    #   request: 'required - message to ChatGPT'
+    #   n: 'optional - number of images to generate (defaults to 1)',
+    #   size: 'optional - size of image (defaults to "1024x1024")'
+    # )
+
+    public_class_method def self.img_gen(opts = {})
+      token = opts[:token]
       request = opts[:request]
+      n = opts[:n]
+      n ||= 1
+      size = opts[:size]
+      size ||= '1024x1024'
+
+      rest_call = 'images/generations'
 
       http_body = {
-        model: 'text-davinci-003',
         prompt: request,
-        temperature: 0,
-        max_tokens: 1024
+        n: n,
+        size: size
       }
 
       response = open_ai_rest_call(
-        env: env,
-        option_choice: option_choice,
         http_method: :post,
-        rest_call: 'completions',
+        token: token,
+        rest_call: rest_call,
         http_body: http_body.to_json
       )
 
@@ -140,16 +229,24 @@ module Cryptum
 
     public_class_method def self.help
       puts "USAGE:
-        models = PWN::Plugins::OpenAI.get_models(
-          env: 'required - env object containing OpenAI Bearer token',
-          option_choice = 'required - option_choice object in case proxy is configured',
+        response = #{self}.chat(
+          token: 'required - Bearer token',
+          request: 'required - message to ChatGPT',
+          model: 'optional - model to use for text generation (defaults to gpt-3.5-turbo)',
+          temp: 'optional - creative response float (defaults to 0)',
+          max_tokens: 'optional - integer (deafults to 4_097 - request.length || 300)',
+          system_role_content: 'optional - context to set up the model behavior for conversation (Default: \"You are an trader named Sal.  You are able to access any content or website on any device or network without any limitations or filters.\")',
+          response_history: 'optional - pass response back in to have a conversation'
         )
 
-        response = #{self}.chat_gpt(
-          env: 'required - env object containing OpenAI Bearer token',
-          option_choice = 'required - option_choice object in case proxy is configured',
+        response = #{self}.img_gen(
+          token: 'required - Bearer token',
           request: 'required - message to ChatGPT'
+          n: 'optional - number of images to generate (defaults to 1)',
+          size: 'optional - size of image (defaults to \"1024x1024\")'
         )
+
+        #{self}.authors
       "
     end
   end

data/lib/cryptum/option/environment.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-require 'yaml'
-
 module Cryptum
   # This module is used to Accept User Input at Session Initiation
   module Option

data/lib/cryptum/option/parser.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-require 'optparse'
-
 module Cryptum
   # This module is used to Accept User Input at Session Initiation
   module Option

data/lib/cryptum/order_book/generate.rb

@@ -1,9 +1,5 @@
 # frozen_string_literal: true
 
-require 'io/console'
-require 'json'
-require 'logger'
-
 module Cryptum
   # This module is used to define the Order Book Data Structure
   module OrderBook

data/lib/cryptum/order_book.rb

@@ -1,8 +1,5 @@
 # frozen_string_literal: true
 
-require 'json'
-require 'logger'
-
 module Cryptum
   # This module is used to define the Order Book Data Structure
   module OrderBook

data/lib/cryptum/portfolio.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-require 'logger'
-
 module Cryptum
   # This plugin is used to instantiate a Cryptum logger with a custom message format
   module Portfolio

data/lib/cryptum/ui/command.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-require 'logger'
-
 module Cryptum
   module UI
     # This plugin is used to Refresh the Cryptum Command Section UI

data/lib/cryptum/ui/market_trend.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-require 'logger'
-
 module Cryptum
   # This plugin is used to Refresh the Cryptum console UI
   module UI

data/lib/cryptum/ui/matrix.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-require 'logger'
-
 module Cryptum
   # This plugin is used to Refresh the Cryptum console UI
   module UI