From bcf90aced49e3f4151270b7f932575870411dc25 Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Mon, 31 Jul 2023 18:00:57 +0800
Subject: [PATCH] Fix output parser

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 llama2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llama2.py b/llama2.py
index ababbde..607be30 100644
--- a/llama2.py
+++ b/llama2.py
@@ -67,7 +67,7 @@ class LlamaCpp(PyOperator):
         return new_messages
     
     def parse_outputs(self, response):
-        return response['choices'][0]['text']
+        return response['choices'][0]['message']['content']
 
     @staticmethod
     def supported_model_names():