agent_pretraining.h

Go to the documentation of this file.

#ifndef YAZE_CLI_SERVICE_AGENT_AGENT_PRETRAINING_H_
#define YAZE_CLI_SERVICE_AGENT_AGENT_PRETRAINING_H_
 
#include <string>
#include <vector>
#include "absl/status/status.h"
 
namespace yaze {
class Rom;
 
namespace cli {
namespace agent {
 
class AgentPretraining {
 public:
  struct KnowledgeModule {
    std::string name;
    std::string content;
    bool required;
  };
  
  static std::vector<KnowledgeModule> GetModules();
  
  static std::string GetRomStructureKnowledge(Rom* rom);
  
  static std::string GetHexAnalysisKnowledge();
  
  static std::string GetMapEditingKnowledge();
  
  static std::string GetToolUsageExamples();
  
  static std::string GeneratePretrainingPrompt(Rom* rom);
};
 
}  // namespace agent
}  // namespace cli
}  // namespace yaze
 
#endif  // YAZE_CLI_SERVICE_AGENT_AGENT_PRETRAINING_H_