server/llm/provider/provider.go

package provider

import (
	"github.com/iomodo/staff/llm"
	"github.com/iomodo/staff/llm/fake"
	"github.com/iomodo/staff/llm/openai"
)

func CreateProvider(config llm.Config) llm.LLMProvider {
	switch config.Provider {
	case llm.ProviderFake:
		return fake.New()
	case llm.ProviderOpenAI:
		return openai.New(config)
	default:
		return fake.New()
	}

}