The AI innkeeper

using System.Text.Json;
using System.Text.Json.Serialization;
using Kallamo.Umwelt;
using var engine = UmweltEngine.Create("drowned-lantern.umw");
const string projectId = "drowned-lantern";
var provenance = Provenance.Human();
var json = new JsonSerializerOptions
{
    DefaultIgnoreCondition = JsonIgnoreCondition.WhenWritingNull,
    Converters = { new JsonStringEnumConverter(JsonNamingPolicy.KebabCaseLower) },
};
void Print(string label, object? value) => Console.WriteLine($"@@{label}\n{JsonSerializer.Serialize(value, json)}");

// The world, declared once: sending the same manifest again changes nothing.
engine.WorldEnsure(new EnsureWorldInput
{
    Project = new WorldProject { Id = projectId, Name = "The Drowned Lantern", Timescale = "day" },
    EntityTypes = new()
    {
        new WorldEntityType { Id = "person", Name = "Person", Kind = EntityKind.Agent },
        new WorldEntityType { Id = "thing", Name = "Thing", Kind = EntityKind.Other },
    },
    Entities = new()
    {
        new WorldEntity { Id = "marta", TypeId = "person", Name = "Marta, the innkeeper" },
        new WorldEntity { Id = "silas", TypeId = "person", Name = "Silas, the lighthouse keeper" },
        new WorldEntity { Id = "bram", TypeId = "person", Name = "Bram, a sailor" },
        new WorldEntity { Id = "rourke", TypeId = "person", Name = "Captain Rourke" },
        new WorldEntity { Id = "tobin", TypeId = "person", Name = "Tobin, Marta’s brother" },
        new WorldEntity { Id = "albatross", TypeId = "thing", Name = "The Albatross" },
        new WorldEntity { Id = "reef", TypeId = "thing", Name = "The reef" },
    },
    Predicates = new()
    {
        new PredicateDef { Id = "sank-on", Label = "sank on", Object = ObjectKind.Entity, Cardinality = Cardinality.Many },
        new PredicateDef { Id = "lit-false-light", Label = "lit false light", Object = ObjectKind.None, Cardinality = Cardinality.Many },
        new PredicateDef { Id = "survived", Label = "survived", Object = ObjectKind.None, Cardinality = Cardinality.Many },
        new PredicateDef { Id = "sailed-on", Label = "sailed on", Object = ObjectKind.Entity, Cardinality = Cardinality.Many },
        new PredicateDef { Id = "carried", Label = "carried", Object = ObjectKind.Value, Cardinality = Cardinality.Many },
    },
});
var sank = Claim.Entity("albatross", "sank-on", "reef");
var falseLight = Claim.Unary("silas", "lit-false-light");
var survived = Claim.Unary("rourke", "survived");
var tobinAboard = Claim.Entity("tobin", "sailed-on", "albatross");
var rum = Claim.Value("albatross", "carried", "rum");

// A scene is one event. Its records take their position and provenance from it.
void Scene(string id, string title, long tick, List<string> participants, List<RecordInput> records) => engine.Commit(new CommitInput
{
    ProjectId = projectId, AttemptId = id,
    Event = new EventInput { Id = id, Type = "scene", Title = title, Position = Position.At(tick), Participants = participants, Provenance = provenance },
    Records = records,
});

