allow users to specify a different server

fsndzomga · Dec 22, 2023 · c38c1f8 · c38c1f8
1 parent da2223d
commit c38c1f8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/mistral_rb.rb b/lib/mistral_rb.rb
@@ -7,13 +7,13 @@
 
 class MistralAPI
   include HTTParty
-  base_uri "https://api.mistral.ai/v1"
 
-  def initialize(api_key)
+  def initialize(api_key, base_uri = "https://api.mistral.ai/v1")
     @headers = {
       "Authorization" => "Bearer #{api_key}",
       "Content-Type" => "application/json"
     }
+    self.class.base_uri base_uri
   end
 
   def create_chat_completion(model, messages, temperature = 0.7, top_p = 1, max_tokens = nil, stream = false, safe_mode = false, random_seed = nil)