gdms commited on
Commit
44a5fa9
·
1 Parent(s): e7d04f7

Acerto chess e submissao

Browse files
Files changed (2) hide show
  1. agent.py +2 -1
  2. tools.py +1 -2
agent.py CHANGED
@@ -64,8 +64,9 @@ class Agent:
64
  },
65
  ):
66
  Agent_Util.pretty_print_messages(chunk, last_message=True)
 
67
 
68
- agent_answer = chunk["supervisor"]["messages"]
69
 
70
  final_answer = re.sub(r"^FINAL ANSWER:\s*", "", agent_answer, flags=re.IGNORECASE)
71
  print(f"Agent returning answer for task {task_id}: {final_answer}")
 
64
  },
65
  ):
66
  Agent_Util.pretty_print_messages(chunk, last_message=True)
67
+ final_chunk = chunk
68
 
69
+ agent_answer = final_chunk["supervisor"]["messages"]
70
 
71
  final_answer = re.sub(r"^FINAL ANSWER:\s*", "", agent_answer, flags=re.IGNORECASE)
72
  print(f"Agent returning answer for task {task_id}: {final_answer}")
tools.py CHANGED
@@ -151,8 +151,7 @@ def chess_fen_get_best_next_move_tool(fen: str, current_player: Literal["black",
151
 
152
  url = CHESS_MOVE_API
153
  payload = {
154
- "fen": fen,
155
- "depth": 1
156
  }
157
 
158
  print(f"Buscando melhor jogada em {CHESS_MOVE_API} - {payload}")
 
151
 
152
  url = CHESS_MOVE_API
153
  payload = {
154
+ "fen": fen
 
155
  }
156
 
157
  print(f"Buscando melhor jogada em {CHESS_MOVE_API} - {payload}")