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

fix: Use Runnable's asynchronous methods to prevent blocking (#9466)

Browse source 
Co-authored-by: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
This commit is contained in:
bingchuan 2025-08-22 01:16:00 +08:00 • committed by GitHub
commit 74080df2af
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/base/models/model.py
View file

@ -252,7 +252,7 @@ class LCModelComponent(Component):
if stream: if stream:
lf_message, result = await self._handle_stream(runnable, inputs) lf_message, result = await self._handle_stream(runnable, inputs)
else: else:
message = runnable.invoke(inputs) message = await runnable.ainvoke(inputs)
result = message.content if hasattr(message, "content") else message result = message.content if hasattr(message, "content") else message
if isinstance(message, AIMessage): if isinstance(message, AIMessage):
status_message = self.build_status_message(message) status_message = self.build_status_message(message)
@ -288,7 +288,7 @@ class LCModelComponent(Component):
else: else:
session_id = None session_id = None
model_message = Message( model_message = Message(
text=runnable.stream(inputs), text=runnable.astream(inputs),
sender=MESSAGE_SENDER_AI, sender=MESSAGE_SENDER_AI,
sender_name="AI", sender_name="AI",
properties={"icon": self.icon, "state": "partial"}, properties={"icon": self.icon, "state": "partial"},
@ -298,7 +298,7 @@ class LCModelComponent(Component):
lf_message = await self.send_message(model_message) lf_message = await self.send_message(model_message)
result = lf_message.text result = lf_message.text
else: else:
message = runnable.invoke(inputs) message = await runnable.ainvoke(inputs)
result = message.content if hasattr(message, "content") else message result = message.content if hasattr(message, "content") else message
return lf_message, result return lf_message, result