Запрос: _invoke () висит при выполнении запроса с плагином DIFY SDK ⇐ Python

[Anonymous]

Запрос: _invoke () висит при выполнении запроса с плагином Dify Sdk
Я в настоящее время разрабатываю плагин, используя плагин DIFY SDK, который подключается к базе данных, выполняет запрос и возвращает результат. Тем не менее, я испытываю проблему с подвеской внутри метода _invoke (), когда я пытаюсь выполнить запрос и вернуть результат. В конце концов, возникает ошибка соединения. < /P>
Ситуация < /p>

Соединение базы данных, выполнение запроса и поиск результатов работает правильно. Время и, наконец, приводит к тайм-ауту или ошибке подключения. < /p>
< /li>
< /ul>
Код, который работает, как и ожидалось < /p>
from collections.abc import Generator
from typing import Any

from dify_plugin import Tool
from dify_plugin.entities.tool import ToolInvokeMessage

from tools.snowflake_client import SnowflakeClient

class SnowflakeQueryTool(Tool):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.sf_client = SnowflakeClient(self.runtime.credentials)

def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage]:
yield self.create_text_message("Hello, world!")
< /code>
Этот основной пример успешно работает без каких-либо проблем.from collections.abc import Generator
from typing import Any

from dify_plugin import Tool
from dify_plugin.entities.tool import ToolInvokeMessage

from tools.snowflake_client import SnowflakeClient

class SnowflakeQueryTool(Tool):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.sf_client = SnowflakeClient(self.runtime.credentials)

def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage]:
res = self.sf_client.execute_query("SELECT CURRENT_VERSION()")
for row in res:
print(row)

yield self.create_text_message("Hello, world!")

< /code>
Print (row) выводит ожидаемый результат, который подтверждает, что соединение снежинки работает должным образом. < /p>
Однако при тестировании плагина через интерфейс тестирования плагина Dify плагин застрял в описании «Запуск» и в конечном итоге сбой с ошибкой соединения. Наблюдения < /p>

Если я вручную нажимаю на остановку, а затем снова запускаю, плагин успешно отвечает. /> Допущения и запрос < /p>

Я подозреваю это. < /p>
< /li>
< /ul>
Может ли это быть проблемой в самой Dify SDK, где вызов урожайности или базы данных мешает реакциям и совет, как решить или решить эту проблему будет отличным.>

Подробнее здесь: https://stackoverflow.com/questions/795 ... plugin-sdk