SDK Go

go get github.com/openai/openai-go/v3

package main

import (
	"context"
	"fmt"
	"os"

	"github.com/openai/openai-go/v3"
	"github.com/openai/openai-go/v3/option"
)

func main() {
	client := openai.NewClient(
		option.WithBaseURL("https://api.uttapen.ir/v1"),
		option.WithAPIKey(os.Getenv("UTTAPEN_API_KEY")), // sk-up-...
	)
	ctx := context.Background()

	resp, err := client.Chat.Completions.New(ctx, openai.ChatCompletionNewParams{
		Model: "openai/gpt-5-mini",
		Messages: []openai.ChatCompletionMessageParamUnion{
			openai.SystemMessage("پاسخ کوتاه و فارسی بده."),
			openai.UserMessage("تفاوت goroutine و thread چیست؟"),
		},
		MaxTokens: openai.Int(300),
	})
	if err != nil {
		panic(err)
	}
	fmt.Println(resp.Choices[0].Message.Content)
	fmt.Println(resp.Usage.PromptTokens, resp.Usage.CompletionTokens)
}

var raw *http.Response
resp, err := client.Chat.Completions.New(ctx, params, option.WithResponseInto(&raw))
if err == nil {
	fmt.Println(raw.Header.Get("X-Uttapen-Cost-Toman"), raw.Header.Get("X-Uttapen-Balance-Toman"))
}

stream := client.Chat.Completions.NewStreaming(ctx, openai.ChatCompletionNewParams{
	Model:    "openai/gpt-5-mini",
	Messages: []openai.ChatCompletionMessageParamUnion{openai.UserMessage("یک تابع اعتبارسنجی کد ملی در Go بنویس.")},
}, option.WithHeader("X-Uttapen-Include-Meta", "1"))

acc := openai.ChatCompletionAccumulator{}
for stream.Next() {
	chunk := stream.Current()
	if chunk.Object == "uttapen.meta" {
		fmt.Println("\nmeta:", chunk.RawJSON()) // cost_toman, balance_toman, hold_toman
		continue
	}
	acc.AddChunk(chunk)
	if len(chunk.Choices) > 0 {
		fmt.Print(chunk.Choices[0].Delta.Content)
	}
}
if err := stream.Err(); err != nil {
	panic(err)
}
fmt.Println("\nتوکن خروجی:", acc.Usage.CompletionTokens)

tools := []openai.ChatCompletionToolUnionParam{
	openai.ChatCompletionFunctionTool(openai.FunctionDefinitionParam{
		Name:        "get_weather",
		Description: openai.String("دمای فعلی یک شهر"),
		Parameters: openai.FunctionParameters{
			"type":       "object",
			"properties": map[string]any{"city": map[string]any{"type": "string"}},
			"required":   []string{"city"},
		},
	}),
}

msgs := []openai.ChatCompletionMessageParamUnion{openai.UserMessage("هوای تهران؟")}
r1, err := client.Chat.Completions.New(ctx, openai.ChatCompletionNewParams{Model: "openai/gpt-5-mini", Messages: msgs, Tools: tools})
if err != nil {
	panic(err)
}
msgs = append(msgs, r1.Choices[0].Message.ToParam())
for _, call := range r1.Choices[0].Message.ToolCalls {
	var args struct{ City string `json:"city"` }
	_ = json.Unmarshal([]byte(call.Function.Arguments), &args)
	result, _ := json.Marshal(map[string]any{"city": args.City, "temp_c": 31})
	msgs = append(msgs, openai.ToolMessage(string(result), call.ID))
}
r2, err := client.Chat.Completions.New(ctx, openai.ChatCompletionNewParams{Model: "openai/gpt-5-mini", Messages: msgs, Tools: tools})
if err != nil {
	panic(err)
}
fmt.Println(r2.Choices[0].Message.Content)

emb, err := client.Embeddings.New(ctx, openai.EmbeddingNewParams{
	Model: "openai/text-embedding-3-small",
	Input: openai.EmbeddingNewParamsInputUnion{OfArrayOfStrings: []string{"قرارداد اجاره", "اجاره‌نامهٔ ملک مسکونی"}},
})
if err != nil {
	panic(err)
}
fmt.Println(len(emb.Data), len(emb.Data[0].Embedding), emb.Usage.PromptTokens)

resp, err := client.Chat.Completions.New(ctx, params,
	option.WithJSONSet("reasoning", map[string]any{"effort": "low"}),
	option.WithJSONSet("models", []string{"openai/o3-mini"}),
)

import "errors"

_, err := client.Chat.Completions.New(ctx, params)
var apierr *openai.Error
if errors.As(err, &apierr) {
	switch apierr.StatusCode {
	case 401:
		log.Fatal("کلید uttapen نامعتبر است")
	case 402:
		log.Printf("موجودی کافی نیست: %s — %s", apierr.Message, apierr.RawJSON()) // بلوک uttapen در RawJSON
	case 429:
		time.Sleep(5 * time.Second)
	default:
		log.Printf("%d %s: %s", apierr.StatusCode, apierr.Code, apierr.Message)
	}
} else if err != nil {
	log.Printf("خطای شبکه: %v", err)
}

import (
	"golang.org/x/sync/errgroup"
	"golang.org/x/sync/semaphore"
)

sem := semaphore.NewWeighted(8) // هشت درخواست هم‌زمان
g, gctx := errgroup.WithContext(ctx)
for _, doc := range docs {
	doc := doc
	g.Go(func() error {
		if err := sem.Acquire(gctx, 1); err != nil {
			return err
		}
		defer sem.Release(1)
		resp, err := client.Chat.Completions.New(gctx, openai.ChatCompletionNewParams{
			Model:     "openai/gpt-5-nano",
			Messages:  []openai.ChatCompletionMessageParamUnion{openai.UserMessage(doc.Text)},
			MaxTokens: openai.Int(200),
		})
		if err != nil {
			return err
		}
		doc.Summary = resp.Choices[0].Message.Content
		return nil
	})
}
if err := g.Wait(); err != nil {
	log.Fatal(err)
}