Fix Azure secret creation after update to azure_llm (#120)

dsg-polymtl · Jan 17, 2025 · 5c4d40b · 5c4d40b
1 parent f499985
commit 5c4d40b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/model_manager/providers/adapters/azure.cpp b/src/model_manager/providers/adapters/azure.cpp
@@ -8,8 +8,7 @@ nlohmann::json AzureProvider::CallComplete(const std::string& prompt, const bool
                                             model_details_.model, model_details_.secret["api_version"], true);
 
     // Create a JSON request payload with the provided parameters
-    nlohmann::json request_payload = {{"model", model_details_.model},
-                                      {"messages", {{{"role", "user"}, {"content", prompt}}}},
+    nlohmann::json request_payload = {{"messages", {{{"role", "user"}, {"content", prompt}}}},
                                       {"max_tokens", model_details_.max_output_tokens},
                                       {"temperature", model_details_.temperature}};
 

diff --git a/src/secret_manager/secret_manager.cpp b/src/secret_manager/secret_manager.cpp
@@ -25,7 +25,7 @@ std::vector<SecretDetails> get_secret_details_list() {
 }
 
 std::unordered_map<std::string, SecretManager::SupportedProviders> SecretManager::providerNames = {
-    {"openai", OPENAI}, {"azure", AZURE}, {"ollama", OLLAMA}};
+    {"openai", OPENAI}, {"azure_llm", AZURE}, {"ollama", OLLAMA}};
 
 SecretManager::SupportedProviders SecretManager::GetProviderType(const std::string& provider) {
     return providerNames[provider];