pw-protobuf/client/ai.go

package client

import (
	"git.ikuban.com/server/kratos-nacos/registry"
	"git.ikuban.com/server/pw-protobuf/api/ai"
	"github.com/go-kratos/kratos/v2/log"
)

func NewAIClient(r *registry.Registry, logger log.Logger) ai.AiClient {
	conn, err := getDialInsecure(r, logger, "discovery://pw/pw-ai")
	if err != nil {
		panic(err)
	}
	if conn == nil {
		return nil
	}
	client := ai.NewAiClient(conn)
	return client
}