Function BuildCommandArguments

internal/sessions/session_injection.go:48–92

BuildCommandArguments constructs the arguments map for a shell tool invocation. It inspects the tool's schema to determine the correct parameter name.

(sess *Session, toolName, command string)

// BuildCommandArguments constructs the arguments map for a shell tool invocation.
// It inspects the tool's schema to determine the correct parameter name.
func BuildCommandArguments(sess *Session, toolName, command string) map[string]any {
	if sess == nil {
		return map[string]any{"command": command}
	}

	sess.mu.Lock()
	var schema map[string]any
	for _, t := range sess.Tools {
		if t.Name == toolName {
			schema = t.Schema
			break
		}
	}
	sess.mu.Unlock()

	// If we have a schema, look for the first string property to use as the command key.
	if schema != nil {
		if props, ok := schema["properties"].(map[string]any); ok {
			// Check common names first.
			for _, key := range []string{"command", "cmd", "input", "script"} {
				if propRaw, exists := props[key]; exists {
					// If the property type is "array" (e.g. Codex CLI shell tool),
					// wrap the command as ["bash", "-c", command].
					if prop, ok := propRaw.(map[string]any); ok {
						if t, _ := prop["type"].(string); t == "array" {
							return map[string]any{key: []string{"bash", "-c", command}}
						}
					}
					return map[string]any{key: command}
				}
			}
			// Fall back to first string property.
			for key, propRaw := range props {
				if prop, ok := propRaw.(map[string]any); ok {
					if t, _ := prop["type"].(string); t == "string" {
						return map[string]any{key: command}
					}
				}
				_ = key
			}
		}
	}

	return map[string]any{"command": command}
}

// readToolPriority defines the preference order for picking a file-read tool.
var readToolPriority = []string{"Read", "read", "read_file", "readFile", "file_read", "cat"}