Skip to content

[pull] main from GoogleCloudPlatform:main#44

Merged
pull[bot] merged 1 commit into
deargvp1:mainfrom
GoogleCloudPlatform:main
Jul 6, 2026
Merged

[pull] main from GoogleCloudPlatform:main#44
pull[bot] merged 1 commit into
deargvp1:mainfrom
GoogleCloudPlatform:main

Conversation

@pull

@pull pull Bot commented Jul 6, 2026

Copy link
Copy Markdown

See Commits and Changes for more details.


Created by pull[bot] (v2.0.0-alpha.4)

Can you help keep this open source service alive? 💖 Please sponsor : )


Summary by cubic

Updates all RAG Python samples to the latest agentplatform RAG module and google.genai for generation, replacing legacy vertexai usage. Aligns examples and tests with new types, client init, and response formats.

  • Refactors

    • Replaced vertexai.rag with agentplatform.Client.rag and agentplatform.types; updated return types and hints across samples.
    • Switched generation to google.genai (models.generate_content) with Tool/Retrieval from google.genai.types; uses MODEL_ID (e.g., gemini-3.5-flash) or gemini-2.5-pro in examples.
    • Migrated corpus and vector DB configs (Vertex Vector Search, Pinecone, Weaviate, Feature Store) to new RagVectorDbConfig shapes.
    • Reworked file import to ImportRagFilesConfig with GcsSource and optional GoogleDriveSource, plus chunking via RagFileTransformationConfig; added async variant via client.aio.rag.
    • Updated list APIs to response wrappers (ListRagCorporaResponse, ListRagFilesResponse); tests now read .rag_corpora and .rag_files.
  • Migration

    • Use agentplatform and google-genai; remove vertexai RAG API calls.
    • Initialize once with agentplatform.Client(project, location) and call via client.rag (or client.aio.rag for async).
    • Set MODEL_ID env var for generation samples; when importing from GCS, use wildcard paths like gs://bucket/dir/*.

Written for commit 8b6869e. Summary will update on new commits.

Review in cubic

… SDK (#14369)

* docs: update RAG quickstart python snippet

* update arg name

* update additional rag samples

* update remaining samples to use latest rag module

* update additional rag samples

* Fix sample typos

* Update imports and return types in RAG samples

* Remove unused imports

* Fix linter errors

* Fix linter errors

* Fix more linter errors

* Fix sample test runner

* fix import order

* fix import error

* Fix typos
@pull pull Bot locked and limited conversation to collaborators Jul 6, 2026
@pull pull Bot added the ⤵️ pull label Jul 6, 2026
@pull pull Bot merged commit 8b6869e into deargvp1:main Jul 6, 2026
1 check was pending
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant