Add Azure OpenAI provider

Author: oxaroky02

lib/ruby_llm/configuration.rb

@@ -24,6 +24,9 @@ class Configuration
                   :bedrock_session_token,
                   :openrouter_api_key,
                   :ollama_api_base,
+                  :azure_openai_api_base,
+                  :azure_openai_api_version,
+                  :azure_openai_api_key,
                   # Default models
                   :default_model,
                   :default_embedding_model,

lib/ruby_llm/providers/azure_openai.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Providers
+    # OpenAI API integration. Handles chat completion, function calling,
+    # and OpenAI's unique streaming format. Supports GPT-4, GPT-3.5,
+    # and other OpenAI models.
+    module AzureOpenAI
+      extend OpenAI
+      extend AzureOpenAI::Chat
+
+      module_function
+
+      def api_base(config)
+        # https://<ENDPOINT>/openai/deployments/<MODEL>/chat/completions?api-version=<APIVERSION>
+        "#{config.azure_openai_api_base}/openai/deployments"
+      end
+
+      def headers(config)
+        {
+          'Authorization' => "Bearer #{config.azure_openai_api_key}"
+        }.compact
+      end
+
+      def capabilities
+        OpenAI::Capabilities
+      end
+
+      def slug
+        'azure_openai'
+      end
+
+      def configuration_requirements
+        %i[azure_openai_api_key azure_openai_api_base azure_openai_api_version]
+      end
+
+      def local?
+        false
+      end
+    end
+  end
+end

lib/ruby_llm/providers/azure_openai/chat.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Providers
+    module AzureOpenAI
+      # Chat methods of the Ollama API integration
+      module Chat
+        extend OpenAI::Chat
+
+        module_function
+
+        def sync_response(connection, payload)
+          # Hold config in instance variable for use in completion_url and stream_url
+          @config = connection.config
+          super
+        end
+
+        def completion_url
+          # https://<ENDPOINT>/openai/deployments/<MODEL>/chat/completions?api-version=<APIVERSION>
+          "#{@model_id}/chat/completions?api-version=#{@config.azure_openai_api_version}"
+        end
+
+        def render_payload(messages, tools:, temperature:, model:, stream: false)
+          # Hold model_id in instance variable for use in completion_url and stream_url
+          @model_id = model
+          super
+        end
+      end
+    end
+  end
+end

lib/ruby_llm/providers/azure_openai/streaming.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Providers
+    module AzureOpenAI
+      # Streaming methods of the OpenAI API integration
+      module Streaming
+        extend OpenAI::Streaming
+
+        module_function
+
+        def stream_response(connection, payload, &block)
+          # Hold config in instance variable for use in completion_url and stream_url
+          @config = connection.config
+          super
+        end
+      end
+    end
+  end
+end