arthur-common 2.1.54__tar.gz → 2.1.55__tar.gz

{arthur_common-2.1.54 → arthur_common-2.1.55}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arthur-common
-Version: 2.1.54
+Version: 2.1.55
 Summary: Utility code common to Arthur platform components.
 License: MIT
 Author: Arthur

{arthur_common-2.1.54 → arthur_common-2.1.55}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "arthur-common"
-version = "2.1.54"
+version = "2.1.55"
 description = "Utility code common to Arthur platform components."
 authors = ["Arthur <engineering@arthur.ai>"]
 license = "MIT"

{arthur_common-2.1.54 → arthur_common-2.1.55}/src/arthur_common/models/shield.py

@@ -602,12 +602,11 @@ class NewMetricRequest(BaseModel):
                 # Default config when none is provided
                 config_values = {"use_llm_judge": True}
             elif isinstance(config_values, dict):
+                relevance_threshold = config_values.get("relevance_threshold")
+                use_llm_judge = config_values.get("use_llm_judge")
+
                 # Handle mutually exclusive parameters
-                if (
-                    "relevance_threshold" in config_values
-                    and "use_llm_judge" in config_values
-                    and config_values["use_llm_judge"]
-                ):
+                if relevance_threshold is not None and use_llm_judge:
                     raise HTTPException(
                         status_code=400,
                         detail="relevance_threshold and use_llm_judge=true are mutually exclusive. Set use_llm_judge=false when using relevance_threshold.",
@@ -615,16 +614,12 @@ class NewMetricRequest(BaseModel):
                     )
 
                 # If relevance_threshold is set but use_llm_judge isn't, set use_llm_judge to false
-                if (
-                    "relevance_threshold" in config_values
-                    and "use_llm_judge" not in config_values
-                ):
+                if relevance_threshold is not None and use_llm_judge is None:
                     config_values["use_llm_judge"] = False
 
                 # If neither is set, default to use_llm_judge=True
-                if (
-                    "relevance_threshold" not in config_values
-                    and "use_llm_judge" not in config_values
+                if relevance_threshold is None and (
+                        use_llm_judge is None or use_llm_judge == False
                 ):
                     config_values["use_llm_judge"] = True