From d035642a136aee85f2474e038c6d94af20b5a088 Mon Sep 17 00:00:00 2001 From: Wenshansilvia Date: Wed, 2 Oct 2024 20:52:18 +0800 Subject: [PATCH] update bleu corpus level and gpt 3.5 --- rageval/utils/check_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rageval/utils/check_utils.py b/rageval/utils/check_utils.py index 2e8b817..cee353c 100644 --- a/rageval/utils/check_utils.py +++ b/rageval/utils/check_utils.py @@ -21,13 +21,13 @@ def text_to_sents(text: str, model_name="nltk") -> List[str]: sentences = [s.strip() for s in sentences if len(s.strip()) >= 3] elif model_name == "gpt-3.5-turbo": - model = OpenAILLM("gpt-3.5-turbo-16k", "OPENAI_API_KEY") # pragma: no cover + model = OpenAILLM("gpt-3.5-turbo", "OPENAI_API_KEY") # pragma: no cover prompt = DOC_TO_SENTENCES_PROMPT # pragma: no cover input_str = prompt.format(doc=text).strip() # pragma: no cover r = model.generate([input_str]) # pragma: no cover sentences = eval(r) # pragma: no cover else: - logger.info("The parameter `model_name` should be in [`nltk`, `gpt-3.5-turbo-16k`]. ") # pragma: no cover + logger.info("The parameter `model_name` should be in [`nltk`, `gpt-3.5-turbo`]. ") # pragma: no cover assert isinstance(sentences, list) return sentences