152 lines
4.0 KiB
Go
152 lines
4.0 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"`
|
|
ToolCall ToolCallInfo `json:"tool_call,omitempty"`
|
|
}
|
|
|
|
type ToolCallInfo struct {
|
|
Name string `json:"name"`
|
|
Args map[string]string `json:"args"`
|
|
}
|
|
|
|
// 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 ToolCallFunction struct {
|
|
Name string `json:"name"`
|
|
Arguments string `json:"arguments"`
|
|
}
|
|
|
|
type ToolCall struct {
|
|
ID string `json:"id"`
|
|
Type string `json:"type"`
|
|
Function ToolCallFunction `json:"function"`
|
|
}
|
|
|
|
type DSRespChoice struct {
|
|
Text string `json:"text"`
|
|
Index int `json:"index"`
|
|
FinishReason string `json:"finish_reason"`
|
|
Message struct {
|
|
Role string `json:"role"`
|
|
Content string `json:"content"`
|
|
ToolCalls []ToolCall `json:"tool_calls,omitempty"`
|
|
} `json:"message,omitempty"`
|
|
}
|
|
|
|
type DSResp struct {
|
|
ID string `json:"id"`
|
|
Choices []DSRespChoice `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"`
|
|
}
|
|
|
|
// === tools models
|
|
|
|
type ToolArgProps struct {
|
|
Type string `json:"type"`
|
|
Description string `json:"description"`
|
|
}
|
|
|
|
type ToolFuncParams struct {
|
|
Type string `json:"type"`
|
|
Properties map[string]ToolArgProps `json:"properties"`
|
|
Required []string `json:"required"`
|
|
}
|
|
|
|
type ToolFunc struct {
|
|
Name string `json:"name"`
|
|
Description string `json:"description"`
|
|
Parameters ToolFuncParams `json:"parameters"`
|
|
}
|
|
|
|
type Tool struct {
|
|
Type string `json:"type"`
|
|
Function ToolFunc `json:"function"`
|
|
}
|
|
|
|
type OpenAIReq struct {
|
|
*ChatBody
|
|
Tools []Tool `json:"tools"`
|
|
}
|
|
|
|
type ChatBody struct {
|
|
Model string `json:"model"`
|
|
Messages []RoleMsg `json:"messages"`
|
|
}
|
|
|
|
type RoleMsg struct {
|
|
Role string `json:"role"`
|
|
Content string `json:"content"`
|
|
}
|
|
|
|
// ===
|