Skip to content

Commit b2f4217

Browse files
authored
docs(samples): Change sample to use Gemini Embedding model (#13368)
1 parent 16ebde7 commit b2f4217

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

genai/embeddings/embeddings_docretrieval_with_txt.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -20,15 +20,11 @@ def embed_content() -> str:
2020

2121
client = genai.Client()
2222
response = client.models.embed_content(
23-
model="text-embedding-005",
24-
contents=[
25-
"How do I get a driver's license/learner's permit?",
26-
"How do I renew my driver's license?",
27-
"How do I change my address on my driver's license?",
28-
],
23+
model="gemini-embedding-001",
24+
contents="How do I get a driver's license/learner's permit?",
2925
config=EmbedContentConfig(
3026
task_type="RETRIEVAL_DOCUMENT", # Optional
31-
output_dimensionality=768, # Optional
27+
output_dimensionality=3072, # Optional
3228
title="Driver's License", # Optional
3329
),
3430
)

0 commit comments

Comments
 (0)