ChatSession.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstAI/LLMClient.hpp"
#include "AstAI/ChatAgent.hpp"
#include "AstAI/LLMConfig.hpp"
#include "AstAI/ChatTool.hpp"
#include "AstAI/ChatMessages.hpp"
#include "AstAI/ChatTools.hpp"
#include "AstAI/AssistantAgent.hpp"
#include <string>
#include <vector>
#include <functional>
 
AST_NAMESPACE_BEGIN
 
class AssistantAgent;
 
class AST_AI_API ChatSession {
public:
    ChatSession();
 
    std::string chat(StringView message, int maxIterForToolCalls = 100);
 
    std::string chatStream(StringView message, ChatEventHandler& handler, int maxIterForToolCalls = 100);
 
    errc_t sendMessage(StringView message);
 
    void setSystemPrompt(StringView systemPrompt);
 
    ChatMessages& messages(){return messages_;}
 
    AssistantAgent& agent() const;
 
    void setAgent(std::unique_ptr<AssistantAgent> agent){agent_ = std::move(agent);}
 
    ChatTools& tools(){return agent().tools();}
 
    LLMClient& client(){return agent().client();}
 
    const std::string& lastError() const {return agent().lastError();}
 
    LLMConfig& config() { return agent().config(); }
    const LLMConfig& config() const { return agent().config(); }
 
    errc_t makeChatCompletion();
 
    void handleToolCalls(const JsonValue& toolCalls);
 
private:
    mutable std::unique_ptr<AssistantAgent> agent_;     
    ChatMessages messages_;                             
};
 
AST_NAMESPACE_END