Added support for Perplexity (#545)

Co-authored-by: Thorsten Sommer <mail@tsommer.org>

app/MindWork AI Studio/Chat/ContentBlockComponent.razor

@@ -1,6 +1,7 @@
 @using AIStudio.Tools
 @using MudBlazor
 @using AIStudio.Components
+@using AIStudio.Provider
 @inherits AIStudio.Components.MSGComponentBase
 <MudCard Class="@this.CardClasses" Outlined="@true">
     <MudCardHeader>
@@ -15,6 +16,14 @@
             </MudText>
         </CardHeaderContent>
         <CardHeaderActions>
+            @if (this.Content.Sources.Count > 0)
+            {
+                <MudTooltip Text="@T("Number of sources")" Placement="Placement.Bottom">
+                    <MudBadge Content="@this.Content.Sources.Count" Color="Color.Primary" Overlap="true" BadgeClass="sources-card-header">
+                        <MudIconButton Icon="@Icons.Material.Filled.Link" />
+                    </MudBadge>
+                </MudTooltip>
+            }
             @if (this.IsSecondToLastBlock && this.Role is ChatRole.USER && this.EditLastUserBlockFunc is not null)
             {
                 <MudTooltip Text="@T("Edit")" Placement="Placement.Bottom">
@@ -72,6 +81,10 @@
                             else
                             {
                                 <MudMarkdown Value="@textContent.Text.RemoveThinkTags().Trim()" Props="Markdown.DefaultConfig" Styling="@this.MarkdownStyling" />
+                                @if (textContent.Sources.Count > 0)
+                                {
+                                    <MudMarkdown Value="@textContent.Sources.ToMarkdown()" Props="Markdown.DefaultConfig" Styling="@this.MarkdownStyling" />
+                                }
                             }
                         }
                     }

app/MindWork AI Studio/Chat/ContentImage.cs

@@ -27,6 +27,9 @@ public sealed class ContentImage : IContent, IImageSource
     [JsonIgnore]
     public Func<Task> StreamingEvent { get; set; } = () => Task.CompletedTask;
 
+    /// <inheritdoc />
+    public List<Source> Sources { get; set; } = [];
+
     /// <inheritdoc />
     public Task<ChatThread> CreateFromProviderAsync(IProvider provider, Model chatModel, IContent? lastPrompt, ChatThread? chatChatThread, CancellationToken token = default)
     {

app/MindWork AI Studio/Chat/ContentText.cs

@@ -24,7 +24,7 @@ public sealed class ContentText : IContent
     public bool InitialRemoteWait { get; set; }
 
     /// <inheritdoc />
-    // [JsonIgnore]
+    [JsonIgnore]
     public bool IsStreaming { get; set; }
 
     /// <inheritdoc />
@@ -35,6 +35,9 @@ public sealed class ContentText : IContent
     [JsonIgnore]
     public Func<Task> StreamingEvent { get; set; } = () => Task.CompletedTask;
 
+    /// <inheritdoc />
+    public List<Source> Sources { get; set; } = [];
+
     /// <inheritdoc />
     public async Task<ChatThread> CreateFromProviderAsync(IProvider provider, Model chatModel, IContent? lastPrompt, ChatThread? chatThread, CancellationToken token = default)
     {
@@ -80,7 +83,7 @@ public sealed class ContentText : IContent
             this.InitialRemoteWait = true;
             
             // Iterate over the responses from the AI:
-            await foreach (var deltaText in provider.StreamChatCompletion(chatModel, chatThread, settings, token))
+            await foreach (var contentStreamChunk in provider.StreamChatCompletion(chatModel, chatThread, settings, token))
             {
                 // When the user cancels the request, we stop the loop:
                 if (token.IsCancellationRequested)
@@ -91,7 +94,10 @@ public sealed class ContentText : IContent
                 this.IsStreaming = true;
                 
                 // Add the response to the text:
-                this.Text += deltaText;
+                this.Text += contentStreamChunk;
+                
+                // Merge the sources:
+                this.Sources.MergeSources(contentStreamChunk.Sources);
                 
                 // Notify the UI that the content has changed,
                 // depending on the energy saving mode:

app/MindWork AI Studio/Chat/IContent.cs

@@ -37,6 +37,12 @@ public interface IContent
     /// </summary>
     [JsonIgnore]
     public Func<Task> StreamingDone { get; set; }
+
+    /// <summary>
+    /// The provided sources, if any.
+    /// </summary>
+    [JsonIgnore]
+    public List<Source> Sources { get; set; }
     
     /// <summary>
     /// Uses the provider to create the content.