File size: 750 Bytes
8d3471e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | package shared
import "ds2api/internal/promptcompat"
func ApplyThinkingInjection(store ConfigReader, stdReq promptcompat.StandardRequest) promptcompat.StandardRequest {
if store == nil || !store.ThinkingInjectionEnabled() || !stdReq.Thinking {
return stdReq
}
messages, changed := promptcompat.AppendThinkingInjectionPromptToLatestUser(stdReq.Messages, store.ThinkingInjectionPrompt())
if !changed {
return stdReq
}
finalPrompt, toolNames := promptcompat.BuildOpenAIPrompt(messages, stdReq.ToolsRaw, "", stdReq.ToolChoice, stdReq.Thinking)
if len(toolNames) == 0 && len(stdReq.ToolNames) > 0 {
toolNames = stdReq.ToolNames
}
stdReq.Messages = messages
stdReq.FinalPrompt = finalPrompt
stdReq.ToolNames = toolNames
return stdReq
}
|