86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
from typing import Optional, Dict, Any
|
|
from models import EvaluationResult
|
|
from constants import MAX_SCORE
|
|
|
|
|
|
def format_evaluation_for_generator(evaluation: Optional[EvaluationResult]) -> str:
|
|
"""
|
|
Format an evaluation result as text for the generator module.
|
|
|
|
Args:
|
|
evaluation: The evaluation result to format
|
|
|
|
Returns:
|
|
Formatted string with category-by-category reasoning and scores
|
|
"""
|
|
if not evaluation or not evaluation.evaluations:
|
|
return ""
|
|
|
|
eval_lines = []
|
|
for eval in evaluation.evaluations:
|
|
eval_lines.append(f"{eval.category} (Score: {eval.score}/{MAX_SCORE}): {eval.reasoning}")
|
|
|
|
return "\n".join(eval_lines)
|
|
|
|
|
|
def build_settings_dict(
|
|
selected_model: str,
|
|
iterations: int,
|
|
patience: int,
|
|
use_cache: bool
|
|
) -> Dict[str, Any]:
|
|
"""
|
|
Build a settings dictionary for saving.
|
|
|
|
Args:
|
|
selected_model: The selected model name
|
|
iterations: Number of optimization iterations
|
|
patience: Patience threshold for early stopping
|
|
use_cache: Whether to use DSPy cache
|
|
|
|
Returns:
|
|
Dictionary containing all settings
|
|
"""
|
|
return {
|
|
"selected_model": selected_model,
|
|
"iterations": iterations,
|
|
"patience": patience,
|
|
"use_cache": use_cache
|
|
}
|
|
|
|
|
|
def truncate_tweet(tweet: str, max_length: int, suffix: str = "...") -> str:
|
|
"""
|
|
Truncate a tweet to the maximum length with a suffix.
|
|
|
|
Args:
|
|
tweet: The tweet text to truncate
|
|
max_length: Maximum allowed length
|
|
suffix: Suffix to add when truncating (default: "...")
|
|
|
|
Returns:
|
|
Truncated tweet text
|
|
"""
|
|
tweet = tweet.strip()
|
|
if len(tweet) <= max_length:
|
|
return tweet
|
|
|
|
truncation_point = max_length - len(suffix)
|
|
return tweet[:truncation_point] + suffix
|
|
|
|
|
|
def truncate_category_display(category: str, max_length: int = 30) -> str:
|
|
"""
|
|
Truncate a category name for display purposes.
|
|
|
|
Args:
|
|
category: The category name
|
|
max_length: Maximum display length (default: 30)
|
|
|
|
Returns:
|
|
Truncated category name with "..." if needed
|
|
"""
|
|
if len(category) <= max_length:
|
|
return category
|
|
return category[:max_length] + "..."
|