Fix: 변경 벡터 검색

검색 정확도 범위를 search_param에 지정
AI-SIP · Oct 10, 2024 · ee6dc8c · ee6dc8c
1 parent c1e8ffe
commit ee6dc8c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/app/main.py b/app/main.py
@@ -422,15 +422,16 @@ async def retrieve(problem_text: str):
         search_params = {
             'metric_type': 'COSINE',
             'params': {
-                'probe': 20
+                'radius': 0.35,
+                'range_filter': 0.9
             },
         }
         dt5 = str(datetime.fromtimestamp(time.time()))
         results = collection.search(
             data=query_embeddings[0],
             anns_field='content_embedding',
             param=search_params,
-            limit=3,
+            limit=2,
             expr=None,
             output_fields=['content', 'subject_name', 'unit_name', 'main_concept']
         )