CalvinBackup/llmsecops-research
1
0 Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-10-04 15:26:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

more for templates

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

No files matched your search

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()