akseljoonas HF Staff commited on
Commit
ea67d9f
·
1 Parent(s): 8541221

gitignore and printing args

Browse files
Files changed (2) hide show
  1. .gitignore +3 -1
  2. agent/main.py +5 -1
.gitignore CHANGED
@@ -13,4 +13,6 @@ wheels/
13
  .claude/
14
  *.jsonl
15
  *.csv
16
- /logs
 
 
 
13
  .claude/
14
  *.jsonl
15
  *.csv
16
+ /logs
17
+ hf-agent-leaderboard/
18
+ .cursor/
agent/main.py CHANGED
@@ -3,6 +3,7 @@ Interactive CLI chat with the agent
3
  """
4
 
5
  import asyncio
 
6
  import os
7
  from dataclasses import dataclass
8
  from pathlib import Path
@@ -64,8 +65,11 @@ async def event_listener(
64
  print(f"\n🤖 Assistant: {content}")
65
  elif event.event_type == "tool_call":
66
  tool_name = event.data.get("tool", "") if event.data else ""
 
67
  if tool_name:
68
- print(f"🔧 Calling tool: {tool_name}")
 
 
69
  elif event.event_type == "tool_output":
70
  output = event.data.get("output", "") if event.data else ""
71
  success = event.data.get("success", False) if event.data else False
 
3
  """
4
 
5
  import asyncio
6
+ import json
7
  import os
8
  from dataclasses import dataclass
9
  from pathlib import Path
 
65
  print(f"\n🤖 Assistant: {content}")
66
  elif event.event_type == "tool_call":
67
  tool_name = event.data.get("tool", "") if event.data else ""
68
+ arguments = event.data.get("arguments", {}) if event.data else {}
69
  if tool_name:
70
+ print(
71
+ f"🔧 Calling tool: {tool_name} with arguments: {json.dumps(arguments)[:100]}..."
72
+ )
73
  elif event.event_type == "tool_output":
74
  output = event.data.get("output", "") if event.data else ""
75
  success = event.data.get("success", False) if event.data else False