provider

package

v0.0.6 Latest Latest Go to latest Published: Mar 18, 2026 License: MIT Imports: 22 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/dimetron/pi-go

Links

Open Source Insights

Documentation ¶

Index ¶

func CheckOllama(baseURL string) error
func NewAnthropic(_ context.Context, modelName, apiKey, baseURL, thinkingLevel string) (model.LLM, error)
func NewGemini(ctx context.Context, modelName, baseURL string) (model.LLM, error)
func NewLLM(ctx context.Context, info Info, apiKey, baseURL, thinkingLevel string) (model.LLM, error)
func NewOpenAI(_ context.Context, modelName, apiKey, baseURL string) (model.LLM, error)
type Info
- func Resolve(modelName string) (Info, error)

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

func CheckOllama ¶ added in v0.0.4

func CheckOllama(baseURL string) error

CheckOllama verifies that the Ollama server at baseURL is reachable. It first checks TCP connectivity on the port, then issues a GET to the root endpoint (Ollama returns "Ollama is running").

func NewAnthropic ¶

func NewAnthropic(_ context.Context, modelName, apiKey, baseURL, thinkingLevel string) (model.LLM, error)

NewAnthropic creates an Anthropic model.LLM. If baseURL is non-empty, it overrides the default API endpoint. When baseURL is set, the API key is optional (for Ollama compatibility). thinkingLevel controls extended thinking: "none", "low", "medium", "high".

func NewGemini ¶

func NewGemini(ctx context.Context, modelName, baseURL string) (model.LLM, error)

NewGemini creates a Gemini model.LLM using ADK Go's native Gemini support. It reads the API key from GOOGLE_API_KEY or GEMINI_API_KEY env vars. If neither is set, it falls back to Application Default Credentials. If baseURL is non-empty, it overrides the default API endpoint.

func NewLLM ¶

func NewLLM(ctx context.Context, info Info, apiKey, baseURL, thinkingLevel string) (model.LLM, error)

NewLLM creates a model.LLM for the given provider info, API key, optional base URL, and thinking level.

func NewOpenAI ¶

func NewOpenAI(_ context.Context, modelName, apiKey, baseURL string) (model.LLM, error)

NewOpenAI creates an OpenAI model.LLM. If baseURL is non-empty, it overrides the default API endpoint.

Types ¶

type Info ¶

type Info struct {
	Provider string
	Model    string
	Ollama   bool // true when model was suffixed with :cloud (Ollama Anthropic-compatible API)
}

Info describes a provider and the model to use.

func Resolve ¶

func Resolve(modelName string) (Info, error)

Resolve determines the provider from a model name. Models ending with ":cloud" are routed through the Anthropic provider using Ollama's Anthropic-compatible API.

Source Files ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL