admin/hermione
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Aggiungi supporto per il modello AzureOpenAI-Preview in ScenarioExecutionService

Browse Source
This commit is contained in:
andrea.terzani
2024-11-14 10:30:40 +01:00
parent 5480a54a31
commit 0ef7ee212a
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/java/com/olympus/hermione/services/ScenarioExecutionService.java
View File

@@ -323,6 +323,20 @@ public class ScenarioExecutionService {
logger.info("AI model used: " + aiModel.getModel()); logger.info("AI model used: " + aiModel.getModel());
return azureOpenaichatModel; 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": case "OpenAI":
OpenAiApi openAiApi = new OpenAiApi(aiModel.getApiKey()); OpenAiApi openAiApi = new OpenAiApi(aiModel.getApiKey());
OpenAiChatOptions openAiChatOptions = OpenAiChatOptions.builder() OpenAiChatOptions openAiChatOptions = OpenAiChatOptions.builder()