raix.rb in raix-0.3.1

- old
+ new
@@ -1,8 +1,54 @@
 # frozen_string_literal: true
 
 require_relative "raix/version"
+require_relative "raix/chat_completion"
+require_relative "raix/function_dispatch"
+require_relative "raix/prompt_declarations"
 
+# The Raix module provides configuration options for the Raix gem.
 module Raix
-  class Error < StandardError; end
-  # Your code goes here...
+  # The Configuration class holds the configuration options for the Raix gem.
+  class Configuration
+    # The temperature option determines the randomness of the generated text.
+    # Higher values result in more random output.
+    attr_accessor :temperature
+
+    # The max_tokens option determines the maximum number of tokens to generate.
+    attr_accessor :max_tokens
+
+    # The model option determines the model to use for text generation. This option
+    # is normally set in each class that includes the ChatCompletion module.
+    attr_accessor :model
+
+    # The openrouter_client option determines the default client to use for communicatio.
+    attr_accessor :openrouter_client
+
+    # The openai_client option determines the OpenAI client to use for communication.
+    attr_accessor :openai_client
+
+    DEFAULT_MAX_TOKENS = 1000
+    DEFAULT_MODEL = "meta-llama/llama-3-8b-instruct:free"
+    DEFAULT_TEMPERATURE = 0.0
+
+    # Initializes a new instance of the Configuration class with default values.
+    def initialize
+      self.temperature = DEFAULT_TEMPERATURE
+      self.max_tokens = DEFAULT_MAX_TOKENS
+      self.model = DEFAULT_MODEL
+    end
+  end
+
+  class << self
+    attr_writer :configuration
+  end
+
+  # Returns the current configuration instance.
+  def self.configuration
+    @configuration ||= Configuration.new
+  end
+
+  # Configures the Raix gem using a block.
+  def self.configure
+    yield(configuration)
+  end
 end