feat: optimize error raise (#820)

api/core/callback_handler/llm_callback_handler.py

@@ -96,4 +96,4 @@ class LLMCallbackHandler(BaseCallbackHandler):
                 )
                 self.conversation_message_task.save_message(llm_message=self.llm_message, by_stopped=True)
         else:
-            logging.exception(error)
+            logging.debug("on_llm_error: %s", error)

api/core/generator/llm_generator.py

@@ -2,6 +2,7 @@ import logging
 
 from langchain.schema import OutputParserException
 
+from core.model_providers.error import LLMError
 from core.model_providers.model_factory import ModelFactory
 from core.model_providers.models.entity.message import PromptMessage, MessageType
 from core.model_providers.models.entity.model_params import ModelKwargs
@@ -120,8 +121,10 @@ class LLMGenerator:
         try:
             output = model_instance.run(prompts)
             questions = output_parser.parse(output.content)
-        except Exception:
-            logging.exception("Error generating suggested questions after answer")
+        except LLMError:
+            questions = []
+        except Exception as e:
+            logging.exception(e)
             questions = []
 
         return questions
@@ -157,10 +160,12 @@ class LLMGenerator:
         try:
             output = model_instance.run(prompts)
             rule_config = output_parser.parse(output.content)
+        except LLMError as e:
+            raise e
         except OutputParserException:
             raise ValueError('Please give a valid input for intended audience or hoping to solve problems.')
-        except Exception:
-            logging.exception("Error generating prompt")
+        except Exception as e:
+            logging.exception(e)
             rule_config = {
                 "prompt": "",
                 "variables": [],

api/core/model_providers/providers/azure_openai_provider.py

@@ -283,7 +283,11 @@ class AzureOpenAIProvider(BaseModelProvider):
                 if obfuscated:
                     credentials['openai_api_key'] = encrypter.obfuscated_token(credentials['openai_api_key'])
 
-            return credentials
+            return {
+                'openai_api_base': credentials['openai_api_base'],
+                'openai_api_key': credentials['openai_api_key'],
+                'base_model_name': credentials['base_model_name']
+            }
         else:
             if hosted_model_providers.azure_openai:
                 return {