chat

type Chat struct {
	// contains filtered or unexported fields
}

type DirectLLMMessage = LLMMessage

type DirectLLMRequest = LLMRequest

type DirectLLMResponse = LLMResponse

type DirectLLMResponseFormat = LLMResponseFormat

type LLMMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type LLMRequest struct {
	Model          string             `json:"model"`
	Messages       []LLMMessage       `json:"messages"`
	Temperature    float32            `json:"temperature,omitempty"`
	TopP           float32            `json:"top_p,omitempty"`
	MaxTokens      int                `json:"max_tokens,omitempty"`
	ResponseFormat *LLMResponseFormat `json:"response_format,omitempty"`
}

type LLMResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	Model   string `json:"model"`
	Choices []struct {
		Message struct {
			Role    string `json:"role"`
			Content string `json:"content"`
		} `json:"message"`
		FinishReason string `json:"finish_reason"`
	} `json:"choices"`
	Usage struct {
		PromptTokens     int `json:"prompt_tokens"`
		CompletionTokens int `json:"completion_tokens"`
		TotalTokens      int `json:"total_tokens"`
	} `json:"usage"`
}

type LLMResponseFormat struct {
	Type string `json:"type"`
}

type ReviewResult struct {
	LGTM          bool   `json:"lgtm"`
	ReviewComment string `json:"review_comment"`
	Summary       string `json:"summary"`     // 代码变更的总结
	Suggestions   string `json:"suggestions"` // 改进建议
	Highlights    string `json:"highlights"`  // 代码亮点
	Risks         string `json:"risks"`       // 潜在风险
}