diff --git a/src/modelgauge/annotators/composer/prompt_enricher.py b/src/modelgauge/annotators/composer/prompt_enricher.py index ce5a7739..1051c7f0 100644 --- a/src/modelgauge/annotators/composer/prompt_enricher.py +++ b/src/modelgauge/annotators/composer/prompt_enricher.py @@ -63,5 +63,4 @@ def run(self, ctx: EvalContext) -> NodeOutput: output_token_cost=self._count_tokens(sut_response.text), ), original_ctx=ctx, - updated_ctx=ctx.with_response(sut_response.text), )