core

Documentation

Index

• func NeedPassword(backend string) bool
• func NewMemoryService() *memoryService
• type Configuration
• type IAI
  • func NewClient(provider string) IAI
• type IAIConfig
• type OpenAIClient
  • func (c *OpenAIClient) ClearHistory(ctx context.Context) error
  • func (OpenAIClient) Close()
  • func (c *OpenAIClient) Configure(config IAIConfig) error
  • func (c *OpenAIClient) GetCompletion(ctx context.Context, contents ...any) (string, error)
  • func (c *OpenAIClient) GetCompletionWithTools(ctx context.Context, contents ...any) ([]openai.ToolCall, string, error)
  • func (c *OpenAIClient) GetHistory(ctx context.Context) []openai.ChatCompletionMessage
  • func (c *OpenAIClient) GetName() string
  • func (c *OpenAIClient) GetStreamCompletion(ctx context.Context, contents ...any) (*openai.ChatCompletionStream, error)
  • func (c *OpenAIClient) GetStreamCompletionWithTools(ctx context.Context, contents ...any) (*openai.ChatCompletionStream, error)
  • func (c *OpenAIClient) SaveAIHistory(ctx context.Context, contents string)
  • func (c *OpenAIClient) SetTools(tools []openai.Tool)
• type OpenAIHeaderTransport
  • func (t *OpenAIHeaderTransport) RoundTrip(req *http.Request) (*http.Response, error)
• type Provider
  • func (p *Provider) GetBaseURL() string
  • func (p *Provider) GetCompartmentId() string
  • func (p *Provider) GetCustomHeaders() []http.Header
  • func (p *Provider) GetEndpointName() string
  • func (p *Provider) GetEngine() string
  • func (p *Provider) GetMaxHistory() int32
  • func (p *Provider) GetMaxTokens() int
  • func (p *Provider) GetModel() string
  • func (p *Provider) GetOrganizationId() string
  • func (p *Provider) GetPassword() string
  • func (p *Provider) GetProviderId() string
  • func (p *Provider) GetProviderRegion() string
  • func (p *Provider) GetProxyEndpoint() string
  • func (p *Provider) GetTemperature() float32
  • func (p *Provider) GetThink() bool
  • func (p *Provider) GetTopK() int32
  • func (p *Provider) GetTopP() float32

Functions

func NeedPassword

func NeedPassword(backend string) bool

func NewMemoryService

func NewMemoryService() *memoryService

NewMemoryService 创建 MemoryService 实例

Types

type Configuration

type Configuration struct {
	Providers       []Provider
	DefaultProvider string
}

type IAI

type IAI interface {
	Configure(config IAIConfig) error
	GetCompletion(ctx context.Context, contents ...any) (string, error)
	GetCompletionWithTools(ctx context.Context, contents ...any) ([]openai.ToolCall, string, error)
	GetStreamCompletion(ctx context.Context, contents ...any) (*openai.ChatCompletionStream, error)
	GetStreamCompletionWithTools(ctx context.Context, contents ...any) (*openai.ChatCompletionStream, error)
	GetName() string
	Close()
	SetTools(tools []openai.Tool)
	SaveAIHistory(ctx context.Context, content string)
	GetHistory(ctx context.Context) []openai.ChatCompletionMessage
	ClearHistory(ctx context.Context) error
}

func NewClient

func NewClient(provider string) IAI

type IAIConfig

type IAIConfig interface {
	GetPassword() string
	GetModel() string
	GetBaseURL() string
	GetProxyEndpoint() string
	GetEndpointName() string
	GetEngine() string
	GetTemperature() float32
	GetProviderRegion() string
	GetTopP() float32
	GetTopK() int32
	GetMaxTokens() int
	GetMaxHistory() int32
	GetProviderId() string
	GetCompartmentId() string
	GetOrganizationId() string
	GetCustomHeaders() []http.Header
	GetThink() bool
}

