mirror of
https://github.com/danielmiessler/Fabric.git
synced 2026-02-12 23:15:05 -05:00
66025d516c
- Part 1 of incorporating `modernize` tool into Fabric.
- Replace `interface{}` with `any` in slice type declarations
- Update map types from `map[string]interface{}` to `map[string]any`
- Change variadic function parameters to use `...any` instead of `...interface{}`
- Modernize JSON unmarshaling variables to `any` for consistency
- Update struct fields and method signatures to prefer `any` alias
- Ensure all type assertions and conversions use `any` throughout codebase
- Add PR guidelines in docs to encourage focused, reviewable changes
46 lines
1.1 KiB
Go
Executable File
46 lines
1.1 KiB
Go
Executable File
package restapi
|
|
|
|
import (
|
|
"github.com/danielmiessler/fabric/internal/plugins/ai"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type ModelsHandler struct {
|
|
vendorManager *ai.VendorsManager
|
|
}
|
|
|
|
func NewModelsHandler(r *gin.Engine, vendorManager *ai.VendorsManager) {
|
|
handler := &ModelsHandler{
|
|
vendorManager: vendorManager,
|
|
}
|
|
|
|
r.GET("/models/names", handler.GetModelNames)
|
|
}
|
|
|
|
func (h *ModelsHandler) GetModelNames(c *gin.Context) {
|
|
vendorsModels, err := h.vendorManager.GetModels()
|
|
if err != nil {
|
|
c.JSON(500, gin.H{"error": "Server failed to retrieve model names"})
|
|
return
|
|
}
|
|
|
|
response := make(map[string]any)
|
|
vendors := make(map[string][]string)
|
|
|
|
for _, groupItems := range vendorsModels.GroupsItems {
|
|
vendors[groupItems.Group] = groupItems.Items
|
|
}
|
|
|
|
response["models"] = h.getAllModelNames(vendorsModels)
|
|
response["vendors"] = vendors
|
|
c.JSON(200, response)
|
|
}
|
|
|
|
func (h *ModelsHandler) getAllModelNames(vendorsModels *ai.VendorsModels) []string {
|
|
var allModelNames []string
|
|
for _, groupItems := range vendorsModels.GroupsItems {
|
|
allModelNames = append(allModelNames, groupItems.Items...)
|
|
}
|
|
return allModelNames
|
|
}
|