9eef569e4b
- Updated the system configuration to include new role-level defaults for Light, Vision, and Audio connectors. - Refactored the resolveSystemConnector function to prioritize per-agent overrides, improving connector resolution logic. - Enhanced LLMConnector integration across various components to streamline settings retrieval and capabilities management. - Improved error handling and logging for connector-related operations, ensuring better diagnostics and user feedback.
58 lines
1.8 KiB
Go
58 lines
1.8 KiB
Go
package assistant
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestResolveSystemConnector_NoConfig(t *testing.T) {
|
|
saved := systemConfig
|
|
systemConfig = nil
|
|
defer func() { systemConfig = saved }()
|
|
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.title"))
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.keyword"))
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.querydsl"))
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.vision"))
|
|
}
|
|
|
|
func TestResolveSystemConnector_PerAgentOverride(t *testing.T) {
|
|
saved := systemConfig
|
|
systemConfig = &SystemConfig{
|
|
Title: "openai.gpt-4o",
|
|
}
|
|
defer func() { systemConfig = saved }()
|
|
|
|
assert.Equal(t, "openai.gpt-4o", resolveSystemConnector("__yao.title"))
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.keyword"))
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.querydsl"))
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.vision"))
|
|
}
|
|
|
|
func TestResolveSystemConnector_RoleLevelOnly(t *testing.T) {
|
|
saved := systemConfig
|
|
systemConfig = &SystemConfig{
|
|
Default: "openai.gpt-4o",
|
|
Light: "openai.gpt-4o-mini",
|
|
}
|
|
defer func() { systemConfig = saved }()
|
|
|
|
// Role-level keys don't produce per-agent overrides
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.title"))
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.keyword"))
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.querydsl"))
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.vision"))
|
|
}
|
|
|
|
func TestResolveSystemConnector_UnknownAgent(t *testing.T) {
|
|
saved := systemConfig
|
|
systemConfig = &SystemConfig{
|
|
Default: "openai.gpt-4o",
|
|
Title: "openai.gpt-4o",
|
|
}
|
|
defer func() { systemConfig = saved }()
|
|
|
|
assert.Equal(t, "", resolveSystemConnector("__yao.nonexistent"))
|
|
assert.Equal(t, "", resolveSystemConnector("custom.agent"))
|
|
}
|