From 535c315e2d978d9b4a3e5c6c74a18ede79b23433 Mon Sep 17 00:00:00 2001 From: Sourcery AI <> Date: Tue, 2 May 2023 07:13:27 +0000 Subject: [PATCH] 'Refactored by Sourcery' --- athena/input_processor.py | 23 +++++++++++------------ 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/athena/input_processor.py b/athena/input_processor.py index 302438a..12c1056 100644 --- a/athena/input_processor.py +++ b/athena/input_processor.py @@ -100,18 +100,17 @@ def process_input( ] } ) + elif completion_callback == fastchat_chat_completion: + logger.debug( + "No intent was detected. Using fast-chat to generate a response." + ) + response = completion_callback( + "fastchat-t5-3b-v1.0", prompt, temperature=0.8, max_tokens=512 + ) else: - if completion_callback == fastchat_chat_completion: - logger.debug( - "No intent was detected. Using fast-chat to generate a response." - ) - response = completion_callback( - "fastchat-t5-3b-v1.0", prompt, temperature=0.8, max_tokens=512 - ) - else: - logger.debug( - "No intent was detected. Using GPT-3 to generate a response." - ) - response = completion_callback(prompt) + logger.debug( + "No intent was detected. Using GPT-3 to generate a response." + ) + response = completion_callback(prompt) logger.info(f"Response generated: {response}") return response