001/*
002 *  Copyright 2025 Anyware Services
003 *
004 *  Licensed under the Apache License, Version 2.0 (the "License");
005 *  you may not use this file except in compliance with the License.
006 *  You may obtain a copy of the License at
007 *
008 *      http://www.apache.org/licenses/LICENSE-2.0
009 *
010 *  Unless required by applicable law or agreed to in writing, software
011 *  distributed under the License is distributed on an "AS IS" BASIS,
012 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 *  See the License for the specific language governing permissions and
014 *  limitations under the License.
015 */
016
017package org.ametys.plugins.ai.provider.impl;
018
019import java.time.Duration;
020
021import org.ametys.runtime.config.Config;
022
023import dev.langchain4j.model.chat.ChatModel;
024import dev.langchain4j.model.embedding.EmbeddingModel;
025import dev.langchain4j.model.image.ImageModel;
026import dev.langchain4j.model.openai.OpenAiChatModel;
027import dev.langchain4j.model.openai.OpenAiEmbeddingModel;
028import dev.langchain4j.model.openai.OpenAiImageModel;
029/**
030 * Implementation of the OpenAi Provider
031 */
032public class OpenAIProvider extends AbstractAIProvider
033{
034    @Override
035    protected ChatModel createChatModel(String modelName)
036    {
037        return OpenAiChatModel.builder()
038                .apiKey(getAPIKey())
039                .modelName(modelName)
040                .build();
041    }
042    
043    @Override
044    protected ImageModel createImageModel(String modelName)
045    {
046        return OpenAiImageModel.builder()
047                .apiKey(getAPIKey())
048                .modelName(modelName)
049                .timeout(Duration.ofSeconds(Config.getInstance().getValue(CONFIG_IMAGE_TIMEOUT)))
050                .build();
051    }
052    
053    @Override
054    protected EmbeddingModel createEmbeddingModel(String modelName)
055    {
056        return OpenAiEmbeddingModel.builder()
057                .apiKey(getAPIKey())
058                .modelName(modelName)
059                .build();
060    }
061}