@@ -34,7 +34,7 @@ public async Task<List<string>> GetCollections()
             return result;
         }
 
-        public async Task<List<string>> Search(string collectionName, float[] vector, int limit = 5)
+        public async Task<List<string>> Search(string collectionName, float[] vector, string returnFieldName, int limit = 5, float confidence = 0.5f)
         {
             var results = _memoryStore.GetNearestMatchesAsync(collectionName, vector, limit);
 
@@ -48,11 +48,12 @@ public async Task<List<string>> Search(string collectionName, float[] vector, in
 
         }
 
-        public async Task Upsert(string collectionName, string id, float[] vector, string text, Dictionary<string, string>? payload = null)
+        public async Task<bool> Upsert(string collectionName, string id, float[] vector, string text, Dictionary<string, string>? payload)
         {
 #pragma warning disable SKEXP0001 // Type is for evaluation purposes only and is subject to change or removal in future updates. Suppress this diagnostic to proceed.
             await _memoryStore.UpsertAsync(collectionName, MemoryRecord.LocalRecord(id.ToString(), text, null, vector));
 #pragma warning restore SKEXP0001 // Type is for evaluation purposes only and is subject to change or removal in future updates. Suppress this diagnostic to proceed.
+            return true;
         }
     }
 }
Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,7 @@ public async Task<List<string>> GetCollections()`
`34`	`34`	`return result;`
`35`	`35`	`}`
`36`	`36`
`37`		`- public async Task<List<string>> Search(string collectionName, float[] vector, int limit = 5)`
	`37`	`+ public async Task<List<string>> Search(string collectionName, float[] vector, string returnFieldName, int limit = 5, float confidence = 0.5f)`
`38`	`38`	`{`
`39`	`39`	`var results = _memoryStore.GetNearestMatchesAsync(collectionName, vector, limit);`
`40`	`40`
`@@ -48,11 +48,12 @@ public async Task<List<string>> Search(string collectionName, float[] vector, in`
`48`	`48`
`49`	`49`	`}`
`50`	`50`
`51`		`- public async Task Upsert(string collectionName, string id, float[] vector, string text, Dictionary<string, string>? payload = null)`
	`51`	`+ public async Task<bool> Upsert(string collectionName, string id, float[] vector, string text, Dictionary<string, string>? payload)`
`52`	`52`	`{`
`53`	`53`	`#pragma warning disable SKEXP0001 // Type is for evaluation purposes only and is subject to change or removal in future updates. Suppress this diagnostic to proceed.`
`54`	`54`	`await _memoryStore.UpsertAsync(collectionName, MemoryRecord.LocalRecord(id.ToString(), text, null, vector));`
`55`	`55`	`#pragma warning restore SKEXP0001 // Type is for evaluation purposes only and is subject to change or removal in future updates. Suppress this diagnostic to proceed.`
	`56`	`+ return true;`
`56`	`57`	`}`
`57`	`58`	`}`
`58`	`59`	`}`