Fix: Deepthink tag recognition & Implement all providers

Author: ShenKSPZ

src/components/chat/MarkdownContent.tsx

@@ -21,7 +21,7 @@ type CodeProps = React.ClassAttributes<HTMLElement> &
 export const MarkdownContent: React.FC<MarkdownContentProps> = ({ content }) => {
   const [processedContent, setProcessedContent] = useState('');
   const [thinkContent, setThinkContent] = useState<string | null>(null);
-  const [isThinkExpanded, setIsThinkExpanded] = useState(false);
+  const [isThinkExpanded, setIsThinkExpanded] = useState(true);
 
   // Process content and check for thinking blocks
   useEffect(() => {
@@ -48,7 +48,13 @@ export const MarkdownContent: React.FC<MarkdownContentProps> = ({ content }) =>
 
       // Process the actual content
       processed = actualContent;
-    } else {
+    } 
+    else if (content.startsWith('<think>')) {
+      setThinkContent(content.substring(7));
+
+      processed = "";
+    }
+    else {
       setThinkContent(null);
     }
 

src/components/models/SelectModelDialog.tsx

@@ -187,7 +187,7 @@ export const SelectModelDialog: React.FC<SelectModelDialogProps> = ({
                             </div>
                           )}
                         </div>
-                        {selectedModelId === model.id && (
+                        {(selectedModelId === model.id && selectedProviderName === model.provider) && (
                           <div className="ml-4 text-white">
                             <Check size={16} />
                           </div>

src/components/settings/ApiManagement.tsx

@@ -1,8 +1,8 @@
 import React, { useState } from 'react';
 import { ChevronRight, Save, AlertCircle } from 'lucide-react';
 import { ProviderSettings } from '../../services/settings-service';
+import { AIProvider } from '../../types/ai-providers';
 
-export type AIProvider = 'Forge' | 'OpenAI' | 'Anthropic' | 'Gemini' | 'Fireworks.ai' | 'Together.ai' | 'OpenRouter' | 'Custom';
 
 interface ApiManagementProps {
   selectedProvider: AIProvider;
@@ -33,7 +33,7 @@ export const ApiManagement: React.FC<ApiManagementProps> = ({
   const currentProviderSettings = providerSettings[selectedProvider] || { apiKey: '' };
 
   const providers: AIProvider[] = [
-    'Forge', 
+    'TensorBlock', 
     'OpenAI', 
     'Anthropic', 
     'Gemini', 

src/components/settings/index.ts

@@ -1,4 +1,3 @@
 export * from './ApiManagement';
 export * from './ModelManagement';
-export * from './ChatSettings';
-export type { AIProvider } from './ApiManagement';
+export * from './ChatSettings';

src/pages/SettingsPage.tsx

@@ -1,7 +1,8 @@
 import React, { useState, useEffect } from 'react';
 import { Server, MessageSquare } from 'lucide-react';
 import { SettingsService, ProviderSettings } from '../services/settings-service';
-import { ApiManagement, ModelManagement, ChatSettings, AIProvider } from '../components/settings';
+import { ApiManagement, ModelManagement, ChatSettings } from '../components/settings';
+import { AIProvider } from '../types/ai-providers';
 import { DatabaseIntegrationService } from '../services/database-integration';
 import { AIService } from '../services/ai-service';
 

src/services/ai-service.ts

@@ -1,8 +1,9 @@
 import { AiServiceProvider, CompletionOptions } from './core/ai-service-provider';
-import { ProviderFactory } from './providers/providers';
+import { ProviderFactory } from './providers/provider-factory';
 import { Message } from '../types/chat';
 import { StreamControlHandler } from './streaming-control';
-import { AIProvider } from '../components/settings/';
+import { AIProvider } from '../types/ai-providers';
+import { SettingsService } from './settings-service';
 
 export interface ModelOption {
   id: string;
@@ -61,22 +62,22 @@ export class AIService {
    */
   private constructor() {
     // Initialize with default providers
-    this.addOpenAIProvider();
-    this.addForgeProvider();
+    this.addProviders();
     this.setupSettingsListener();
   }
 
-  private addOpenAIProvider(): void {
-    const openaiProvider = ProviderFactory.getProvider('OpenAI');
-    if (openaiProvider) {
-      this.providers.set('OpenAI', openaiProvider);
-    }
-  }
-
-  private addForgeProvider(): void {
-    const forgeProvider = ProviderFactory.getProvider('Forge');
-    if (forgeProvider) {
-      this.providers.set('Forge', forgeProvider);
+  private addProviders(): void {
+    const settingsService = SettingsService.getInstance();
+    const settings = settingsService.getSettings();
+    for (const provider of Object.keys(settings.providers)) {
+      const providerSettings = settings.providers[provider];
+      console.log('Provider: ', provider, ' Provider settings: ', providerSettings);
+      if (providerSettings && providerSettings.apiKey && providerSettings.apiKey.length > 0) {
+        const providerInstance = ProviderFactory.getNewProvider(provider as AIProvider);
+        if (providerInstance) {
+          this.providers.set(provider, providerInstance);
+        }
+      }
     }
   }
 
@@ -85,7 +86,7 @@ export class AIService {
    */
   private setupSettingsListener(): void {
     const handleSettingsChange = () => {
-      ProviderFactory.refreshProviders();
+      // ProviderFactory.refreshProviders();
       // Refresh models when settings change
       this.refreshModels();
     };
@@ -165,7 +166,7 @@ export class AIService {
     }
 
     // If provider not in cache, try to create it
-    const provider = ProviderFactory.getProvider(name as AIProvider);
+    const provider = ProviderFactory.getNewProvider(name as AIProvider);
     if (provider) {
       this.providers.set(name, provider);
       return provider;
@@ -383,8 +384,6 @@ export class AIService {
         name: model,
         provider: providerName
       }));
-      
-      console.log('Models for provider', providerName, modelOptions);
 
       // Cache results
       this.modelCache.set(providerName, modelOptions);

src/services/chat-service.ts

@@ -455,7 +455,7 @@ export class ChatService {
       // Update UI
       conversationUpdate(this.conversations);
 
-      const settingsService = SettingsService.getInstance();
+      // const settingsService = SettingsService.getInstance();
 
       // TODO: Uncomment this when we have a way to handle streaming
 

src/services/core/ai-service-provider.ts

@@ -52,7 +52,7 @@ export interface AiServiceProvider {
   /**
    * Set up authentication for this provider
    */
-  setupAuthentication(): void;
+  recreateClient(): void;
 
   /**
    * Check if the provider has a valid API key

src/services/database-integration.ts

@@ -3,7 +3,7 @@ import { SettingsService, ProviderSettings } from './settings-service';
 import { Conversation, Message } from '../types/chat';
 import { ApiSettings } from './api-settings';
 import { v4 as uuidv4 } from 'uuid';
-import { AIProvider } from '../components/settings';
+import { AIProvider } from '../types/ai-providers';
 
 /**
  * Service for integrating database operations with app components