@@ -103,10 +103,10 @@ def __init__(
103
103
)
104
104
105
105
# 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 ()
107
107
# If rag_retrieval_config is specified, populate the default config.
108
108
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
110
110
# Set vector_distance_threshold to config value if specified
111
111
if rag_retrieval_config .filter :
112
112
# Check if both vector_distance_threshold and
@@ -121,10 +121,10 @@ def __init__(
121
121
" vector_similarity_threshold can be specified at a time"
122
122
" in rag_retrieval_config."
123
123
)
124
- api_retrival_config .filter .vector_distance_threshold = (
124
+ api_retrieval_config .filter .vector_distance_threshold = (
125
125
rag_retrieval_config .filter .vector_distance_threshold
126
126
)
127
- api_retrival_config .filter .vector_similarity_threshold = (
127
+ api_retrieval_config .filter .vector_similarity_threshold = (
128
128
rag_retrieval_config .filter .vector_similarity_threshold
129
129
)
130
130
@@ -135,10 +135,10 @@ def __init__(
135
135
)
136
136
self ._raw_vertex_rag_store = gapic_tool_types .VertexRagStore (
137
137
rag_resources = [gapic_rag_resource ],
138
- rag_retreival_config = api_retrival_config ,
138
+ rag_retrieval_config = api_retrieval_config ,
139
139
)
140
140
else :
141
141
self ._raw_vertex_rag_store = gapic_tool_types .VertexRagStore (
142
142
rag_corpora = [rag_corpus_name ],
143
- rag_retreival_config = api_retrival_config ,
143
+ rag_retrieval_config = api_retrieval_config ,
144
144
)
0 commit comments