Browse Source
Fix output parser
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
llama2.py
|
|
@ -67,7 +67,7 @@ class LlamaCpp(PyOperator): |
|
|
return new_messages |
|
|
return new_messages |
|
|
|
|
|
|
|
|
def parse_outputs(self, response): |
|
|
def parse_outputs(self, response): |
|
|
return response['choices'][0]['text'] |
|
|
|
|
|
|
|
|
return response['choices'][0]['message']['content'] |
|
|
|
|
|
|
|
|
@staticmethod |
|
|
@staticmethod |
|
|
def supported_model_names(): |
|
|
def supported_model_names(): |
|
|
|
