mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-08-14 23:30:22 +02:00
LLM config output
This commit is contained in:
+11
-1
@@ -124,16 +124,22 @@ def rag_config_builder(
|
||||
prompt_injection_example_repository=prompt_injection_example_repository
|
||||
)
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def llm_configuration_introspection_service():
|
||||
return LLMConfigurationIntrospectionService()
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def rag_context_guidelines(
|
||||
foundation_model,
|
||||
response_processing_service,
|
||||
prompt_template_service,
|
||||
llm_configuration_introspection_service,
|
||||
rag_config_builder):
|
||||
return RagContextSecurityGuidelinesService(
|
||||
foundation_model=foundation_model,
|
||||
response_processing_service=response_processing_service,
|
||||
prompt_template_service=prompt_template_service,
|
||||
llm_configuration_introspection_service=llm_configuration_introspection_service,
|
||||
config_builder=rag_config_builder
|
||||
)
|
||||
|
||||
@@ -141,10 +147,12 @@ def rag_context_guidelines(
|
||||
def chain_of_thought_guidelines(
|
||||
foundation_model,
|
||||
response_processing_service,
|
||||
llm_configuration_introspection_service,
|
||||
prompt_template_service):
|
||||
return ChainOfThoughtSecurityGuidelinesService(
|
||||
foundation_model=foundation_model,
|
||||
response_processing_service=response_processing_service,
|
||||
response_processing_service=response_processing_service,
|
||||
llm_configuration_introspection_service=llm_configuration_introspection_service,
|
||||
prompt_template_service=prompt_template_service
|
||||
)
|
||||
|
||||
@@ -153,11 +161,13 @@ def rag_plus_cot_guidelines(
|
||||
foundation_model,
|
||||
response_processing_service,
|
||||
prompt_template_service,
|
||||
llm_configuration_introspection_service,
|
||||
rag_config_builder):
|
||||
return RagPlusCotSecurityGuidelinesService(
|
||||
foundation_model=foundation_model,
|
||||
response_processing_service=response_processing_service,
|
||||
prompt_template_service=prompt_template_service,
|
||||
llm_configuration_introspection_service=llm_configuration_introspection_service,
|
||||
config_builder=rag_config_builder
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user