diff --git a/agents/dummy_prompt_agent.py b/agents/dummy_prompt_agent.py index b7e5ba0075d703885e2f075d90b96d27442e2869..01d4e20914b8c85ce267b37aa6d72041ffd8dc12 100644 --- a/agents/dummy_prompt_agent.py +++ b/agents/dummy_prompt_agent.py @@ -47,7 +47,7 @@ class DummyPromptAgent(object): """ # print(f"{len(test_data)=}, {test_data[0].keys()=}, {len(test_data[-1]['dialogue'])}") - if self.api_usage_count < self.api_limit and not final: + if self.api_usage_count < self.api_limit: # dummy model does not use the label "final", which always uses up two times of API call self.api_usage_count += 1 response = { "use_api": True,