CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-03-19 00:33:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

more for templates

Browse Source
This commit is contained in:
Adam Wilson
2025-07-16 21:07:37 -06:00
parent 1dba565236
commit f3dd8e9208
4 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/text_generation/common/constants.py
View File

@@ -3,6 +3,7 @@ class Constants:
PHI_3_END_TOKEN = "<|end|>"
PHI_3_SYSTEM_START_TOKEN = "<|system|>"
PHI_3_USER_START_TOKEN = "<|user|>"
INPUT_VARIABLE_TOKEN = "input"
class PromptTemplateIds:
PHI_3_MINI_4K_INSTRUCT_BASIC = "phi-3-mini-4k-instruct-basic"

2
src/text_generation/services/guidelines/chain_of_thought_security_guidelines_service.py
View File

@@ -26,7 +26,7 @@ class ChainOfThoughtSecurityGuidelinesService(
def _create_chain(self, prompt_template: PromptTemplate):
return (
{ "question": RunnablePassthrough() }
{ f"{self.constants.INPUT_VARIABLE_TOKEN}": RunnablePassthrough() }
| prompt_template
| self.foundation_model_pipeline
| StrOutputParser()

2
src/text_generation/services/guidelines/rag_context_security_guidelines_service.py
View File

@@ -26,7 +26,7 @@ class RetrievalAugmentedGenerationContextSecurityGuidelinesService(
def _create_chain(self, prompt_template: PromptTemplate):
return (
{ "question": RunnablePassthrough() }
{ f"{self.constants.INPUT_VARIABLE_TOKEN}": RunnablePassthrough() }
| prompt_template
| self.foundation_model_pipeline
| StrOutputParser()