learned_knowledge_service.h

Go to the documentation of this file.

#ifndef YAZE_CLI_SERVICE_AGENT_LEARNED_KNOWLEDGE_SERVICE_H_
#define YAZE_CLI_SERVICE_AGENT_LEARNED_KNOWLEDGE_SERVICE_H_
 
#include <filesystem>
#include <map>
#include <optional>
#include <string>
#include <vector>
 
#include "absl/status/status.h"
#include "absl/status/statusor.h"
 
#ifdef YAZE_WITH_JSON
#include "nlohmann/json.hpp"
#endif
 
namespace yaze {
namespace cli {
namespace agent {
 
class LearnedKnowledgeService {
 public:
  LearnedKnowledgeService();
  explicit LearnedKnowledgeService(const std::filesystem::path& data_dir);
  
  // Initialize the service and load existing data
  absl::Status Initialize();
  
  // Save all data to disk
  absl::Status SaveAll();
  
  // === Preference Management ===
  
  absl::Status SetPreference(const std::string& key, const std::string& value);
  
  std::optional<std::string> GetPreference(const std::string& key) const;
  
  std::map<std::string, std::string> GetAllPreferences() const;
  
  absl::Status RemovePreference(const std::string& key);
  
  // === ROM Pattern Learning ===
  
  struct ROMPattern {
    std::string pattern_type;  // e.g., "sprite_location", "tile_usage"
    std::string rom_hash;
    std::string pattern_data;  // JSON encoded
    float confidence = 1.0f;   // 0.0 to 1.0
    int64_t learned_at = 0;
    int access_count = 0;
  };
  
  absl::Status LearnPattern(const std::string& type, 
                            const std::string& rom_hash,
                            const std::string& data,
                            float confidence = 1.0f);
  
  std::vector<ROMPattern> QueryPatterns(const std::string& type,
                                        const std::string& rom_hash = "") const;
  
  absl::Status UpdatePatternConfidence(const std::string& type,
                                      const std::string& rom_hash,
                                      float new_confidence);
  
  // === Project Context ===
  
  struct ProjectContext {
    std::string project_name;
    std::string rom_hash;
    std::string context_data;  // JSON encoded: description, goals, custom labels
    int64_t last_accessed = 0;
  };
  
  absl::Status SaveProjectContext(const std::string& project_name,
                                  const std::string& rom_hash,
                                  const std::string& context);
  
  std::optional<ProjectContext> GetProjectContext(const std::string& project_name) const;
  
  std::vector<ProjectContext> GetAllProjects() const;
  
  // === Conversation Memory ===
  
  struct ConversationMemory {
    std::string id;
    std::string topic;
    std::string summary;
    std::vector<std::string> key_facts;
    int64_t created_at = 0;
    int access_count = 0;
  };
  
  absl::Status StoreConversationSummary(const std::string& topic,
                                        const std::string& summary,
                                        const std::vector<std::string>& key_facts);
  
  std::vector<ConversationMemory> SearchMemories(const std::string& query) const;
  
  std::vector<ConversationMemory> GetRecentMemories(int limit = 10) const;
  
  // === Import/Export ===
  
  absl::StatusOr<std::string> ExportToJSON() const;
  
  absl::Status ImportFromJSON(const std::string& json_data);
  
  absl::Status ClearAll();
  
  // === Statistics ===
  
  struct Stats {
    int preference_count = 0;
    int pattern_count = 0;
    int project_count = 0;
    int memory_count = 0;
    int64_t first_learned_at = 0;
    int64_t last_updated_at = 0;
  };
  
  Stats GetStats() const;
  
 private:
  std::filesystem::path data_dir_;
  std::filesystem::path prefs_file_;
  std::filesystem::path patterns_file_;
  std::filesystem::path projects_file_;
  std::filesystem::path memories_file_;
  
  std::map<std::string, std::string> preferences_;
  std::vector<ROMPattern> patterns_;
  std::vector<ProjectContext> projects_;
  std::vector<ConversationMemory> memories_;
  
  bool initialized_ = false;
  
  // Internal helpers
  absl::Status LoadPreferences();
  absl::Status LoadPatterns();
  absl::Status LoadProjects();
  absl::Status LoadMemories();
  
  absl::Status SavePreferences();
  absl::Status SavePatterns();
  absl::Status SaveProjects();
  absl::Status SaveMemories();
  
  std::string GenerateID() const;
};
 
}  // namespace agent
}  // namespace cli
}  // namespace yaze
 
#endif  // YAZE_CLI_SERVICE_AGENT_LEARNED_KNOWLEDGE_SERVICE_H_