From 0ef7ee212a3bdab0e11ef3efb67f2c37c7264990 Mon Sep 17 00:00:00 2001
From: "andrea.terzani" <andrea.terzani@accenture.com>
Date: Thu, 14 Nov 2024 10:30:40 +0100
Subject: [PATCH] Aggiungi supporto per il modello AzureOpenAI-Preview in
 ScenarioExecutionService

---
 .../services/ScenarioExecutionService.java         | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/main/java/com/olympus/hermione/services/ScenarioExecutionService.java b/src/main/java/com/olympus/hermione/services/ScenarioExecutionService.java
index 0e505ef..20855e5 100644
--- a/src/main/java/com/olympus/hermione/services/ScenarioExecutionService.java
+++ b/src/main/java/com/olympus/hermione/services/ScenarioExecutionService.java
@@ -322,6 +322,20 @@ public class ScenarioExecutionService {
                 AzureOpenAiChatModel azureOpenaichatModel = new AzureOpenAiChatModel(openAIClient, openAIChatOptions);
                 logger.info("AI model used: " + aiModel.getModel());
                 return azureOpenaichatModel;
+
+            case "AzureOpenAI-Preview":
+                OpenAIClient openAIClient = new OpenAIClientBuilder()
+                    .credential(new AzureKeyCredential(aiModel.getApiKey()))
+                    .endpoint(aiModel.getEndpoint())
+                    .buildClient();
+                AzureOpenAiChatOptions openAIChatOptions = AzureOpenAiChatOptions.builder()
+                    .withDeploymentName(aiModel.getModel())                        
+                    .withTemperature(aiModel.getTemperature())
+                    .build();
+    
+                AzureOpenAiChatModel azureOpenaichatModel = new AzureOpenAiChatModel(openAIClient, openAIChatOptions);
+                logger.info("AI model used: " + aiModel.getModel());
+                return azureOpenaichatModel;
     
             case "OpenAI":
                 OpenAiApi openAiApi = new OpenAiApi(aiModel.getApiKey());