Skip to content

Commit baa1563

Browse files
committed
fix: Add required ENV vars
1 parent 719ce2a commit baa1563

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

genai/content_cache/test_content_cache_examples.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,13 +12,21 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
import os
16+
1517
import contentcache_create_with_txt_gcs_pdf
1618
import contentcache_delete
1719
import contentcache_list
1820
import contentcache_update
1921
import contentcache_use_with_txt
2022

2123

24+
os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
25+
os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
26+
# The project name is included in the CICD pipeline
27+
# os.environ['GOOGLE_CLOUD_PROJECT'] = "add-your-project-name"
28+
29+
2230
def test_content_cache() -> None:
2331
# Create a Cache
2432
cache_name = contentcache_create_with_txt_gcs_pdf.create_content_cache()

0 commit comments

Comments
 (0)