CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-06-07 23:43:56 +02: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
src/text_generation/services/guidelines/chain_of_thought_security_guidelines_service.py
+1 -1
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()
src/text_generation/services/guidelines/rag_context_security_guidelines_service.py
+1 -1
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()