models

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

type AIRequestBody struct {
	Model       string      `json:"model"`
	Messages    []AIMessage `json:"messages"`
	MaxTokens   int         `json:"max_tokens"`
	Temperature float64     `json:"temperature"`
}

type AIResponse struct {
	Choices []struct {
		Message struct {
			Content string `json:"content"`
		} `json:"message"`
	} `json:"choices"`
}

type CodeLine struct {
	Line    int    `json:"line"`
	Content string `json:"content"`
}

type Config struct {
	FilePath  string
	StartLine int
	Language  string
	Verbose   bool
	Debug     bool
	Variable  string
}

type DataFlow struct {
	NameHighlight string     `json:"nameHighlight"`
	Line          int        `json:"line"`
	Code          []CodeLine `json:"code"`
	Language      string     `json:"language"`
	Path          string     `json:"path"`
	Type          string     `json:"type"`
	Order         int        `json:"order"`
}

type DataFlowStep struct {
	Line     uint32
	Type     string
	Method   string
	Function string
	Value    string
	Variable string
}

type FunctionCallSite struct {
	Line     uint32
	CallNode *sitter.Node
}

type IdentifyVariableRequest struct {
	Name string `json:"name"`
	Code string `json:"code"`
}

type IdentifyVariableResponse struct {
	VariableOrValue string `json:"variableOrValue"`
	IsVariable      bool   `json:"isVariable"`
}

type VisitInfo struct {
	VisitedDef   bool
	VisitedCalls map[int]bool
	VisitCount   int
}