From 5e3bddf60e500053da4eb18d74d433249f20ac40 Mon Sep 17 00:00:00 2001
From: Jon Michael Aanes <jonjmaa@gmail.com>
Date: Fri, 9 May 2025 01:38:12 +0200
Subject: [PATCH] Show parameters for tool calls

---
 main.py  | 5 +++--
 tools.py | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index fb44133..0a3945c 100644
--- a/main.py
+++ b/main.py
@@ -40,10 +40,11 @@ def main():
         result = llm.invoke({
             'messages': messages,
         })
-        for msg in result['messages']:
+        messages = result['messages']
+        for msg in messages[prev_idx:]:
             print(f'{msg.type}: {msg.content}')
-            messages.append(msg)
             del msg
+        prev_idx = len(messages)
 
 
 if __name__ == '__main__':
diff --git a/tools.py b/tools.py
index 3895e20..33a41dd 100644
--- a/tools.py
+++ b/tools.py
@@ -21,7 +21,7 @@ def wrap(class_, method):
     logger.debug("Wrapping %s.%s", class_.__name__, method.__name__)
     is_iterator = str(method.__annotations__.get('return', '')).startswith('collections.abc.Iterator')
     def wrapper(input):
-        logger.info("AI called %s.%s", class_.__name__, method.__name__)
+        logger.info("AI called %s.%s(%s)", class_.__name__, method.__name__, input)
         result = method(input)
         if is_iterator:
             result = list(result)