mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-09-27 02:13:36 +00:00
19 lines
803 B
C#
19 lines
803 B
C#
using AIStudio.Tools.PluginSystem;
|
|
|
|
namespace AIStudio.Tools.Security;
|
|
|
|
public static class PromptInjectionSourceKindExtensions
|
|
{
|
|
private static string TB(string fallbackEN) => I18N.I.T(fallbackEN, typeof(PromptInjectionSourceKindExtensions).Namespace, nameof(PromptInjectionSourceKindExtensions));
|
|
|
|
public static string GetDisplayName(this PromptInjectionSourceKind kind) => kind switch
|
|
{
|
|
PromptInjectionSourceKind.WEB_CONTENT => TB("Web content"),
|
|
PromptInjectionSourceKind.FILE_CONTENT => TB("File content"),
|
|
PromptInjectionSourceKind.CHAT_ATTACHMENT => TB("Chat attachment"),
|
|
PromptInjectionSourceKind.RETRIEVAL_CONTEXT => TB("Retrieved context"),
|
|
PromptInjectionSourceKind.MAIL_CONTENT => TB("Outlook Mail"),
|
|
_ => TB("Unknown"),
|
|
};
|
|
}
|