diff --git a/.semversioner/next-release/patch-20250121205226363912.json b/.semversioner/next-release/patch-20250121205226363912.json new file mode 100644 index 0000000000..7597e07533 --- /dev/null +++ b/.semversioner/next-release/patch-20250121205226363912.json @@ -0,0 +1,4 @@ +{ + "type": "patch", + "description": "Fix DRIFT search on Azure AI Search." +} diff --git a/graphrag/vector_stores/azure_ai_search.py b/graphrag/vector_stores/azure_ai_search.py index 02e3a35c41..535cfc05c1 100644 --- a/graphrag/vector_stores/azure_ai_search.py +++ b/graphrag/vector_stores/azure_ai_search.py @@ -111,6 +111,7 @@ def load_documents( name="vector", type=SearchFieldDataType.Collection(SearchFieldDataType.Single), searchable=True, + hidden=False, # DRIFT needs to return the vector for client-side similarity vector_search_dimensions=self.vector_size, vector_search_profile_name=self.vector_search_profile_name, ), diff --git a/tests/fixtures/min-csv/config.json b/tests/fixtures/min-csv/config.json index 86b4c49a11..4e750db6c7 100644 --- a/tests/fixtures/min-csv/config.json +++ b/tests/fixtures/min-csv/config.json @@ -134,6 +134,10 @@ { "query": "What is the major conflict in this story and who are the protagonist and antagonist?", "method": "global" + }, + { + "query": "Who is Jordan Hayes?", + "method": "basic" } ], "slow": false diff --git a/tests/fixtures/text/config.json b/tests/fixtures/text/config.json index cf4180857c..81dd8667b3 100644 --- a/tests/fixtures/text/config.json +++ b/tests/fixtures/text/config.json @@ -152,6 +152,10 @@ { "query": "What is the major conflict in this story and who are the protagonist and antagonist?", "method": "global" + }, + { + "query": "Who is Jordan Hayes?", + "method": "basic" } ], "slow": false