Merge branch 'main' into backend_biome_up

Author: Artuomka

README.md

@@ -100,5 +100,6 @@ This section provides clear and concise instructions for installing RocketAdmin
 
 ## Usage
 
-1. After installation rocketadmin will create a user with email admin@email.local and autogenerated password. The message will be `Admin user created with email: "admin@email.local" and password: "<password>"`
-2. You can sign in using these credentials. We recommend to change email and password after first login
+1. After installation, open RocketAdmin in your browser at `http://localhost:8080`. You will be redirected to the setup page.
+2. On the setup page, create your admin account by entering your email and password.
+3. After completing the setup, sign in with your new credentials.

frontend/src/app/components/dashboard/db-table-view/db-table-ai-panel/db-table-ai-panel.component.html

@@ -23,7 +23,7 @@ <h2 class="mat-heading-2 ai-panel-sidebar__title">
             <div class="ai-message-chain">
                 <div *ngFor="let message of messagesChain" class="{{message.type}}-message">
                     <span style="white-space: pre-wrap" *ngIf="message.type == 'user'">{{message.text}}</span>
-                    <markdown *ngIf="message.type == 'ai'" mermaid>{{message.text}}</markdown>
+                    <markdown *ngIf="message.type == 'ai'" mermaid [data]="message.text"></markdown>
                 </div>
             </div>
         </div>

frontend/src/app/components/dashboard/db-table-view/db-table-ai-panel/db-table-ai-panel.component.spec.ts

@@ -3,31 +3,35 @@ import { ComponentFixture, TestBed } from '@angular/core/testing';
 import { MatIconTestingModule } from '@angular/material/icon/testing';
 import { BrowserAnimationsModule } from '@angular/platform-browser/animations';
 import { Angulartics2Module } from 'angulartics2';
-import { MarkdownService } from 'ngx-markdown';
-import { of, throwError } from 'rxjs';
+import { of } from 'rxjs';
+import { AiService } from 'src/app/services/ai.service';
 import { ConnectionsService } from 'src/app/services/connections.service';
 import { TableStateService } from 'src/app/services/table-state.service';
 import { TablesService } from 'src/app/services/tables.service';
 import { DbTableAiPanelComponent } from './db-table-ai-panel.component';
 
