chore: boost, basic tool calls proxying, logger revamp

Author: av

boost/src/chat.py

@@ -12,10 +12,12 @@ class Chat:
   tail: ChatNode
   llm: Optional['llm.LLM']
 
+  @staticmethod
   def from_conversation(messages):
     tail = ChatNode.from_conversation(messages)
     return Chat(tail=tail)
 
+  @staticmethod
   def from_tail(chat: 'Chat'):
     new_tail = ChatNode(role=chat.tail.role, content=chat.tail.content)
     return Chat(tail=new_tail)
@@ -52,6 +54,16 @@ def add_message(self, role, content):
 
     return self.tail
 
+  def tool_call(self, tool_call):
+    self.add_message('assistant', '')
+    self.tail.tool_calls = [tool_call]
+    return self.tail
+
+  def tool(self, id, content):
+    self.add_message('tool', content)
+    self.tail.tool_call_id = id
+    return self.tail
+
   def user(self, content):
     return self.add_message('user', content)
 

boost/src/chat_node.py

@@ -10,6 +10,8 @@ class ChatNode:
   id: str
   content: str
   role: str
+  tool_call_id: Optional[str]
+  tool_calls: Optional[List[dict]]
 
   parent: Optional['ChatNode']
   children: List['ChatNode']
@@ -18,6 +20,7 @@ class ChatNode:
   value: float
   meta: dict
 
+  @staticmethod
   def from_conversation(messages):
     root_message = messages[0]
     node = ChatNode(role=root_message['role'], content=root_message['content'])
@@ -44,6 +47,9 @@ def __init__(self, **kwargs):
 
     self.meta = kwargs.get('meta', {})
 
+    self.tool_call_id = kwargs.get('tool_call_id', None)
+    self.tool_calls = kwargs.get('tool_calls', [])
+
   def add_parent(self, new_parent: 'ChatNode'):
     # Guard against None and self-reference
     if new_parent is None or self == new_parent:
@@ -111,11 +117,19 @@ def parents(self):
     return parents[::-1]
 
   def message(self):
-    return {
+    result = {
       "role": self.role,
       "content": self.content,
     }
 
+    if self.tool_call_id is not None:
+      result["tool_call_id"] = self.tool_call_id
+
+    if self.tool_calls:
+      result["tool_calls"] = self.tool_calls
+
+    return result
+
   def ancestor(self):
     node = self
     while node.parent: