Browse Source

Download image implementation

pull/653/head
Zoli Somogyi 2 years ago
parent
commit
44a82b0f3f
1 changed files with 9 additions and 2 deletions
  1. +9
    -2
      LLama/LLamaInteractExecutor.cs

+ 9
- 2
LLama/LLamaInteractExecutor.cs View File

@@ -11,6 +11,7 @@ using System.Threading.Tasks;
using LLama.Exceptions;
using LLama.Extensions;
using Microsoft.Extensions.Logging;
using System.Net.Http;

namespace LLama
{
@@ -154,7 +155,7 @@ namespace LLama
{
if (image.Type == ImageData.DataType.ImagePath && image.Data != null)
{
_imageEmbedHandles.Add(SafeLlavaImageEmbedHandle.CreateFromFileName(ClipModel.NativeHandle, Context, image.Data.ToString()));
_imageEmbedHandles.Add(SafeLlavaImageEmbedHandle.CreateFromFileName(ClipModel.NativeHandle, Context, (string)image.Data));
}
else if (image.Type == ImageData.DataType.ImageBytes && image.Data != null)
{
@@ -162,7 +163,13 @@ namespace LLama
}
else if (image.Type == ImageData.DataType.ImageURL && image.Data != null)
{
throw new NotImplementedException();
using var httpClient = new HttpClient();
var uri = new Uri((string)image.Data);
var imageBytes = httpClient.GetByteArrayAsync(uri).Result;
if (imageBytes != null && imageBytes.Length > 0)
{
_imageEmbedHandles.Add(SafeLlavaImageEmbedHandle.CreateFromMemory(ClipModel.NativeHandle, Context, imageBytes));
}
}
}



Loading…
Cancel
Save