initial commit
This commit is contained in:
41
backend/internal/client/convert.go
Normal file
41
backend/internal/client/convert.go
Normal file
@@ -0,0 +1,41 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
|
||||
"github.com/openai/openai-go/v3"
|
||||
)
|
||||
|
||||
func fromOpenAIModel(m openai.Model) Model {
|
||||
newModel := Model{
|
||||
Model: m,
|
||||
Name: m.ID,
|
||||
}
|
||||
|
||||
extraFields := make(map[string]any)
|
||||
for k, v := range m.JSON.ExtraFields {
|
||||
var val any
|
||||
if err := json.Unmarshal([]byte(v.Raw()), &val); err != nil {
|
||||
continue
|
||||
}
|
||||
extraFields[k] = val
|
||||
}
|
||||
|
||||
// Extract Name
|
||||
if rawName, found := extraFields["name"]; found {
|
||||
if name, ok := rawName.(string); ok {
|
||||
newModel.Name = name
|
||||
}
|
||||
}
|
||||
|
||||
// Extract Meta
|
||||
if rawMeta, found := extraFields["meta"]; found {
|
||||
if parsedMeta, ok := rawMeta.(map[string]any); ok {
|
||||
if llamaMeta, ok := parsedMeta["llamaswap"].(map[string]any); ok {
|
||||
newModel.Meta = llamaMeta
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return newModel
|
||||
}
|
||||
Reference in New Issue
Block a user