fixes

Azure · Nov 1, 2024 · f92650b · f92650b
1 parent d80a9b0
commit f92650b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 11 deletions.
diff --git a/sdk/provisioning/Azure.Provisioning.CloudMachine/src/AzureSdkExtensions/OpenAIFeature.cs b/sdk/provisioning/Azure.Provisioning.CloudMachine/src/AzureSdkExtensions/OpenAIFeature.cs
@@ -2,13 +2,8 @@
 // Licensed under the MIT License.
 
 using System;
-using System.ClientModel;
-using Azure.AI.OpenAI;
-using Azure.Core;
 using Azure.Provisioning.Authorization;
 using Azure.Provisioning.CognitiveServices;
-using OpenAI.Chat;
-using OpenAI.Embeddings;
 
 namespace Azure.Provisioning.CloudMachine.OpenAI;
 

diff --git a/sdk/provisioning/Azure.Provisioning.CloudMachine/tests/CloudMachineTests.cs b/sdk/provisioning/Azure.Provisioning.CloudMachine/tests/CloudMachineTests.cs
@@ -23,12 +23,12 @@ public void Provisioning(string[] args)
     {
         CloudMachineInfrastructure.Configure(args, (cm) =>
         {
-            //cm.AddFeature(new KeyVaultFeature());
-            //cm.AddFeature(new OpenAIFeature() // TODO: rework it such that models can be added as features
-            //{
-            //    Chat = new AIModel("gpt-35-turbo", "0125"),
-            //    Embeddings = new AIModel("text-embedding-ada-002", "2")
-            //});
+            cm.AddFeature(new KeyVaultFeature());
+            cm.AddFeature(new OpenAIFeature() // TODO: rework it such that models can be added as features
+            {
+                Chat = new AIModel("gpt-35-turbo", "0125"),
+                Embeddings = new AIModel("text-embedding-ada-002", "2")
+            });
         });
     }
 }