@@ -92,7 +92,7 @@ class MyDataModelList:
92
92
# function which returns the store.
93
93
# Using a function allows for lazy initialization of the store,
94
94
# so that settings for unused stores do not cause validation errors.
95
- services : dict [str , Callable [[], VectorStoreRecordCollection ]] = {
95
+ collections : dict [str , Callable [[], VectorStoreRecordCollection ]] = {
96
96
"ai_search" : lambda : AzureAISearchCollection [MyDataModel ](
97
97
data_model_type = MyDataModel ,
98
98
),
@@ -133,7 +133,7 @@ async def main(service: str, use_azure_openai: bool, embedding_model: str):
133
133
else :
134
134
embedder = OpenAITextEmbedding (service_id = service_id , ai_model_id = embedding_model )
135
135
kernel .add_service (embedder )
136
- async with services [service ]() as record_collection :
136
+ async with collections [service ]() as record_collection :
137
137
print (f"Creating { service } collection!" )
138
138
await record_collection .create_collection_if_not_exists ()
139
139
@@ -184,7 +184,7 @@ async def main(service: str, use_azure_openai: bool, embedding_model: str):
184
184
argparse .ArgumentParser ()
185
185
186
186
parser = argparse .ArgumentParser ()
187
- parser .add_argument ("--service" , default = "in_memory" , choices = services .keys (), help = "What store to use." )
187
+ parser .add_argument ("--service" , default = "in_memory" , choices = collections .keys (), help = "What store to use." )
188
188
# Option of whether to use OpenAI or Azure OpenAI.
189
189
parser .add_argument ("--use-azure-openai" , action = "store_true" , help = "Use Azure OpenAI instead of OpenAI." )
190
190
# Model
0 commit comments