MindWorkAI/AI-Studio
1
0 Fork 0
mirror of https://github.com/MindWorkAI/AI-Studio.git synced 2026-10-14 01:33:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Finished the function for importing arbitrary documents (#516)

Browse Source
This commit is contained in:
Thorsten Sommer authored and GitHub committed 2025-06-30 21:51:02 +02:00
1 parent aaedf667fe
commit 68f5bb1512
5 files changed
+123 -47
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

app/MindWork AI Studio/Tools/Services/RustService.Retrieval.cs
+7 -4
View File
@@ -5,9 +5,10 @@ namespace AIStudio.Tools.Services;
public sealed partial class RustService
{
public async Task<string> ReadArbitraryFileData(string path, string streamId, int maxChunks)
public async Task<string> ReadArbitraryFileData(string path, int maxChunks, bool extractImages = false)
{
var requestUri = $"/retrieval/fs/extract?path={Uri.EscapeDataString(path)}&stream_id={streamId}";
var streamId = Guid.NewGuid().ToString();
var requestUri = $"/retrieval/fs/extract?path={Uri.EscapeDataString(path)}&stream_id={streamId}&extract_images={extractImages}";
var request = new HttpRequestMessage(HttpMethod.Get, requestUri);
var response = await this.http.SendAsync(request, HttpCompletionOption.ResponseHeadersRead);
@@ -36,8 +37,10 @@ public sealed partial class RustService
var sseEvent = JsonSerializer.Deserialize<ContentStreamSseEvent>(jsonContent);
if (sseEvent is not null)
{
var content = ContentStreamSseHandler.ProcessEvent(sseEvent, false);
resultBuilder.Append(content);
var content = ContentStreamSseHandler.ProcessEvent(sseEvent, extractImages);
if(content is not null)
resultBuilder.AppendLine(content);
chunkCount++;
}
}