Skip to content

feat: Add SUB_AGENT_BACKGROUND_DEFAULT environment variable #26

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
mudler merged 7 commits into from
Mar 5, 2026
Merged

feat: Add SUB_AGENT_BACKGROUND_DEFAULT environment variable #26

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
1626e42
feat: add sub-agent MCP server with background processing support
localai-bot Feb 27, 2026
761c3d8
feat: add SUB_AGENT_BACKGROUND_DEFAULT environment variable
localai-bot Feb 27, 2026
cc9c636
Remove binary file as requested
localai-bot Mar 5, 2026
d60f40f
Rebase: update sub-agent files to resolve conflicts with upstream/master
localai-bot Mar 5, 2026
0dfc952
Merge upstream/master to resolve conflicts and update branch
localai-bot Mar 5, 2026
156029f
feat: add SUB_AGENT_BACKGROUND_DEFAULT environment variable
localai-bot Mar 5, 2026
f8e551b
fix: remove sub-agent go.mod and go.sum, use root go.mod instead
localai-bot Mar 5, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 19 additions & 5 deletions sub-agent/main.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ import (
"context"
"log"
"os"
"strconv"
"sync"
"time"

Expand Down Expand Up @@ -98,12 +99,19 @@ func (s *SubAgentStore) List() []*SubAgentResult {
}

var (
store *SubAgentStore
openaiClient *openai.Client
openaiModel string
store *SubAgentStore
openaiClient *openai.Client
openaiModel string
backgroundDefault bool
)

func init() {
// SUB_AGENT_BACKGROUND_DEFAULT can be "true" or "false"
backgroundDefault = false
if bgEnv := os.Getenv("SUB_AGENT_BACKGROUND_DEFAULT"); bgEnv != "" {
backgroundDefault, _ = strconv.ParseBool(bgEnv)
}

// Initialize TTL (default 1 hour)
ttl := 1 * time.Hour
if ttlStr := os.Getenv("TTL"); ttlStr != "" {
Expand Down Expand Up @@ -134,7 +142,7 @@ func init() {
// SendInput represents the input for sub_agent_send
type SendInput struct {
Message string `json:"message" jsonschema:"the message to send to the OpenAI endpoint"`
Background bool `json:"background,omitempty" jsonschema:"whether to run the request in the background (default: false)"`
Background bool `json:"background,omitempty" jsonschema:"whether to process in background (defaults to SUB_AGENT_BACKGROUND_DEFAULT env var, or false if not set)"`
Model string `json:"model,omitempty" jsonschema:"override the default model for this request"`
}

Expand Down Expand Up @@ -195,7 +203,13 @@ func SendChatCompletion(ctx context.Context, req *mcp.CallToolRequest, input Sen
result = resp.Choices[0].Message.Content
}

if input.Background {
// Use backgroundDefault as the default, but allow request-level override
background := input.Background
if !input.Background && backgroundDefault {
// If request doesn't explicitly set background, use the default
background = backgroundDefault
}
if background {
taskID := uuid.New().String()
subResult := &SubAgentResult{
ID: taskID,
Expand Down
Loading