chore: fix tests and citations

Azure-Samples · Feb 13, 2024 · a7840b3 · a7840b3
1 parent a8ce365
commit a7840b3
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/packages/chat-component/src/components/chat-component.ts b/packages/chat-component/src/components/chat-component.ts
@@ -56,7 +56,7 @@ export class ChatComponent extends LitElement {
   }
 
   @property({ type: String, attribute: 'data-api-url' })
-  set apiUrl(value: string = chatHttpOptions.url) {
+  set apiUrl(value: string) {
     this.chatContext.apiUrl = value;
   }
 
@@ -172,8 +172,14 @@ export class ChatComponent extends LitElement {
 
   handleCitationClick(event: CustomEvent): void {
     event?.preventDefault();
-    this.chatContext.selectedCitation = event?.detail?.citation;
-    this.chatContext.selectedChatEntry = event?.detail?.chatThreadEntry;
+    const citation = event?.detail?.citation;
+    const chatThreadEntry = event?.detail?.chatThreadEntry;
+    if (citation) {
+      this.chatContext.selectedCitation = citation;
+    }
+    if (chatThreadEntry) {
+      this.chatContext.selectedChatEntry = chatThreadEntry;
+    }
     this.chatContext.setState('showCitations', true);
   }
 

diff --git a/packages/chat-component/src/components/chat-context.ts b/packages/chat-component/src/components/chat-context.ts
@@ -1,4 +1,5 @@
 import { type ReactiveController, type ReactiveControllerHost } from 'lit';
+import { chatHttpOptions } from '../config/global-config.js';
 
 export class ChatContextController implements ReactiveController {
   host: ReactiveControllerHost;
@@ -19,7 +20,7 @@ export class ChatContextController implements ReactiveController {
 
   private _selectedChatEntry: ChatThreadEntry | undefined = undefined;
 
-  private _apiUrl: string = '';
+  private _apiUrl: string = chatHttpOptions.url;
 
   private _interactionModel: 'ask' | 'chat' = 'chat';
 

diff --git a/tests/e2e/webapp.spec.ts b/tests/e2e/webapp.spec.ts
@@ -438,7 +438,6 @@ test.describe('generate answer', () => {
 
   test('follow up questions', async ({ page }) => {
     const followupQuestions = page.getByTestId('followUpQuestion');
-    await followupQuestions.waitFor();
 
     await expect(followupQuestions).toHaveCount(3);