import dev.langchain4j.model.anthropic.AnthropicChatModel;
import dev.langchain4j.model.chat.ChatModel;

/**
 * Anthropic 模型
 */
public class AnthropicIntegration {

    /**
     * 创建 Anthropic 模型
     */
    public static ChatModel createAnthropicModel(String apiKey) {
        return AnthropicChatModel.builder()
                .apiKey(apiKey)
                .modelName(AnthropicChatModel.Models.CLAUDE_3_SONNET)
                .temperature(0.7)
                .maxTokens(1000)
                .timeout(Duration.ofSeconds(60))
                .build();
    }

    /**
     * Anthropic 模型列表
     */
    public static class AnthropicModels {
        public static final String CLAUDE_3_OPUS = "claude-3-opus";
        public static final String CLAUDE_3_SONNET = "claude-3-sonnet";
        public static final String CLAUDE_2_1 = "claude-2.1";
        public static final String CLAUDE_2 = "claude-2";
        public static final String CLAUDE_INSTANT_1_2 = "claude-instant-1.2";
    }
}

import dev.langchain4j.model.azure.AzureOpenAiChatModel;
import dev.langchain4j.model.chat.ChatModel;

/**
 * Azure OpenAI 模型
 */
public class AzureOpenAiIntegration {

    /**
     * 创建 Azure OpenAI 模型
     */
    public static ChatModel createAzureModel(String endpoint, 
                                                    String apiKey, 
                                                    String deploymentName) {
        return AzureOpenAiChatModel.builder()
                .endpoint(endpoint)
                .apiKey(apiKey)
                .deploymentName(deploymentName)
                .modelName(deploymentName)
                .temperature(0.7)
                .maxTokens(1000)
                .timeout(Duration.ofSeconds(60))
                .logRequests(true)
                .logResponses(true)
                .build();
    }

    /**
     * Azure 配置
     */
    public static class AzureConfig {
        private final String endpoint;
        private final String apiKey;
        private final String deploymentName;

        public AzureConfig(String endpoint, String apiKey, String deploymentName) {
            this.endpoint = endpoint;
            this.apiKey = apiKey;
            this.deploymentName = deploymentName;
        }

        public String getEndpoint() { return endpoint; }
        public String getApiKey() { return apiKey; }
        public String getDeploymentName() { return deploymentName; }
    }
}

import dev.langchain4j.model.huggingface.HuggingFaceChatModel;
import dev.langchain4j.model.huggingface.HuggingFaceEmbeddingModel;

/**
 * Hugging Face 模型
 */
public class HuggingFaceIntegration {

    /**
     * 创建 Hugging Face 聊天模型
     */
    public static HuggingFaceChatModel createChatModel(String modelId) {
        return HuggingFaceChatModel.builder()
                .modelId(modelId)
                .temperature(0.7)
                .maxNewTokens(256)
                .build();
    }

    /**
     * 创建 Hugging Face 嵌入模型
     */
    public static HuggingFaceEmbeddingModel createEmbeddingModel(String modelId) {
        return HuggingFaceEmbeddingModel.builder()
                .modelId(modelId)
                .build();
    }

    /**
     * 推荐的开源模型
     */
    public static class RecommendedModels {
        public static final String LAMA_2_7B = "meta-llama/Llama-2-7b-chat-hf";
        public static final String MISTRAL_7B = "mistralai/Mistral-7B-Instruct-v0.2";
        public static final String QWEN_7B = "Qwen/Qwen-7B-Chat";
        public static final String BAICHUAN_7B = "baichuan-inc/Baichuan-7B";
    }
}