mirror of
https://github.com/langgenius/dify.git
synced 2026-05-09 04:36:31 +08:00
Apply suggestions from code review
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
parent
bc82676d93
commit
e819a9a5f7
@ -523,7 +523,8 @@ class AdvancedChatAppGenerator(MessageBasedAppGenerator):
|
||||
with Session(bind=db.engine, expire_on_commit=False) as session:
|
||||
workflow = _refresh_model(session=session, model=workflow)
|
||||
message = _refresh_model(session=session, model=message)
|
||||
assert message is not None
|
||||
if message is None:
|
||||
raise RuntimeError("Failed to refresh Message; _refresh_model returned None.")
|
||||
# workflow_ = session.get(Workflow, workflow.id)
|
||||
# assert workflow_ is not None
|
||||
# workflow = workflow_
|
||||
|
||||
@ -166,8 +166,8 @@ class ChatAppGenerator(MessageBasedAppGenerator):
|
||||
|
||||
# init generate records
|
||||
(conversation, message) = self._init_generate_records(application_generate_entity, conversation)
|
||||
assert conversation is not None
|
||||
assert message is not None
|
||||
if conversation is None or message is None:
|
||||
raise RuntimeError("_init_generate_records() returned None for conversation or message")
|
||||
generated_conversation_id = str(conversation.id)
|
||||
generated_message_id = str(message.id)
|
||||
|
||||
|
||||
@ -149,8 +149,11 @@ class CompletionAppGenerator(MessageBasedAppGenerator):
|
||||
|
||||
# init generate records
|
||||
(conversation, message) = self._init_generate_records(application_generate_entity)
|
||||
assert conversation is not None
|
||||
assert message is not None
|
||||
if conversation is None or message is None:
|
||||
raise RuntimeError(
|
||||
"_init_generate_records() returned None for conversation or message, "
|
||||
"which is required to proceed with generation."
|
||||
)
|
||||
|
||||
# init queue manager
|
||||
queue_manager = MessageBasedAppQueueManager(
|
||||
|
||||
Loading…
Reference in New Issue
Block a user