(resp *api.ListResponse)
| 16 | ) |
| 17 | |
| 18 | func parseAPIResponse(resp *api.ListResponse) []Model { |
| 19 | logging.DebugLogger.Println("Fetching models from API") |
| 20 | |
| 21 | models := make([]Model, len(resp.Models)) |
| 22 | for i, modelResp := range resp.Models { |
| 23 | models[i] = Model{ |
| 24 | Name: modelResp.Name, |
| 25 | ID: truncate(modelResp.Digest, 7), |
| 26 | Size: float64(modelResp.Size) / (1024 * 1024 * 1024), // Convert bytes to GB |
| 27 | QuantizationLevel: modelResp.Details.QuantizationLevel, |
| 28 | Family: modelResp.Details.Family, |
| 29 | Modified: modelResp.ModifiedAt, |
| 30 | ParameterSize: modelResp.Details.ParameterSize, |
| 31 | } |
| 32 | } |
| 33 | logging.DebugLogger.Println("Models:", models) |
| 34 | return models |
| 35 | } |
| 36 | |
| 37 | func normalizeSize(size float64) float64 { |
| 38 | return size // Sizes are already in GB in the API response |
no test coverage detected