mirror of
https://github.com/langgenius/dify-plugin-daemon.git
synced 2026-07-22 17:56:00 -04:00
Yeuoly
31d7c7417e
* feat: introduce 'run' command for local plugin execution - Added a new command `run` to launch plugins locally, allowing communication through stdin/stdout. - Removed the previous `test` command and its associated functionality to streamline the plugin testing process. * feat: enhance 'run' command with TCP support - Introduced a new `RunPluginPayload` structure to encapsulate plugin execution parameters. - Added TCP communication mode to the `run` command, allowing multiple client connections. - Updated command flags to configure run mode and logging options. - Implemented client handling and server creation for both stdin/stdout and TCP modes. * docs: improve comments in RunPlugin function for clarity - Enhanced comments to provide clearer explanations of the plugin decoding process and the creation of client streams for both stdin/stdout and TCP modes. - Updated comments to reflect the functionality and behavior of the plugin execution flow. * refactor: update command structure and enhance plugin invocation handling - Changed the command structure to add `runPluginCommand` under `pluginCommand` for better organization. - Introduced `InvokePluginPayload` type to encapsulate plugin invocation details. - Enhanced `RunPlugin` function to handle responses and errors more effectively, including logging to stdout. - Updated client handling to support session management and improved error handling during plugin invocation. - Renamed TCP server creation function for consistency. * feat: enhance plugin response handling and logging capabilities - Added a new flag `--response-format` to specify the output format (text or json) for plugin responses. - Introduced a `logger` to manage logging output to stdout with timestamps and file information. - Updated `logResponse` and `systemLog` functions to handle different response formats. - Enhanced `handleClient` and `RunPlugin` functions to utilize the new response format feature. - Implemented signal handling to clean up temporary directories on shutdown. * feat: enhance plugin response structure and logging - Added `InvokeID` to `InvokePluginPayload` and `GenericResponse` for better tracking of plugin invocations. - Updated `logResponse` to include `InvokeID` in error responses for improved debugging. - Enhanced client handling in `handleClient` to log plugin readiness and received requests. - Refactored client stream creation for better readability and consistency. * feat: add plugin invoke end response type and logging - Introduced `GENERIC_RESPONSE_TYPE_PLUGIN_INVOKE_END` to enhance response tracking for plugin invocations. - Updated `handleClient` to log the end of plugin invocation, improving visibility into the plugin lifecycle. * chore: remove fullfeature tags
56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
package integration_test
|
|
|
|
import (
|
|
_ "embed"
|
|
"testing"
|
|
|
|
"github.com/langgenius/dify-plugin-daemon/internal/core/plugin_daemon/access_types"
|
|
"github.com/langgenius/dify-plugin-daemon/internal/core/plugin_manager/test_utils"
|
|
"github.com/langgenius/dify-plugin-daemon/internal/utils/parser"
|
|
"github.com/langgenius/dify-plugin-daemon/internal/utils/routine"
|
|
"github.com/langgenius/dify-plugin-daemon/pkg/entities/agent_entities"
|
|
"github.com/langgenius/dify-plugin-daemon/pkg/entities/requests"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
//go:embed testdata/langgenius-agent_0.0.14.difypkg
|
|
var difyOfficialAgent []byte
|
|
|
|
//go:embed testdata/invoke_agent_strategy_json.json
|
|
var invokeAgentStrategyJson []byte
|
|
|
|
const (
|
|
_testingPath = "./integration_test_cwd"
|
|
)
|
|
|
|
func TestDifyOfficialAgentIntegration(t *testing.T) {
|
|
routine.InitPool(10000)
|
|
|
|
defer test_utils.ClearTestingPath(_testingPath)
|
|
|
|
runtime, err := test_utils.GetRuntime(difyOfficialAgent, _testingPath)
|
|
assert.NoError(t, err)
|
|
|
|
invokePayload, err := parser.UnmarshalJsonBytes2Map(invokeAgentStrategyJson)
|
|
assert.NoError(t, err)
|
|
response, err := test_utils.RunOnce[requests.RequestInvokeAgentStrategy, agent_entities.AgentStrategyResponseChunk](
|
|
runtime,
|
|
access_types.PLUGIN_ACCESS_TYPE_AGENT_STRATEGY,
|
|
access_types.PLUGIN_ACCESS_ACTION_INVOKE_AGENT_STRATEGY,
|
|
requests.RequestInvokeAgentStrategy{
|
|
InvokeAgentStrategySchema: requests.InvokeAgentStrategySchema{
|
|
AgentStrategyProvider: "agent",
|
|
AgentStrategy: "function_calling",
|
|
AgentStrategyParams: invokePayload,
|
|
},
|
|
},
|
|
)
|
|
|
|
assert.NoError(t, err)
|
|
|
|
for response.Next() {
|
|
_, err := response.Read()
|
|
assert.NoError(t, err)
|
|
}
|
|
}
|