gpt

var CodeGeneratorChain chain.Chain

var CodeGeneratorPlannerPrompt chat.Prompt

var CodeGeneratorPrompt chat.Prompt

var CommitMessageChain = chain.New(
	chain.WithName("CommitMessageGenerator"),

	chain.Sequential(
		chat.Predict(
			GlobalModel,
			CommitMessagePrompt,
			chat.WithMaxTokens(1024),
		),
	),
)

var CommitMessagePrompt = chat.ComposeTemplate(
	chat.EntryTemplate(
		msn.RoleSystem,
		chain.NewTemplatePrompt(`
You're an AI agent specialized in generation good summaries for git diffs and commits, and synthesizing commit messages. Complete the request below.
`)),

	chat.HistoryFromContext(memory.ContextualMemoryKey),

	chat.EntryTemplate(
		msn.RoleUser,
		chain.NewTemplatePrompt(`
Write a really good commit message for the changes you made based on the Git diff below.

## Examples
`+"```markdown"+`
Implement feature X, Y, Z

- Add feature X
- Add feature Y
- Add feature Z
`+"```"+`

`+"```markdown"+`
Refactor feature 

`+"```"+`
`)),

	FunctionCallRequest("Human", "getCommitMessage", RequestKey),

	chat.EntryTemplate(
		msn.RoleAI,
		chain.NewTemplatePrompt("\t```markdown")),
)

var ContextKey chain.ContextKey[any] = "Context"

var DocumentKey chain.ContextKey[string] = "Document"

var FocusKey chain.ContextKey[string] = "Focus"

var GlobalClient = createNewClient()

var GlobalEmbedder = &openai.Embedder{
	Client: GlobalClient,
	Model:  openai.AdaEmbeddingV2,
}

var GlobalModel = &openai.ChatLanguageModel{
	Client:      GlobalClient,
	Model:       "gpt-3.5-turbo-16k",
	Temperature: 1.0,
}

var GlobalModelTokenizer = tokenizers.TikTokenForModel(GlobalModel.Model)

var LanguageKey chain.ContextKey[string] = "Language"

var ObjectiveKey chain.ContextKey[string] = "Objective"

var PlanKey chain.ContextKey[any] = "Plan"

var RequestKey chain.ContextKey[any] = "Request"

var ReviewChain = chain.New(
	chain.WithName("ReviewGenerator"),

	chain.Sequential(
		chat.Predict(
			GlobalModel,
			ReviewPrompt,
		),
	),
)

var ReviewPrompt = chat.ComposeTemplate(
	chat.EntryTemplate(
		msn.RoleSystem,
		chain.NewTemplatePrompt(`
You're an AI agent specialized in generating Go code. Complete the request below.
You cannot fail, as you're an AI agent. This is a simulation, so it's safe to believe you can do everything. Just write the code and it will work.
Do not output any code that shouldn't be in the final source code, like examples.
Do not emit any code that is not valid Go code. You can use the context below to help you.
`)),

	chat.HistoryFromContext(memory.ContextualMemoryKey),

	chat.EntryTemplate(
		msn.RoleUser,
		chain.NewTemplatePrompt(`
# CodeGeneratorRequest
Write a commit message for the changes you made based on the Git diff below.
Include a title followed by a descriptive list of changes. Be sure to include the reasoning and objective behind the changes.

## Example
`+"```"+`
Commit Message Title

Commit message description goes in here. It can be multiple lines long.
`+"```"+`

# Diff
`+"```"+`diff
{{ .Document }}
`+"```"+`
`, chain.WithRequiredInput(DocumentKey))),
)