feat: Implement advanced routing system for agent tool responses
- Introduced AdvancedRouter class for routing hex analysis, map editing, and palette analysis responses. - Added methods for generating GUI automation scripts and synthesizing multi-tool responses. - Implemented knowledge modules for agent pretraining, covering ROM structure, hex analysis patterns, and tool usage examples. - Enhanced data handling with structured responses, including summaries, detailed data, and next steps for user guidance. - Refactored project build process to utilize std::filesystem for cross-platform patch application.
This commit is contained in:
scawful
208
src/cli/service/agent/advanced_routing.cc
Normal file
208
src/cli/service/agent/advanced_routing.cc
Normal file
@@ -0,0 +1,208 @@
|
||||
#include "cli/service/agent/advanced_routing.h"
|
||||
|
||||
#include <algorithm>
|
||||
#include <sstream>
|
||||
#include "absl/strings/str_format.h"
|
||||
#include "absl/strings/str_join.h"
|
||||
|
||||
namespace yaze {
|
||||
namespace cli {
|
||||
namespace agent {
|
||||
|
||||
AdvancedRouter::RoutedResponse AdvancedRouter::RouteHexAnalysis(
|
||||
const std::vector<uint8_t>& data,
|
||||
uint32_t address,
|
||||
const RouteContext& ctx) {
|
||||
|
||||
RoutedResponse response;
|
||||
|
||||
// Infer data type
|
||||
std::string data_type = InferDataType(data);
|
||||
auto patterns = ExtractPatterns(data);
|
||||
|
||||
// Summary for user
|
||||
response.summary = absl::StrFormat(
|
||||
"Address 0x%06X contains %s (%zu bytes)",
|
||||
address, data_type, data.size());
|
||||
|
||||
// Detailed data for agent with structure hints
|
||||
std::ostringstream detailed;
|
||||
detailed << absl::StrFormat("Raw hex at 0x%06X:\n", address);
|
||||
for (size_t i = 0; i < data.size(); i += 16) {
|
||||
detailed << absl::StrFormat("%06X: ", address + i);
|
||||
for (size_t j = i; j < std::min(i + 16, data.size()); ++j) {
|
||||
detailed << absl::StrFormat("%02X ", data[j]);
|
||||
}
|
||||
detailed << " | ";
|
||||
for (size_t j = i; j < std::min(i + 16, data.size()); ++j) {
|
||||
char c = data[j];
|
||||
detailed << (isprint(c) ? c : '.');
|
||||
}
|
||||
detailed << "\n";
|
||||
}
|
||||
|
||||
if (!patterns.empty()) {
|
||||
detailed << "\nDetected patterns:\n";
|
||||
for (const auto& pattern : patterns) {
|
||||
detailed << "- " << pattern << "\n";
|
||||
}
|
||||
}
|
||||
|
||||
response.detailed_data = detailed.str();
|
||||
|
||||
// Next steps based on data type
|
||||
if (data_type.find("sprite") != std::string::npos) {
|
||||
response.next_steps = "Use resource-list --type=sprite to identify sprite IDs";
|
||||
} else if (data_type.find("tile") != std::string::npos) {
|
||||
response.next_steps = "Use overworld-find-tile to see where this tile appears";
|
||||
} else if (data_type.find("palette") != std::string::npos) {
|
||||
response.next_steps = "Use palette-get-colors to see full palette";
|
||||
} else {
|
||||
response.next_steps = "Use hex-search to find similar patterns in ROM";
|
||||
}
|
||||
|
||||
return response;
|
||||
}
|
||||
|
||||
AdvancedRouter::RoutedResponse AdvancedRouter::RouteMapEdit(
|
||||
const std::string& edit_intent,
|
||||
const RouteContext& ctx) {
|
||||
|
||||
RoutedResponse response;
|
||||
|
||||
// Parse intent and generate action sequence
|
||||
response.summary = "Preparing map edit operation";
|
||||
response.needs_approval = true;
|
||||
|
||||
// Generate GUI automation steps
|
||||
response.gui_actions = {
|
||||
"Click(\"Overworld Editor\")",
|
||||
"Wait(500)",
|
||||
"Click(canvas, x, y)",
|
||||
"SelectTile(tile_id)",
|
||||
"Click(target_x, target_y)",
|
||||
"Wait(100)",
|
||||
"Screenshot(\"after_edit.png\")",
|
||||
};
|
||||
|
||||
response.detailed_data = GenerateGUIScript(response.gui_actions);
|
||||
response.next_steps = "Review proposed changes, then approve or modify";
|
||||
|
||||
return response;
|
||||
}
|
||||
|
||||
AdvancedRouter::RoutedResponse AdvancedRouter::RoutePaletteAnalysis(
|
||||
const std::vector<uint16_t>& colors,
|
||||
const RouteContext& ctx) {
|
||||
|
||||
RoutedResponse response;
|
||||
|
||||
// Analyze color relationships
|
||||
int unique_colors = 0;
|
||||
std::map<uint16_t, int> color_counts;
|
||||
for (uint16_t c : colors) {
|
||||
color_counts[c]++;
|
||||
}
|
||||
unique_colors = color_counts.size();
|
||||
|
||||
response.summary = absl::StrFormat(
|
||||
"Palette has %zu colors (%d unique)",
|
||||
colors.size(), unique_colors);
|
||||
|
||||
// Detailed breakdown
|
||||
std::ostringstream detailed;
|
||||
detailed << "Color breakdown:\n";
|
||||
for (size_t i = 0; i < colors.size(); ++i) {
|
||||
uint16_t snes = colors[i];
|
||||
uint8_t r = (snes & 0x1F) << 3;
|
||||
uint8_t g = ((snes >> 5) & 0x1F) << 3;
|
||||
uint8_t b = ((snes >> 10) & 0x1F) << 3;
|
||||
detailed << absl::StrFormat(" [%zu] $%04X = #%02X%02X%02X\n", i, snes, r, g, b);
|
||||
}
|
||||
|
||||
if (color_counts.size() < colors.size()) {
|
||||
detailed << "\nDuplicates found - optimization possible\n";
|
||||
}
|
||||
|
||||
response.detailed_data = detailed.str();
|
||||
response.next_steps = "Use palette-set-color to modify colors";
|
||||
|
||||
return response;
|
||||
}
|
||||
|
||||
AdvancedRouter::RoutedResponse AdvancedRouter::SynthesizeMultiToolResponse(
|
||||
const std::vector<std::string>& tool_results,
|
||||
const RouteContext& ctx) {
|
||||
|
||||
RoutedResponse response;
|
||||
|
||||
// Combine results intelligently
|
||||
response.summary = absl::StrFormat("Analyzed %zu data sources", tool_results.size());
|
||||
response.detailed_data = absl::StrJoin(tool_results, "\n---\n");
|
||||
|
||||
// Generate insights
|
||||
response.next_steps = "Analysis complete. " + ctx.user_intent;
|
||||
|
||||
return response;
|
||||
}
|
||||
|
||||
std::string AdvancedRouter::GenerateGUIScript(
|
||||
const std::vector<std::string>& actions) {
|
||||
std::ostringstream script;
|
||||
script << "# Generated GUI Automation Script\n";
|
||||
script << "test: \"Automated Edit\"\n";
|
||||
script << "steps:\n";
|
||||
for (const auto& action : actions) {
|
||||
script << " - " << action << "\n";
|
||||
}
|
||||
return script.str();
|
||||
}
|
||||
|
||||
std::string AdvancedRouter::InferDataType(const std::vector<uint8_t>& data) {
|
||||
if (data.size() == 8) return "tile16 data";
|
||||
if (data.size() % 3 == 0 && data.size() <= 48) return "sprite data";
|
||||
if (data.size() == 32) return "palette data (16 colors)";
|
||||
if (data.size() > 1000) return "compressed data block";
|
||||
return "unknown data";
|
||||
}
|
||||
|
||||
std::vector<std::string> AdvancedRouter::ExtractPatterns(
|
||||
const std::vector<uint8_t>& data) {
|
||||
std::vector<std::string> patterns;
|
||||
|
||||
// Check for repeating bytes
|
||||
if (data.size() > 2) {
|
||||
bool all_same = true;
|
||||
for (size_t i = 1; i < data.size(); ++i) {
|
||||
if (data[i] != data[0]) {
|
||||
all_same = false;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (all_same) {
|
||||
patterns.push_back(absl::StrFormat("Repeating byte: 0x%02X", data[0]));
|
||||
}
|
||||
}
|
||||
|
||||
// Check for ascending/descending sequences
|
||||
if (data.size() > 3) {
|
||||
bool ascending = true, descending = true;
|
||||
for (size_t i = 1; i < data.size(); ++i) {
|
||||
if (data[i] != data[i-1] + 1) ascending = false;
|
||||
if (data[i] != data[i-1] - 1) descending = false;
|
||||
}
|
||||
if (ascending) patterns.push_back("Ascending sequence");
|
||||
if (descending) patterns.push_back("Descending sequence");
|
||||
}
|
||||
|
||||
return patterns;
|
||||
}
|
||||
|
||||
std::string AdvancedRouter::FormatForAgent(const std::string& raw_data) {
|
||||
// Format data for easy agent consumption
|
||||
return "```\n" + raw_data + "\n```";
|
||||
}
|
||||
|
||||
} // namespace agent
|
||||
} // namespace cli
|
||||
} // namespace yaze
|
||||
Reference in New Issue
Block a user