feat: update prompt generate (#6516)

2024-07-23 19:52:14 +08:00
parent 0f6a064c08
commit 8123a00e97
5 changed files with 219 additions and 95 deletions
--- a/api/core/model_runtime/model_providers/tongyi/llm/llm.py
+++ b/api/core/model_runtime/model_providers/tongyi/llm/llm.py
@@ -262,6 +262,10 @@ You should also complete the text started with ``` but not tell ``` directly.
        :param prompt_messages: prompt messages
        :return: llm response
        """
+        if response.status_code != 200 and response.status_code != HTTPStatus.OK:
+            raise ServiceUnavailableError(
+                response.message
+            )
        # transform assistant message to prompt message
        assistant_prompt_message = AssistantPromptMessage(
            content=response.output.choices[0].message.content,