agent_context.hpp

Go to the documentation of this file.

#pragma once
 
#include <map>
 
#include "async_result.hpp"
#include "callback.hpp"
#include "config.hpp"
#include "push.hpp"
#include "types.hpp"
 
typedef struct lb_agent_context_t lb_agent_context_t;
 
namespace longbridge {
namespace agent {
 
using AnswersByToolCall = std::map<std::string, std::map<std::string, std::string>>;
 
class AgentContext
{
private:
  const lb_agent_context_t* ctx_;
 
public:
  AgentContext();
  AgentContext(const lb_agent_context_t* ctx);
  AgentContext(const AgentContext& ctx);
  AgentContext(AgentContext&& ctx);
  ~AgentContext();
 
  AgentContext& operator=(const AgentContext& ctx);
 
  static AgentContext create(const Config& config);
 
  void workspaces(AsyncCallback<AgentContext, WorkspacesResponse> callback) const;
 
  void agents(const std::string& workspace_id,
             const std::optional<GetAgentsOptions>& opts,
             AsyncCallback<AgentContext, AgentsResponse> callback) const;
 
  void conversation(const std::string& agent_id,
                    const std::string& query,
                    const std::optional<std::string>& chat_uid,
                    AsyncCallback<AgentContext, ConversationResponse> callback) const;
 
  void continue_conversation(const std::string& agent_id,
                             const std::string& chat_uid,
                             const std::string& message_id,
                             const AnswersByToolCall& answers,
                             AsyncCallback<AgentContext, ConversationResponse> callback) const;
 
  void conversation_streamed(
    const std::string& agent_id,
    const std::string& query,
    const std::optional<std::string>& chat_uid,
    PushCallback<AgentContext, ConversationStreamEvent> on_event,
    AsyncCallback<AgentContext, ConversationResponse> callback) const;
 
  void continue_conversation_streamed(
    const std::string& agent_id,
    const std::string& chat_uid,
    const std::string& message_id,
    const AnswersByToolCall& answers,
    PushCallback<AgentContext, ConversationStreamEvent> on_event,
    AsyncCallback<AgentContext, ConversationResponse> callback) const;
};
 
} // namespace agent
} // namespace longbridge