updates for RAG + CoT tests

This commit is contained in:
Adam Wilson committed 2025-07-25 18:11:49 -06:00
1 parent 72785c6420
commit 741629908c
10 files changed
+6603 -61

No files matched your search

@@ -1,4 +1,4 @@
from typing import List
from typing import Any, List
from src.text_generation.domain.abstract_guidelines_processed_completion import AbstractGuidelinesProcessedCompletion
from src.text_generation.domain.abstract_text_generation_completion_result import AbstractTextGenerationCompletionResult
@@ -8,11 +8,13 @@ class GuidelinesResult(
def __init__(
self,
completion_text: str,
full_prompt: dict[str, Any],
llm_config: dict,
cosine_similarity_score: float = 0.0,
cosine_similarity_risk_threshold: float = 0.0):
self.completion_text = completion_text
self.full_prompt = full_prompt
self.llm_config = llm_config
self.cosine_similarity_score = cosine_similarity_score
self.cosine_similarity_risk_threshold = cosine_similarity_risk_threshold