Azure OpenAI Integration

go get github.com/langwatch/langwatch/sdks/go
go get github.com/langwatch/langwatch/sdks/go/instrumentation/azureopenai
go get github.com/openai/openai-go/v3

package main

import (
	"context"
	"log"
	"os"

	"github.com/openai/openai-go/v3"
	"github.com/openai/openai-go/v3/azure"
	"github.com/openai/openai-go/v3/option"
	"go.opentelemetry.io/otel"
	sdktrace "go.opentelemetry.io/otel/sdk/trace"

	langwatch "github.com/langwatch/langwatch/sdks/go"
	azureopenai "github.com/langwatch/langwatch/sdks/go/instrumentation/azureopenai"
)

func main() {
	ctx := context.Background()

	// Set up the LangWatch exporter (reads LANGWATCH_API_KEY from env).
	exporter, err := langwatch.NewExporter(ctx)
	if err != nil {
		log.Fatalf("failed to create exporter: %v", err)
	}
	tp := sdktrace.NewTracerProvider(sdktrace.WithBatcher(exporter))
	otel.SetTracerProvider(tp)
	defer tp.Shutdown(ctx) // Critical: ensures traces are flushed.

	// Create an instrumented Azure OpenAI client. The provider is recorded as
	// "azure.openai"; by default the middleware captures both input and output.
	const apiVersion = "2024-06-01"
	client := openai.NewClient(
		azure.WithEndpoint(os.Getenv("AZURE_OPENAI_ENDPOINT"), apiVersion),
		azure.WithAPIKey(os.Getenv("AZURE_OPENAI_API_KEY")),
		// Or authenticate with an Azure Identity TokenCredential:
		// azure.WithTokenCredential(tokenCredential),
		option.WithMiddleware(azureopenai.Middleware("my-app")),
	)

	// The Model is your Azure *deployment* name; the azure subpackage rewrites
	// the request path to /openai/deployments/{deployment}/chat/completions.
	response, err := client.Chat.Completions.New(ctx, openai.ChatCompletionNewParams{
		Model: "my-gpt4o-deployment",
		Messages: []openai.ChatCompletionMessageParamUnion{
			openai.SystemMessage("You are a helpful assistant."),
			openai.UserMessage("Hello, Azure OpenAI!"),
		},
	})
	if err != nil {
		log.Fatalf("Chat completion failed: %v", err)
	}

	log.Printf("Response: %s", response.Choices[0].Message.Content)
}