graphdb-workbench-tests 3.4.0-TR2 → 3.4.0-TR4

package/{e2e-legacy → integration}/ttyg/agent-select-menu.spec.js

@@ -2,6 +2,7 @@ import {TTYGViewSteps} from "../../steps/ttyg/ttyg-view-steps";
 import {TTYGStubs} from "../../stubs/ttyg/ttyg-stubs";
 import {RepositoriesStubs} from "../../stubs/repositories/repositories-stubs";
 import {ModalDialogSteps} from "../../steps/modal-dialog-steps";
+import {RepositoriesStub} from "../../stubs/repositories-stub";
 
 describe('TTYG agent select menu', () => {
 
@@ -9,7 +10,7 @@ describe('TTYG agent select menu', () => {
 
     beforeEach(() => {
         RepositoriesStubs.stubRepositories(0, '/repositories/get-ttyg-repositories.json');
-        RepositoriesStubs.stubBaseEndpoints(repositoryId);
+        RepositoriesStub.stubBaseEndpoints(repositoryId);
         cy.presetRepository(repositoryId);
         TTYGStubs.stubChatsListGetNoResults();
     });
@@ -28,30 +29,6 @@ describe('TTYG agent select menu', () => {
         ]);
     });
 
-    it('should not allow selecting an incompatible agent from the menu', () => {
-        TTYGStubs.stubAgentListWithIncompatibleGet();
-        // Given: I have opened the ttyg page
-        TTYGViewSteps.visit();
-        // And: The agent dropdown menu is not visible
-        TTYGViewSteps.getAgentsDropdownMenu().should('not.be.visible');
-        // And: The first agent is selected
-        TTYGViewSteps.getAgentsMenuToggleButton().should('contain', 'agent-1');
-
-        // When: I open the agent selection menu
-        TTYGViewSteps.openAgentsMenu();
-        // Then: I should see that the first agent is marked as incompatible
-        TTYGViewSteps.getAgentFromMenu(0)
-            .trigger('mouseover')
-            .should('have.css', 'cursor', 'not-allowed');
-
-        // When: I attempt to select an incompatible agent from the menu
-        TTYGViewSteps.selectAgent(0);
-        // Then: The incompatible agent should not be selected
-        TTYGViewSteps.getAgentsMenuToggleButton().should('contain', 'agent-1');
-        // And: The dropdown menu should remain open
-        TTYGViewSteps.getAgentsDropdownMenu().should('be.visible');
-    });
-
     it('Should be able to select agent from the menu', () => {
         TTYGStubs.stubAgentListGet();
         // Given I have opened the ttyg page

package/{e2e-legacy → integration}/ttyg/chat-list.spec.js

@@ -6,12 +6,13 @@ import {ApplicationSteps} from "../../steps/application-steps";
 import HomeSteps from "../../steps/home-steps";
 import {ChatPanelSteps} from "../../steps/ttyg/chat-panel-steps";
 import {AlertDialogSteps} from "../../steps/alert-dialog-steps";
+import {RepositoriesStub} from "../../stubs/repositories-stub";
 
 describe('TTYG chat list', () => {
 
     beforeEach(() => {
         RepositoriesStubs.stubRepositories(0, '/repositories/get-ttyg-repositories.json');
-        RepositoriesStubs.stubBaseEndpoints('starwars');
+        RepositoriesStub.stubBaseEndpoints('starwars');
         cy.presetRepository('starwars');
     });
 
@@ -61,7 +62,12 @@ describe('TTYG chat list', () => {
         TTYGViewSteps.getChatByDayGroups().should('have.length', 0);
     });
 
-    it('Should be able to edit an existing chat name by double click on the chat in the list', () => {
+    it('Should be able to edit an existing chat name by double click on the chat in the list', {
+        retries: {
+            runMode: 1,
+            openMode: 0
+        }
+    }, () => {
         TTYGStubs.stubChatsListGet();
         TTYGStubs.stubChatGet();
         TTYGStubs.stubAgentListGet();
@@ -85,7 +91,12 @@ describe('TTYG chat list', () => {
         TTYGViewSteps.getChatFromGroup(1, 0).should('contain', 'New chat name');
     });
 
-    it('Should be able to edit an existing chat name through the action menu', () => {
+    it('Should be able to edit an existing chat name through the action menu', {
+        retries: {
+            runMode: 1,
+            openMode: 0
+        }
+    }, () => {
         TTYGStubs.stubChatsListGet();
         TTYGStubs.stubChatGet();
         TTYGStubs.stubAgentListGet();

package/{e2e-legacy → integration}/ttyg/chat-panel.spec.js

@@ -3,15 +3,14 @@ import {TTYGStubs} from "../../stubs/ttyg/ttyg-stubs";
 import {TTYGViewSteps} from "../../steps/ttyg/ttyg-view-steps";
 import {ChatPanelSteps} from "../../steps/ttyg/chat-panel-steps";
 import {ApplicationSteps} from "../../steps/application-steps";
-import {TtygAgentSettingsModalSteps} from '../../steps/ttyg/ttyg-agent-settings-modal.steps.js';
-import {SparqlStubs} from '../../stubs/sparql-stubs.js';
+import {RepositoriesStub} from "../../stubs/repositories-stub";
 
 describe('Ttyg ChatPanel', () => {
 
     beforeEach(() => {
         // Create an actual repository to prevent stubbing all background requests that are not related to the ttyg view
         RepositoriesStubs.stubRepositories(0, '/repositories/get-ttyg-repositories.json');
-        RepositoriesStubs.stubBaseEndpoints('starwars');
+        RepositoriesStub.stubBaseEndpoints('starwars');
         cy.presetRepository('starwars');
         TTYGStubs.stubChatsListGet();
         TTYGStubs.stubAgentListGet();
@@ -43,7 +42,7 @@ describe('Ttyg ChatPanel', () => {
         // When the new question input is empty.
         // The "Ask" button must be disabled.
         ChatPanelSteps.getAskButtonElement().should('be.disabled');
-        ChatPanelSteps.getQuestionInputElement().should('have.attr', 'disabled');
+        ChatPanelSteps.getQuestionInputElement().should('be.disabled');
 
         // Then I expect the "Ask" button be not active because agent is not selected
         ChatPanelSteps.getAskButtonElement().should('not.be.enabled');
@@ -55,8 +54,7 @@ describe('Ttyg ChatPanel', () => {
         // When I type a question
         ChatPanelSteps.getQuestionInputElement()
             .should('be.visible')
-            .and('not.have.attr', 'disabled');
-        ChatPanelSteps.getQuestionInputElement()
+            .and('not.be.disabled')
             .type('Who is Han Solo?');
 
         // Then I expect the "Ask" button be active.
@@ -70,22 +68,13 @@ describe('Ttyg ChatPanel', () => {
         ChatPanelSteps.getChatDetailsElements().should('have.length', 3);
         ChatPanelSteps.getChatDetailQuestionElement(2).contains('Who is Han Solo?');
         // and input field be empty,
-        ChatPanelSteps.getQuestionInputElement().should('not.have.attr', 'disabled');
+        ChatPanelSteps.getQuestionInputElement().should('be.enabled');
         ChatPanelSteps.getQuestionInputElement().should('have.value', '');
         // and "Ask" button be disabled.
         ChatPanelSteps.getAskButtonElement().should('be.disabled');
         // and only the actions for the last message are visible.
         ChatPanelSteps.getChatDetailActions(2, 0).should('not.be.visible');
         ChatPanelSteps.getChatDetailActions(2, 1).should('be.visible');
-        // When: I hover over the token usage info button
-        TTYGViewSteps.hoverTokenUsageInfoButton(0);
-        // Then: I expect the token usage info popover to be displayed.
-        TTYGViewSteps.getTokenUsageInfoPopover()
-            .should("exist")
-            .and('contain', '10,246')
-            .and('contain', 'prompt tokens')
-            .and('contain', '82')
-            .and('contain', 'completion tokens');
 
         // When I click on regenerate button on the last response => +2 messages
         TTYGStubs.stubAnswerQuestion();
@@ -140,22 +129,14 @@ describe('Ttyg ChatPanel', () => {
 
         // When I click on first explain response button
         TTYGViewSteps.getExplainResponseButton(0).realHover();
-        // Then I expect to explain button to be visible and not disabled.
+        // Then I expect to explain button to be visible.
         TTYGViewSteps.getExplainResponseButton(0).should('be.visible');
-        TTYGViewSteps.getExplainResponseButton(0).should('not.be.disabled');
 
         // When I click on the button.
         TTYGStubs.stubExplainResponse();
         TTYGViewSteps.clickOnExplainResponse(0);
         // Then I expect the hint message to not exist (it should only exist for the last 'Explain' response).
         TTYGViewSteps.getHowDeliverAnswerButton().should('not.exist');
-        // When I click on the open in sparql editor button for the explain response
-        SparqlStubs.spyAddKnownPrefixes();
-        TtygAgentSettingsModalSteps.clickOpenQueryInSparqlEditor(0);
-        // Then I expect a request for appending prefixes to be sent with the correct query.
-        cy.wait('@addKnownPrefixes').its('request.body').then((body) => {
-            expect(body).to.contain('SELECT ?name ?height WHERE {\n  ?character voc:height ?height;\n             rdfs:label ?name.\n  FILTER(?name = "Luke Skywalker" || ?name = "Leia Organa")\n}');
-        });
 
         // Then when I click on explain response button on the second answer
         TTYGStubs.stubExplainResponse('/ttyg/chats/explain-response-2.json');
@@ -183,55 +164,6 @@ describe('Ttyg ChatPanel', () => {
         TTYGViewSteps.getExplainQueryQueryElement(1, 2).contains("PREFIX rdfs: <http://www.w3.org/2000/01/rdf-sch");
     });
 
-    it('Should display info about used tokens for response', () => {
-        // Given: I visit the TTYG page, a chat with two questions and answers is loaded.
-        // Wait to chat be loaded
-        ChatPanelSteps.getChatDetailsElements().should('have.length', 2);
-
-        // When: I hover over the token usage info button
-        TTYGViewSteps.hoverTokenUsageInfoButton(0);
-        // Then: I expect the token usage info popover to be displayed.
-        TTYGViewSteps.getTokenUsageInfoPopover()
-            .should("exist")
-            .and('contain', '10,245')
-            .and('contain', 'prompt tokens')
-            .and('contain', '81')
-            .and('contain', 'completion tokens');
-    });
-
-    it('Should abort question', () => {
-        // When I type a question
-        ChatPanelSteps.getQuestionInputElement()
-            .should('be.visible')
-            .and('not.have.attr', 'disabled');
-        ChatPanelSteps.getQuestionInputElement()
-            .type('Who is Han Solo?');
-
-        // Then I expect the "Ask" button be active
-        ChatPanelSteps.getAskButtonElement().should('be.enabled');
-
-        // When I click on "Ask" button
-        TTYGStubs.stubAnswerQuestionWithDelay();
-        TTYGStubs.stubCancelQuestion();
-        ChatPanelSteps.askQuestion();
-        // Then the Cancel button should become visible and the Ask button should be hidden
-        ChatPanelSteps.getAskButtonElement().should('not.exist');
-        ChatPanelSteps.getCancelButton().should('be.visible');
-        // When I cancel the question
-        ChatPanelSteps.cancelQuestion();
-        // Then the Cancel button should be replaced with the Ask button
-        ChatPanelSteps.getCancelButton().should('not.exist');
-        ChatPanelSteps.getAskButtonElement().should('be.visible');
-        // Then the default Agent response should be visible
-        ChatPanelSteps.getAssistantAnswer(2).invoke('text').then((text) => {
-            expect(text.trim()).to.equal('Request cancelled by the user.');
-        });
-        ChatPanelSteps.getAssistantIcon(2).should('not.exist');
-        // And I expect the explain response button to be visible but disabled
-        TTYGViewSteps.getExplainResponseButton(2).should('be.visible');
-        TTYGViewSteps.getExplainResponseButton(2).should('be.disabled');
-    });
-
     // Can't test this on CI
     it.skip('Should copy an answer when click on copy button', () => {
         // When I click on copy button

package/{e2e-legacy → integration}/ttyg/clone-agent.spec.js

@@ -2,13 +2,14 @@ import {RepositoriesStubs} from "../../stubs/repositories/repositories-stubs";
 import {TTYGViewSteps} from "../../steps/ttyg/ttyg-view-steps";
 import {TTYGStubs} from "../../stubs/ttyg/ttyg-stubs";
 import {TtygAgentSettingsModalSteps} from "../../steps/ttyg/ttyg-agent-settings-modal.steps";
+import {RepositoriesStub} from "../../stubs/repositories-stub";
 
 describe('TTYG clone an agent', () => {
     const repositoryId = 'starwars';
 
     beforeEach(() => {
         RepositoriesStubs.stubRepositories(0, '/repositories/get-ttyg-repositories.json');
-        RepositoriesStubs.stubBaseEndpoints(repositoryId);
+        RepositoriesStub.stubBaseEndpoints(repositoryId);
         cy.presetRepository(repositoryId);
         TTYGStubs.stubAgentDefaultsGet();
     });
@@ -17,7 +18,6 @@ describe('TTYG clone an agent', () => {
         TTYGStubs.stubAgentListGet();
         TTYGStubs.stubChatsListGet();
         TTYGStubs.stubChatGet();
-        RepositoriesStubs.stubRepositoryModel(repositoryId);
         // Given I have opened the ttyg page
         TTYGViewSteps.visit();
         cy.wait('@get-all-repositories');
@@ -46,13 +46,11 @@ describe('TTYG clone an agent', () => {
         TTYGStubs.stubAgentCreate();
         TtygAgentSettingsModalSteps.saveAgent();
         cy.wait('@create-agent').then((interception) => {
-            // eslint-disable-next-line no-undef
             assert.deepEqual(interception.request.body, {
                 "id": "asst_gAPcrHQQ9ZIxD5eXWH2BNFfo",
                 "name": "agent-11",
                 "repositoryId": "starwars",
                 "model": "gpt-4o",
-                "contextSize": 128000,
                 "temperature": 0,
                 "topP": 0,
                 "seed": 0,
@@ -74,24 +72,4 @@ describe('TTYG clone an agent', () => {
             });
         });
     });
-
-    it('should not be able to clone an agent if an extraction method is selected but the precondition has failed', () => {
-        TTYGStubs.stubAgentListGet();
-        TTYGStubs.stubChatsListGet();
-        TTYGStubs.stubChatGet();
-        RepositoriesStubs.stubFtsSearchDisabled(repositoryId);
-
-        // GIVEN: I have opened the TTYG page
-        TTYGViewSteps.visit();
-        cy.wait('@get-all-repositories');
-        cy.wait('@get-agent-list');
-        cy.wait('@get-chat');
-
-        // WHEN: I try to clone an agent that has the FTS extraction method enabled, but FTS is disabled in the repository
-        TTYGViewSteps.expandAgentsSidebar();
-        TTYGViewSteps.triggerCloneAgentActionMenu(0);
-
-        // THEN: The Save Agent button should be disabled
-        TtygAgentSettingsModalSteps.getSaveAgentButton().should('be.disabled');
-    });
 });

package/{e2e-legacy → integration}/ttyg/create-agent.spec.js

@@ -2,22 +2,28 @@ import {TTYGViewSteps} from "../../steps/ttyg/ttyg-view-steps";
 import {TTYGStubs} from "../../stubs/ttyg/ttyg-stubs";
 import {RepositoriesStubs} from "../../stubs/repositories/repositories-stubs";
 import {TtygAgentSettingsModalSteps} from "../../steps/ttyg/ttyg-agent-settings-modal.steps";
+import {SimilarityIndexStubs} from "../../stubs/similarity-index-stubs";
 import {ConnectorStubs} from "../../stubs/connector-stubs";
-import {ModalDialogSteps} from "../../steps/modal-dialog-steps"
+import {ModalDialogSteps} from "../../steps/modal-dialog-steps";
+import {RepositoriesStub} from "../../stubs/repositories-stub";
 import {AlertDialogSteps} from "../../steps/alert-dialog-steps";
-import {ApplicationSteps} from '../../steps/application-steps.js';
 
 describe('TTYG create new agent', () => {
     const repositoryId = 'starwars';
 
     beforeEach(() => {
         RepositoriesStubs.stubRepositories(0, '/repositories/get-ttyg-repositories.json');
-        RepositoriesStubs.stubBaseEndpoints(repositoryId);
+        RepositoriesStub.stubBaseEndpoints(repositoryId);
         cy.presetRepository(repositoryId);
         TTYGStubs.stubAgentDefaultsGet();
     });
 
-    it('Should be able to cancel the new agent creation on the no agents view', () => {
+    it('Should be able to cancel the new agent creation on the no agents view', {
+        retries: {
+            runMode: 1,
+            openMode: 0
+        }
+    }, () => {
         TTYGStubs.stubChatsListGetNoResults();
         TTYGStubs.stubAgentListGet('/ttyg/agent/get-agent-list-0.json');
         // Given I have opened the ttyg page
@@ -33,7 +39,12 @@ describe('TTYG create new agent', () => {
         TTYGViewSteps.getNoAgentsView().should('be.visible');
     });
 
-    it('Should be able to create a new agent with SPARQL extraction method on the no agents view', () => {
+    it('Should be able to create a new agent with SPARQL extraction method on the no agents view', {
+        retries: {
+            runMode: 1,
+            openMode: 0
+        }
+    }, () => {
         TTYGStubs.stubChatsListGetNoResults();
         TTYGStubs.stubAgentListGet('/ttyg/agent/get-agent-list-0.json', 1000);
         // Given I have opened the ttyg page
@@ -57,12 +68,10 @@ describe('TTYG create new agent', () => {
         // enable SPARQL extraction method and disable it again to check the error message for the extraction methods
         TtygAgentSettingsModalSteps.enableSparqlExtractionMethod();
         // The component here is the bootstrap collapse component, so we need to wait for the animation to finish, otherwise the test might fail randomly
-        // eslint-disable-next-line cypress/no-unnecessary-waiting
         cy.wait(1000);
         TtygAgentSettingsModalSteps.getSparqlExtractionMethodPanel().should('be.visible');
         TtygAgentSettingsModalSteps.disableSparqlExtractionMethod();
         // The component here is the bootstrap collapse component, so we need to wait for the animation to finish, otherwise the test might fail randomly
-        // eslint-disable-next-line cypress/no-unnecessary-waiting
         cy.wait(1000);
         TtygAgentSettingsModalSteps.getSparqlExtractionMethodPanel().should('not.exist');
         TtygAgentSettingsModalSteps.getSaveAgentButton().should('be.disabled');
@@ -96,12 +105,12 @@ describe('TTYG create new agent', () => {
 
         // Validate the other agent settings
 
-        // LLM model
-        TtygAgentSettingsModalSteps.getLLMModelField().should('have.value', 'gpt-4o');
-        TtygAgentSettingsModalSteps.clearLLMModel();
+        // gpt model
+        TtygAgentSettingsModalSteps.getGptModelField().should('have.value', 'gpt-4o');
+        TtygAgentSettingsModalSteps.clearGptModel();
         TtygAgentSettingsModalSteps.getSaveAgentButton().should('be.disabled');
-        TtygAgentSettingsModalSteps.getLLMModelError().should('be.visible').and('contain', 'This field is required');
-        TtygAgentSettingsModalSteps.typeLLMModel('gpt-4o');
+        TtygAgentSettingsModalSteps.getGptModelError().should('be.visible').and('contain', 'This field is required');
+        TtygAgentSettingsModalSteps.typeGptModel('gpt-4o');
 
         // temperature
         TtygAgentSettingsModalSteps.setTemperature('0.2');
@@ -134,13 +143,11 @@ describe('TTYG create new agent', () => {
         TtygAgentSettingsModalSteps.saveAgent();
         TtygAgentSettingsModalSteps.getCreatingAgentLoader().should('be.visible');
         cy.wait('@create-agent').then((interception) => {
-            // eslint-disable-next-line no-undef
             assert.deepEqual(interception.request.body, {
                 "id": "id",
                 "name": "Test Agent",
                 "repositoryId": "starwars",
                 "model": "gpt-4o",
-                "contextSize": 128000,
                 "temperature": "0.2",
                 "topP": "0.2",
                 "seed": 0,
@@ -172,7 +179,12 @@ describe('TTYG create new agent', () => {
         TTYGViewSteps.getAgent(0).should('contain', 'Test Agent').and('contain', 'starwars');
     });
 
-    it('Should require FTS to be enabled for selected repository when creating agent with FTS extraction method', () => {
+    it('Should require FTS to be enabled for selected repository when creating agent with FTS extraction method', {
+        retries: {
+            runMode: 1,
+            openMode: 0
+        }
+    }, () => {
         RepositoriesStubs.stubGetRepositoryConfig(repositoryId, '/repositories/get-repository-config-starwars-disabled-fts.json');
         TTYGStubs.stubChatsListGetNoResults();
         TTYGStubs.stubAgentListGet('/ttyg/agent/get-agent-list-0.json');
@@ -189,7 +201,12 @@ describe('TTYG create new agent', () => {
         TtygAgentSettingsModalSteps.getFtsDisabledHelp().should('be.visible');
     });
 
-    it('Should be able to create agent with FTS extraction method', () => {
+    it('Should be able to create agent with FTS extraction method', {
+        retries: {
+            runMode: 1,
+            openMode: 0
+        }
+    }, () => {
         RepositoriesStubs.stubGetRepositoryConfig(repositoryId, '/repositories/get-repository-config-starwars-enabled-fts.json');
         TTYGStubs.stubChatsListGetNoResults();
         TTYGStubs.stubAgentListGet('/ttyg/agent/get-agent-list-0.json');
@@ -225,10 +242,15 @@ describe('TTYG create new agent', () => {
         TTYGViewSteps.getAgent(0).should('contain', 'Test Agent').and('contain', 'starwars');
     });
 
-    it('Should require similarity index in order to create agent with similarity search method', () => {
+    it('Should require similarity index in order to create agent with similarity search method', {
+        retries: {
+            runMode: 1,
+            openMode: 0
+        }
+    }, () => {
         TTYGStubs.stubChatsListGetNoResults();
         TTYGStubs.stubAgentListGet('/ttyg/agent/get-agent-list-0.json');
-        TTYGStubs.getSimilarityIndexesForRepo();
+        SimilarityIndexStubs.stubGetSimilarityIndexes('/similarity/get-similarity-indexes-0.json');
         // Given I have opened the ttyg page
         TTYGViewSteps.visit();
         cy.wait('@get-all-repositories');
@@ -243,10 +265,15 @@ describe('TTYG create new agent', () => {
         TtygAgentSettingsModalSteps.getSaveAgentButton().should('be.disabled');
     });
 
-    it('Should be able to configure and create agent with similarity index search method', () => {
+    it('Should be able to configure and create agent with similarity index search method', {
+        retries: {
+            runMode: 1,
+            openMode: 0
+        }
+    }, () => {
         TTYGStubs.stubChatsListGetNoResults();
         TTYGStubs.stubAgentListGet('/ttyg/agent/get-agent-list-0.json');
-        TTYGStubs.getSimilarityIndexesForRepo(repositoryId);
+        SimilarityIndexStubs.stubGetSimilarityIndexes();
         // Given I have opened the ttyg page
         TTYGViewSteps.visit();
         cy.wait('@get-all-repositories');
@@ -258,10 +285,8 @@ describe('TTYG create new agent', () => {
         TtygAgentSettingsModalSteps.selectRepository(repositoryId);
         // And I enable the similarity search extraction method
         TtygAgentSettingsModalSteps.enableSimilaritySearchMethodPanel();
-        // Then I expect the first similarity search index instance to be selected
-        TtygAgentSettingsModalSteps.getSimilarityIndexSelectedOption().should('have.text', 'otkg-vector-new');
-        // And I expect that no vector fields from that instance should be selected
-        TtygAgentSettingsModalSteps.getSimilarityIndexSelectedVectorFieldsValue().should('contain', 'Select field');
+        // Then I expect similarity index to be selected
+        TtygAgentSettingsModalSteps.getSimilarityIndexField().should('have.value', '0');
         // Then agent save button should be enabled
         TtygAgentSettingsModalSteps.getSaveAgentButton().should('be.enabled');
         // When I set the similarity index threshold
@@ -271,7 +296,6 @@ describe('TTYG create new agent', () => {
         TtygAgentSettingsModalSteps.getSimilarityIndexMaxTriplesField().should('have.value', '');
         TtygAgentSettingsModalSteps.setSimilarityIndexMaxTriples('100');
         // When I save the agent
-        // !!! This stub is not for agent with similarity search method, but it's ok for this test !!!
         TTYGStubs.stubAgentCreate();
         TTYGStubs.stubAgentListGet('/ttyg/agent/get-agent-list-new-agent.json');
         TtygAgentSettingsModalSteps.saveAgent();
@@ -286,7 +310,12 @@ describe('TTYG create new agent', () => {
         TTYGViewSteps.getAgent(0).should('contain', 'Test Agent').and('contain', 'starwars');
     });
 
-    it('Should require retrieval connector in order to create an agent with GPT retrieval connector method', () => {
+    it('Should require retrieval connector in order to create an agent with GPT retrieval connector method', {
+        retries: {
+            runMode: 1,
+            openMode: 0
+        }
+    }, () => {
         TTYGStubs.stubChatsListGetNoResults();
         TTYGStubs.stubAgentListGet('/ttyg/agent/get-agent-list-0.json');
         ConnectorStubs.stubGetConnectors();
@@ -308,7 +337,12 @@ describe('TTYG create new agent', () => {
         TtygAgentSettingsModalSteps.getSaveAgentButton().should('be.disabled');
     });
 
-    it('Should be able to configure and create an agent with retrieval connector method', () => {
+    it('Should be able to configure and create an agent with retrieval connector method', {
+        retries: {
+            runMode: 1,
+            openMode: 0
+        }
+    }, () => {
         TTYGStubs.stubChatsListGetNoResults();
         TTYGStubs.stubAgentListGet('/ttyg/agent/get-agent-list-0.json');
         ConnectorStubs.stubGetConnectors();
@@ -397,7 +431,7 @@ describe('TTYG create new agent', () => {
         TTYGStubs.stubChatsListGetNoResults();
         TTYGStubs.stubAgentListGet('/ttyg/agent/get-agent-list-0.json');
         ConnectorStubs.stubGetConnectors();
-        TTYGStubs.getSimilarityIndexesForRepo(repositoryId );
+        SimilarityIndexStubs.stubTTYGSimilarityIndexes();
         // Given I have opened the ttyg page
         TTYGViewSteps.visit();
         cy.wait('@get-all-repositories');
@@ -409,21 +443,16 @@ describe('TTYG create new agent', () => {
 
         // When I open Similarity index name panel
         TtygAgentSettingsModalSteps.enableSimilaritySearchMethodPanel();
-        cy.wait('@get-similarity-indexes');
         // Then I expect to see the first index selected.
-        TtygAgentSettingsModalSteps.verifySimilarityIndexSelected('otkg-vector-new');
+        TtygAgentSettingsModalSteps.verifySimilarityIndexSelected('similarity_index_starwars_one');
 
         // When I select another repository that have similarity connectors
-        TTYGStubs.getSimilarityIndexesForRepo('biomarkers', '/ttyg/agent/get-similarity-indexes-for-another-repo.json');
         TtygAgentSettingsModalSteps.selectRepository('biomarkers');
-        cy.wait('@get-similarity-indexes');
         // Then I expect to see the first similarity index from new repository selected.
-        TtygAgentSettingsModalSteps.verifySimilarityIndexSelected('biomarkers-vector');
+        TtygAgentSettingsModalSteps.verifySimilarityIndexSelected('similarity_index_biomarkers_one');
 
         // When I select a repository that not have similarity indexes
         TtygAgentSettingsModalSteps.selectRepository('ttyg-repo-1725518186812');
-        // Then I expect an error message to be open for similarity search index missing
-        ApplicationSteps.getErrorNotifications().should('be.visible');
         // Then I expect help message to be open
         TtygAgentSettingsModalSteps.getSimilaritySearchIndexMissingHelp().should('be.visible');
 
@@ -451,7 +480,7 @@ describe('TTYG create new agent', () => {
         TtygAgentSettingsModalSteps.setTemperature('1.2');
         TtygAgentSettingsModalSteps.getTemperatureSliderField().should('have.value', '1.2');
         // Then I should see a warning message
-        TtygAgentSettingsModalSteps.scrollToTemperatureWarning().should('be.visible');
+        TtygAgentSettingsModalSteps.getTemperatureWarning().should('be.visible');
         TtygAgentSettingsModalSteps.getTemperatureField().should('have.class', 'has-warning');
         // When I change the temperature to value below 1.0
         TtygAgentSettingsModalSteps.setTemperature('0.9');
@@ -492,7 +521,7 @@ describe('TTYG create new agent', () => {
         // When I open agent settings dialog and make all steps so the create button became enabled.
         TTYGStubs.stubChatsListGetNoResults();
         TTYGStubs.stubAgentListGet('/ttyg/agent/get-agent-list-0.json');
-        TTYGStubs.getSimilarityIndexesForRepo();
+        SimilarityIndexStubs.stubGetSimilarityIndexes('/similarity/get-similarity-indexes-0.json');
         TTYGViewSteps.visit();
         cy.wait('@get-all-repositories');
         TTYGViewSteps.createFirstAgent();

package/{e2e-legacy → integration}/ttyg/create-chat.spec.js

@@ -1,4 +1,5 @@
 import {RepositoriesStubs} from "../../stubs/repositories/repositories-stubs";
+import {RepositoriesStub} from "../../stubs/repositories-stub";
 import {TTYGStubs} from "../../stubs/ttyg/ttyg-stubs";
 import {TTYGViewSteps} from "../../steps/ttyg/ttyg-view-steps";
 import {ChatPanelSteps} from "../../steps/ttyg/chat-panel-steps";
@@ -8,7 +9,7 @@ describe('TTYG create chat', () => {
     beforeEach(() => {
         // Create an actual repository to prevent stubbing all background requests that are not related to the ttyg view
         RepositoriesStubs.stubRepositories(0, '/repositories/get-ttyg-repositories.json');
-        RepositoriesStubs.stubBaseEndpoints('starwars');
+        RepositoriesStub.stubBaseEndpoints('starwars');
         cy.presetRepository('starwars');
         TTYGStubs.stubChatsListGet("/ttyg/chats/create/get-chats-before-create.json");
         TTYGStubs.stubAgentListGet();
@@ -37,19 +38,16 @@ describe('TTYG create chat', () => {
         // When I type a question
         ChatPanelSteps.getQuestionInputElement()
             .should('be.visible')
-            .and('not.have.attr', 'disabled');
-        ChatPanelSteps.getQuestionInputElement()
+            .and('not.be.disabled')
             .type('Who is Han Solo?');
 
-
         // Then I expect the "Ask" button be active.
         ChatPanelSteps.getAskButtonElement().should('be.enabled');
 
         // When I click on "Ask" button.
-        TTYGStubs.stubCreateNewChat();
+        TTYGStubs.stubCrateNewChat();
         ChatPanelSteps.getAskButtonElement().scrollIntoView().click();
         cy.wait('@create-chat');
-        cy.wait('@ask-first-chat-question');
 
         // Then I expect new chat to be created in a new group "Today" and be selected
         TTYGViewSteps.getChatGroup(0).should('contain', 'Today');

package/{e2e-legacy → integration}/ttyg/delete-agent.spec.js

@@ -2,13 +2,14 @@ import {TTYGViewSteps} from "../../steps/ttyg/ttyg-view-steps";
 import {TTYGStubs} from "../../stubs/ttyg/ttyg-stubs";
 import {RepositoriesStubs} from "../../stubs/repositories/repositories-stubs";
 import {ModalDialogSteps} from "../../steps/modal-dialog-steps";
+import {RepositoriesStub} from "../../stubs/repositories-stub";
 
 describe('TTYG delete agent', () => {
     const repositoryId = 'starwars';
 
     beforeEach(() => {
         RepositoriesStubs.stubRepositories(0, '/repositories/get-ttyg-repositories.json');
-        RepositoriesStubs.stubBaseEndpoints(repositoryId);
+        RepositoriesStub.stubBaseEndpoints(repositoryId);
         cy.presetRepository(repositoryId);
     });