micro--go-micro
Asim Aslam
56 行
1.1 KiB
Go
56 行
1.1 KiB
Go
package ai
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/urfave/cli/v2"
|
|
goai "go-micro.dev/v6/ai"
|
|
_ "go-micro.dev/v6/ai/anthropic"
|
|
_ "go-micro.dev/v6/ai/atlascloud"
|
|
_ "go-micro.dev/v6/ai/gemini"
|
|
_ "go-micro.dev/v6/ai/groq"
|
|
_ "go-micro.dev/v6/ai/mistral"
|
|
_ "go-micro.dev/v6/ai/openai"
|
|
_ "go-micro.dev/v6/ai/together"
|
|
"go-micro.dev/v6/cmd"
|
|
)
|
|
|
|
func init() {
|
|
cmd.Register(&cli.Command{
|
|
Name: "ai",
|
|
Usage: "Inspect AI provider support",
|
|
Subcommands: []*cli.Command{{
|
|
Name: "providers",
|
|
Usage: "Print the registered AI provider capability matrix",
|
|
Action: providersAction,
|
|
}},
|
|
})
|
|
}
|
|
|
|
func providersAction(c *cli.Context) error {
|
|
writeProviderMatrix(c.App.Writer, goai.CapabilityRows())
|
|
return nil
|
|
}
|
|
|
|
func writeProviderMatrix(w io.Writer, rows []goai.CapabilityRow) {
|
|
const check = "✓"
|
|
fmt.Fprintln(w, "Provider Model Image Video")
|
|
fmt.Fprintln(w, "-------- ----- ----- -----")
|
|
for _, row := range rows {
|
|
fmt.Fprintf(w, "%-11s %-6s %-6s %-6s\n",
|
|
row.Provider,
|
|
mark(row.Model, check),
|
|
mark(row.Image, check),
|
|
mark(row.Video, check),
|
|
)
|
|
}
|
|
}
|
|
|
|
func mark(ok bool, value string) string {
|
|
if ok {
|
|
return value
|
|
}
|
|
return "-"
|
|
}
|