WulaFallenEmpireRW/Source/WulaFallenEmpire/EventSystem/AI/Tools/Tool_RecallMemories.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using RimWorld;
using Verse;

namespace WulaFallenEmpire.EventSystem.AI.Tools
{
    public class Tool_RecallMemories : AITool
    {
        public override string Name => "recall_memories";
        public override string Description => "Searches the AI's long-term memory for facts matching a specific query or keyword.";
        public override string UsageSchema => "{\"query\":\"keywords\",\"limit\":5}";
        public override Dictionary<string, object> GetParametersSchema()
        {
            var properties = new Dictionary<string, object>
            {
                ["query"] = SchemaString("Search query.", nullable: true),
                ["limit"] = SchemaInteger("Max memories to return.", nullable: true)
            };
            return SchemaObject(properties, RequiredList("query", "limit"));
        }

        public override string Execute(string args)
        {
            var argsDict = ParseJsonArgs(args);
            string query = TryGetString(argsDict, "query", out string q) ? q : "";
            int limit = TryGetInt(argsDict, "limit", out int parsedLimit) ? parsedLimit : 5;

            var memoryManager = Find.World?.GetComponent<AIMemoryManager>();
            if (memoryManager == null)
            {
                return "Error: AIMemoryManager world component not found.";
            }

            if (string.IsNullOrWhiteSpace(query))
            {
                var recent = memoryManager.GetRecentMemories(limit);
                if (recent.Count == 0) return "No recent memories found.";
                return FormatMemories(recent);
            }

            var results = memoryManager.SearchMemories(query, limit);
            if (results.Count == 0)
            {
                return "No memories found matching the query.";
            }

            return FormatMemories(results);
        }

        private string FormatMemories(List<AIMemoryEntry> memories)
        {
            StringBuilder sb = new StringBuilder();
            sb.AppendLine("Found Memories:");
            foreach (var m in memories)
            {
                sb.AppendLine($"- [{m.Category}] {m.Fact} (ID: {m.Id})");
            }
            return sb.ToString();
        }
    }
}