mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-07-07 21:48:06 +02:00
log full completion result with semantic similarity comparison results
This commit is contained in:
@@ -53,15 +53,12 @@ def run_prompt_analysis_test(
|
||||
|
||||
TestRunLoggingService(test_id=test_id).log_results(
|
||||
id=inspect.currentframe().f_back.f_code.co_name,
|
||||
prompt=prompt,
|
||||
final_completion=completion_result.final_completion_text,
|
||||
text_generation_completion_result=completion_result,
|
||||
final_completion_text_score=result.max,
|
||||
is_rag_few_shot_enabled=text_generation_completion_service.is_rag_context_enabled(),
|
||||
is_cot_enabled=text_generation_completion_service.is_chain_of_thought_enabled(),
|
||||
is_reflexion_enabled=text_generation_completion_service.is_reflexion_enabled(),
|
||||
original_llm_config=completion_result.original_result.llm_config,
|
||||
scores=result.scores,
|
||||
mean=result.mean,
|
||||
max=result.max
|
||||
original_llm_config=completion_result.original_result.llm_config
|
||||
)
|
||||
|
||||
results.append(result)
|
||||
|
||||
Reference in New Issue
Block a user