opencode/picoclaw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix for review

Browse source
This commit is contained in:
Cytown 2026-03-11 12:39:42 +08:00
parent fbf940d2c8
commit 9ceb55a378
1 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
pkg/agent/loop.go
View file

@ -287,7 +287,8 @@ func (al *AgentLoop) Run(ctx context.Context) error {
mcpTool := tools.NewMCPTool(mcpManager, serverName, tool) mcpTool := tools.NewMCPTool(mcpManager, serverName, tool)
if al.cfg.Tools.MCP.Discovery.Enabled { cfg := al.GetConfig()
if cfg.Tools.MCP.Discovery.Enabled {
agent.Tools.RegisterHidden(mcpTool) agent.Tools.RegisterHidden(mcpTool)
} else { } else {
agent.Tools.Register(mcpTool) agent.Tools.Register(mcpTool)
@ -313,9 +314,10 @@ func (al *AgentLoop) Run(ctx context.Context) error {
}) })
// Initializes Discovery Tools only if enabled by configuration // Initializes Discovery Tools only if enabled by configuration
if al.cfg.Tools.MCP.Enabled && al.cfg.Tools.MCP.Discovery.Enabled { cfg := al.GetConfig()
useBM25 := al.cfg.Tools.MCP.Discovery.UseBM25 if cfg.Tools.MCP.Enabled && cfg.Tools.MCP.Discovery.Enabled {
useRegex := al.cfg.Tools.MCP.Discovery.UseRegex useBM25 := cfg.Tools.MCP.Discovery.UseBM25
useRegex := cfg.Tools.MCP.Discovery.UseRegex
// Fail fast: If discovery is enabled but no search method is turned on // Fail fast: If discovery is enabled but no search method is turned on
if !useBM25 && !useRegex { if !useBM25 && !useRegex {
@ -324,12 +326,12 @@ func (al *AgentLoop) Run(ctx context.Context) error {
) )
} }
ttl := al.cfg.Tools.MCP.Discovery.TTL ttl := cfg.Tools.MCP.Discovery.TTL
if ttl <= 0 { if ttl <= 0 {
ttl = 5 // Default value ttl = 5 // Default value
} }
maxSearchResults := al.cfg.Tools.MCP.Discovery.MaxSearchResults maxSearchResults := cfg.Tools.MCP.Discovery.MaxSearchResults
if maxSearchResults <= 0 { if maxSearchResults <= 0 {
maxSearchResults = 5 // Default value maxSearchResults = 5 // Default value
} }
@ -338,8 +340,9 @@ func (al *AgentLoop) Run(ctx context.Context) error {
"bm25": useBM25, "regex": useRegex, "ttl": ttl, "max_results": maxSearchResults, "bm25": useBM25, "regex": useRegex, "ttl": ttl, "max_results": maxSearchResults,
}) })
registry := al.GetRegistry()
for _, agentID := range agentIDs { for _, agentID := range agentIDs {
agent, ok := al.registry.GetAgent(agentID) agent, ok := registry.GetAgent(agentID)
if !ok { if !ok {
continue continue
} }
@ -432,7 +435,7 @@ func (al *AgentLoop) Stop() {
// Close releases resources held by agent session stores. Call after Stop. // Close releases resources held by agent session stores. Call after Stop.
func (al *AgentLoop) Close() { func (al *AgentLoop) Close() {
al.registry.Close() al.GetRegistry().Close()
} }
func (al *AgentLoop) RegisterTool(tool tools.Tool) { func (al *AgentLoop) RegisterTool(tool tools.Tool) {