Spaces:
Sleeping
Sleeping
Update src/utils/response_processor.py
Browse files
src/utils/response_processor.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
from typing import Dict, Any, List, Optional
|
| 2 |
import time
|
| 3 |
from ..knowledge_base.grounding_truth import GroundingTruth
|
|
|
|
| 4 |
|
| 5 |
# Try to import generic responder for multi-perspective optimization
|
| 6 |
GENERIC_RESPONDER_AVAILABLE = False
|
|
@@ -42,6 +43,7 @@ class ResponseProcessor:
|
|
| 42 |
self.context_history = []
|
| 43 |
self.generic_responder = get_generic_responder() if GENERIC_RESPONDER_AVAILABLE else None
|
| 44 |
self.natural_enhancer = get_natural_enhancer() if NATURAL_ENHANCER_AVAILABLE else None
|
|
|
|
| 45 |
self.user_id = "anonymous" # Can be set per session
|
| 46 |
|
| 47 |
def process_response(self, query: str, response: str, context: Optional[str] = None,
|
|
@@ -162,6 +164,8 @@ class ResponseProcessor:
|
|
| 162 |
return f"{statement} (possibly)"
|
| 163 |
elif confidence >= 0.2:
|
| 164 |
return f"I'm uncertain, but {statement.lower()}"
|
|
|
|
|
|
|
| 165 |
else:
|
| 166 |
return f"I'm not sure about this, but {statement.lower()}"
|
| 167 |
|
|
@@ -187,6 +191,8 @@ class ResponseProcessor:
|
|
| 187 |
# Add general note if there are uncertain statements (less obtrusive)
|
| 188 |
if uncertain_statements and not verified_statements:
|
| 189 |
response_parts.insert(0, "Based on my current understanding:")
|
|
|
|
|
|
|
| 190 |
|
| 191 |
return "\n".join(response_parts).strip()
|
| 192 |
|
|
|
|
| 1 |
from typing import Dict, Any, List, Optional
|
| 2 |
import time
|
| 3 |
from ..knowledge_base.grounding_truth import GroundingTruth
|
| 4 |
+
from .response_templates import get_response_templates
|
| 5 |
|
| 6 |
# Try to import generic responder for multi-perspective optimization
|
| 7 |
GENERIC_RESPONDER_AVAILABLE = False
|
|
|
|
| 43 |
self.context_history = []
|
| 44 |
self.generic_responder = get_generic_responder() if GENERIC_RESPONDER_AVAILABLE else None
|
| 45 |
self.natural_enhancer = get_natural_enhancer() if NATURAL_ENHANCER_AVAILABLE else None
|
| 46 |
+
self.response_templates = get_response_templates()
|
| 47 |
self.user_id = "anonymous" # Can be set per session
|
| 48 |
|
| 49 |
def process_response(self, query: str, response: str, context: Optional[str] = None,
|
|
|
|
| 164 |
return f"{statement} (possibly)"
|
| 165 |
elif confidence >= 0.2:
|
| 166 |
return f"I'm uncertain, but {statement.lower()}"
|
| 167 |
+
uncertain = self.response_templates.get_uncertain_prefix()
|
| 168 |
+
return f"{uncertain}{statement.lower()}"
|
| 169 |
else:
|
| 170 |
return f"I'm not sure about this, but {statement.lower()}"
|
| 171 |
|
|
|
|
| 191 |
# Add general note if there are uncertain statements (less obtrusive)
|
| 192 |
if uncertain_statements and not verified_statements:
|
| 193 |
response_parts.insert(0, "Based on my current understanding:")
|
| 194 |
+
reflection = self.response_templates.get_reflection_prefix()
|
| 195 |
+
response_parts.insert(0, reflection)
|
| 196 |
|
| 197 |
return "\n".join(response_parts).strip()
|
| 198 |
|