Update schema store index

Author: BenConstable9

deploy_ai_search/text_2_sql_schema_store.py

@@ -155,7 +155,7 @@ def get_index_fields(self) -> list[SearchableField]:
                 ],
             ),
             SimpleField(
-                name="CompleteEntityRelationshipGraph",
+                name="CompleteEntityRelationshipsGraph",
                 type=SearchFieldDataType.String,
                 collection=True,
             ),
@@ -270,6 +270,14 @@ def get_indexer(self) -> SearchIndexer:
                     source_field_name="/document/EntityName",
                     target_field_name="EntityName",
                 ),
+                FieldMapping(
+                    source_field_name="/document/Database",
+                    target_field_name="Database",
+                ),
+                FieldMapping(
+                    source_field_name="/document/Warehouse",
+                    target_field_name="Warehouse",
+                ),
                 FieldMapping(
                     source_field_name="/document/Definition",
                     target_field_name="Definition",
@@ -291,12 +299,12 @@ def get_indexer(self) -> SearchIndexer:
                     target_field_name="ColumnDefinitions",
                 ),
                 FieldMapping(
-                    source_field_name="/document/ImmediateEntityRelationships",
-                    target_field_name="ImmediateEntityRelationships",
+                    source_field_name="/document/EntityRelationships",
+                    target_field_name="EntityRelationships",
                 ),
                 FieldMapping(
-                    source_field_name="/document/CompleteEntityRelationships",
-                    target_field_name="CompleteEntityRelationships",
+                    source_field_name="/document/CompleteEntityRelationshipsGraph",
+                    target_field_name="CompleteEntityRelationshipsGraph",
                 ),
                 FieldMapping(
                     source_field_name="/document/DateLastModified",