ollama

Documentation

Index

• type CreateEmbeddingRequest
  • func (c *CreateEmbeddingRequest) JSON() (string, error)
• type CreateEmbeddingResponse
• type OllamaClient
  • func NewOllamaClient(opts ...Option) (*OllamaClient, error)
• type OllamaEmbeddingFunction
  • func NewOllamaEmbeddingFunction(option ...Option) (*OllamaEmbeddingFunction, error)
  • func (e *OllamaEmbeddingFunction) EmbedDocuments(ctx context.Context, documents []string) ([]*types.Embedding, error)
  • func (e *OllamaEmbeddingFunction) EmbedQuery(ctx context.Context, document string) (*types.Embedding, error)
  • func (e *OllamaEmbeddingFunction) EmbedRecords(ctx context.Context, records []*types.Record, force bool) error
• type Option
  • func WithBaseURL(baseURL string) Option
  • func WithModel(model string) Option

Types

type CreateEmbeddingRequest

type CreateEmbeddingRequest struct {
	Model  string `json:"model"`
	Prompt string `json:"prompt"`
}

func (*CreateEmbeddingRequest) JSON

func (c *CreateEmbeddingRequest) JSON() (string, error)

type CreateEmbeddingResponse

type CreateEmbeddingResponse struct {
	Embedding []float32 `json:"embedding"`
}

type OllamaClient

type OllamaClient struct {
	BaseURL        string
	Model          string
	Client         *http.Client
	DefaultHeaders map[string]string
}

func NewOllamaClient

func NewOllamaClient(opts ...Option) (*OllamaClient, error)

type OllamaEmbeddingFunction

type OllamaEmbeddingFunction struct {
	// contains filtered or unexported fields
}

func NewOllamaEmbeddingFunction

func NewOllamaEmbeddingFunction(option ...Option) (*OllamaEmbeddingFunction, error)

func (*OllamaEmbeddingFunction) EmbedDocuments

func (e *OllamaEmbeddingFunction) EmbedDocuments(ctx context.Context, documents []string) ([]*types.Embedding, error)

func (*OllamaEmbeddingFunction) EmbedQuery

func (e *OllamaEmbeddingFunction) EmbedQuery(ctx context.Context, document string) (*types.Embedding, error)

func (*OllamaEmbeddingFunction) EmbedRecords

func (e *OllamaEmbeddingFunction) EmbedRecords(ctx context.Context, records []*types.Record, force bool) error

type Option

type Option func(p *OllamaClient) error

func WithBaseURL

func WithBaseURL(baseURL string) Option

func WithModel

func WithModel(model string) Option