AssistantAgent.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstAI/ChatAgent.hpp"
#include "AstAI/ChatEventHandler.hpp"
#include "AstAI/LLMClient.hpp"
#include "AstAI/LLMConfig.hpp"
#include "AstAI/ChatTool.hpp"
#include "AstAI/ChatTools.hpp"
#include "AstAI/ChatMessage.hpp"
#include "AstAI/ChatMessages.hpp"
#include <string>
#include <memory>
 
AST_NAMESPACE_BEGIN
 
class AST_AI_API AssistantAgent : public ChatAgent
{
public:
    // —— 工厂方法 ——
 
    static AssistantAgent DefaultAgent();
 
    static AssistantAgent* NewDefaultAgent();
 
    static AssistantAgent SpaceEngineer();
 
    static AssistantAgent* NewSpaceEngineer();
 
public:
    AssistantAgent() = default;
    explicit AssistantAgent(StringView systemPrompt);
    ~AssistantAgent() = default;
    AssistantAgent(AssistantAgent&&) = default;
    AssistantAgent& operator=(AssistantAgent&&) = default;
 
    // —— 身份 ——
 
    void setSystemPrompt(StringView prompt);
    const std::string& systemPrompt() const;
 
    // —— 能力 ——
 
    template<typename Func>
    ChatTool* addTool(Func func)
    {
       return tools_.addTool(func);
    }
 
    void addTool(std::unique_ptr<ChatTool> tool);
 
    ChatTools& tools();
 
    // —— 配置 ——
 
    LLMConfig& config();
    const LLMConfig& config() const;
 
    // —— 客户端 ——
 
    void setClient(std::unique_ptr<LLMClient> client);
    LLMClient& client();
 
    // —— 交互 ——
 
    errc_t run(ChatMessages& messages) override;
 
 
    errc_t run(ChatMessages& messages, int maxSteps);
 
    errc_t runOneStep(ChatMessages& messages);
 
 
    void handleToolCalls(const JsonValue& toolCalls, ChatMessages& messages);
 
    std::string handleToolCall(const JsonValue& toolCall);
 
    // —— 流式交互 ——
    errc_t runStream(ChatMessages& messages, ChatEventHandler& handler) override;
 
 
    errc_t runStream(ChatMessages& messages, ChatEventHandler& handler, int maxSteps);
 
 
    errc_t runOneStepStream(ChatMessages& messages, ChatEventHandler& handler);
 
    const std::string& lastError() const { return lastError_; }
private:
    errc_t runOneStep(const ChatMessages& messages, ChatMessage& response);
 
    JsonValue buildRequestJson(const ChatMessages& messages, bool stream) const;
 
    errc_t parseResponseMessage(JsonValue& response, ChatMessage& outMessage);
 
private:
    std::string                    systemPrompt_;
    ChatTools                      tools_;
    LLMConfig                      config_;
    std::unique_ptr<LLMClient>     client_;                 
    std::string                    lastError_;              
    int                            maxToolIterations_{10};  
};
 
AST_NAMESPACE_END