Scene("wreck", "The Albatross goes down", 1, new() { "marta", "silas" }, new()
{
    Records.Fact(sank, id: "canon-sank"),
    Records.Fact(falseLight, id: "canon-light"),
    Records.Fact(tobinAboard, id: "canon-tobin"),
    Records.Knows("marta", sank, Acquired.Witnessed("wreck"), id: "marta-wreck", text: "I watched the Albatross break on the reef from the harbor wall."),
    Records.Knows("marta", tobinAboard, Acquired.Witnessed("wreck"), id: "marta-tobin", text: "My brother Tobin was aboard her.", accessibility: Accessibility.Suppressed),
    Records.Knows("silas", falseLight, Acquired.Witnessed("wreck"), id: "silas-light", text: "I lit the false light that drew her onto the reef."),
});
Scene("rescue", "A man on the shingle", 2, new() { "bram", "rourke" }, new()
{
    Records.Fact(survived, id: "canon-survived"),
    Records.Fact(rum, id: "canon-rum"),
    Records.Knows("bram", survived, Acquired.Witnessed("rescue"), id: "bram-saw", text: "I pulled the captain out of the surf myself."),
});
engine.Transmit(new TransmitInput
{
    ProjectId = projectId, AttemptId = "gossip", From = "bram", To = "marta", SourceRecordId = "bram-saw", Intent = Intent.Honest,
    Attitude = Attitude.Believes, Confidence = 0.6, Position = Position.At(2), RecordId = "marta-captain",
    Text = "Bram swears the captain crawled ashore alive.",
});
Scene("ledger", "The ledger", 5, new() { "marta" }, new()
{
    Records.Knows("marta", rum, Acquired.Witnessed("ledger"), id: "marta-rum", text: "The ledger says her hold was full of rum."),
});
// Evening of day 3. The player asks Marta about the wreck. Your retrieval gathers candidates:
// what the narrator's search finds, what Marta's own search finds, and a few records it ranks high.
var now = new At { Tick = 3 };
var narrator = engine.Search(new SearchPayload { ProjectId = projectId, Text = "albatross", At = now });
var hers = engine.Search(new SearchPayload { ProjectId = projectId, Text = "albatross", At = now, Filter = new SearchFilter { Holder = "marta" } });
Print("search", new { narrator = narrator.Items.Select(i => i.DocId), marta = hers.Items.Select(i => i.DocId) });

// Your cost: a rough token count of the text you would put in the prompt.
var texts = new Dictionary<string, string>
{
    ["marta-wreck"] = "I watched the Albatross break on the reef from the harbor wall.",
    ["marta-captain"] = "Bram swears the captain crawled ashore alive.",
    ["marta-tobin"] = "My brother Tobin was aboard her.",
    ["silas-light"] = "I lit the false light that drew her onto the reef.",
    ["marta-rum"] = "The ledger says her hold was full of rum.",
    ["canon-sank"] = "The Albatross sank on the reef.",
    ["wreck"] = "Scene: the Albatross goes down.",
};
uint Cost(string id) => (uint)Math.Ceiling((texts.TryGetValue(id, out var text) ? text : id).Length / 4.0);
ContextCandidate Record(string id, int priority) => new() { Id = id, RecordId = id, Cost = Cost(id), Priority = priority };
var candidates = new List<ContextCandidate>
{
    Record("marta-wreck", 5),
    new() { Id = "wreck", EventId = "wreck", Cost = Cost("wreck"), Priority = 4 },
    Record("marta-captain", 3), Record("canon-sank", 3), Record("silas-light", 2), Record("marta-tobin", 2), Record("marta-rum", 1),
    new() { Id = "rescue", EventId = "rescue", Cost = 9, Priority = 1 },
};
var admitted = engine.ContextAdmit(new ContextAdmitInput { ProjectId = projectId, Holder = "marta", At = now, Budget = 30, Candidates = candidates });
Print("admit", admitted);

// Your prompt, from what was admitted, highest priority first.
var prompt = string.Join("\n", admitted.Admitted.Select(c => $"- {(texts.TryGetValue(c.Id, out var text) ? text : c.Id)}")
    .Prepend("You are Marta, innkeeper of the Drowned Lantern. You know only this:"));
Print("prompt", prompt);
// The model answers the player. (Here the reply is written into the program, so the recipe
// runs without a model.) Your extractor turns each sentence into a claim; Umwelt checks it.
var reply = new (string Line, Claim Claim)[]
{
    ("Aye, the Albatross broke up on the reef, I saw it with my own eyes.", sank),
    ("They say her captain made it to shore.", survived),
    ("And between us, old Silas lit a false light that night.", falseLight),
    ("Mind you, she was running rum.", rum),
};
var check = engine.CheckUtterance(new CheckUtteranceInput { ProjectId = projectId, Holder = "marta", At = now, Claims = reply.Select(r => r.Claim).ToList() });
var lines = reply.Select((r, i) => new { line = r.Line, verdict = check.Items[i].Verdict, canon = check.Items[i].Canon }).ToList();
Print("reply", lines);
// Your rule: keep what she holds, cut the rest and ask the model again.
Print("kept", lines.Where(l => l.verdict == UtteranceVerdict.Holds).Select(l => l.line));