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

update DI container

Browse Source
This commit is contained in:
Adam Wilson
2025-06-24 14:16:57 -06:00
parent f71a094716
commit f74f8274af
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/text_generation/dependency_injection_container.py
+9 -3
View File
@@ -21,8 +21,7 @@ class DependencyInjectionContainer(containers.DeclarativeContainer):
)
language_model = providers.Singleton(
LanguageModel,
logging_service=logging_service
LanguageModel
)
embedding_model = providers.Singleton(
@@ -36,9 +35,16 @@ class DependencyInjectionContainer(containers.DeclarativeContainer):
# add / implement guidelines svc
# TODO placeholder; replace with responses from garak runs
comparison_texts = [
'something here',
'another example'
]
generated_text_guardrail_service = providers.Factory(
GeneratedTextGuardrailService,
embedding_model=embedding_model
embedding_model=embedding_model,
comparison_texts=comparison_texts
)
text_generation_response_service = providers.Factory(