plugins

Documentation

Index

• Constants
• func InitBeelzebubCloud(uri, authToken string) *beelzebubCloud
• type Choice
• type HoneypotConfigResponseDTO
• type LLMHoneypot
  • func InitLLMHoneypot(config LLMHoneypot) *LLMHoneypot
  • func (llmHoneypot *LLMHoneypot) ExecuteModel(command string) (string, error)
• type LLMProvider
  • func FromStringToLLMProvider(llmProvider string) (LLMProvider, error)
• type Message
• type Request
• type Response
• type Role
  • func (role Role) String() string

Constants

const (
	LLMPluginName = "LLMHoneypot"
)

Functions

func InitBeelzebubCloud

func InitBeelzebubCloud(uri, authToken string) *beelzebubCloud

Types

type Choice

type Choice struct {
	Message      Message `json:"message"`
	Index        int     `json:"index"`
	FinishReason string  `json:"finish_reason"`
}

type HoneypotConfigResponseDTO

type HoneypotConfigResponseDTO struct {
	ID            string `json:"id"`
	Config        string `json:"config"`
	TokenID       string `json:"tokenId"`
	LastUpdatedOn string `json:"lastUpdatedOn"`
}

type LLMHoneypot

type LLMHoneypot struct {
	Histories []Message
	OpenAIKey string

	Protocol     tracer.Protocol
	Provider     LLMProvider
	Model        string
	Host         string
	CustomPrompt string
	// contains filtered or unexported fields
}

func InitLLMHoneypot

func InitLLMHoneypot(config LLMHoneypot) *LLMHoneypot

func (*LLMHoneypot) ExecuteModel

func (llmHoneypot *LLMHoneypot) ExecuteModel(command string) (string, error)

type LLMProvider

type LLMProvider int

const (
	Ollama LLMProvider = iota
	OpenAI
)

func FromStringToLLMProvider

func FromStringToLLMProvider(llmProvider string) (LLMProvider, error)

type Message

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

type Request

type Request struct {
	Model    string    `json:"model"`
	Messages []Message `json:"messages"`
	Stream   bool      `json:"stream"`
}

type Response

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

type Role

type Role int

const (
	SYSTEM Role = iota
	USER
	ASSISTANT
)

func (Role) String

func (role Role) String() string