openai go

Golang OpenAI API 클라이언트

OpenAI API와 쉽게 상호작용하기 위한 Golang 기본 구현입니다.

https://beta.openai.com/docs/api-reference/

용법

환경 변수를 사용하여 API 비밀 키를 저장할 수 있습니다

 export OPENAI_KEY=[YOUR_KEY]

엔진을 초기화하려면 다음을 사용하십시오.

 e := openai . New ( os . Getenv ( "OPENAI_KEY" ))

팁

모델

가장 강력한 모델을 사용하여 텍스트 출력을 생성하려면 "text-davinci-003"을 사용하고 있는지 확인하세요. 이 모델은 상수 openai.ModelTextDavinci003 으로 정의됩니다.

텍스트 에디션

Completion+Edit 번들을 사용하여 마지막 컨텍스트를 기반으로 응답을 다시 생성할 수 있습니다.

 e := openai . New ( os . Getenv ( "OPENAI_KEY" ))
ctx := context . Background ()
completionResp , err := e . Completion ( ctx , & openai. CompletionOptions {
	// Choose model, you can see list of available models in models.go file
	Model : openai . ModelTextDavinci001 ,
	// Number of completion tokens to generate response. By default - 1024
	MaxTokens : 1200 ,
	// Text to completion
	Prompt : [] string { "Write a little bit of Wikipedia. What is that?" },
})

editResp , err := e . Edit ( ctx , & EditOptions {
	Model :       ModelTextDavinci001 ,
	Input :       completionResp . Choices [ 0 ],
	Instruction : "Please rewrite a bit more and add more information about Wikipedia in different aspects. Please build based on that for 4 topics" ,
})

텍스트 완성 예시

프롬프트가 주어지면 모델은 하나 이상의 예상 완료를 반환합니다.

참고 : 완료 토큰의 기본 수는 1024입니다. 이 제한을 늘리거나 줄이려면 MaxTokens 매개변수를 변경해야 합니다.

 e := openai . New ( os . Getenv ( "OPENAI_KEY" ))
r , err := e . Completion ( context . Background (), & openai. CompletionOptions {
	// Choose model, you can see list of available models in models.go file
	Model : openai . ModelTextDavinci001 ,
	// Number of completion tokens to generate response. By default - 1024
	MaxTokens : 1200 ,
	// Text to completion
	Prompt : [] string { "Write a little bit of Wikipedia. What is that?" },
})

다음 출력을 얻게 됩니다.

{
  "id" : " cmpl-6SrcYDLCVT7xyHKVNuSLNuhRvwOJ1 " ,
  "object" : " text_completion " ,
  "created" : 1672337322 ,
  "model" : " text-davinci-001 " ,
  "choices" : [
    {
      "text" : " nn Wikipedia is a free online encyclopedia, created and edited by volunteers. " ,
      "index" : 0 ,
      "finish_reason" : " stop "
    }
  ],
  "usage" : {
    "prompt_tokens" : 11 ,
    "completion_tokens" : 15 ,
    "total_tokens" : 26
  }
}

텍스트만 얻으려면 다음 코드를 사용해야 합니다:

 fmt . Println ( r . Choices [ 0 ]. Text )

따라서 전체 코드는 다음과 같습니다.

 package main 


import (
	"context"
	"encoding/json"
	"log"
	"os"
	"testing"
  "github.com/0x9ef/openai-go"
)

func main () {
	e := openai . New ( os . Getenv ( "OPENAI_KEY" ))
	r , err := e . Completion ( context . Background (), & openai. CompletionOptions {
		// Choose model, you can see list of available models in models.go file
		Model : openai . ModelTextDavinci001 ,
		// Text to completion
		Prompt : [] string { "Write a little bit of Wikipedia. What is that?" },
	})

	if b , err := json . MarshalIndent ( r , "" , "  " ); err != nil {
		panic ( err )
	} else {
		fmt . Println ( string ( b ))
	}

	// Wikipedia is a free online encyclopedia, created and edited by volunteers.
	fmt . Println ( "What is the Wikipedia?" , r . Choices [ 0 ]. Text )
}

모델 목록/검색

현재 사용 가능한 모델을 나열하고 소유자, 가용성 등 각 모델에 대한 기본 정보를 제공합니다.

 e := openai . New ( os . Getenv ( "OPENAI_KEY" ))
r , err := e . ListModels ( context . Background ())
if err != nil {
	log . Fatal ( err )
}

다음 출력을 얻게 됩니다.

{
  "data" : [
    {
      "id" : " babbage " ,
      "object" : " model " ,
      "owned_by" : " openai "
    },
    {
      "id" : " ada " ,
      "object" : " model " ,
      "owned_by" : " openai "
    },
    {
      "id" : " text-davinci-002 " ,
      "object" : " model " ,
      "owned_by" : " openai "
    },
    {
      "id" : " davinci " ,
      "object" : " model " ,
      "owned_by" : " openai "
    },
    ...
  ]
}

모든 모델 대신 지정된 모델에 대한 정보를 검색하려면 다음을 수행할 수 있습니다.

 e := openai . New ( os . Getenv ( "OPENAI_KEY" ))
r , err := e . RetrieveModel ( context . Background (), & openai. RetrieveModelOptions {
	ID : openai . ModelDavinci ,
})
if err != nil {
	log . Fatal ( err )
}

특허

MIT