Files
flowframes/Code/Main/AiModels.cs

37 lines
1.0 KiB
C#

using Flowframes.Data;
using Flowframes.IO;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace Flowframes.Main
{
class AiModels
{
public static ModelCollection GetModels (AI ai)
{
string pkgPath = Path.Combine(Paths.GetPkgPath(), ai.pkgDir);
string modelsFile = Path.Combine(pkgPath, "models.json");
ModelCollection modelCollection = new ModelCollection(ai, modelsFile);
return modelCollection;
}
public static ModelCollection.ModelInfo GetModelByName(AI ai, string modelName)
{
Logger.Log($"looking for model '{modelName}'");
ModelCollection modelCollection = GetModels(ai);
foreach(ModelCollection.ModelInfo model in modelCollection.models)
{
if (model.name == modelName)
return model;
}
return null;
}
}
}