mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-09-27 00:33:38 +00:00
18 lines
824 B
C#
18 lines
824 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.DATA_SOURCE_DESCRIPTION => TB("Data source description"),
|
|
_ => TB("Unknown"),
|
|
};
|
|
} |