genie

Documentation

Index

• func Init(c *Config)
• func Run()
• type ChatChoice
• type ChatCompletionResponse
• type CompletionChoice
• type CompletionResponse
• type Config
• type Language
• type Message

Functions

func Init

func Init(c *Config)

func Run

func Run()

Types

type ChatChoice

type ChatChoice struct {
	Index   int `json:"index"`
	Message struct {
		Role    string `json:"role"`
		Content string `json:"content"`
	} `json:"message"`
	FinishReason string `json:"finish_reason"`
}

type ChatCompletionResponse

type ChatCompletionResponse struct {
	ID      string       `json:"id"`
	Object  string       `json:"object"`
	Created int          `json:"created"`
	Choices []ChatChoice `json:"choices"`
	Usage   struct {
		PromptTokens     int `json:"prompt_tokens"`
		CompletionTokens int `json:"completion_tokens"`
		TotalTokens      int `json:"total_tokens"`
	} `json:"usage"`
}

type CompletionChoice

type CompletionChoice struct {
	Text         string      `json:"text"`
	Index        int         `json:"index"`
	Logprobs     interface{} `json:"logprobs"`
	FinishReason string      `json:"finish_reason"`
}

type CompletionResponse

type CompletionResponse struct {
	ID      string             `json:"id"`
	Object  string             `json:"object"`
	Created int                `json:"created"`
	Model   string             `json:"model"`
	Choices []CompletionChoice `json:"choices"`
	Usage   struct {
		PromptTokens     int `json:"prompt_tokens"`
		CompletionTokens int `json:"completion_tokens"`
		TotalTokens      int `json:"total_tokens"`
	} `json:"usage"`
}

type Config

type Config struct {
	OpenAiApiHost  string
	OpenAiApiToken string
	Suggestions    string
	Length         string
	Max_tokens     string
	Skipedit       bool
	Language       string
	Signoff        bool
	Debug          bool
	Diffcontext    string
}

type Language

type Language struct {
	Language string `yaml:"language"`
	System   string `yaml:"system"`
	User     string `yaml:"user"`
}

type Message

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}