last ...
This commit is contained in:
@@ -0,0 +1,145 @@
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Threading.Tasks;
|
||||
using xCommons.Extensions;
|
||||
|
||||
namespace xAiApi.Helpers
|
||||
{
|
||||
public class XOllamaHelper
|
||||
{
|
||||
/// <summary>
|
||||
/// Get Version ...
|
||||
/// </summary>
|
||||
/// <returns></returns>
|
||||
public static async Task<string> GetVersion()
|
||||
{
|
||||
//
|
||||
var cmd = "ollama";
|
||||
var args = "--version";
|
||||
var cmdResult = await XOsHelper.Execute(cmd, args);
|
||||
|
||||
//
|
||||
var result =
|
||||
cmdResult.Error.IsNullOrEmpty()
|
||||
? cmdResult.Result
|
||||
: cmdResult.Error;
|
||||
|
||||
//
|
||||
result = result.Replace("\n", "");
|
||||
|
||||
//
|
||||
return result;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Get Exists Models ...
|
||||
/// </summary>
|
||||
/// <returns></returns>
|
||||
public static async Task<IEnumerable<string>> GetModels()
|
||||
{
|
||||
//
|
||||
var cmd = "ollama";
|
||||
var args = "list";
|
||||
var cmdResult = await XOsHelper.Execute(cmd, args);
|
||||
|
||||
//
|
||||
var cmdResultText =
|
||||
cmdResult.Error.IsNullOrEmpty()
|
||||
? cmdResult.Result
|
||||
: cmdResult.Error;
|
||||
|
||||
//
|
||||
var cmdResultList = cmdResultText
|
||||
.Split("\n")
|
||||
.ToList();
|
||||
if (cmdResultList.Count > 0)
|
||||
{
|
||||
cmdResultList.RemoveAt(0);
|
||||
}
|
||||
|
||||
//
|
||||
var result = new List<string>();
|
||||
if (cmdResultList.Count > 0)
|
||||
{
|
||||
//
|
||||
cmdResultList
|
||||
.ForEach(i =>
|
||||
{
|
||||
//
|
||||
if (!i.IsNullOrEmpty())
|
||||
{
|
||||
//
|
||||
var d = i.Split(" ");
|
||||
if (d.Length > 0)
|
||||
{
|
||||
result.Add(d[0]);
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
//
|
||||
return result;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Check Specified Model Exists ...
|
||||
/// </summary>
|
||||
/// <param name="model"></param>
|
||||
/// <returns></returns>
|
||||
public static async Task<bool> HasModel(string model)
|
||||
{
|
||||
//
|
||||
var result = !model.IsNullOrEmpty();
|
||||
if (!result)
|
||||
{
|
||||
return result;
|
||||
}
|
||||
|
||||
//
|
||||
var models = await GetModels();
|
||||
result = models.HasChild();
|
||||
if (!result)
|
||||
{
|
||||
return result;
|
||||
}
|
||||
|
||||
//
|
||||
result = models
|
||||
.Any(n => n == model || n.ToNormalString().Contains(model.ToNormalString()));
|
||||
return result;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Search Available Models ...
|
||||
/// </summary>
|
||||
/// <param name="model"></param>
|
||||
/// <returns></returns>
|
||||
public static async Task<IEnumerable<string>> FindModel(string model)
|
||||
{
|
||||
//
|
||||
var result = new List<string>();
|
||||
|
||||
//
|
||||
if (model.IsNullOrEmpty())
|
||||
{
|
||||
return result;
|
||||
}
|
||||
|
||||
//
|
||||
var models = await GetModels();
|
||||
if (!models.HasChild())
|
||||
{
|
||||
return result;
|
||||
}
|
||||
|
||||
//
|
||||
result = models
|
||||
.Where(n => n == model || n.ToNormalString().Contains(model.ToNormalString()))
|
||||
.ToList();
|
||||
|
||||
//
|
||||
return result;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user