Commit: cfbf97cd1d5d10fa0fd77ef16b2b582c5463d7c8 - zirpu (git)

Emergent generative agents

Commit MetaInfo

• Revision: cfbf97cd1d5d10fa0fd77ef16b2b582c5463d7c8 (tree)
• Zeit: 2023-05-20 04:04:24
• Autor: Corbin <cds@corb...>
• Commiter: Corbin

Log Message

Well, it works. But it sucks.

Fuck it. Nobody gives a shit anyway.

Ändern Zusammenfassung

• modified: src/agent.py

Diff

--- a/src/agent.py
+++ b/src/agent.py
@@ -132,13 +132,15 @@ class Clock(Agent):
 class LeftBrainInterpreter(Agent):
     tag = "lbi"
     events = 0
-    def __init__(self, mind): self.mind = mind
+    def __init__(self, mind, idle):
+        self.mind = mind
+        self.idle = idle
     def overhear(self, tag, subtag, s):
-        # XXX should skip thoughts, maybe?
-        self.events += 1
+        if tag != "thoughts": self.events += 1
         if self.events >= 20:
             self.events = 0
-            return self.mind.infer(self.tag, self.subtag, "")
+            d = self.mind.infer(self.tag, self.subtag, "")
+            d.addCallback(lambda s: s.strip() or self.idle())
 
 choices = {
     "irc": "Chat on IRC, channel {subtag}",
@@ -146,6 +148,7 @@ choices = {
 }
 def choicify(options):
     return " ".join(f"({c}) {o}" for c, o in zip(ascii_uppercase, options))
+def indexify(c): return ord(c[0].upper()) - ord('A')
 
 class ChoiceMaker(Agent):
     tag = "choice"
@@ -165,11 +168,9 @@ class ChoiceMaker(Agent):
 
         prompt = choicify([choices[tag].format(subtag=subtag)
                            for tag, subtag in possibilities])
-        self.broadcast(prompt)
-        d = self.mind.forceChoice(self.tag, self.subtag, "My choice: ",
+        d = self.mind.forceChoice(self.tag, self.subtag, prompt + " My choice: ",
                                   ascii_uppercase[:len(possibilities)])
-        d.addCallback(lambda s: self.dispatch(*possibilities[chr(ord(s[0]) -
-                                                                 ord('A'))]))
+        d.addCallback(lambda s: self.dispatch(*possibilities[indexify(s)]))
         return d
 
 class ChainOfThoughts(Agent):
@@ -269,7 +270,7 @@ class IRCFactory(ClientFactory):
 def go():
     print("~ Starting tasks…")
     clock = Clock()
-    LoopingCall(clock.go).start(60 * 10, now=False)
+    LoopingCall(clock.go).start(60 * 30, now=True)
 
     for logpath in sys.argv[2:]:
         character = load_character(logpath)
@@ -282,14 +283,14 @@ def go():
         with Timer("initial warmup"):
             mind.logits, mind.state = gen.feedForward(gen.tokenize(firstStatement), None, None)
 
-        lbi = LeftBrainInterpreter(mind)
-        clock.listeners += lbi, mind
-
         cm = ChoiceMaker(mind)
 
+        lbi = LeftBrainInterpreter(mind, cm.idle)
+        clock.listeners += lbi, mind
+
         thoughts = ChainOfThoughts(mind, thoughtIndex, firstStatement, cm.idle)
         thoughts.listeners = lbi, mind
-        LoopingCall(thoughts.go).start(60, now=False)
+        LoopingCall(thoughts.go).start(60 * 3, now=True)
 
         print("~ Thought index:", thoughtIndex.size(), "thoughts")
         factory = IRCFactory(mind, cm, (thoughts, lbi, cm, mind),