@@ -1247,161 +1247,3 @@ def cli_deploy_gke(
1247
1247
)
1248
1248
except Exception as e :
1249
1249
click .secho (f"Deploy failed: { e } " , fg = "red" , err = True )
1250
-
1251
-
1252
- @deploy .command ("gke" )
1253
- @click .option (
1254
- "--project" ,
1255
- type = str ,
1256
- help = (
1257
- "Required. Google Cloud project to deploy the agent. When absent,"
1258
- " default project from gcloud config is used."
1259
- ),
1260
- )
1261
- @click .option (
1262
- "--region" ,
1263
- type = str ,
1264
- help = (
1265
- "Required. Google Cloud region to deploy the agent. When absent,"
1266
- " gcloud run deploy will prompt later."
1267
- ),
1268
- )
1269
- @click .option (
1270
- "--cluster_name" ,
1271
- type = str ,
1272
- help = "Required. The name of the GKE cluster." ,
1273
- )
1274
- @click .option (
1275
- "--service_name" ,
1276
- type = str ,
1277
- default = "adk-default-service-name" ,
1278
- help = (
1279
- "Optional. The service name to use in GKE (default:"
1280
- " 'adk-default-service-name')."
1281
- ),
1282
- )
1283
- @click .option (
1284
- "--app_name" ,
1285
- type = str ,
1286
- default = "" ,
1287
- help = (
1288
- "Optional. App name of the ADK API server (default: the folder name"
1289
- " of the AGENT source code)."
1290
- ),
1291
- )
1292
- @click .option (
1293
- "--port" ,
1294
- type = int ,
1295
- default = 8000 ,
1296
- help = "Optional. The port of the ADK API server (default: 8000)." ,
1297
- )
1298
- @click .option (
1299
- "--trace_to_cloud" ,
1300
- is_flag = True ,
1301
- show_default = True ,
1302
- default = False ,
1303
- help = "Optional. Whether to enable Cloud Trace for GKE." ,
1304
- )
1305
- @click .option (
1306
- "--with_ui" ,
1307
- is_flag = True ,
1308
- show_default = True ,
1309
- default = False ,
1310
- help = (
1311
- "Optional. Deploy ADK Web UI if set. (default: deploy ADK API server"
1312
- " only)"
1313
- ),
1314
- )
1315
- @click .option ( # This is the crucial missing piece
1316
- "--verbosity" ,
1317
- type = LOG_LEVELS ,
1318
- help = "Deprecated. Use --log_level instead." ,
1319
- )
1320
- @click .option (
1321
- "--log_level" ,
1322
- type = LOG_LEVELS ,
1323
- default = "INFO" ,
1324
- help = "Optional. Set the logging level" ,
1325
- )
1326
- @click .option (
1327
- "--temp_folder" ,
1328
- type = str ,
1329
- default = os .path .join (
1330
- tempfile .gettempdir (),
1331
- "gke_deploy_src" ,
1332
- datetime .now ().strftime ("%Y%m%d_%H%M%S" ),
1333
- ),
1334
- help = (
1335
- "Optional. Temp folder for the generated GKE source files"
1336
- " (default: a timestamped folder in the system temp directory)."
1337
- ),
1338
- )
1339
- @click .argument (
1340
- "agent" ,
1341
- type = click .Path (
1342
- exists = True , dir_okay = True , file_okay = False , resolve_path = True
1343
- ),
1344
- )
1345
- @click .option (
1346
- "--adk_version" ,
1347
- type = str ,
1348
- default = version .__version__ ,
1349
- show_default = True ,
1350
- help = (
1351
- "Optional. The ADK version used in GKE deployment. (default: the"
1352
- " version in the dev environment)"
1353
- ),
1354
- )
1355
- @adk_services_options ()
1356
- @deprecated_adk_services_options ()
1357
- def cli_deploy_gke (
1358
- agent : str ,
1359
- project : Optional [str ],
1360
- region : Optional [str ],
1361
- cluster_name : str ,
1362
- service_name : str ,
1363
- app_name : str ,
1364
- temp_folder : str ,
1365
- port : int ,
1366
- trace_to_cloud : bool ,
1367
- with_ui : bool ,
1368
- verbosity : str ,
1369
- adk_version : str ,
1370
- log_level : Optional [str ] = None ,
1371
- session_service_uri : Optional [str ] = None ,
1372
- artifact_service_uri : Optional [str ] = None ,
1373
- memory_service_uri : Optional [str ] = None ,
1374
- session_db_url : Optional [str ] = None , # Deprecated
1375
- artifact_storage_uri : Optional [str ] = None , # Deprecated
1376
- ):
1377
- """Deploys an agent to GKE.
1378
-
1379
- AGENT: The path to the agent source code folder.
1380
-
1381
- Example:
1382
-
1383
- adk deploy gke --project=[project] --region=[region] --cluster_name=[cluster_name] path/to/my_agent
1384
- """
1385
- session_service_uri = session_service_uri or session_db_url
1386
- artifact_service_uri = artifact_service_uri or artifact_storage_uri
1387
- try :
1388
- cli_deploy .to_gke (
1389
- agent_folder = agent ,
1390
- project = project ,
1391
- region = region ,
1392
- cluster_name = cluster_name ,
1393
- service_name = service_name ,
1394
- app_name = app_name ,
1395
- temp_folder = temp_folder ,
1396
- port = port ,
1397
- trace_to_cloud = trace_to_cloud ,
1398
- with_ui = with_ui ,
1399
- verbosity = verbosity ,
1400
- log_level = log_level ,
1401
- adk_version = adk_version ,
1402
- session_service_uri = session_service_uri ,
1403
- artifact_service_uri = artifact_service_uri ,
1404
- memory_service_uri = memory_service_uri ,
1405
- )
1406
- except Exception as e :
1407
- click .secho (f"Deploy failed: { e } " , fg = "red" , err = True )
0 commit comments