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

Rinomina variabili per il supporto del modello AzureOpenAI-Preview in ScenarioExecutionService

Browse Source
This commit is contained in:
andrea.terzani
2024-11-14 11:52:55 +01:00
parent 0ef7ee212a
commit db92b5d2dd
2 changed files with 2 additions and 13 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,19 +323,7 @@ public class ScenarioExecutionService {
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());

1
src/main/java/com/olympus/hermione/stepSolvers/AdvancedAIPromptSolver.java
View File

@@ -71,6 +71,7 @@ public class AdvancedAIPromptSolver extends StepSolver {
if( chatModel.getDefaultOptions().getModel().contains("o1-") ){
//Non funziona
logger.info("Using o1- model => " + chatModel.getDefaultOptions().getModel());
String completePrompt = this.qai_system_prompt_template + userText;