gen_models

Documentation

Index

• Constants
• func GeminiGen(content string) (*genai.GenerateContentResponse, error)
• func RetrieveResponse(parts []genai.Part) string
• type OllamaRequest
• type OllamaResponse
  • func OllamaGen(article scrapers.ScrapedResult) OllamaResponse

Constants

const (
	Wizard = "wizardlm2:7b-q4_K_M"
	Qwen   = "qwen2:latest"
	Phi    = "phi3:latest"
	LLama3 = "llama3:latest"
)

Functions

func GeminiGen

func GeminiGen(content string) (*genai.GenerateContentResponse, error)

func RetrieveResponse

func RetrieveResponse(parts []genai.Part) string

Types

type OllamaRequest

type OllamaRequest struct {
	Model  string `json:"model"`
	Prompt string `json:"prompt"`
	Stream bool   `json:"stream"`
}

type OllamaResponse

type OllamaResponse struct {
	Model              string `json:"model,omitempty"`
	Created            string `json:"created_at,omitempty"`
	Response           string `json:"response"`
	Done               bool   `json:"done,omitempty"`
	Context            []int  `json:"context"`
	TotalDuration      int64  `json:"total_duration,omitempty"`
	LoadDuration       int64  `json:"load_duration,omitempty"`
	PromptEvalDuration int64  `json:"prompt_eval_duration,omitempty"`
	EvalCount          int64  `json:"eval_count,omitempty"`
	EvalDuration       int64  `json:"eval_duration,omitempty"`
}

func OllamaGen

func OllamaGen(article scrapers.ScrapedResult) OllamaResponse