tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

Support modifying AZURE_OPENAI_API_VERSION (#1042)

Browse Source 
Co-authored-by: Jean-FrançoisMillet <JMillet@barbizon.fr>
This commit is contained in:
TechnoHouse
2025-02-12 15:01:32 +08:00
committed by GitHub
parent 255c849e5e
commit 5e8a8bae3d
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/providers/azure.rs
+13 -2
View File
@@ -15,7 +15,7 @@ use mcp_core::tool::Tool;
pub const AZURE_DEFAULT_MODEL: &str = "gpt-4o"; pub const AZURE_DEFAULT_MODEL: &str = "gpt-4o";
pub const AZURE_DOC_URL: &str = pub const AZURE_DOC_URL: &str =
"https://learn.microsoft.com/en-us/azure/ai-services/openai/concepts/models"; "https://learn.microsoft.com/en-us/azure/ai-services/openai/concepts/models";
pub const AZURE_API_VERSION: &str = "2024-10-21"; pub const AZURE_DEFAULT_API_VERSION: &str = "2024-10-21";
pub const AZURE_OPENAI_KNOWN_MODELS: &[&str] = &["gpt-4o", "gpt-4o-mini", "gpt-4"]; pub const AZURE_OPENAI_KNOWN_MODELS: &[&str] = &["gpt-4o", "gpt-4o-mini", "gpt-4"];
#[derive(Debug, serde::Serialize)] #[derive(Debug, serde::Serialize)]
@@ -25,6 +25,7 @@ pub struct AzureProvider {
endpoint: String, endpoint: String,
api_key: String, api_key: String,
deployment_name: String, deployment_name: String,
api_version: String,
model: ModelConfig, model: ModelConfig,
} }
@@ -41,6 +42,9 @@ impl AzureProvider {
let api_key: String = config.get_secret("AZURE_OPENAI_API_KEY")?; let api_key: String = config.get_secret("AZURE_OPENAI_API_KEY")?;
let endpoint: String = config.get("AZURE_OPENAI_ENDPOINT")?; let endpoint: String = config.get("AZURE_OPENAI_ENDPOINT")?;
let deployment_name: String = config.get("AZURE_OPENAI_DEPLOYMENT_NAME")?; let deployment_name: String = config.get("AZURE_OPENAI_DEPLOYMENT_NAME")?;
let api_version: String = config
.get("AZURE_OPENAI_API_VERSION")
.unwrap_or_else(|_| AZURE_DEFAULT_API_VERSION.to_string());
let client = Client::builder() let client = Client::builder()
.timeout(Duration::from_secs(600)) .timeout(Duration::from_secs(600))
@@ -51,6 +55,7 @@ impl AzureProvider {
endpoint, endpoint,
api_key, api_key,
deployment_name, deployment_name,
api_version,
model, model,
}) })
} }
@@ -63,7 +68,7 @@ impl AzureProvider {
"openai/deployments/{}/chat/completions", "openai/deployments/{}/chat/completions",
self.deployment_name self.deployment_name
)); ));
base_url.set_query(Some(&format!("api-version={}", AZURE_API_VERSION))); base_url.set_query(Some(&format!("api-version={}", self.api_version)));
let response: reqwest::Response = self let response: reqwest::Response = self
.client .client
@@ -99,6 +104,12 @@ impl Provider for AzureProvider {
false, false,
Some("Name of your Azure OpenAI deployment"), Some("Name of your Azure OpenAI deployment"),
), ),
ConfigKey::new(
"AZURE_OPENAI_API_VERSION",
false,
false,
Some("Azure OpenAI API version, default: 2024-10-21"),
),
], ],
) )
} }