Correct return type

This commit is contained in:
Konstantin Gukov 2023-05-25 17:53:48 +02:00 committed by Richard Guo
parent e362e18431
commit 136a5ac30d

View File

@ -173,7 +173,7 @@ class GPT4All():
default_prompt_footer: bool = True, default_prompt_footer: bool = True,
verbose: bool = True, verbose: bool = True,
streaming: bool = True, streaming: bool = True,
**generate_kwargs) -> str: **generate_kwargs) -> dict:
""" """
Format list of message dictionaries into a prompt and call model Format list of message dictionaries into a prompt and call model
generate on prompt. Returns a response dictionary with metadata and generate on prompt. Returns a response dictionary with metadata and