Migrate to v2

1
--- v1.go
2
+++ v2.go
3
@@ -5,7 +5,8 @@ import (
4
 	"log"
5
 	"os"
6
 
7
-	entitymanagerv1 "github.com/anduril/lattice-sdk-go/src/anduril/entitymanager/v1"
8
+	"buf.build/gen/go/anduril/lattice-sdk/grpc/go/anduril/entitymanager/v1/entitymanagerv1grpc"
9
+	entitymanagerv1 "buf.build/gen/go/anduril/lattice-sdk/protocolbuffers/go/anduril/entitymanager/v1"
10
 	"google.golang.org/grpc"
11
 	"google.golang.org/grpc/credentials"
12
 )
13
@@ -30,15 +31,16 @@ func main() {
14
 	if err != nil {
15
 		log.Fatalf("did not connect: %v", err)
16
 	}
17
+	defer conn.Close()
18
 
19
-	client := entitymanagerv1.NewEntityManagerAPIClient(conn)
20
+	client := entitymanagerv1grpc.NewEntityManagerAPIClient(conn)
21
 
22
-	request := &entitymanagerv1.GetEntityRequest{
23
+	response, err := client.GetEntity(ctx, &entitymanagerv1.GetEntityRequest{
24
 		EntityId: "<ENTITY_ID>",
25
-	}
26
-	response, err := client.GetEntity(ctx, request)
27
+	})
28
 
29
 	if err != nil {
30
		log.Fatalf("Error fetching entity: %v", err)
31
 	}
32
 	log.Printf("Response: %v", response)

1
--- grpc_v1.go	2025-08-01 10:11:20
2
+++ rest_v2.go	2025-08-01 12:07:40
3
@@ -5,39 +5,26 @@
4
 	"log"
5
 	"os"
6
 
7
-	entitymanagerv1 "github.com/anduril/lattice-sdk-go/src/anduril/entitymanager/v1"
8
-	"google.golang.org/grpc"
9
-	"google.golang.org/grpc/credentials"
10
+	"github.com/anduril/lattice-sdk-go/v2/client"
11
+	"github.com/anduril/lattice-sdk-go/v2/option"
12
 )
13
 
14
 func main() {
15
 	ctx := context.Background()
16
 	latticeEndpoint := os.Getenv("LATTICE_ENDPOINT")
17
	environmentToken := os.Getenv("ENVIRONMENT_TOKEN")
18
 
19
 	if latticeEndpoint == "" || environmentToken == "" {
20
 		log.Fatalf("Missing required environment variables")
21
 	}
22
-	auth := &BearerTokenAuth{
23
-		Token: environmentToken,
24
-	}
25
 
26
-	opts := []grpc.DialOption{
27
-		grpc.WithTransportCredentials(credentials.NewClientTLSFromCert(nil, "")),
28
-		grpc.WithPerRPCCredentials(auth),
29
-	}
30
-	conn, err := grpc.NewClient(latticeEndpoint, opts...)
31
-	if err != nil {
32
-		log.Fatalf("did not connect: %v", err)
33
-	}
34
+	client := client.NewClient(
35
+		option.WithToken(environmentToken),
36
+		option.WithBaseURL(latticeEndpoint),
37
+	)
38
 
39
-	client := entitymanagerv1.NewEntityManagerAPIClient(conn)
40
+	response, err := client.Entities.GetEntity(ctx, "<ENTITY_ID>")
41
 
42
-	request := &entitymanagerv1.GetEntityRequest{
43
-		EntityId: "<ENTITY_ID>",
44
-	}
45
-	response, err := client.GetEntity(ctx, request)
46
-
47
 	if err != nil {
48
 		log.Fatalf("Error fetching entity: %v", err)
49
 	}