02e813af0b
- Updated `logrus` dependency from v1.9.3 to v1.9.4 and `golang.org/x/sys` from v0.38.0 to v0.40.0 for improved functionality and security. - Introduced a `Fork` method in the agent context to create child contexts for concurrent agent and LLM calls, preventing race conditions on shared state during batch operations. - Enhanced the `Orchestrator` methods to utilize forked contexts, ensuring independent execution of agent calls without interference.
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package testutils
|
|
|
|
import (
|
|
"testing"
|
|
|
|
_ "github.com/yaoapp/gou/encoding"
|
|
"github.com/yaoapp/gou/model"
|
|
"github.com/yaoapp/gou/query"
|
|
"github.com/yaoapp/gou/query/gou"
|
|
_ "github.com/yaoapp/gou/text"
|
|
"github.com/yaoapp/xun/capsule"
|
|
"github.com/yaoapp/yao/agent"
|
|
"github.com/yaoapp/yao/agent/caller"
|
|
"github.com/yaoapp/yao/agent/llm"
|
|
"github.com/yaoapp/yao/config"
|
|
"github.com/yaoapp/yao/kb"
|
|
"github.com/yaoapp/yao/test"
|
|
|
|
// Import assistant to trigger init() which registers AgentGetterFunc
|
|
_ "github.com/yaoapp/yao/agent/assistant"
|
|
)
|
|
|
|
// Prepare prepare the test environment with optional V8 mode configuration
|
|
// Usage:
|
|
//
|
|
// testutils.Prepare(t) // standard mode (default)
|
|
// testutils.Prepare(t, test.PrepareOption{V8Mode: "performance"}) // performance mode for benchmarks
|
|
func Prepare(t *testing.T, opts ...interface{}) {
|
|
test.Prepare(t, config.Conf, opts...)
|
|
|
|
// Load KB (required for agent KB features)
|
|
_, err := kb.Load(config.Conf)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Load agent
|
|
err = agent.Load(config.Conf)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Ensure JSAPI factories are registered (may be called multiple times, idempotent)
|
|
// This is needed because Go's init() order is not guaranteed across packages
|
|
caller.SetJSAPIFactory()
|
|
llm.SetJSAPIFactory()
|
|
|
|
// Register default query engine (required for DB search)
|
|
// capsule.Global is initialized by test.Prepare
|
|
if _, has := query.Engines["default"]; !has && capsule.Global != nil {
|
|
query.Register("default", &gou.Query{
|
|
Query: capsule.Query(),
|
|
GetTableName: func(s string) string {
|
|
if mod, has := model.Models[s]; has {
|
|
return mod.MetaData.Table.Name
|
|
}
|
|
return s
|
|
},
|
|
AESKey: config.Conf.DB.AESKey,
|
|
})
|
|
}
|
|
}
|
|
|
|
// Clean clean the test environment
|
|
func Clean(t *testing.T) {
|
|
test.Clean()
|
|
}
|