spark

Documentation

Index

• func Create(id, name string, v *Config, workers map[eosc.RequireId]eosc.IWorker) (eosc.IWorker, error)
• func NewFactory() eosc.IExtenderDriverFactory
• func Register(register eosc.IExtenderDriverRegister)
• type Chat
  • func NewChat() *Chat
  • func (c *Chat) Endpoint() string
  • func (c *Chat) RequestConvert(ctx eocontext.EoContext, extender map[string]interface{}) error
  • func (c *Chat) ResponseConvert(ctx eocontext.EoContext) error
• type ClientRequest
• type Config
• type Converter
  • func (c *Converter) RequestConvert(ctx eocontext.EoContext, extender map[string]interface{}) error
  • func (c *Converter) ResponseConvert(ctx eocontext.EoContext) error
• type Error
• type IModelMode
• type Message
• type ModelConfig
• type Response
• type ResponseChoice

Functions

func Create

func Create(id, name string, v *Config, workers map[eosc.RequireId]eosc.IWorker) (eosc.IWorker, error)

Create 创建驱动实例

func NewFactory

func NewFactory() eosc.IExtenderDriverFactory

NewFactory 创建service_http驱动工厂

func Register

func Register(register eosc.IExtenderDriverRegister)

Register 注册驱动

Types

type Chat

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

func NewChat

func NewChat() *Chat

func (*Chat) Endpoint

func (c *Chat) Endpoint() string

func (*Chat) RequestConvert

func (c *Chat) RequestConvert(ctx eocontext.EoContext, extender map[string]interface{}) error

func (*Chat) ResponseConvert

func (c *Chat) ResponseConvert(ctx eocontext.EoContext) error

type ClientRequest

type ClientRequest struct {
	Messages []*Message `json:"messages"`
}

type Config

type Config struct {
	APIPassword string `json:"api_password"`
}

type Converter

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

func (*Converter) RequestConvert

func (c *Converter) RequestConvert(ctx eocontext.EoContext, extender map[string]interface{}) error

func (*Converter) ResponseConvert

func (c *Converter) ResponseConvert(ctx eocontext.EoContext) error

type Error

type Error struct {
	Message string      `json:"message"`
	Type    string      `json:"type"`
	Param   interface{} `json:"param"`
	Code    interface{} `json:"code"`
}

type IModelMode

type IModelMode interface {
	Endpoint() string
	convert.IConverter
}

type Message

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

type ModelConfig

type ModelConfig struct {
	FrequencyPenalty float64 `json:"frequency_penalty"`
	MaxTokens        int     `json:"max_tokens"`
	PresencePenalty  float64 `json:"presence_penalty"`
	ResponseFormat   string  `json:"response_format"`
	Temperature      float64 `json:"temperature"`
	TopP             float64 `json:"top_p"`
	TopK             float64 `json:"top_k"`
}

type Response

type Response struct {
	Code    int              `json:"code"`
	Message string           `json:"message"`
	Choices []ResponseChoice `json:"choices"`
	Error   *Error           `json:"error"`
}

type ResponseChoice

type ResponseChoice struct {
	Index   int     `json:"index"`
	Message Message `json:"message"`
}