acecode/src/session/local_session_client.hpp at master · tmoonlight/acecode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
#pragma once

// LocalSessionClient: SessionClient 的同进程实现。所有方法直接转发到
// SessionRegistry + AgentLoop + AsyncPrompter。daemon HTTP handler 通过这个
// 客户端跟 worker 沟通。
//
// 浏览器侧的 RemoteSessionClient(走 HTTP/WebSocket 的 wire format)由
// add-web-chat-ui change 落地,接口契约相同。

#include "session_client.hpp"
#include "session_registry.hpp"

#include <memory>

namespace acecode {

class LocalSessionClient : public SessionClient {
public:
    explicit LocalSessionClient(SessionRegistry& registry) : registry_(registry) {}

    std::string create_session(const SessionOptions& opts) override;
    bool resume_session(const std::string& id, const SessionOptions& opts = {}) override;
    std::vector<SessionInfo> list_sessions() override;
    void destroy_session(const std::string& id) override;
    SubscriptionId subscribe(const std::string& session_id,
                              EventListener on_event,
                              std::uint64_t since_seq = 0) override;
    void unsubscribe(const std::string& session_id, SubscriptionId sub) override;
    bool send_input(const std::string& session_id, const std::string& text) override;
    bool send_input(const std::string& session_id,
                      const std::string& text,
                      const std::string& display_text) override;
    bool send_input(const std::string& session_id, const UserInput& input) override;
    TurnSteerResult steer_input(const std::string& session_id,
                                const std::string& expected_turn_id,
                                const UserInput& input) override;
    TurnSteerResult interrupt_turn(const std::string& session_id,
                                   const std::string& expected_turn_id,
                                   const UserInput& input) override;
    BuiltinCommandResult execute_builtin_command(
        const std::string& session_id,
        const BuiltinCommandRequest& request) override;
    void respond_permission(const std::string& session_id,
                              const PermissionDecision& decision) override;
    QuestionResponseStatus respond_question(
        const std::string& session_id,
        const std::string& request_id,
        const AskUserQuestionResponse& response) override;
    std::optional<std::vector<PendingQuestionRequestSnapshot>>
    snapshot_pending_questions(const std::string& session_id) override;
    void abort(const std::string& session_id) override;

private:
    SessionRegistry& registry_;
};

} // namespace acecode