ChatConsole.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstAI/ChatEventHandler.hpp"
// #include "AstUtil/Markdown.hpp"
#include "AstUtil/MarkdownRenderer.hpp"
 
AST_NAMESPACE_BEGIN
 
class AST_AI_API ChatConsole : public ChatEventHandler
{
public:
    using RendererType = MarkdownRenderer;
    //using RendererType = Markdown;
 
 
    ChatConsole() = default;
    ~ChatConsole() = default;
 
    void onContentChunk(const std::string& text) override;
 
    void onReasoningChunk(const std::string& reasoning) override;
 
    void onToolCallRequest(const std::string& /*toolCallId*/,
                           const std::string& functionName,
                           const std::string& arguments) override;
 
    void onToolCallResult(const std::string& /*toolCallId*/,
                          const std::string& functionName,
                          const std::string& result) override;
 
    void onComplete() override;
 
    void onError(const std::string& error) override;
 
    void setRawOutput(bool raw) { rawOutput_ = raw; }
 
    bool isRawOutput() const { return rawOutput_; }
 
private:
    void outputText(const std::string& text);
 
    RendererType renderer_;
    bool     rawOutput_ = false;
};
 
AST_NAMESPACE_END