SSEParser.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstUtil/NetworkStreamReceiver.hpp"
#include "AstUtil/JsonValue.hpp"
#include "AstAI/ChatEventHandler.hpp"
#include <map>
#include <string>
 
AST_NAMESPACE_BEGIN
 
 
class AST_AI_API SSEParser : public NetworkStreamReceiver
{
public:
    explicit SSEParser(ChatEventHandler& handler);
 
    // ── NetworkStreamReceiver 接口 ──
 
    void onHeaders(int statusCode, const std::map<std::string, std::string>& headers) override;
    errc_t onData(const char* data, size_t size) override;
    void onComplete() override;
    void onError(errc_t error) override;
 
    // ── 状态 ──
 
    bool hasError() const { return !error_.empty(); }
 
    const std::string& errorMessage() const { return error_; }
 
    // ── 累积结果 ──
 
    JsonValue buildResult() const;
 
private:
    struct AccumulatedToolCall
    {
        std::string id;
        std::string type;
        std::string functionName;
        std::string functionArguments;
    };
 
    void processBuffer();
    void processEvent(const std::string& event);
    void processToolCallDeltas(const JsonValue& toolCalls);
 
    ChatEventHandler& handler_;
    bool thoughtCompleted_{false};   // 是否已完成推理/思考内容
    std::string       buffer_;
    std::string       error_;
 
    std::string id_;
    std::string model_;
    std::string accumulatedContent_;
    std::string accumulatedReasoning_;
    std::string finishReason_;
    std::map<int, AccumulatedToolCall> toolCallsByIndex_;
};
 
 
AST_NAMESPACE_END