openai

Documentation

Index

• type ChatRequest
• type ChatResponse
• type Client
  • func NewClient(apiKey string) *Client
  • func (c *Client) GeneratePostContent(prompt string) (string, error)
• type Message

Types

type ChatRequest

type ChatRequest struct {
	Model       string    `json:"model"`
	Messages    []Message `json:"messages"`
	MaxTokens   int       `json:"max_tokens,omitempty"`
	Temperature float64   `json:"temperature,omitempty"`
}

ChatRequest represents the request body for chat completions API

type ChatResponse

type ChatResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	Choices []struct {
		Message struct {
			Role    string `json:"role"`
			Content string `json:"content"`
		} `json:"message"`
		FinishReason string `json:"finish_reason"`
	} `json:"choices"`
}

ChatResponse represents the response from chat completions API

type Client

type Client struct {
	APIKey     string
	HTTPClient *http.Client
	Model      string
}

Client represents an OpenAI API client

func NewClient

func NewClient(apiKey string) *Client

NewClient creates a new OpenAI client

func (*Client) GeneratePostContent

func (c *Client) GeneratePostContent(prompt string) (string, error)

GeneratePostContent creates content based on a prompt

type Message

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

Message represents a chat message