align AI provider typing with ProviderName union

chatapi/src/models/chat.model.ts

@@ -1,4 +1,4 @@
-type ProviderName = 'openai' | 'perplexity' | 'deepseek' | 'gemini';
+export type ProviderName = 'openai' | 'perplexity' | 'deepseek' | 'gemini';
 
 export interface AIProvider {
   name: ProviderName;

chatapi/src/models/db-doc.model.ts

@@ -1,12 +1,12 @@
-import { ChatItem } from './chat.model';
+import { ChatItem, ProviderName } from './chat.model';
 
 export interface DbDoc {
   _id: string;
   _rev: string;
   user: any;
   title: string;
   createdDate: number;
-  aiProvider: string;
+  aiProvider?: ProviderName;
   conversations: ChatItem[];
 }
 

chatapi/src/services/chat.service.ts

@@ -3,7 +3,7 @@ import { DocumentInsertResponse } from 'nano';
 import { chatDB } from '../config/nano.config';
 import { retrieveChatHistory } from '../utils/db.utils';
 import { aiChat } from '../utils/chat.utils';
-import { AIProvider, ChatMessage } from '../models/chat.model';
+import { AIProvider, ChatMessage, ProviderName } from '../models/chat.model';
 
 function handleChatError(error: any) {
   if (error.response) {
@@ -26,7 +26,11 @@ export async function chat(data: any, stream?: boolean, callback?: (response: st
 } | undefined> {
   const { content, ...dbData } = data;
   const messages: ChatMessage[] = [];
-  const aiProvider = dbData.aiProvider as AIProvider || { 'name': 'openai' };
+  const aiProviderDetails = dbData.aiProvider as AIProvider | ProviderName | undefined;
+  const fallbackProvider: ProviderName = 'openai';
+  const aiProvider: AIProvider = typeof aiProviderDetails === 'object' && aiProviderDetails !== null
+    ? { name: aiProviderDetails.name, model: aiProviderDetails.model }
+    : { name: aiProviderDetails ?? fallbackProvider };
 
   if (!content || typeof content !== 'string') {
     throw new Error('"data.content" is a required non-empty string field');

src/app/chat/chat-sidebar/chat-sidebar.component.ts

@@ -38,7 +38,7 @@ export class ChatSidebarComponent implements OnInit, OnDestroy {
   selectedConversation: Conversation;
   lastRenderedConversation: number;
   isEditing: boolean;
-  provider: AIProvider;
+  provider: AIProvider = { name: 'openai' };
   fullTextSearch = false;
   searchType: 'questions' | 'responses';
   overlayOpen = false;

src/app/chat/chat-window/chat-window.component.ts

@@ -30,7 +30,7 @@ export class ChatWindowComponent implements OnInit, OnDestroy, AfterViewInit {
   streaming: boolean;
   disabled = false;
   clearChat = true;
-  provider: AIProvider;
+  provider: AIProvider = { name: 'openai' };
   fallbackConversation: any[] = [];
   selectedConversationId: any;
   promptForm: FormGroup<PromptForm>;

src/app/chat/chat.component.ts

@@ -2,15 +2,15 @@ import { Component, OnInit } from '@angular/core';
 import { ActivatedRoute, Router } from '@angular/router';
 
 import { ChatService } from '../shared/chat.service';
-import { AIProvider } from './chat.model';
+import { AIProvider, ProviderName } from './chat.model';
 
 @Component({
   selector: 'planet-chat',
   templateUrl: './chat.component.html',
   styleUrls: [ './chat.scss' ]
 })
 export class ChatComponent implements OnInit {
-  activeService: string;
+  activeService?: ProviderName;
   aiServices: AIProvider[] = [];
   displayToggle: boolean;
 
@@ -23,9 +23,11 @@ export class ChatComponent implements OnInit {
   ngOnInit() {
     this.chatService.listAIProviders().subscribe((providers) => {
       this.aiServices = providers;
-      this.activeService = this.aiServices[0]?.model;
+      this.activeService = this.aiServices[0]?.name;
       this.displayToggle = this.aiServices.length > 0;
-      this.chatService.toggleAIServiceSignal(this.activeService);
+      if (this.activeService) {
+        this.chatService.toggleAIServiceSignal(this.activeService);
+      }
     });
     this.subscribeToAIService();
   }
@@ -45,7 +47,9 @@ export class ChatComponent implements OnInit {
   }
 
   toggleAIService(): void {
-    this.chatService.toggleAIServiceSignal(this.activeService);
+    if (this.activeService) {
+      this.chatService.toggleAIServiceSignal(this.activeService);
+    }
   }
 
 }

src/app/chat/chat.model.ts

@@ -28,7 +28,7 @@ export interface Message {
 export type ProviderName = 'openai' | 'perplexity' | 'deepseek' | 'gemini';
 
 export interface AIProvider {
-  name: string;
+  name: ProviderName;
   model?: string;
 }
 

src/app/shared/chat.service.ts

@@ -6,7 +6,7 @@ import { catchError, map } from 'rxjs/operators';
 import { environment } from '../../environments/environment';
 import { findDocuments, inSelector } from '../shared/mangoQueries';
 import { CouchService } from '../shared/couchdb.service';
-import { AIServices, AIProvider } from '../chat/chat.model';
+import { AIServices, AIProvider, ProviderName } from '../chat/chat.model';
 
 @Injectable({
   providedIn: 'root'
@@ -20,14 +20,14 @@ import { AIServices, AIProvider } from '../chat/chat.model';
   private errorSubject: Subject<string> = new Subject<string>();
   private newChatAdded: Subject<void> = new Subject<void>();
   private newChatSelected: Subject<void> = new Subject<void>();
-  private toggleAIService = new Subject<string>();
+  private toggleAIService = new Subject<ProviderName>();
   private selectedConversationIdSubject = new BehaviorSubject<object | null>(null);
   private aiProvidersSubject = new BehaviorSubject<Array<AIProvider>>([]);
   private currentChatAIProvider = new BehaviorSubject<AIProvider>(undefined);
 
   newChatAdded$ = this.newChatAdded.asObservable();
   newChatSelected$ = this.newChatSelected.asObservable();
-  toggleAIService$ = this.toggleAIService.asObservable();
+  toggleAIService$: Observable<ProviderName> = this.toggleAIService.asObservable();
   aiProviders$ = this.aiProvidersSubject.asObservable();
   selectedConversationId$: Observable<object | null> = this.selectedConversationIdSubject.asObservable();
   currentChatAIProvider$: Observable<AIProvider> = this.currentChatAIProvider.asObservable();
@@ -70,7 +70,7 @@ import { AIServices, AIProvider } from '../chat/chat.model';
         }),
         map((services: AIServices) => {
           if (services) {
-            return Object.entries(services)
+            return (Object.entries(services) as [ ProviderName, boolean ][])
               .filter(([ _, model ]) => model === true)
               .map(([ key ]) => ({ name: key, model: key }));
           } else {
@@ -129,7 +129,7 @@ import { AIServices, AIProvider } from '../chat/chat.model';
     this.newChatSelected.next();
   }
 
-  toggleAIServiceSignal(aiService: string) {
+  toggleAIServiceSignal(aiService: ProviderName) {
     this.toggleAIService.next(aiService);
   }