diff --git a/Carrot-Assistant/evaluation/pipelines.py b/Carrot-Assistant/evaluation/pipelines.py index 62d228a..50b066c 100644 --- a/Carrot-Assistant/evaluation/pipelines.py +++ b/Carrot-Assistant/evaluation/pipelines.py @@ -17,4 +17,4 @@ def run(self, input: Dict[str, str]) -> str: prompt = self.prompt_template.render(input) return self._model.create_chat_completion( messages=[{"role": "user", "content": prompt}] - )["choices"][0]["message"] + )["choices"][0]["message"]["content"]