xwl2023/Yi.Framework
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 完成支持functioncall功能

Browse Source
This commit is contained in:
ccnetcore
2025-07-18 23:12:20 +08:00
parent d6836b8bcf
commit 5d7217b775
3 changed files with 25 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/OpenAi/ThorChatCompletionsRequest.cs
View File

@@ -14,8 +14,7 @@ public class ThorChatCompletionsRequest
Messages = new List<ThorChatMessage>(); Messages = new List<ThorChatMessage>();
} }
[JsonPropertyName("store")] [JsonPropertyName("store")] public bool? Store { get; set; }
public bool? Store { get; set; }
/// <summary> /// <summary>
/// 表示对话中支持的模态类型数组。可以为 null。 /// 表示对话中支持的模态类型数组。可以为 null。
@@ -26,7 +25,8 @@ public class ThorChatCompletionsRequest
/// <summary> /// <summary>
/// 表示对话中的音频请求参数。可以为 null。 /// 表示对话中的音频请求参数。可以为 null。
/// </summary> /// </summary>
[JsonPropertyName("audio")] public ThorChatAudioRequest? Audio { get; set; } [JsonPropertyName("audio")]
public ThorChatAudioRequest? Audio { get; set; }
/// <summary> /// <summary>
/// 包含迄今为止对话的消息列表 /// 包含迄今为止对话的消息列表
@@ -229,18 +229,25 @@ public class ThorChatCompletionsRequest
{ {
if (value is JsonElement jsonElement) if (value is JsonElement jsonElement)
{ {
if (jsonElement.ValueKind == JsonValueKind.String) // if (jsonElement.ValueKind == JsonValueKind.String)
{ // {
ToolChoice = new ThorToolChoice // ToolChoice = new ThorToolChoice
{ // {
Type = jsonElement.GetString() // Type = jsonElement.GetString()
}; // };
} // }
else if (jsonElement.ValueKind == JsonValueKind.Object) if (jsonElement.ValueKind == JsonValueKind.Object)
{ {
ToolChoice = jsonElement.Deserialize<ThorToolChoice>(); ToolChoice = jsonElement.Deserialize<ThorToolChoice>();
} }
} }
else if (value is string text)
{
ToolChoice = new ThorToolChoice
{
Type = text
};
}
else else
{ {
ToolChoice = (ThorToolChoice)value; ToolChoice = (ThorToolChoice)value;

2
Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Chat/MessageAggregateRoot.cs
View File

@@ -44,7 +44,7 @@ public class MessageAggregateRoot : FullAuditedAggregateRoot<Guid>
public Guid? SessionId { get; set; } public Guid? SessionId { get; set; }
[SugarColumn(ColumnDataType = StaticConfig.CodeFirst_BigString)] [SugarColumn(ColumnDataType = StaticConfig.CodeFirst_BigString)]
public string Content { get; set; } public string? Content { get; set; }
public string Role { get; set; } public string Role { get; set; }
public string ModelId { get; set; } public string ModelId { get; set; }

10
Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/AiGateWayManager.cs
View File

@@ -109,8 +109,7 @@ public class AiGateWayManager : DomainService
_specialCompatible.Compatible(request); _specialCompatible.Compatible(request);
var response = httpContext.Response; var response = httpContext.Response;
// 设置响应头，声明是 json // 设置响应头，声明是 json
response.ContentType = "application/json; charset=UTF-8"; //response.ContentType = "application/json; charset=UTF-8";
await using var writer = new StreamWriter(response.Body, Encoding.UTF8, leaveOpen: true);
var modelDescribe = await GetModelAsync(request.Model); var modelDescribe = await GetModelAsync(request.Model);
var chatService = var chatService =
LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName); LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName);
@@ -137,12 +136,7 @@ public class AiGateWayManager : DomainService
data.Usage.OutputTokens ?? 0); data.Usage.OutputTokens ?? 0);
} }
var body = JsonConvert.SerializeObject(data, new JsonSerializerSettings await response.WriteAsJsonAsync(data, cancellationToken);
{
ContractResolver = new CamelCasePropertyNamesContractResolver()
});
await writer.WriteLineAsync(body);
await writer.FlushAsync(cancellationToken);
} }
/// <summary> /// <summary>