• Home
• org.wildfly
• wildfly-ai-subsystem
• 0.2.0
• source code
• AIAttachements.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.wildfly.extension.ai.deployment.AIAttachements Maven / Gradle / Ivy

/*
 * Copyright The WildFly Authors
 * SPDX-License-Identifier: Apache-2.0
 */
package org.wildfly.extension.ai.deployment;

import dev.langchain4j.model.chat.ChatLanguageModel;
import dev.langchain4j.model.embedding.EmbeddingModel;
import dev.langchain4j.rag.content.retriever.ContentRetriever;
import dev.langchain4j.store.embedding.EmbeddingStore;
import org.jboss.as.server.deployment.AttachmentKey;
import org.jboss.as.server.deployment.AttachmentList;

/**
 *
 * @author Emmanuel Hugonnet (c) 2024 Red Hat, Inc.
 */
public class AIAttachements {

    static final AttachmentKey> CHAT_MODELS = AttachmentKey.createList(ChatLanguageModel.class);
    static final AttachmentKey> CHAT_MODEL_KEYS = AttachmentKey.createList(String.class);
    static final AttachmentKey> EMBEDDING_MODELS = AttachmentKey.createList(EmbeddingModel.class);
    static final AttachmentKey> EMBEDDING_MODEL_KEYS = AttachmentKey.createList(String.class);
    static final AttachmentKey> EMBEDDING_STORES = AttachmentKey.createList(EmbeddingStore.class);
    static final AttachmentKey> EMBEDDING_STORE_KEYS = AttachmentKey.createList(String.class);
    static final AttachmentKey> CONTENT_RETRIEVERS = AttachmentKey.createList(ContentRetriever.class);
    static final AttachmentKey> CONTENT_RETRIEVER_KEYS = AttachmentKey.createList(String.class);
}