Show parameters for tool calls

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__':

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)