go on

Author: liujunliuhong

.gitignore

@@ -0,0 +1,21 @@
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/
+
+# Auth token for tests
+.openai-token
+.idea
+
+src/key.go

README.md

@@ -0,0 +1,3 @@
+# go-openai
+
+OpenAI wrapper for Go

src/completions/param.go

@@ -0,0 +1,20 @@
+package completions
+
+type Param struct {
+	Model            string         `json:"model"`
+	Prompt           any            `json:"prompt,omitempty"`
+	Suffix           string         `json:"suffix,omitempty"`
+	MaxTokens        int            `json:"max_tokens,omitempty"`
+	Temperature      float32        `json:"temperature,omitempty"`
+	TopP             float32        `json:"top_p,omitempty"`
+	N                int            `json:"n,omitempty"`
+	Stream           bool           `json:"stream,omitempty"`
+	LogProbs         int            `json:"logprobs,omitempty"`
+	Echo             bool           `json:"echo,omitempty"`
+	Stop             any            `json:"stop,omitempty"`
+	PresencePenalty  float32        `json:"presence_penalty,omitempty"`
+	FrequencyPenalty float32        `json:"frequency_penalty,omitempty"`
+	BestOf           int            `json:"best_of,omitempty"`
+	LogitBias        map[string]int `json:"logit_bias,omitempty"`
+	User             string         `json:"user,omitempty"`
+}

src/completions/provider.go

@@ -0,0 +1,12 @@
+package completions
+
+import (
+	"configuration"
+	"request_handler"
+)
+
+func Create(c *configuration.Configuration, p Param) (response Response, err error) {
+	r := NewRequest(c, p)
+	err = request_handler.Perform(r, &response)
+	return
+}

src/completions/request.go

@@ -0,0 +1,56 @@
+package completions
+
+import (
+	"bytes"
+	"configuration"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"request"
+)
+
+// Request https://platform.openai.com/docs/api-reference/completions
+type Request struct {
+	c *configuration.Configuration
+	p Param
+}
+
+func NewRequest(c *configuration.Configuration, p Param) *Request {
+	return &Request{c: c, p: p}
+}
+
+func (c *Request) Method() string {
+	return http.MethodPost
+}
+
+func (c *Request) Host() string {
+	return request.BaseURL
+}
+
+func (c *Request) Path() string {
+	return "/v1/completions"
+}
+
+func (c *Request) URL() string {
+	return c.Host() + c.Path()
+}
+
+func (c *Request) Headers() map[string]string {
+	headers := map[string]string{
+		"Content-Type": "application/json",
+	}
+	for k, v := range c.c.Headers() {
+		headers[k] = v
+	}
+	return headers
+}
+
+func (c *Request) Body() io.Reader {
+	marshal, err := json.Marshal(c.p)
+	if err != nil {
+		return nil
+	}
+	fmt.Println(string(marshal))
+	return bytes.NewBuffer(marshal)
+}

src/completions/response.go

@@ -0,0 +1,30 @@
+package completions
+
+type Response struct {
+	Id      string    `json:"id"`
+	Object  string    `json:"object"`
+	Created int       `json:"created"`
+	Model   string    `json:"model"`
+	Usage   Usage     `json:"usage,omitempty"`
+	Choices []Choices `json:"choices"`
+}
+
+type Choices struct {
+	Text         string   `json:"text,omitempty"`
+	Index        int      `json:"index,omitempty"`
+	Logprobs     Logprobs `json:"logprobs,omitempty"`
+	FinishReason string   `json:"finish_reason,omitempty"`
+}
+
+type Logprobs struct {
+	Tokens        []string             `json:"tokens,omitempty"`
+	TokenLogprobs []float32            `json:"token_logprobs,omitempty"`
+	TopLogprobs   []map[string]float32 `json:"top_logprobs,omitempty"`
+	TextOffset    []int                `json:"text_offset,omitempty"`
+}
+
+type Usage struct {
+	PromptTokens     int `json:"prompt_tokens,omitempty"`
+	CompletionTokens int `json:"completion_tokens,omitempty"`
+	TotalTokens      int `json:"total_tokens,omitempty"`
+}

src/configuration/configuration.go

@@ -0,0 +1,31 @@
+package configuration
+
+import "fmt"
+
+type Configuration struct {
+	ApiKey       string
+	Organization string
+}
+
+func NewConfiguration(apiKey string) *Configuration {
+	return &Configuration{
+		ApiKey: apiKey,
+	}
+}
+
+func NewConfigurationWithOrg(apiKey string, organization string) *Configuration {
+	return &Configuration{
+		ApiKey:       apiKey,
+		Organization: organization,
+	}
+}
+
+func (c *Configuration) Headers() map[string]string {
+	var header = map[string]string{
+		"Authorization": fmt.Sprintf("Bearer %s", c.ApiKey),
+	}
+	if len(c.Organization) != 0 {
+		header["OpenAI-Organization"] = c.Organization
+	}
+	return header
+}

src/main.go

