diff --git a/pilot/model/proxy/proxy_llms/bard.py b/pilot/model/proxy/proxy_llms/bard.py index dcdb291d9..770a1ec6a 100644 --- a/pilot/model/proxy/proxy_llms/bard.py +++ b/pilot/model/proxy/proxy_llms/bard.py @@ -34,6 +34,5 @@ def bard_generate_stream(model, tokenizer, params, device, context_len=2048): response = bardapi.core.Bard(token).get_answer(last_user_input["content"]) if response is not None and response.get("content") is not None: yield str(response["content"]) - yield f"bard response error: {str(response)}" - - + else: + yield f"bard response error: {str(response)}" diff --git a/pilot/model/proxy/security/sens_data_recognition.py b/pilot/model/proxy/security/sens_data_recognition.py index a0e83cd8e..62a2978a9 100644 --- a/pilot/model/proxy/security/sens_data_recognition.py +++ b/pilot/model/proxy/security/sens_data_recognition.py @@ -1,3 +1,3 @@ """ a tool to discovery sensitive data -""" \ No newline at end of file +"""