1
0
mirror of https://github.com/coalaura/whiskr.git synced 2025-12-02 20:22:52 +00:00

show pricing and better title format

This commit is contained in:
Laura
2025-09-30 22:25:01 +02:00
parent bb8b333efa
commit 592262f95c
4 changed files with 54 additions and 6 deletions

View File

@@ -3,6 +3,7 @@ package main
import (
"context"
"sort"
"strconv"
"strings"
"sync"
"time"
@@ -10,12 +11,18 @@ import (
"github.com/revrost/go-openrouter"
)
type ModelPricing struct {
Input float64 `json:"input"`
Output float64 `json:"output"`
}
type Model struct {
ID string `json:"id"`
Created int64 `json:"created"`
Name string `json:"name"`
Description string `json:"description"`
Tags []string `json:"tags,omitempty"`
ID string `json:"id"`
Created int64 `json:"created"`
Name string `json:"name"`
Description string `json:"description"`
Pricing ModelPricing `json:"pricing"`
Tags []string `json:"tags,omitempty"`
Reasoning bool `json:"-"`
Vision bool `json:"-"`
@@ -90,11 +97,19 @@ func LoadModels(initial bool) error {
name = name[index+2:]
}
input, _ := strconv.ParseFloat(model.Pricing.Prompt, 64)
output, _ := strconv.ParseFloat(model.Pricing.Completion, 64)
m := &Model{
ID: model.ID,
Created: model.Created,
Name: name,
Description: model.Description,
Pricing: ModelPricing{
Input: input * 1000000,
Output: output * 1000000,
},
}
GetModelTags(model, m)