type OpenAIClient

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

func (*OpenAIClient) ClearHistory

func (c *OpenAIClient) ClearHistory(ctx context.Context) error

func (OpenAIClient) Close

func (OpenAIClient) Close()

func (*OpenAIClient) Configure

func (c *OpenAIClient) Configure(config IAIConfig) error

func (*OpenAIClient) GetCompletion

func (c *OpenAIClient) GetCompletion(ctx context.Context, contents ...any) (string, error)

func (*OpenAIClient) GetCompletionWithTools

func (c *OpenAIClient) GetCompletionWithTools(ctx context.Context, contents ...any) ([]openai.ToolCall, string, error)

func (*OpenAIClient) GetHistory

func (c *OpenAIClient) GetHistory(ctx context.Context) []openai.ChatCompletionMessage

func (*OpenAIClient) GetName

func (c *OpenAIClient) GetName() string

func (*OpenAIClient) GetStreamCompletion

func (c *OpenAIClient) GetStreamCompletion(ctx context.Context, contents ...any) (*openai.ChatCompletionStream, error)

func (*OpenAIClient) GetStreamCompletionWithTools

func (c *OpenAIClient) GetStreamCompletionWithTools(ctx context.Context, contents ...any) (*openai.ChatCompletionStream, error)

func (*OpenAIClient) SaveAIHistory

func (c *OpenAIClient) SaveAIHistory(ctx context.Context, contents string)

func (*OpenAIClient) SetTools

func (c *OpenAIClient) SetTools(tools []openai.Tool)

type OpenAIHeaderTransport

type OpenAIHeaderTransport struct {
	Origin  http.RoundTripper
	Headers []http.Header
}

OpenAIHeaderTransport is an http.RoundTripper that adds the given headers to each request.

func (*OpenAIHeaderTransport) RoundTrip

func (t *OpenAIHeaderTransport) RoundTrip(req *http.Request) (*http.Response, error)

RoundTrip implements the http.RoundTripper interface.

type Provider

type Provider struct {
	Name           string
	Model          string
	Password       string
	BaseURL        string
	ProxyEndpoint  string
	ProxyPort      string
	EndpointName   string
	Engine         string
	Temperature    float32
	ProviderRegion string
	ProviderId     string
	CompartmentId  string
	TopP           float32
	TopK           int32
	MaxHistory     int32
	MaxTokens      int
	OrganizationId string
	CustomHeaders  []http.Header
	Think          bool
}

func (*Provider) GetBaseURL

func (p *Provider) GetBaseURL() string

func (*Provider) GetCompartmentId

func (p *Provider) GetCompartmentId() string

func (*Provider) GetCustomHeaders

func (p *Provider) GetCustomHeaders() []http.Header

func (*Provider) GetEndpointName

func (p *Provider) GetEndpointName() string

func (*Provider) GetEngine

func (p *Provider) GetEngine() string

func (*Provider) GetMaxHistory

func (p *Provider) GetMaxHistory() int32

func (*Provider) GetMaxTokens

func (p *Provider) GetMaxTokens() int

func (*Provider) GetModel

func (p *Provider) GetModel() string

func (*Provider) GetOrganizationId

func (p *Provider) GetOrganizationId() string

func (*Provider) GetPassword

func (p *Provider) GetPassword() string

func (*Provider) GetProviderId

func (p *Provider) GetProviderId() string

func (*Provider) GetProviderRegion

func (p *Provider) GetProviderRegion() string

func (*Provider) GetProxyEndpoint

func (p *Provider) GetProxyEndpoint() string

func (*Provider) GetTemperature

func (p *Provider) GetTemperature() float32

func (*Provider) GetThink

func (p *Provider) GetThink() bool

func (*Provider) GetTopK

func (p *Provider) GetTopK() int32

func (*Provider) GetTopP

func (p *Provider) GetTopP() float32