chore: add tests and update url construction for azure provider (#988)

crates/goose/tests/providers.rs

@@ -3,7 +3,7 @@ use dotenv::dotenv;
 use goose::message::{Message, MessageContent};
 use goose::providers::base::Provider;
 use goose::providers::errors::ProviderError;
-use goose::providers::{anthropic, databricks, google, groq, ollama, openai, openrouter};
+use goose::providers::{anthropic, azure, databricks, google, groq, ollama, openai, openrouter};
 use mcp_core::content::Content;
 use mcp_core::tool::Tool;
 use std::collections::HashMap;
@@ -359,6 +359,21 @@ async fn test_openai_provider() -> Result<()> {
     .await
 }
 
+#[tokio::test]
+async fn test_azure_provider() -> Result<()> {
+    test_provider(
+        "Azure",
+        &[
+            "AZURE_OPENAI_API_KEY",
+            "AZURE_OPENAI_ENDPOINT",
+            "AZURE_OPENAI_DEPLOYMENT_NAME",
+        ],
+        None,
+        azure::AzureProvider::default,
+    )
+    .await
+}
+
 #[tokio::test]
 async fn test_databricks_provider() -> Result<()> {
     test_provider(

crates/goose/tests/truncate_agent.rs

@@ -7,13 +7,15 @@ use goose::message::Message;
 use goose::model::ModelConfig;
 use goose::providers::base::Provider;
 use goose::providers::{anthropic::AnthropicProvider, databricks::DatabricksProvider};
-use goose::providers::{google::GoogleProvider, groq::GroqProvider};
 use goose::providers::{
-    ollama::OllamaProvider, openai::OpenAiProvider, openrouter::OpenRouterProvider,
+    azure::AzureProvider, ollama::OllamaProvider, openai::OpenAiProvider,
+    openrouter::OpenRouterProvider,
 };
+use goose::providers::{google::GoogleProvider, groq::GroqProvider};
 
 #[derive(Debug)]
 enum ProviderType {
+    Azure,
     OpenAi,
     Anthropic,
     Databricks,
@@ -26,6 +28,11 @@ enum ProviderType {
 impl ProviderType {
     fn required_env(&self) -> &'static [&'static str] {
         match self {
+            ProviderType::Azure => &[
+                "AZURE_OPENAI_API_KEY",
+                "AZURE_OPENAI_ENDPOINT",
+                "AZURE_OPENAI_DEPLOYMENT_NAME",
+            ],
             ProviderType::OpenAi => &["OPENAI_API_KEY"],
             ProviderType::Anthropic => &["ANTHROPIC_API_KEY"],
             ProviderType::Databricks => &["DATABRICKS_HOST"],
@@ -56,6 +63,7 @@ impl ProviderType {
 
     fn create_provider(&self, model_config: ModelConfig) -> Result<Box<dyn Provider>> {
         Ok(match self {
+            ProviderType::Azure => Box::new(AzureProvider::from_env(model_config)?),
             ProviderType::OpenAi => Box::new(OpenAiProvider::from_env(model_config)?),
             ProviderType::Anthropic => Box::new(AnthropicProvider::from_env(model_config)?),
             ProviderType::Databricks => Box::new(DatabricksProvider::from_env(model_config)?),
@@ -172,6 +180,16 @@ mod tests {
         .await
     }
 
+    #[tokio::test]
+    async fn test_truncate_agent_with_azure() -> Result<()> {
+        run_test_with_config(TestConfig {
+            provider_type: ProviderType::Azure,
+            model: "gpt-4o-mini",
+            context_window: 128_000,
+        })
+        .await
+    }
+
     #[tokio::test]
     async fn test_truncate_agent_with_anthropic() -> Result<()> {
         run_test_with_config(TestConfig {