alihan/chatgpt-cli
1
0
Fork 0
mirror of https://github.com/kardolus/chatgpt-cli.git synced 2024-09-08 23:15:00 +03:00
Code Issues Packages Projects Releases Wiki Activity

Refactor Query function to include token usage in response

Browse Source
This commit is contained in:
kardolus
2024-04-20 12:08:33 -04:00
parent 154f0331a4
commit ca2e544603
5 changed files with 64 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
types/completions.go
View File

@@ -17,18 +17,20 @@ type Message struct {
}
type CompletionsResponse struct {
ID string `json:"id"`
Object string `json:"object"`
Created int `json:"created"`
Model string `json:"model"`
Usage struct {
PromptTokens int `json:"prompt_tokens"`
CompletionTokens int `json:"completion_tokens"`
TotalTokens int `json:"total_tokens"`
} `json:"usage"`
ID string `json:"id"`
Object string `json:"object"`
Created int `json:"created"`
Model string `json:"model"`
Usage Usage `json:"usage"`
Choices []Choice `json:"choices"`
}
type Usage struct {
PromptTokens int `json:"prompt_tokens"`
CompletionTokens int `json:"completion_tokens"`
TotalTokens int `json:"total_tokens"`
}
type Choice struct {
Message Message `json:"message"`
FinishReason string `json:"finish_reason"`