LLMEmbedder.cs

Go to the documentation of this file.

 
using System.Threading.Tasks;
using UnityEngine;
 
namespace LLMUnity
{
    [DefaultExecutionOrder(-2)]
    public class LLMEmbedder : LLMClient
    {
        protected override async Task SetLLM(LLM llmSet)
        {
            await base.SetLLM(llmSet);
            if (llmSet != null && !llmSet.embeddingsOnly)
            {
                LLMUnitySetup.LogWarning($"The LLM {llmSet.name} set for LLMEmbeddings {gameObject.name} is not an embeddings-only model, accuracy may be sub-optimal");
            }
        }
 
        public override bool IsAutoAssignableLLM(LLM llmSet)
        {
            return llmSet.embeddingsOnly;
        }
    }
}