ChatSession.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstAI/OpenAI.hpp"
#include "AstAI/ChatTool.hpp"
#include "AstAI/ChatMessages.hpp"
#include "AstAI/ChatTools.hpp"
#include <string>
#include <vector>
#include <functional>
 
AST_NAMESPACE_BEGIN
 
class AST_AI_API ChatSession {
public:
    ChatSession();
 
    std::string sendMessage(StringView message);
 
    void setSystemPrompt(StringView systemPrompt);
 
    ChatMessages& messages(){return messages_;}
 
    ChatTools& tools(){return tools_;}
 
    OpenAI& client();
private:
    std::string makeChatCompletion(int maxInteractions=20);
 
    void handleToolCalls(const JsonValue& toolCalls, int maxInteractions = 20);
 
    std::string handleToolCall(const JsonValue& toolCall);
 
private:
    OpenAI* client_{nullptr};               
    OpenAI internalClient_;                 
    ChatMessages messages_;                 
    ChatTools tools_;                       
};
 
AST_NAMESPACE_END