Add support for listing models

Author: oxaroky02

lib/ruby_llm/providers/azure_openai.rb

@@ -8,12 +8,13 @@ module AzureOpenAI
       extend OpenAI
       extend AzureOpenAI::Chat
       extend AzureOpenAI::Streaming
+      extend AzureOpenAI::Models
 
       module_function
 
       def api_base(config)
         # https://<ENDPOINT>/openai/deployments/<MODEL>/chat/completions?api-version=<APIVERSION>
-        "#{config.azure_openai_api_base}/openai/deployments"
+        "#{config.azure_openai_api_base}/openai"
       end
 
       def headers(config)

lib/ruby_llm/providers/azure_openai/chat.rb

@@ -17,7 +17,7 @@ def sync_response(connection, payload)
 
         def completion_url
           # https://<ENDPOINT>/openai/deployments/<MODEL>/chat/completions?api-version=<APIVERSION>
-          "#{@model_id}/chat/completions?api-version=#{@config.azure_openai_api_version}"
+          "deployments/#{@model_id}/chat/completions?api-version=#{@config.azure_openai_api_version}"
         end
 
         def render_payload(messages, tools:, temperature:, model:, stream: false)

lib/ruby_llm/providers/azure_openai/models.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Providers
+    module AzureOpenAI
+      # Models methods of the OpenAI API integration
+      module Models
+        extend OpenAI::Models
+
+        module_function
+
+        def models_url
+          'models?api-version=2024-10-21'
+        end
+      end
+    end
+  end
+end

lib/tasks/models_update.rake

@@ -24,10 +24,17 @@ def configure_from_env
     config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY', nil)
     config.openrouter_api_key = ENV.fetch('OPENROUTER_API_KEY', nil)
     configure_bedrock(config)
+    configure_azure_openai(config)
     config.request_timeout = 30
   end
 end
 
+def configure_azure_openai(config)
+  config.azure_openai_api_base = ENV.fetch('AZURE_OPENAI_ENDPOINT', nil)
+  config.azure_openai_api_key = ENV.fetch('AZURE_OPENAI_API_KEY', nil)
+  config.azure_openai_api_version = ENV.fetch('AZURE_OPENAI_API_VER', nil)
+end
+
 def configure_bedrock(config)
   config.bedrock_api_key = ENV.fetch('AWS_ACCESS_KEY_ID', nil)
   config.bedrock_secret_key = ENV.fetch('AWS_SECRET_ACCESS_KEY', nil)