1import * as GCP from "@/GCP";
2import * as Test from "@/Test/Alchemy";
3import * as dialogflow from "@distilled.cloud/gcp/dialogflow_v3";
4import { expect } from "alchemy-test";
5import * as Effect from "effect/Effect";
6import { MinimumLogLevel } from "effect/References";
7import * as Schedule from "effect/Schedule";
8import { deleteAgent, ensureAgent, quotaTolerant } from "./parent.ts";
9import { GcpEnvironment } from "@/GCP/Environment";
10
11const { test } = Test.make({ providers: GCP.providers() });
12
13const logLevel = Effect.provideService(
14 MinimumLogLevel,
15 process.env.DEBUG ? "Debug" : "Info",
16);
17
18const runLifecycle = !process.env.FAST;
19const agentDisplayName = "alch-df-tool";
20
21const waitUntilGone = (name: string) =>
22 dialogflow.getProjectsLocationsAgentsTools({ name }).pipe(
23 Effect.as("found" as const),
24 Effect.catchTag("NotFound", () => Effect.succeed("gone" as const)),
25 Effect.repeat({
26 schedule: Schedule.spaced("2 seconds"),
27 until: (status) => status === "gone",
28 times: 10,
29 }),
30 );
31
32test.provider(
33 "getProjectsLocationsAgentsTools on a missing tool fails with a typed tag",
34 (stack) =>
35 Effect.gen(function* () {
36 const { project } = yield* GcpEnvironment.current;
37 yield* stack.destroy();
38
39 const error = yield* Effect.flip(
40 dialogflow.getProjectsLocationsAgentsTools({
41 name: `projects/${project}/locations/global/agents/missing/tools/missing`,
42 }),
43 );
44 expect(error._tag).toEqual("NotFound");
45
46 yield* stack.destroy();
47 }).pipe(logLevel, quotaTolerant),
48 {
49 tags: ["provider:gcp", "provider:gcp:dialogflow", "live"],
50 timeout: 90_000,
51 },
52);
53
54test.provider.skipIf(!runLifecycle)(
55 "create, update, and delete a tool",
56 (stack) =>
57 Effect.gen(function* () {
58 const { project } = yield* GcpEnvironment.current;
59 yield* stack.destroy();
60
61 const agent = yield* ensureAgent(project, agentDisplayName);
62 const agentName = agent.name ?? "";
63
64 const created = yield* stack.deploy(
65 Effect.gen(function* () {
66 return yield* GCP.Dialogflow.AgentsTool("Lookup", {
67 agent: agentName,
68 displayName: "lookup",
69 description: "Look up an order.",
70 functionSpec: {
71 inputSchema: {
72 type: "object",
73 properties: { orderId: { type: "string" } },
74 },
75 outputSchema: {
76 type: "object",
77 properties: { status: { type: "string" } },
78 },
79 },
80 });
81 }),
82 );
83
84 expect(created.name).toContain("/tools/");
85 expect(created.agent).toEqual(agentName);
86 expect(created.displayName).toEqual("lookup");
87 expect(created.description).toEqual("Look up an order.");
88 expect(created.functionSpec).toBeDefined();
89
90 const fetched = yield* dialogflow.getProjectsLocationsAgentsTools({
91 name: created.name,
92 });
93 expect(fetched.name).toEqual(created.name);
94
95 const updated = yield* stack.deploy(
96 Effect.gen(function* () {
97 return yield* GCP.Dialogflow.AgentsTool("Lookup", {
98 agent: agentName,
99 toolId: created.toolId,
100 displayName: "lookup",
101 description: "Look up an order by id.",
102 functionSpec: {
103 inputSchema: {
104 type: "object",
105 properties: {
106 orderId: { type: "string" },
107 includeHistory: { type: "boolean" },
108 },
109 },
110 outputSchema: {
111 type: "object",
112 properties: { status: { type: "string" } },
113 },
114 },
115 });
116 }),
117 );
118
119 expect(updated.name).toEqual(created.name);
120 expect(updated.description).toEqual("Look up an order by id.");
121
122 yield* stack.destroy();
123
124 const gone = yield* waitUntilGone(created.name);
125 expect(gone).toEqual("gone");
126
127 yield* deleteAgent(agentName);
128 }).pipe(logLevel, quotaTolerant),
129 {
130 tags: ["provider:gcp", "provider:gcp:dialogflow", "live"],
131 timeout: 120_000,
132 },
133);