001/*
002 * SPDX-License-Identifier: Apache-2.0
003 *
004 * Copyright 2025-2026 The Enola <https://enola.dev> Authors
005 *
006 * Licensed under the Apache License, Version 2.0 (the "License");
007 * you may not use this file except in compliance with the License.
008 * You may obtain a copy of the License at
009 *
010 *     https://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018package dev.enola.ai.adk.iri;
019
020import com.google.adk.models.BaseLlm;
021import com.google.adk.models.langchain4j.LangChain4j;
022
023import dev.enola.ai.iri.ModelConfig;
024import dev.enola.ai.iri.OllamaModelProvider;
025import dev.langchain4j.model.ollama.OllamaChatModel;
026import dev.langchain4j.model.ollama.OllamaStreamingChatModel;
027
028public class OllamaLlmProvider extends OllamaModelProvider<BaseLlm> {
029
030    @Override
031    protected BaseLlm create(String baseURL, String modelName, ModelConfig config) {
032        var sync =
033                OllamaChatModel.builder()
034                        .logRequests(true)
035                        .logResponses(true)
036                        .baseUrl(baseURL)
037                        .modelName(modelName)
038                        .build();
039
040        var streaming =
041                OllamaStreamingChatModel.builder()
042                        .logRequests(true)
043                        .logResponses(true)
044                        .baseUrl(baseURL)
045                        .modelName(modelName)
046                        .build();
047
048        var langChain4jLlm = new LangChain4j(sync, streaming, modelName);
049        return WrappedBaseLlm.wrap(langChain4jLlm, config);
050    }
051}