Method InjectContext

pkg/context/auto_inject.go:38–94

InjectContext 注入上下文到系统提示词

(ctx context.Context, systemPrompt string)

// InjectContext 注入上下文到系统提示词
func (ci *ContextInjector) InjectContext(ctx context.Context, systemPrompt string) string {
	var additions []string

	// 注入日期时间
	if ci.config.EnableDateTime {
		additions = append(additions, ci.buildDateTimeContext())
	}

	// 注入位置信息
	if ci.config.EnableLocation && ci.config.LocationProvider != nil {
		if locCtx := ci.buildLocationContext(ctx); locCtx != "" {
			additions = append(additions, locCtx)
		}
	}

	// 注入 Session 状态
	if ci.config.EnableSessionState && ci.config.SessionStateProvider != nil {
		if stateCtx := ci.buildSessionStateContext(ctx); stateCtx != "" {
			additions = append(additions, stateCtx)
		}
	}

	// 注入用户信息
	if ci.config.EnableUserInfo && ci.config.UserInfoProvider != nil {
		if userCtx := ci.buildUserInfoContext(ctx); userCtx != "" {
			additions = append(additions, userCtx)
		}
	}

	// 注入系统信息
	if ci.config.EnableSystemInfo {
		additions = append(additions, ci.buildSystemInfoContext())
	}

	// 注入自定义上下文
	if ci.config.CustomContext != "" {
		additions = append(additions, ci.config.CustomContext)
	}

	// 如果没有需要注入的内容，直接返回原始提示词
	if len(additions) == 0 {
		return systemPrompt
	}

	// 构建完整的系统提示词
	var builder strings.Builder
	builder.WriteString(systemPrompt)

	if !strings.HasSuffix(systemPrompt, "\n") {
		builder.WriteString("\n")
	}

	builder.WriteString("\n## Context Information\n\n")
	builder.WriteString(strings.Join(additions, "\n\n"))

	return builder.String()
}