Skip to content

Commit 07d6973

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
chore: RAG Fixed typos in rag_store
PiperOrigin-RevId: 702387440
1 parent 565c800 commit 07d6973

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

vertexai/rag/rag_store.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -103,10 +103,10 @@ def __init__(
103103
)
104104

105105
# If rag_retrieval_config is not specified, set it to default values.
106-
api_retrival_config = aiplatform_v1.RagRetrievalConfig()
106+
api_retrieval_config = aiplatform_v1.RagRetrievalConfig()
107107
# If rag_retrieval_config is specified, populate the default config.
108108
if rag_retrieval_config:
109-
api_retrival_config.top_k = rag_retrieval_config.top_k
109+
api_retrieval_config.top_k = rag_retrieval_config.top_k
110110
# Set vector_distance_threshold to config value if specified
111111
if rag_retrieval_config.filter:
112112
# Check if both vector_distance_threshold and
@@ -121,10 +121,10 @@ def __init__(
121121
" vector_similarity_threshold can be specified at a time"
122122
" in rag_retrieval_config."
123123
)
124-
api_retrival_config.filter.vector_distance_threshold = (
124+
api_retrieval_config.filter.vector_distance_threshold = (
125125
rag_retrieval_config.filter.vector_distance_threshold
126126
)
127-
api_retrival_config.filter.vector_similarity_threshold = (
127+
api_retrieval_config.filter.vector_similarity_threshold = (
128128
rag_retrieval_config.filter.vector_similarity_threshold
129129
)
130130

@@ -135,10 +135,10 @@ def __init__(
135135
)
136136
self._raw_vertex_rag_store = gapic_tool_types.VertexRagStore(
137137
rag_resources=[gapic_rag_resource],
138-
rag_retreival_config=api_retrival_config,
138+
rag_retrieval_config=api_retrieval_config,
139139
)
140140
else:
141141
self._raw_vertex_rag_store = gapic_tool_types.VertexRagStore(
142142
rag_corpora=[rag_corpus_name],
143-
rag_retreival_config=api_retrival_config,
143+
rag_retrieval_config=api_retrieval_config,
144144
)

0 commit comments

Comments
 (0)