package com.zy.ai.service;

import com.zy.ai.entity.AiPromptTemplate;
import com.zy.ai.entity.ChatCompletionRequest;

import java.util.List;
import java.util.Map;

public interface AiChatStoreService {

    List<Map<String, Object>> listChats();

    boolean deleteChat(String chatId);

    List<ChatCompletionRequest.Message> getChatHistory(String chatId);

    void saveConversation(String chatId,
                          String title,
                          ChatCompletionRequest.Message userMessage,
                          ChatCompletionRequest.Message assistantMessage,
                          AiPromptTemplate promptTemplate,
                          long promptTokens,
                          long completionTokens,
                          long totalTokens,
                          int llmCallCount);
}