Update vector store layout

Author: BenConstable9

deploy_ai_search/text_2_sql_query_cache.py

@@ -49,7 +49,7 @@ def get_index_fields(self) -> list[SearchableField]:
                 vector_search_profile_name=self.vector_search_profile_name,
             ),
             ComplexField(
-                name="Queries",
+                name="Decomposition",
                 collection=True,
                 fields=[
                     SearchableField(
@@ -71,11 +71,11 @@ def get_index_fields(self) -> list[SearchableField]:
                                 collection=True,
                                 fields=[
                                     SearchableField(
-                                        name="ColumnName",
+                                        name="Name",
                                         type=SearchFieldDataType.String,
                                     ),
                                     SearchableField(
-                                        name="ColumnDefinition",
+                                        name="Definition",
                                         type=SearchFieldDataType.String,
                                     ),
                                     SearchableField(

deploy_ai_search/text_2_sql_schema_store.py

@@ -70,6 +70,14 @@ def get_index_fields(self) -> list[SearchableField]:
                 type=SearchFieldDataType.String,
                 analyzer_name="keyword",
             ),
+            SearchableField(
+                name="Database",
+                type=SearchFieldDataType.String,
+            ),
+            SearchableField(
+                name="Warehouse",
+                type=SearchFieldDataType.String,
+            ),
             SearchableField(
                 name="Definition",
                 type=SearchFieldDataType.String,
@@ -129,11 +137,7 @@ def get_index_fields(self) -> list[SearchableField]:
                 collection=True,
                 fields=[
                     SearchableField(
-                        name="ImmediateRelationshipName",
-                        type=SearchFieldDataType.String,
-                    ),
-                    SearchableField(
-                        name="ImmediateRelationshipEntity",
+                        name="Name",
                         type=SearchFieldDataType.String,
                     ),
                     ComplexField(