jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Message => str (#25876)

Browse source
This commit is contained in:
crazywoola 2025-09-18 17:57:57 +08:00 • committed by GitHub
commit 878420463c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/controllers/web/message.py
View file

@ -261,6 +261,8 @@ class MessageSuggestedQuestionApi(WebApiResource):
questions = MessageService.get_suggested_questions_after_answer( questions = MessageService.get_suggested_questions_after_answer(
app_model=app_model, user=end_user, message_id=message_id, invoke_from=InvokeFrom.WEB_APP app_model=app_model, user=end_user, message_id=message_id, invoke_from=InvokeFrom.WEB_APP
) )
# questions is a list of strings, not a list of Message objects
# so we can directly return it
except MessageNotExistsError: except MessageNotExistsError:
raise NotFound("Message not found") raise NotFound("Message not found")
except ConversationNotExistsError: except ConversationNotExistsError:

4
api/services/message_service.py
View file

@ -217,7 +217,7 @@ class MessageService:
@classmethod @classmethod
def get_suggested_questions_after_answer( def get_suggested_questions_after_answer(
cls, app_model: App, user: Union[Account, EndUser] | None, message_id: str, invoke_from: InvokeFrom cls, app_model: App, user: Union[Account, EndUser] | None, message_id: str, invoke_from: InvokeFrom
) -> list[Message]: ) -> list[str]:
if not user: if not user:
raise ValueError("user cannot be None") raise ValueError("user cannot be None")
@ -288,7 +288,7 @@ class MessageService:
) )
with measure_time() as timer: with measure_time() as timer:
questions: list[Message] = LLMGenerator.generate_suggested_questions_after_answer( questions: list[str] = LLMGenerator.generate_suggested_questions_after_answer(
tenant_id=app_model.tenant_id, histories=histories tenant_id=app_model.tenant_id, histories=histories
) )