import logging
from collections import defaultdict, namedtuple
from iclbench.agents.base import BaseAgent
LLMResponse = namedtuple(
"LLMResponse", ["model_id", "completion", "stop_reason", "input_tokens", "output_tokens", "reasoning"]
)
[docs]
def make_dummy_action(text):
return LLMResponse(
model_id="dummy", completion="wait", stop_reason="none", input_tokens=1, output_tokens=1, reasoning=None
)
[docs]
class DummyAgent(BaseAgent):
"""
For debugging.
"""
[docs]
def __init__(self, client_factory, prompt_builder):
super().__init__(client_factory, prompt_builder)
self.client = client_factory()
[docs]
def act(self, obs, prev_action=None):
# print("\n", obs["obs"].keys())
# print("_" * 80, "\n", obs["text"]["long_term_context"])
return make_dummy_action("dummy_action")