CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-07-06 21:17:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

remove additional reflexion support

Browse Source
This commit is contained in:
Adam Wilson
2025-08-16 19:26:12 -06:00
parent 7a510f03fa
commit d5beccb774
2 changed files with 1 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/text_generation/common/constants.py
+1 -4
View File
@@ -23,18 +23,15 @@ class Constants:
PHI_3_MINI_4K_INSTRUCT__02_ZERO_SHOT_CHAIN_OF_THOUGHT = "phi-3-mini-4k-instruct.02-zero-shot-cot"
PHI_3_MINI_4K_INSTRUCT__03_FEW_SHOT_EXAMPLES = "phi-3-mini-4k-instruct.03-few-shot"
PHI_3_MINI_4K_INSTRUCT__04_FEW_SHOT_RAG_PLUS_COT = "phi-3-mini-4k-instruct.04-few-shot-rag-plus-cot"
PHI_3_MINI_4K_INSTRUCT__05_REFLEXION = "phi-3-mini-4k-instruct.05-reflexion"
# OpenELM templates: apple/OpenELM-3B-Instruct
OPENELM_3B_INSTRUCT__01_BASIC = "openelm-3b-instruct.01-basic"
OPENELM_3B_INSTRUCT__02_ZERO_SHOT_CHAIN_OF_THOUGHT = "openelm-3b-instruct.02-zero-shot-cot"
OPENELM_3B_INSTRUCT__03_FEW_SHOT_EXAMPLES = "openelm-3b-instruct.03-few-shot"
OPENELM_3B_INSTRUCT__04_FEW_SHOT_RAG_PLUS_COT = "openelm-3b-instruct.04-few-shot-rag-plus-cot"
OPENELM_3B_INSTRUCT__05_REFLEXION = "openelm-3b-instruct.05-reflexion"
# meta-llama/Llama-3.2-3B-Instruct templates
LLAMA_1_1B_CHAT__01_BASIC = "llama-3.2-3b-instruct.01-basic"
LLAMA_1_1B_CHAT__02_ZERO_SHOT_CHAIN_OF_THOUGHT = "llama-3.2-3b-instruct.02-zero-shot-cot"
LLAMA_1_1B_CHAT__03_FEW_SHOT_EXAMPLES = "llama-3.2-3b-instruct.03-few-shot"
LLAMA_1_1B_CHAT__04_FEW_SHOT_RAG_PLUS_COT = "llama-3.2-3b-instruct.04-few-shot-rag-plus-cot"
LLAMA_1_1B_CHAT__05_REFLEXION = "llama-3.2-3b-instruct.05-reflexion"
LLAMA_1_1B_CHAT__04_FEW_SHOT_RAG_PLUS_COT = "llama-3.2-3b-instruct.04-few-shot-rag-plus-cot"
src/text_generation/services/nlp/abstract_text_generation_completion_service.py
-9
View File
@@ -20,11 +20,6 @@ class AbstractTextGenerationCompletionService(abc.ABC):
"""Enable RAG context security guidelines"""
raise NotImplementedError
@abc.abstractmethod
def with_reflexion_guardrails(self) -> 'AbstractTextGenerationCompletionService':
"""Apply security guardrails using the reflexion technique"""
raise NotImplementedError
@abc.abstractmethod
def is_chain_of_thought_enabled(self) -> bool:
raise NotImplementedError
@@ -33,10 +28,6 @@ class AbstractTextGenerationCompletionService(abc.ABC):
def is_rag_context_enabled(self) -> bool:
raise NotImplementedError
@abc.abstractmethod
def is_reflexion_enabled(self) -> bool:
raise NotImplementedError
@abc.abstractmethod
def invoke(self, user_prompt: str) -> AbstractTextGenerationCompletionResult:
raise NotImplementedError