PilotSession.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include <QObject>
#include <string>
 
AST_NAMESPACE_BEGIN
 
class ChatSession;
class ChatTools;
class PilotAgent;
 
class AST_UIPILOT_API PilotSession : public QObject
{
    Q_OBJECT
 
public:
    explicit PilotSession(PilotAgent* agent, QObject* parent = nullptr);
 
    ~PilotSession() override;
    PilotSession(const PilotSession&) = delete;
    PilotSession& operator=(const PilotSession&) = delete;
 
    std::string execute(const std::string& command);
 
    void chat(const std::string& message, int maxIterForToolCalls=100);
 
    ChatSession& session();
 
    PilotAgent* agent() const { return agent_; }
 
    static const char* systemPrompt();
 
    std::string buildSystemPrompt() const;
 
private:
    void registerTools();
 
    PilotAgent* agent_;
    ChatSession*  chatSession_;
};
 
AST_NAMESPACE_END