请提供需要提交的变更内容或简要说明(例如:做了什么改动、涉及哪些模块)。
我将按你给定的规范生成对应的提交标题和说明。
This commit is contained in:
@@ -0,0 +1,143 @@
|
||||
using System.ComponentModel;
|
||||
using System.Text;
|
||||
using System.Text.Json;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Volo.Abp.DependencyInjection;
|
||||
using Yi.Framework.AiHub.Domain.Shared.Attributes;
|
||||
|
||||
namespace Yi.Framework.AiHub.Domain.Mcp;
|
||||
|
||||
[YiAgentTool]
|
||||
public class HttpRequestTool : ISingletonDependency
|
||||
{
|
||||
private readonly IHttpClientFactory _httpClientFactory;
|
||||
private readonly ILogger<HttpRequestTool> _logger;
|
||||
|
||||
public HttpRequestTool(
|
||||
IHttpClientFactory httpClientFactory,
|
||||
ILogger<HttpRequestTool> logger)
|
||||
{
|
||||
_httpClientFactory = httpClientFactory;
|
||||
_logger = logger;
|
||||
}
|
||||
|
||||
[YiAgentTool("HTTP请求"), DisplayName("HttpRequest"),
|
||||
Description("发送HTTP请求,支持GET/POST/PUT/DELETE等方法,获取指定URL的响应内容")]
|
||||
public async Task<string> HttpRequest(
|
||||
[Description("请求的URL地址")] string url,
|
||||
[Description("HTTP方法:GET、POST、PUT、DELETE等")] string method = "GET",
|
||||
[Description("请求体内容(JSON字符串),POST/PUT时使用")] string? body = null,
|
||||
[Description("请求头,格式:key1:value1,key2:value2")] string? headers = null)
|
||||
{
|
||||
if (string.IsNullOrWhiteSpace(url))
|
||||
{
|
||||
return "URL不能为空";
|
||||
}
|
||||
|
||||
if (string.IsNullOrWhiteSpace(method))
|
||||
{
|
||||
method = "GET";
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
var client = _httpClientFactory.CreateClient();
|
||||
var request = new HttpRequestMessage(new HttpMethod(method.ToUpper()), url);
|
||||
|
||||
// 添加请求体
|
||||
if (!string.IsNullOrWhiteSpace(body))
|
||||
{
|
||||
request.Content = new StringContent(body, Encoding.UTF8, "application/json");
|
||||
}
|
||||
|
||||
// 添加自定义请求头
|
||||
if (!string.IsNullOrWhiteSpace(headers))
|
||||
{
|
||||
AddHeaders(request, headers);
|
||||
}
|
||||
|
||||
var response = await client.SendAsync(request);
|
||||
return await FormatResponse(response);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
_logger.LogError(ex, "HTTP {Method}请求失败: {Url}", method, url);
|
||||
return $"请求失败: {ex.Message}";
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 添加请求头
|
||||
/// </summary>
|
||||
private void AddHeaders(HttpRequestMessage request, string headers)
|
||||
{
|
||||
var headerPairs = headers.Split(',', StringSplitOptions.RemoveEmptyEntries | StringSplitOptions.TrimEntries);
|
||||
foreach (var pair in headerPairs)
|
||||
{
|
||||
var parts = pair.Split(':', 2, StringSplitOptions.RemoveEmptyEntries | StringSplitOptions.TrimEntries);
|
||||
if (parts.Length == 2)
|
||||
{
|
||||
request.Headers.TryAddWithoutValidation(parts[0], parts[1]);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 格式化响应结果
|
||||
/// </summary>
|
||||
private async Task<string> FormatResponse(HttpResponseMessage response)
|
||||
{
|
||||
var sb = new StringBuilder();
|
||||
sb.AppendLine($"状态码: {(int)response.StatusCode} {response.StatusCode}");
|
||||
sb.AppendLine($"Content-Type: {response.Content.Headers.ContentType?.ToString() ?? "未知"}");
|
||||
sb.AppendLine();
|
||||
|
||||
var content = await response.Content.ReadAsStringAsync();
|
||||
if (string.IsNullOrWhiteSpace(content))
|
||||
{
|
||||
sb.AppendLine("响应内容为空");
|
||||
}
|
||||
else
|
||||
{
|
||||
// 尝试格式化JSON
|
||||
if (IsJsonContentType(response.Content.Headers.ContentType?.MediaType))
|
||||
{
|
||||
try
|
||||
{
|
||||
var jsonDoc = JsonDocument.Parse(content);
|
||||
sb.AppendLine("响应内容(JSON格式化):");
|
||||
sb.AppendLine(JsonSerializer.Serialize(jsonDoc, new JsonSerializerOptions
|
||||
{
|
||||
WriteIndented = true
|
||||
}));
|
||||
}
|
||||
catch
|
||||
{
|
||||
sb.AppendLine("响应内容:");
|
||||
sb.AppendLine(content);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
sb.AppendLine("响应内容:");
|
||||
sb.AppendLine(content);
|
||||
}
|
||||
}
|
||||
|
||||
return sb.ToString();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 判断是否为JSON内容类型
|
||||
/// </summary>
|
||||
private bool IsJsonContentType(string? contentType)
|
||||
{
|
||||
if (string.IsNullOrWhiteSpace(contentType))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
return contentType.Contains("application/json", StringComparison.OrdinalIgnoreCase) ||
|
||||
contentType.Contains("text/json", StringComparison.OrdinalIgnoreCase);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user