mirror of
https://github.com/ollama/ollama.git
synced 2025-11-11 03:07:43 +01:00
55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
package renderers
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/ollama/ollama/api"
|
|
)
|
|
|
|
type Renderer interface {
|
|
Render(messages []api.Message, tools []api.Tool, think *api.ThinkValue) (string, error)
|
|
}
|
|
|
|
type (
|
|
RendererConstructor func() Renderer
|
|
RendererRegistry struct {
|
|
renderers map[string]RendererConstructor
|
|
}
|
|
)
|
|
|
|
func (r *RendererRegistry) Register(name string, renderer RendererConstructor) {
|
|
r.renderers[name] = renderer
|
|
}
|
|
|
|
var registry = RendererRegistry{
|
|
renderers: make(map[string]RendererConstructor),
|
|
}
|
|
|
|
func Register(name string, renderer RendererConstructor) {
|
|
registry.Register(name, renderer)
|
|
}
|
|
|
|
func RenderWithRenderer(name string, msgs []api.Message, tools []api.Tool, think *api.ThinkValue) (string, error) {
|
|
renderer := rendererForName(name)
|
|
if renderer == nil {
|
|
return "", fmt.Errorf("unknown renderer %q", name)
|
|
}
|
|
return renderer.Render(msgs, tools, think)
|
|
}
|
|
|
|
func rendererForName(name string) Renderer {
|
|
if constructor, ok := registry.renderers[name]; ok {
|
|
return constructor()
|
|
}
|
|
switch name {
|
|
case "qwen3-coder":
|
|
renderer := &Qwen3CoderRenderer{}
|
|
return renderer
|
|
case "qwen3-vl-instruct":
|
|
renderer := &Qwen3VLRenderer{false}
|
|
return renderer
|
|
default:
|
|
return nil
|
|
}
|
|
}
|