Skip to content

Commit 8f77cc6

Browse files
Removing OAuth from tuning for REST (google-gemini#579)
* Removing OAuth from tuning for REST * remove key * Fix line continuation Change-Id: I639aa453699766bf31f7722a59f4157a53cf7de4 --------- Co-authored-by: Mark Daoust <[email protected]>
1 parent 4354af6 commit 8f77cc6

File tree

1 file changed

+12
-28
lines changed

1 file changed

+12
-28
lines changed

samples/rest/tuned_models.sh

+12-28
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,9 @@
11
set -eu
22

3-
access_token=$(gcloud auth application-default print-access-token)
4-
5-
63
echo "[START tuned_models_create]"
74
# [START tuned_models_create]
8-
curl -X POST https://generativelanguage.googleapis.com/v1beta/tunedModels \
5+
curl -X POST "https://generativelanguage.googleapis.com/v1beta/tunedModels?key=$GOOGLE_API_KEY" \
96
-H 'Content-Type: application/json' \
10-
-H "Authorization: Bearer ${access_token}" \
11-
-H "x-goog-user-project: ${project_id}" \
127
-d '
138
{
149
"display_name": "number generator model",
@@ -82,10 +77,9 @@ tuning_done=false
8277
while [[ "$tuning_done" != "true" ]];
8378
do
8479
sleep 5
85-
curl -X GET https://generativelanguage.googleapis.com/v1/${operation} \
80+
curl -X GET "https://generativelanguage.googleapis.com/v1/${operation}?key=$GOOGLE_API_KEY" \
8681
-H 'Content-Type: application/json' \
87-
-H "Authorization: Bearer ${access_token}" \
88-
-H "x-goog-user-project: ${project_id}" 2> /dev/null > tuning_operation.json
82+
2> /dev/null > tuning_operation.json
8983

9084
complete=$(jq .metadata.completedPercent < tuning_operation.json)
9185
tput cuu1
@@ -96,21 +90,17 @@ done
9690

9791
# Or get the TunedModel and check it's state. The model is ready to use if the state is active.
9892
modelname=$(cat tunemodel.json | jq ".metadata.tunedModel" | tr -d '"')
99-
curl -X GET https://generativelanguage.googleapis.com/v1beta/${modelname} \
100-
-H 'Content-Type: application/json' \
101-
-H "Authorization: Bearer ${access_token}" \
102-
-H "x-goog-user-project: ${project_id}" > tuned_model.json
93+
curl -X GET https://generativelanguage.googleapis.com/v1beta/${modelname}?key=$GOOGLE_API_KEY \
94+
-H 'Content-Type: application/json' > tuned_model.json
10395

10496
cat tuned_model.json | jq ".state"
10597
# [END tuned_models_create]
10698

10799

108100
echo "[START tuned_models_generate_content]"
109101
# [START tuned_models_generate_content]
110-
curl -X POST https://generativelanguage.googleapis.com/v1beta/$modelname:generateContent \
102+
curl -X POST https://generativelanguage.googleapis.com/v1beta/$modelname:generateContent?key=$GOOGLE_API_KEY \
111103
-H 'Content-Type: application/json' \
112-
-H "Authorization: Bearer ${access_token}" \
113-
-H "x-goog-user-project: ${project_id}" \
114104
-d '{
115105
"contents": [{
116106
"parts": [{
@@ -122,10 +112,8 @@ curl -X POST https://generativelanguage.googleapis.com/v1beta/$modelname:generat
122112

123113
echo "[START tuned_models_get]"
124114
# [START tuned_models_get]
125-
curl -X GET https://generativelanguage.googleapis.com/v1beta/${modelname} \
126-
-H 'Content-Type: application/json' \
127-
-H "Authorization: Bearer ${access_token}" \
128-
-H "x-goog-user-project: ${project_id}" | grep state
115+
curl -X GET https://generativelanguage.googleapis.com/v1beta/${modelname}?key=$GOOGLE_API_KEY \
116+
-H 'Content-Type: application/json' | grep state
129117
# [END tuned_models_get]
130118

131119
echo "[START tuned_models_list]"
@@ -142,18 +130,14 @@ jq .tunedModels[].name < tuned_models.json
142130
page_token=$(jq .nextPageToken < tuned_models.json | tr -d '"')
143131

144132
if [[ "$page_token" != "null"" ]]; then
145-
curl -X GET https://generativelanguage.googleapis.com/v1beta/tunedModels?page_size=5\&page_token=${page_token} \
146-
-H "Content-Type: application/json" \
147-
-H "Authorization: Bearer ${access_token}" \
148-
-H "x-goog-user-project: ${project_id}" > tuned_models2.json
133+
curl -X GET https://generativelanguage.googleapis.com/v1beta/tunedModels?page_size=5\&page_token=${page_token}?key=$GOOGLE_API_KEY \
134+
-H "Content-Type: application/json" > tuned_models2.json
149135
jq .tunedModels[].name < tuned_models.json
150136
fi
151137
# [END tuned_models_list]
152138
153139
echo "[START tuned_models_delete]"
154140
# [START tuned_models_delete]
155-
curl -X DELETE https://generativelanguage.googleapis.com/v1beta/${modelname} \
156-
-H 'Content-Type: application/json' \
157-
-H "Authorization: Bearer ${access_token}" \
158-
-H "x-goog-user-project: ${project_id}"
141+
curl -X DELETE https://generativelanguage.googleapis.com/v1beta/${modelname}?key=$GOOGLE_API_KEY \
142+
-H 'Content-Type: application/json'
159143
# [END tuned_models_delete]

0 commit comments

Comments
 (0)