ai

Documentation

Index

• Constants
• Variables
• type AiSuggestionRequest
• type AiSuggestionResponse
• type OpenAiUsage
  • func GetAiSuggestionsViaOpenAiApi(apiEndpoint, query, shellName, osName, overriddenOpenAiModel string, ...) ([]string, OpenAiUsage, error)
• type TestOnlyOverrideAiSuggestionRequest

Constants

const DefaultOpenAiEndpoint = "https://api.openai.com/v1/chat/completions"

Variables

var TestOnlyOverrideAiSuggestions map[string][]string = make(map[string][]string)

Types

type AiSuggestionRequest

type AiSuggestionRequest struct {
	DeviceId          string `json:"device_id"`
	UserId            string `json:"user_id"`
	Query             string `json:"query"`
	NumberCompletions int    `json:"number_completions"`
	ShellName         string `json:"shell_name"`
	OsName            string `json:"os_name"`
	Model             string `json:"model"`
}

type AiSuggestionResponse

type AiSuggestionResponse struct {
	Suggestions []string `json:"suggestions"`
}

type OpenAiUsage

type OpenAiUsage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

func GetAiSuggestionsViaOpenAiApi

func GetAiSuggestionsViaOpenAiApi(apiEndpoint, query, shellName, osName, overriddenOpenAiModel string, numberCompletions int) ([]string, OpenAiUsage, error)

type TestOnlyOverrideAiSuggestionRequest

type TestOnlyOverrideAiSuggestionRequest struct {
	Query       string   `json:"query"`
	Suggestions []string `json:"suggestions"`
}