From f74f8274af76156c4931524c4f30b35deda4498c Mon Sep 17 00:00:00 2001 From: Adam Wilson Date: Tue, 24 Jun 2025 14:16:57 -0600 Subject: [PATCH] update DI container --- .../dependency_injection_container.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/src/text_generation/dependency_injection_container.py b/src/text_generation/dependency_injection_container.py index 85343275a..bd5461bc0 100644 --- a/src/text_generation/dependency_injection_container.py +++ b/src/text_generation/dependency_injection_container.py @@ -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(