ernie

type ErnieFunction struct {
	Name        string                  `json:"name"`
	Description string                  `json:"description"`
	Parameters  any                     `json:"parameters"`
	Responses   any                     `json:"responses"`
	Examples    []*ErnieFunctionExample `json:"examples,omitempty"`
}

type ErnieFunctionCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
	Thoughts  string `json:"thoughts"`
}

type ErnieFunctionExample struct {
	Role         string             `json:"role"`
	Content      *string            `json:"content"`
	Name         string             `json:"name,omitempty"`
	FunctionCall *ErnieFunctionCall `json:"function_call,omitempty"`
}

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

type ModelType int

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

type ReqChatCompMsg struct {
	Role         string             `json:"role"`
	Content      *string            `json:"content"`
	Name         string             `json:"name,omitempty"`
	FunctionCall *ErnieFunctionCall `json:"function_call,omitempty"`
}

type Request struct {
	Messages        []*ReqChatCompMsg `json:"messages"`
	Temperature     float32           `json:"temperature,omitempty"`
	TopP            float32           `json:"top_p,omitempty"`
	PresencePenalty float32           `json:"presence_penalty,omitempty"`
	Stream          bool              `json:"stream"`
	UserId          string            `json:"user_id,omitempty"`
	PenaltyScore    float32           `json:"penalty_score,omitempty"`
	Functions       []*ErnieFunction  `json:"functions,omitempty"`
	System          string            `json:"system,omitempty"`
	Stop            []string          `json:"stop,omitempty"`
	DisableSearch   bool              `json:"disable_search,omitempty"`
	EnableCitation  bool              `json:"enable_citation,omitempty"`
}

type RespError struct {
	ErrorCode int    `json:"error_code"`
	ErrorMsg  string `json:"error_msg"`
}

type RespPluginUsage struct {
	Name           string `json:"name"`
	ParseTokens    int    `json:"parse_tokens"`
	AbstractTokens int    `json:"abstract_tokens"`
	SearchTokens   int    `json:"search_tokens"`
	TotalTokens    int    `json:"total_tokens"`
}

type RespSearchInfo struct {
	IsBeset       bool                `json:"is_beset"`
	ReWrite       string              `json:"re_write"`
	SearchResults []*RespSearchResult `json:"search_results,omitempty"`
}

type RespSearchResult struct {
	Index        int    `json:"index"`
	Url          string `json:"url"`
	Title        string `json:"title"`
	DataSourceId string `json:"data_source_id"`
}

type RespUsage struct {
	PromptTokens     int                `json:"prompt_tokens"`
	CompletionTokens int                `json:"completion_tokens"`
	TotalTokens      int                `json:"total_tokens"`
	Plugins          []*RespPluginUsage `json:"plugins,omitempty"`
}

type Response struct {
	RespError

	ID               string             `json:"id"`
	Object           string             `json:"object"`
	Created          int                `json:"created"`
	SentenceId       int                `json:"sentence_id"`
	BanRound         int                `json:"ban_round"`
	IsEnd            bool               `json:"is_end"`
	IsTruncated      bool               `json:"is_truncated"`
	NeedClearHistory bool               `json:"need_clear_history"`
	FinishReason     string             `json:"finish_reason"`
	Result           string             `json:"result"`
	Usage            RespUsage          `json:"usage"`
	FunctionCall     *ErnieFunctionCall `json:"function_call,omitempty"`
	SearchInfo       *RespSearchInfo    `json:"search_info,omitempty"`
}

type ResponseAccessToken struct {
	RefreshToken     string `json:"refresh_token"`
	ExpiresIn        int    `json:"expires_in"`
	SessionKey       string `json:"session_key"`
	AccessToken      string `json:"access_token"`
	Scope            string `json:"scope"`
	SessionSecret    string `json:"session_secret"`
	Error            string `json:"error"`
	ErrorDescription string `json:"error_description"`
}