+async function* mockStream(...chunks: string[]): AsyncGenerator<string> {
+	for (const chunk of chunks) {
+		yield chunk;
+	}
+}
+
 describe('DbTableAiPanelComponent', () => {
 	let component: DbTableAiPanelComponent;
 	let fixture: ComponentFixture<DbTableAiPanelComponent>;
-	let tablesService: TablesService;
 	let tableStateService: TableStateService;
 
-	const mockMarkdownService = {
-		parse: vi.fn().mockReturnValue('parsed markdown'),
-	};
-
 	const mockConnectionsService = {
 		currentConnectionID: '12345678',
 	};
 
 	const mockTablesService = {
 		currentTableName: 'users',
-		createAIthread: vi.fn(),
-		requestAImessage: vi.fn(),
+	};
+
+	const mockAiService: Partial<AiService> = {
+		createThread: vi.fn(),
+		sendMessage: vi.fn(),
 	};
 
 	const mockTableStateService = {
@@ -41,16 +45,15 @@ describe('DbTableAiPanelComponent', () => {
 			imports: [Angulartics2Module.forRoot(), DbTableAiPanelComponent, BrowserAnimationsModule, MatIconTestingModule],
 			providers: [
 				provideHttpClient(),
-				{ provide: MarkdownService, useValue: mockMarkdownService },
 				{ provide: ConnectionsService, useValue: mockConnectionsService },
 				{ provide: TablesService, useValue: mockTablesService },
+				{ provide: AiService, useValue: mockAiService },
 				{ provide: TableStateService, useValue: mockTableStateService },
 			],
 		}).compileComponents();
 
 		fixture = TestBed.createComponent(DbTableAiPanelComponent);
 		component = fixture.componentInstance;
-		tablesService = TestBed.inject(TablesService);
 		tableStateService = TestBed.inject(TableStateService);
 		fixture.detectChanges();
 	});
@@ -77,7 +80,10 @@ describe('DbTableAiPanelComponent', () => {
 	});
 
 	it('should create thread on Enter key when no thread exists', () => {
-		mockTablesService.createAIthread.mockReturnValue(of({ threadId: 'thread-123', responseMessage: 'AI response' }));
+		(mockAiService.createThread as ReturnType<typeof vi.fn>).mockResolvedValue({
+			threadId: 'thread-123',
+			stream: mockStream('AI response'),
+		});
 
 		component.message = 'Test message';
 		component.threadID = null;
@@ -86,11 +92,16 @@ describe('DbTableAiPanelComponent', () => {
 		Object.defineProperty(event, 'preventDefault', { value: vi.fn() });
 		component.onKeydown(event);
 
-		expect(mockTablesService.createAIthread).toHaveBeenCalledWith('12345678', 'users', 'Test message');
+		expect(mockAiService.createThread).toHaveBeenCalledWith(
+			'12345678',
+			'users',
+			'Test message',
+			expect.any(AbortSignal),
+		);
 	});
 
 	it('should send message on Enter key when thread exists', () => {
-		mockTablesService.requestAImessage.mockReturnValue(of('AI response'));
+		(mockAiService.sendMessage as ReturnType<typeof vi.fn>).mockResolvedValue(mockStream('AI response'));
 
 		component.message = 'Follow up message';
 		component.threadID = 'existing-thread';
@@ -99,55 +110,65 @@ describe('DbTableAiPanelComponent', () => {
 		Object.defineProperty(event, 'preventDefault', { value: vi.fn() });
 		component.onKeydown(event);
 
-		expect(mockTablesService.requestAImessage).toHaveBeenCalledWith(
+		expect(mockAiService.sendMessage).toHaveBeenCalledWith(
 			'12345678',
 			'users',
 			'existing-thread',
 			'Follow up message',
+			expect.any(AbortSignal),
 		);
 	});
 
-	it('should add user message to chain when creating thread', () => {
-		mockTablesService.createAIthread.mockReturnValue(of({ threadId: 'thread-123', responseMessage: 'AI response' }));
+	it('should add user message to chain when creating thread', async () => {
+		(mockAiService.createThread as ReturnType<typeof vi.fn>).mockResolvedValue({
+			threadId: 'thread-123',
+			stream: mockStream('AI response'),
+		});
 
 		component.message = 'User question';
-		component.createThread();
+		await component.createThread();
 
 		expect(component.messagesChain[0]).toEqual({
 			type: 'user',
 			text: 'User question',
 		});
 	});
 
-	it('should add AI response to chain after thread creation', () => {
-		mockTablesService.createAIthread.mockReturnValue(of({ threadId: 'thread-123', responseMessage: 'AI response' }));
+	it('should add AI response to chain after thread creation', async () => {
+		(mockAiService.createThread as ReturnType<typeof vi.fn>).mockResolvedValue({
+			threadId: 'thread-123',
+			stream: mockStream('AI response'),
+		});
 
 		component.message = 'User question';
-		component.createThread();
+		await component.createThread();
 
 		expect(component.threadID).toBe('thread-123');
 		expect(component.messagesChain[1]).toEqual({
 			type: 'ai',
-			text: 'parsed markdown',
+			text: 'AI response',
 		});
 	});
 
-	it('should handle error when creating thread', () => {
-		mockTablesService.createAIthread.mockReturnValue(throwError(() => 'Error message'));
+	it('should handle error when creating thread', async () => {
+		(mockAiService.createThread as ReturnType<typeof vi.fn>).mockRejectedValue('Error message');
 
 		component.message = 'User question';
-		component.createThread();
+		await component.createThread();
 
 		expect(component.messagesChain[1]).toEqual({
 			type: 'ai-error',
 			text: 'Error message',
 		});
 	});
 
-	it('should use suggested message when provided to createThread', () => {
-		mockTablesService.createAIthread.mockReturnValue(of({ threadId: 'thread-123', responseMessage: 'AI response' }));
+	it('should use suggested message when provided to createThread', async () => {
+		(mockAiService.createThread as ReturnType<typeof vi.fn>).mockResolvedValue({
+			threadId: 'thread-123',
+			stream: mockStream('AI response'),
+		});
 
-		component.createThread('Suggested question');
+		await component.createThread('Suggested question');
 
 		expect(component.messagesChain[0].text).toBe('Suggested question');
 	});
@@ -157,12 +178,12 @@ describe('DbTableAiPanelComponent', () => {
 		expect(mockTableStateService.handleViewAIpanel).toHaveBeenCalled();
 	});
 
-	it('should clear message after sending', () => {
-		mockTablesService.requestAImessage.mockReturnValue(of('AI response'));
+	it('should clear message after sending', async () => {
+		(mockAiService.sendMessage as ReturnType<typeof vi.fn>).mockResolvedValue(mockStream('AI response'));
 
 		component.message = 'Test message';
 		component.threadID = 'thread-123';
-		component.sendMessage();
+		await component.sendMessage();
 
 		expect(component.message).toBe('');
 		expect(component.charactrsNumber).toBe(0);