@@ -0,0 +1,44 @@
+package main
+
+import (
+	"completions"
+	"configuration"
+	"encoding/json"
+	"fmt"
+)
+
+func main() {
+	fmt.Println("Hello, World")
+
+	key := "sk-xxxx"
+	c := configuration.NewConfiguration(key)
+
+	p := completions.Param{
+		Model:       "text-davinci-003",
+		Prompt:      "今天是几号？",
+		Temperature: 1,
+		MaxTokens:   1024,
+	}
+
+	response, err := completions.Create(c, p)
+	if err != nil {
+		fmt.Println(err.Error())
+		return
+	}
+	marshal, err := json.Marshal(response)
+	if err != nil {
+		return
+	}
+	fmt.Println(string(marshal))
+
+	//
+	//// list, err := models.List(c)
+	//model, err := models.Retrieve(c, "babbage")
+	//
+	//if err != nil {
+	//	fmt.Println(err.Error())
+	//	return
+	//}
+	//fmt.Println(model)
+
+}

src/models/list_models_request.go

@@ -0,0 +1,47 @@
+package models
+
+import (
+	"configuration"
+	"io"
+	"net/http"
+	"request"
+)
+
+// ListModelsRequest https://platform.openai.com/docs/api-reference/models/list
+type ListModelsRequest struct {
+	c *configuration.Configuration
+}
+
+func NewListModelsRequest(c *configuration.Configuration) *ListModelsRequest {
+	return &ListModelsRequest{c: c}
+}
+
+func (l *ListModelsRequest) Method() string {
+	return http.MethodGet
+}
+
+func (l *ListModelsRequest) Host() string {
+	return request.BaseURL
+}
+
+func (l *ListModelsRequest) Path() string {
+	return "/v1/models"
+}
+
+func (l *ListModelsRequest) URL() string {
+	return l.Host() + l.Path()
+}
+
+func (l *ListModelsRequest) Headers() map[string]string {
+	headers := map[string]string{
+		"Content-Type": "application/json",
+	}
+	for k, v := range l.c.Headers() {
+		headers[k] = v
+	}
+	return headers
+}
+
+func (l *ListModelsRequest) Body() io.Reader {
+	return nil
+}

src/models/provider.go

@@ -0,0 +1,18 @@
+package models
+
+import (
+	"configuration"
+	"request_handler"
+)
+
+func List(c *configuration.Configuration) (response Response, err error) {
+	r := NewListModelsRequest(c)
+	err = request_handler.Perform(r, &response)
+	return
+}
+
+func Retrieve(c *configuration.Configuration, id string) (model Model, err error) {
+	r := NewRetrieveModelRequest(c, id)
+	err = request_handler.Perform(r, &model)
+	return
+}

src/models/response.go

@@ -0,0 +1,30 @@
+package models
+
+type Model struct {
+	Id         string       `json:"id"`
+	Object     string       `json:"object"`
+	Created    int64        `json:"created"`
+	OwnedBy    string       `json:"owned_by"`
+	Root       string       `json:"root"`
+	Parent     string       `json:"parent"`
+	Permission []Permission `json:"permission"`
+}
+
+type Permission struct {
+	Id                 string      `json:"id"`
+	Object             string      `json:"object"`
+	Created            int         `json:"created"`
+	AllowCreateEngine  bool        `json:"allow_create_engine"`
+	AllowSampling      bool        `json:"allow_sampling"`
+	AllowLogprobs      bool        `json:"allow_logprobs"`
+	AllowSearchIndices bool        `json:"allow_search_indices"`
+	AllowView          bool        `json:"allow_view"`
+	AllowFineTuning    bool        `json:"allow_fine_tuning"`
+	Organization       string      `json:"organization"`
+	Group              interface{} `json:"group"`
+	IsBlocking         bool        `json:"is_blocking"`
+}
+
+type Response struct {
+	Models []Model `json:"data"`
+}

src/models/retrieve_model_request.go

@@ -0,0 +1,49 @@
+package models
+
+import (
+	"configuration"
+	"fmt"
+	"io"
+	"net/http"
+	"request"
+)
+
+// RetrieveModelRequest https://platform.openai.com/docs/api-reference/models/retrieve
+type RetrieveModelRequest struct {
+	c       *configuration.Configuration
+	modelID string
+}
+
+func NewRetrieveModelRequest(c *configuration.Configuration, modelID string) *RetrieveModelRequest {
+	return &RetrieveModelRequest{c: c, modelID: modelID}
+}
+
+func (r *RetrieveModelRequest) Method() string {
+	return http.MethodGet
+}
+
+func (r *RetrieveModelRequest) Host() string {
+	return request.BaseURL
+}
+
+func (r *RetrieveModelRequest) Path() string {
+	return fmt.Sprintf("/v1/models/%s", r.modelID)
+}
+
+func (r *RetrieveModelRequest) URL() string {
+	return r.Host() + r.Path()
+}
+
+func (r *RetrieveModelRequest) Headers() map[string]string {
+	headers := map[string]string{
+		"Content-Type": "application/json",
+	}
+	for k, v := range r.c.Headers() {
+		headers[k] = v
+	}
+	return headers
+}
+
+func (r *RetrieveModelRequest) Body() io.Reader {
+	return nil
+}