Files
grailbench/models/models.go
Grail Finder e499a1ae37 Feat: rp mode
2025-08-09 11:46:33 +03:00

88 lines
2.5 KiB
Go

package models
type Question struct {
ID string `json:"id"`
Topic string `json:"topic"`
Question string `json:"question"`
}
type Answer struct {
Q Question
Answer string `json:"answer"`
Model string `json:"model"`
// resp time?
}
// type OpenRouterResp struct {
// ID string `json:"id"`
// Provider string `json:"provider"`
// Model string `json:"model"`
// Object string `json:"object"`
// Created int `json:"created"`
// Choices []struct {
// Logprobs any `json:"logprobs"`
// FinishReason string `json:"finish_reason"`
// NativeFinishReason string `json:"native_finish_reason"`
// Index int `json:"index"`
// Message struct {
// Role string `json:"role"`
// Content string `json:"content"`
// Refusal any `json:"refusal"`
// Reasoning any `json:"reasoning"`
// } `json:"message"`
// } `json:"choices"`
// Usage struct {
// PromptTokens int `json:"prompt_tokens"`
// CompletionTokens int `json:"completion_tokens"`
// TotalTokens int `json:"total_tokens"`
// } `json:"usage"`
// }
type DSResp struct {
ID string `json:"id"`
Choices []struct {
Text string `json:"text"`
Index int `json:"index"`
FinishReason string `json:"finish_reason"`
} `json:"choices"`
Created int `json:"created"`
Model string `json:"model"`
SystemFingerprint string `json:"system_fingerprint"`
Object string `json:"object"`
}
type LLMResp struct {
Index int `json:"index"`
Content string `json:"content"`
Tokens []any `json:"tokens"`
IDSlot int `json:"id_slot"`
Stop bool `json:"stop"`
Model string `json:"model"`
TokensPredicted int `json:"tokens_predicted"`
TokensEvaluated int `json:"tokens_evaluated"`
Prompt string `json:"prompt"`
HasNewLine bool `json:"has_new_line"`
Truncated bool `json:"truncated"`
StopType string `json:"stop_type"`
StoppingWord string `json:"stopping_word"`
TokensCached int `json:"tokens_cached"`
}
type LCPResp struct {
Content string `json:"content"`
}
type RPScenario struct {
Name string `json:"name"`
CharSysPrompt string `json:"char_sys_prompt"`
UserSysPrompt string `json:"user_sys_prompt"`
CharName string `json:"char_name"`
UserName string `json:"user_name"`
FirstMsg string `json:"first_msg"`
}
type RPMessage struct {
Author string `json:"author"`
Content string `json:"content"`
}