rubyml-ai 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: ab0d3ab12880405fa0846a88e13f2fc4ccd1a189d1e05bc4bafc2851bff95c1a
+  data.tar.gz: 7cdea16e0a71730a64a868a77b4c54c9d4bb959ff41763168a554e1ade761294
+SHA512:
+  metadata.gz: eccb769dbb5341a6094bea951c06f5ad5fdc4f297067edb249ab7f1f31905036703245eb47c6734240653ebd645625cfed37b4caecf0478bc9dcd6457ed001af
+  data.tar.gz: 8c7a15fc5e5f57e64b07d0513b3f6f47e50a5cec19cce056e7db7ec5e91388c780d700e9ae0f3fea86068cd82088671b40bbeb6aba8fde19818268875e3f590e

data/lib/train/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Train
+  VERSION = "0.1.0"
+end

data/lib/train.rb

@@ -0,0 +1,99 @@
+
+# frozen_string_literal: true
+
+require "net/http"
+require "json"
+require "uri"
+
+module Train
+  module MODELS
+    @default_map = {
+      "GPT-3.5 Turbo" => "gpt-3.5-turbo",
+      "GPT-4o"        => "gpt-4o",
+      "Gemini 2.5 Pro"=> "google/gemini-1.5-pro-latest",
+      "Gemini 1.5"    => "google/gemini-1.5-flash-latest",
+      "DeepSeek R1"   => "deepseek/deepseek-r1-0528-qwen3-8b:free"
+    }
+
+    def self.get(name)
+      @default_map[name]
+    end
+
+    def self.all
+      @default_map
+    end
+  end
+end
+
+
+
+  ##
+  # Train::Chat::INPUT for interactive terminal loop
+  module Chat
+    INPUT = :__interactive__
+
+    def self.start(model)
+      puts "🚂 Ruby AI Train Interactive Mode"
+      puts "Type a message. Empty input or Ctrl+C to exit.\n\n"
+
+      loop do
+        print "Prompt: "
+        prompt = $stdin.gets&.chomp
+        break if prompt.nil? || prompt.strip.empty?
+
+        puts "AI Answer: #{model.chat(prompt)}\n\n"
+      end
+    rescue Interrupt
+      puts "\nGoodbye!"
+    end
+  end
+
+  ##
+  # Train::Model represents a connection to an AI model endpoint
+class Train::Model
+  attr_accessor :api_key, :url, :model_name
+
+  def initialize(api_key:, url:, model_name:, model_id_changed: nil)
+    @api_key = api_key
+    @url = URI.parse(url)
+
+    # 🔁 override if custom model ID provided
+    default_id = Train::MODELS.get(model_name)
+    @model_name = model_id_changed || default_id || model_name
+  end
+
+
+  ##
+    # Sends a chat message to the AI model
+    #
+    # @param prompt [String or Symbol] User message or Train::Chat::INPUT for interactive mode
+    # @param system [String] Optional system instruction
+    # @return [String] AI-generated response or fallback message
+    def chat(prompt, system: "You are a helpful AI.")
+      return Chat.start(self) if prompt == Chat::INPUT
+
+      http = Net::HTTP.new(url.host, url.port)
+      http.use_ssl = (url.scheme == "https")
+
+      headers = {
+        "Content-Type" => "application/json",
+        "Authorization" => "Bearer #{api_key}"
+      }
+
+      body = {
+        model: model_name,
+        messages: [
+          { role: "system", content: system },
+          { role: "user", content: prompt }
+        ]
+      }.to_json
+
+      response = http.post(url.path, body, headers)
+      json = JSON.parse(response.body)
+
+      json["choices"]&.first&.dig("message", "content") || "⚠️ No response"
+    rescue => e
+      "❌ Train error: #{e.message}"
+    end
+  end
+

metadata

@@ -0,0 +1,42 @@
+--- !ruby/object:Gem::Specification
+name: rubyml-ai
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Cortlet
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies: []
+description: A smarter ride into the AI future. 'train' brings Ruby back to domination.
+email:
+- kkuppuluri0@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/train.rb
+- lib/train/version.rb
+homepage: https://github.com/Cortlet/train
+licenses:
+- MIT
+metadata: {}
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.6'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: Ruby AI Train – Chat with LLMs from Ruby
+test_files: []