Skip to content

Commit

Permalink
fix: nit param
Browse files Browse the repository at this point in the history
  • Loading branch information
ishaansehgal99 committed Jan 12, 2025
1 parent 8e6acf9 commit 54c1165
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion presets/ragengine/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def validate_params(cls, values: Dict[str, Any]) -> Dict[str, Any]:
# Validate LLM parameters
if "temperature" in llm_params and not (0.0 <= llm_params["temperature"] <= 1.0):
raise ValueError("Temperature must be between 0.0 and 1.0.")

# TODO: More LLM Param Validations here
# Validate rerank parameters
top_k = values.get("top_k")
if "top_n" in rerank_params and rerank_params["top_n"] > top_k:
Expand Down

0 comments on commit 54c1165

Please sign in to comment.