67 lines
2.1 KiB
C#
67 lines
2.1 KiB
C#
using System.Text.Json;
|
||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
||
using Yi.Framework.AiHub.Domain.Shared.Extensions;
|
||
|
||
namespace Yi.Framework.AiHub.Domain.Shared.Dtos.Gemini;
|
||
|
||
public static class GeminiGenerateContentAcquirer
|
||
{
|
||
public static ThorUsageResponse? GetUsage(JsonElement response)
|
||
{
|
||
var usage = response.GetPath("usageMetadata");
|
||
if (!usage.HasValue)
|
||
{
|
||
return null;
|
||
}
|
||
|
||
var inputTokens = usage.Value.GetPath("promptTokenCount").GetInt();
|
||
var outputTokens = usage.Value.GetPath("candidatesTokenCount").GetInt()
|
||
+ usage.Value.GetPath("cachedContentTokenCount").GetInt()
|
||
+ usage.Value.GetPath("thoughtsTokenCount").GetInt()
|
||
+ usage.Value.GetPath("toolUsePromptTokenCount").GetInt();
|
||
|
||
|
||
return new ThorUsageResponse
|
||
{
|
||
PromptTokens = inputTokens,
|
||
InputTokens = inputTokens,
|
||
OutputTokens = outputTokens,
|
||
CompletionTokens = outputTokens,
|
||
TotalTokens = inputTokens + outputTokens,
|
||
};
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取图片url，包含前缀
|
||
/// </summary>
|
||
/// <param name="response"></param>
|
||
/// <returns></returns>
|
||
public static string GetImagePrefixBase64(JsonElement response)
|
||
{
|
||
// 获取 candidates[0].content.parts[0].text
|
||
var text = response.GetPath("candidates", 0, "content", "parts", 0, "text").GetString();
|
||
if (string.IsNullOrEmpty(text))
|
||
{
|
||
return string.Empty;
|
||
}
|
||
|
||
// 解析 markdown 图片格式: 
|
||
// 提取括号内的 data:image/xxx;base64,xxx 部分
|
||
var startMarker = "(data:image/";
|
||
var startIndex = text.IndexOf(startMarker);
|
||
if (startIndex < 0)
|
||
{
|
||
return string.Empty;
|
||
}
|
||
|
||
// 从 "data:" 开始
|
||
startIndex += 1; // 跳过 "("
|
||
var endIndex = text.IndexOf(')', startIndex);
|
||
if (endIndex < 0)
|
||
{
|
||
return string.Empty;
|
||
}
|
||
|
||
return text.Substring(startIndex, endIndex - startIndex);
|
||
}
|
||
} |