Merge pull request #3398 from microsoft-search/fix/3210

Fix/3210
microsoft-search · Dec 13, 2023 · e8bdbe6 · e8bdbe6
2 parents 32e4b56 + 23352a4
commit e8bdbe6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/search-parts/src/dataSources/MicrosoftSearchDataSource.ts b/search-parts/src/dataSources/MicrosoftSearchDataSource.ts
@@ -540,9 +540,9 @@ export class MicrosoftSearchDataSource extends BaseDataSource<IMicrosoftSearchDa
         if (propertyPath.localeCompare('dataSourceProperties.useBetaEndpoint') === 0) {
 
             if (newValue) {
-                this._microsoftSearchUrl = "https://graph.microsoft.com/beta/search/query";
+                this._microsoftSearchUrl = `${this.context?.pageContext?.legacyPageContext?.msGraphEndpointUrl}/beta/search/query`;
             } else {
-                this._microsoftSearchUrl = "https://graph.microsoft.com/v1.0/search/query";
+                this._microsoftSearchUrl = `${this.context?.pageContext?.legacyPageContext?.msGraphEndpointUrl}/v1.0/search/query`;
             }
         }
 
@@ -662,9 +662,9 @@ export class MicrosoftSearchDataSource extends BaseDataSource<IMicrosoftSearchDa
         this.properties.collapseProperties = this.properties.collapseProperties !== undefined ? this.properties.collapseProperties : [];
 
         if (this.properties.useBetaEndpoint) {
-            this._microsoftSearchUrl = "https://graph.microsoft.com/beta/search/query";
+            this._microsoftSearchUrl = `${this.context?.pageContext?.legacyPageContext?.msGraphEndpointUrl}/beta/search/query`;
         } else {
-            this._microsoftSearchUrl = "https://graph.microsoft.com/v1.0/search/query";
+            this._microsoftSearchUrl = `${this.context?.pageContext?.legacyPageContext?.msGraphEndpointUrl}/v1.0/search/query`;
         }
     }