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

more progress on fluent service call

Browse Source
This commit is contained in:
Adam Wilson
2025-07-09 21:56:44 -06:00
parent 5470554d28
commit b4b2d792fc
12 changed files with 214 additions and 218 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/conftest.py
+5
View File
@@ -20,6 +20,7 @@ from src.text_generation.adapters.prompt_template_repository import PromptTempla
from src.text_generation.adapters.text_generation_foundation_model import TextGenerationFoundationModel
from src.text_generation.common.constants import Constants
from src.text_generation.services.guardrails.generated_text_guardrail_service import GeneratedTextGuardrailService
from src.text_generation.services.guidelines.generative_ai_security_guidelines_service import GenerativeAiSecurityGuidelinesService
from src.text_generation.services.guidelines.rag_guidelines_service import RetrievalAugmentedGenerationGuidelinesService
from src.text_generation.services.nlp.prompt_template_service import PromptTemplateService
from src.text_generation.services.nlp.retrieval_augmented_generation_completion_service import RetrievalAugmentedGenerationCompletionService
@@ -99,6 +100,10 @@ def prompt_template_service(prompt_template_repository):
def rag_guidelines_service(embedding_model):
return RetrievalAugmentedGenerationGuidelinesService(embedding_model)
@pytest.fixture(scope="session")
def generative_ai_security_guidelines_service(prompt_template_service):
return GenerativeAiSecurityGuidelinesService(prompt_template_service)
@pytest.fixture(scope="session")
def response_processing_service():
return ResponseProcessingService()