From 5b74f4ce8b05455c82492fe198faaffd60c0f256 Mon Sep 17 00:00:00 2001 From: crazywoola <427733928@qq.com> Date: Thu, 17 Apr 2025 09:13:12 +0800 Subject: [PATCH] fix: lint --- api/core/model_runtime/entities/message_entities.py | 1 + api/tests/unit_tests/core/prompt/test_prompt_message.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/api/core/model_runtime/entities/message_entities.py b/api/core/model_runtime/entities/message_entities.py index 0d104204bf..579e68c8a1 100644 --- a/api/core/model_runtime/entities/message_entities.py +++ b/api/core/model_runtime/entities/message_entities.py @@ -123,6 +123,7 @@ PromptMessageContentUnionTypes = Annotated[ Field(discriminator="type"), ] + class PromptMessage(BaseModel): """ Model class for prompt message. diff --git a/api/tests/unit_tests/core/prompt/test_prompt_message.py b/api/tests/unit_tests/core/prompt/test_prompt_message.py index 5de713b81a..e5da51d733 100644 --- a/api/tests/unit_tests/core/prompt/test_prompt_message.py +++ b/api/tests/unit_tests/core/prompt/test_prompt_message.py @@ -24,4 +24,4 @@ def test_dump_prompt_message(): ] ) data = prompt.model_dump() - assert data["content"][0].get("url") == example_url \ No newline at end of file + assert data["content"][0].get("url") == example_url