From a5b4a3131df3319ed2cdc1f1cb63ee95a1fbc517 Mon Sep 17 00:00:00 2001 From: James Mitchell-White Date: Fri, 18 Oct 2024 16:36:40 +0100 Subject: [PATCH] Update pipelines.py Modified to pass tests - now returns a string --- Carrot-Assistant/evaluation/pipelines.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"]