639f0c59fc
- Implemented AssistantReloadFunc to enable hot-reloading of assistants after deployment, improving deployment flexibility. - Enhanced gRPC authProvider to include workspace and sandbox IDs from incoming context metadata, enriching the authentication context. - Updated tools to support new agent-related functionalities, including listing, downloading, deploying, and connecting agents. - Expanded system tools documentation to include new agent commands, ensuring comprehensive guidance for users.
63 lines
1.6 KiB
Go
63 lines
1.6 KiB
Go
package agent
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/yaoapp/gou/connector"
|
|
"github.com/yaoapp/gou/process"
|
|
"github.com/yaoapp/yao/llmprovider"
|
|
"github.com/yaoapp/yao/openapi/oauth/authorized"
|
|
)
|
|
|
|
// ConnectorsHandler handles the agent_connectors tool.
|
|
// No input args. Returns the current user's LLM connector matrix without keys.
|
|
func ConnectorsHandler(proc *process.Process) interface{} {
|
|
authInfo := authorized.ProcessAuthInfo(proc)
|
|
|
|
if llmprovider.Global == nil {
|
|
return map[string]interface{}{"error": "llmprovider not initialized"}
|
|
}
|
|
|
|
var roles map[string]llmprovider.RoleTarget
|
|
var err error
|
|
|
|
if authInfo != nil && authInfo.UserID != "" {
|
|
roles, err = llmprovider.Global.ListRolesByUser(authInfo.UserID)
|
|
} else {
|
|
roles, err = llmprovider.Global.ListRoles()
|
|
}
|
|
if err != nil {
|
|
return map[string]interface{}{"error": fmt.Sprintf("failed to list roles: %s", err.Error())}
|
|
}
|
|
|
|
result := make(map[string]interface{}, len(roles))
|
|
for role, target := range roles {
|
|
connID := target.Provider
|
|
info := map[string]interface{}{
|
|
"id": connID,
|
|
"model": target.Model,
|
|
}
|
|
|
|
conn, exists := connector.Connectors[connID]
|
|
if exists {
|
|
setting := conn.Setting()
|
|
meta := conn.GetMetaInfo()
|
|
if meta.Label != "" {
|
|
info["name"] = meta.Label
|
|
}
|
|
if model, ok := setting["model"]; ok && info["model"] == "" {
|
|
info["model"] = model
|
|
}
|
|
if caps, ok := setting["capabilities"]; ok {
|
|
info["capabilities"] = sanitizeCapabilities(caps)
|
|
}
|
|
if t := settingStr(setting, "auth_mode"); t != "" {
|
|
info["type"] = "openai"
|
|
}
|
|
}
|
|
|
|
result[role] = info
|
|
}
|
|
|
|
return result
|
|
}
|