feat(openai-assistant): assistants streaming events added

Author: sebastianmusial

apps/spa/src/app/components/chat/chat-audio/chat-audio.component.ts

@@ -1,6 +1,9 @@
 import { Component, Input, OnInit } from '@angular/core';
 import { ChatClientService } from '../../../modules/+chat/shared/chat-client.service';
-import { ChatMessage, SpeechVoice } from '../../../modules/+chat/shared/chat.model';
+import {
+  ChatMessage,
+  SpeechVoice,
+} from '../../../modules/+chat/shared/chat.model';
 import { environment } from '../../../../environments/environment';
 import { MatIconModule } from '@angular/material/icon';
 import { delay } from 'rxjs';

apps/spa/src/app/components/chat/chat-message/chat-message.component.ts

@@ -1,5 +1,8 @@
 import { Component, HostBinding, Input } from '@angular/core';
-import { ChatRole, ChatMessage } from '../../../modules/+chat/shared/chat.model';
+import {
+  ChatRole,
+  ChatMessage,
+} from '../../../modules/+chat/shared/chat.model';
 import { MarkdownComponent } from 'ngx-markdown';
 import { ChatAudioComponent } from '../chat-audio/chat-audio.component';
 import { NgClass } from '@angular/common';

apps/spa/src/app/modules/+chat/containers/chat-iframe/chat-iframe.component.html

@@ -8,19 +8,19 @@
 
   <ai-spinner [isActive]="isLoading()" />
   @if (isConfigEnabled && !threadId()) {
-  <ai-configuration-form class="chat__content" />
+    <ai-configuration-form class="chat__content" />
   } @else {
-  <ai-chat-messages
-    [messages]="messages()"
-    [isTyping]="isTyping()"
-    [tips]="tips"
-    (tipSelected$)="chatService.sendMessage($event)"
-    class="chat__content" />
-  <ai-chat-footer
-    [isDisabled]="isTyping()"
-    [isTranscriptionEnabled]="isTranscriptionEnabled"
-    [isAttachmentEnabled]="isAttachmentEnabled"
-    (sendMessage$)="chatService.sendMessage($event)"
-    (sendAudio$)="chatService.sendAudio($event)" />
+    <ai-chat-messages
+      [messages]="messages()"
+      [isTyping]="isTyping()"
+      [tips]="tips"
+      (tipSelected$)="chatService.sendMessage($event)"
+      class="chat__content" />
+    <ai-chat-footer
+      [isDisabled]="isTyping()"
+      [isTranscriptionEnabled]="isTranscriptionEnabled"
+      [isAttachmentEnabled]="isAttachmentEnabled"
+      (sendMessage$)="chatService.sendMessage($event)"
+      (sendAudio$)="chatService.sendAudio($event)" />
   }
 </ai-card>

apps/spa/src/app/modules/+chat/shared/chat-gateway.service.ts

@@ -10,12 +10,12 @@ export class ChatGatewayService {
   private socket = io(environment.websocketUrl);
 
   sendMessage(payload: ChatCallDto): void {
-    this.socket.emit(ChatEvents.SendMessage, payload);
+    this.socket.emit(ChatEvents.CallStart, payload);
   }
 
   getMessages(): Observable<ChatCallDto> {
     return new Observable<ChatCallDto>(observer => {
-      this.socket.on(ChatEvents.MessageReceived, data => observer.next(data));
+      this.socket.on(ChatEvents.CallDone, data => observer.next(data));
       return () => this.socket.disconnect();
     });
   }

apps/spa/src/app/modules/+chat/shared/chat.model.ts

@@ -15,8 +15,21 @@ export interface ChatMessage {
 }
 
 export enum ChatEvents {
-  SendMessage = 'send_message',
-  MessageReceived = 'message_received',
+  CallStart = 'callStart',
+  CallDone = 'callDone',
+  TextDone = 'textCreated',
+  TextCreated = 'textDelta',
+  TextDelta = 'textDone',
+  MessageCreated = 'messageCreated',
+  MessageDelta = 'messageDelta',
+  MessageDone = 'messageDone',
+  ImageFileDone = 'imageFileDone',
+  ToolCallCreated = 'toolCallCreated',
+  ToolCallDelta = 'toolCallDelta',
+  ToolCallDone = 'toolCallDone',
+  RunStepCreated = 'runStepCreated',
+  RunStepDelta = 'runStepDelta',
+  RunStepDone = 'runStepDone',
 }
 
 export enum ChatMessageStatus {

apps/spa/src/app/modules/+chat/shared/chat.service.ts

@@ -19,7 +19,6 @@ import { OpenAiFile, GetThreadResponseDto } from '@boldare/openai-assistant';
 import { Message } from 'openai/resources/beta/threads/messages';
 import { TextContentBlock } from 'openai/resources/beta/threads/messages/messages';
 
-
 @Injectable({ providedIn: 'root' })
 export class ChatService {
   isLoading$ = new BehaviorSubject<boolean>(false);

apps/spa/src/app/modules/+chat/shared/thread.service.ts

@@ -1,5 +1,12 @@
 import { Injectable } from '@angular/core';
-import { BehaviorSubject, catchError, Observable, Subject, take, tap } from 'rxjs';
+import {
+  BehaviorSubject,
+  catchError,
+  Observable,
+  Subject,
+  take,
+  tap,
+} from 'rxjs';
 import { environment } from '../../../../environments/environment';
 import { ThreadClientService } from './thread-client.service';
 import { ConfigurationFormService } from '../../+configuration/shared/configuration-form.service';
@@ -43,11 +50,9 @@ export class ThreadService {
   }
 
   getThread(id: string): Observable<GetThreadResponseDto> {
-    return this.threadClientService
-      .getThread(id)
-      .pipe(
-        take(1),
-        catchError(() => this.start()),
-      );
+    return this.threadClientService.getThread(id).pipe(
+      take(1),
+      catchError(() => this.start()),
+    );
   }
 }

libs/openai-assistant/src/lib/agent/agent.mock.ts

@@ -1,10 +1,10 @@
-import { AssistantCreateParams } from 'openai/resources/beta';
+import { FunctionTool } from 'openai/resources/beta';
 
 export const agentNameMock = 'agent-name';
 
 export const agentMock = async () => 'agent-result';
 
-export const definitionMock: AssistantCreateParams.AssistantToolsFunction = {
+export const definitionMock: FunctionTool = {
   type: 'function',
   function: { name: agentNameMock },
 };

libs/openai-assistant/src/lib/agent/agent.service.ts

@@ -7,10 +7,7 @@ export class AgentService {
   public agents: Agents = {};
   public tools: FunctionTool[] = [];
 
-  add(
-    definition: FunctionTool,
-    fn: Agent,
-  ): void {
+  add(definition: FunctionTool, fn: Agent): void {
     this.tools.push(definition);
     this.agents[definition.function.name] = fn;
   }

libs/openai-assistant/src/lib/chat/chat.gateway.spec.ts

@@ -32,7 +32,7 @@ describe('ChatGateway', () => {
 
       await chatGateway.listenForMessages(request, {} as Socket);
 
-      expect(chatService.call).toHaveBeenCalledWith(request);
+      expect(chatService.call).toHaveBeenCalled();
     });
   });