tool_registry.h

Go to the documentation of this file.

#ifndef YAZE_CLI_SERVICE_AGENT_TOOL_REGISTRY_H_
#define YAZE_CLI_SERVICE_AGENT_TOOL_REGISTRY_H_
 
#include <functional>
#include <map>
#include <memory>
#include <optional>
#include <string>
#include <vector>
 
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "cli/service/ai/common.h" // For ToolCall
#include "cli/service/resources/command_handler.h"
 
namespace yaze {
class Rom;
namespace cli {
namespace agent {
 
struct ToolDefinition {
  std::string name;
  std::string category;
  std::string description;
  std::string usage;
  std::vector<std::string> examples;
  bool requires_rom = false;
  bool requires_project = false; // New: distinguish project vs ROM dependency
};
 
class ToolRegistry {
 public:
  // Factory function to create a handler instance
  using HandlerFactory = std::function<std::unique_ptr<resources::CommandHandler>()>;
 
  static ToolRegistry& Get();
 
  // Registration
  void RegisterTool(const ToolDefinition& def, HandlerFactory factory);
 
  // Discovery
  std::vector<ToolDefinition> GetAllTools() const;
  std::optional<ToolDefinition> GetToolDefinition(const std::string& name) const;
  std::vector<ToolDefinition> GetToolsByCategory(const std::string& category) const;
 
  // Execution
  absl::StatusOr<std::unique_ptr<resources::CommandHandler>> CreateHandler(
      const std::string& tool_name) const;
 
 private:
  ToolRegistry() = default;
  
  struct ToolEntry {
    ToolDefinition def;
    HandlerFactory factory;
  };
 
  std::map<std::string, ToolEntry> tools_;
};
 
// Helper macro for static registration
#define REGISTER_AGENT_TOOL(Name, Category, Desc, Usage, Examples, ReqRom, ReqProject, HandlerClass) \
  static struct ToolReg_##HandlerClass { \
    ToolReg_##HandlerClass() { \
      yaze::cli::agent::ToolRegistry::Get().RegisterTool( \
        {Name, Category, Desc, Usage, Examples, ReqRom, ReqProject}, \
        []() { return std::make_unique<HandlerClass>(); } \
      ); \
    } \
  } tool_reg_##HandlerClass;
 
}  // namespace agent
}  // namespace cli
}  // namespace yaze
 
#endif  // YAZE_CLI_SERVICE_AGENT_TOOL_REGISTRY_H_