Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(nodes): revert invocation_complete url hack

Browse Source
This commit is contained in:
psychedelicious 2023-05-22 12:46:16 +10:00 committed by Kent Keirsey
parent b77ccfaf32
commit 11bd932cba
2 changed files with 1 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/app/services/events.py
View File

@ -50,8 +50,6 @@ class EventServiceBase:
result: dict, result: dict,
node: dict, node: dict,
source_node_id: str, source_node_id: str,
image_url: Optional[str] = None,
thumbnail_url: Optional[str] = None,
) -> None: ) -> None:
"""Emitted when an invocation has completed""" """Emitted when an invocation has completed"""
self.__emit_session_event( self.__emit_session_event(
@ -61,8 +59,6 @@ class EventServiceBase:
node=node, node=node,
source_node_id=source_node_id, source_node_id=source_node_id,
result=result, result=result,
image_url=image_url,
thumbnail_url=thumbnail_url
), ),
) )

20
invokeai/app/services/processor.py
View File

@ -91,30 +91,12 @@ class DefaultInvocationProcessor(InvocationProcessorABC):
graph_execution_state graph_execution_state
) )
def is_image_output(obj: Any) -> TypeGuard[ImageOutput]:
return obj.__class__ == ImageOutput
outputs_dict = outputs.dict()
if is_image_output(outputs):
image_url = self.__invoker.services.images_new.get_url(
ImageType.RESULT, outputs.image.image_name
)
thumbnail_url = self.__invoker.services.images_new.get_url(
ImageType.RESULT, outputs.image.image_name, True
)
else:
image_url = None
thumbnail_url = None
# Send complete event # Send complete event
self.__invoker.services.events.emit_invocation_complete( self.__invoker.services.events.emit_invocation_complete(
graph_execution_state_id=graph_execution_state.id, graph_execution_state_id=graph_execution_state.id,
node=invocation.dict(), node=invocation.dict(),
source_node_id=source_node_id, source_node_id=source_node_id,
result=outputs_dict, result=outputs.dict(),
image_url=image_url,
thumbnail_url=thumbnail_url,
) )
except KeyboardInterrupt: except KeyboardInterrupt: