judgeval 0.4.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

judgeval/__init__.py

@@ -2,6 +2,7 @@
 from judgeval.clients import client, together_client
 from judgeval.judgment_client import JudgmentClient
 from judgeval.version_check import check_latest_version
+from judgeval.local_eval_queue import LocalEvaluationQueue
 
 check_latest_version()
 
@@ -10,4 +11,5 @@ __all__ = [
     "client",
     "together_client",
     "JudgmentClient",
+    "LocalEvaluationQueue",
 ]

judgeval/clients.py

@@ -2,7 +2,6 @@ import os
 from dotenv import load_dotenv
 from openai import OpenAI
 from typing import Optional
-from together import Together, AsyncTogether
 
 PATH_TO_DOTENV = os.path.join(os.path.dirname(__file__), ".env")
 load_dotenv(dotenv_path=PATH_TO_DOTENV)
@@ -28,6 +27,8 @@ async_together_client: Optional["AsyncTogether"] = None
 together_api_key = os.getenv("TOGETHERAI_API_KEY") or os.getenv("TOGETHER_API_KEY")
 if together_api_key:
     try:
+        from together import Together, AsyncTogether
+
         together_client = Together(api_key=together_api_key)
         async_together_client = AsyncTogether(api_key=together_api_key)
     except Exception:

judgeval/common/api/api.py

@@ -53,8 +53,7 @@ from judgeval.common.api.constants import (
     CheckExampleKeysPayload,
 )
 from judgeval.utils.requests import requests
-
-import orjson
+from judgeval.common.api.json_encoder import json_encoder
 
 
 class JudgmentAPIException(exceptions.HTTPError):
@@ -111,7 +110,7 @@ class JudgmentApiClient:
             r = requests.request(
                 method,
                 url,
-                data=self._serialize(payload),
+                json=json_encoder(payload),
                 headers=self._headers(),
                 **self._request_kwargs(),
             )
@@ -368,16 +367,3 @@ class JudgmentApiClient:
             "verify": True,
             "timeout": 30,
         }
-
-    def _serialize(self, data: Any) -> str:
-        def fallback_encoder(obj):
-            try:
-                return repr(obj)
-            except Exception:
-                try:
-                    return str(obj)
-                except Exception as e:
-                    return f"<Unserializable object of type {type(obj).__name__}: {e}>"
-
-        # orjson returns bytes, so we need to decode to str
-        return orjson.dumps(data, default=fallback_encoder).decode("utf-8")

judgeval/common/api/json_encoder.py

@@ -0,0 +1,242 @@
+"""
+
+This is a modified version of https://docs.powertools.aws.dev/lambda/python/2.35.1/api/event_handler/openapi/encoders.html
+
+"""
+
+import dataclasses
+import datetime
+from collections import defaultdict, deque
+from decimal import Decimal
+from enum import Enum
+from pathlib import Path, PurePath
+from re import Pattern
+from types import GeneratorType
+from typing import Any, Callable, Dict, List, Literal, Optional, Tuple, Type, Union
+from uuid import UUID
+
+from pydantic import BaseModel
+from pydantic.types import SecretBytes, SecretStr
+
+
+"""
+This module contains the encoders used by jsonable_encoder to convert Python objects to JSON serializable data types.
+"""
+
+
+def _model_dump(
+    model: BaseModel, mode: Literal["json", "python"] = "json", **kwargs: Any
+) -> Any:
+    return model.model_dump(mode=mode, **kwargs)
+
+
+def json_encoder(
+    obj: Any,
+    custom_serializer: Optional[Callable[[Any], str]] = None,
+) -> Any:
+    """
+    JSON encodes an arbitrary Python object into JSON serializable data types.
+
+    This is a modified version of fastapi.encoders.jsonable_encoder that supports
+    encoding of pydantic.BaseModel objects.
+
+    Parameters
+    ----------
+    obj : Any
+        The object to encode
+    custom_serializer : Callable, optional
+        A custom serializer to use for encoding the object, when everything else fails.
+
+    Returns
+    -------
+    Any
+        The JSON serializable data types
+    """
+    # Pydantic models
+    if isinstance(obj, BaseModel):
+        return _dump_base_model(
+            obj=obj,
+        )
+
+    # Dataclasses
+    if dataclasses.is_dataclass(obj):
+        obj_dict = dataclasses.asdict(obj)
+        return json_encoder(
+            obj_dict,
+        )
+
+    # Enums
+    if isinstance(obj, Enum):
+        return obj.value
+
+    # Paths
+    if isinstance(obj, PurePath):
+        return str(obj)
+
+    # Scalars
+    if isinstance(obj, (str, int, float, type(None))):
+        return obj
+
+    # Dictionaries
+    if isinstance(obj, dict):
+        return _dump_dict(
+            obj=obj,
+        )
+
+    # Sequences
+    if isinstance(obj, (list, set, frozenset, GeneratorType, tuple, deque)):
+        return _dump_sequence(
+            obj=obj,
+        )
+
+    # Other types
+    if type(obj) in ENCODERS_BY_TYPE:
+        return ENCODERS_BY_TYPE[type(obj)](obj)
+
+    for encoder, classes_tuple in encoders_by_class_tuples.items():
+        if isinstance(obj, classes_tuple):
+            return encoder(obj)
+
+    # Use custom serializer if present
+    if custom_serializer:
+        return custom_serializer(obj)
+
+    # Default
+    return _dump_other(
+        obj=obj,
+    )
+
+
+def _dump_base_model(
+    *,
+    obj: Any,
+):
+    """
+    Dump a BaseModel object to a dict, using the same parameters as jsonable_encoder
+    """
+    obj_dict = _model_dump(
+        obj,
+        mode="json",
+    )
+    if "__root__" in obj_dict:
+        obj_dict = obj_dict["__root__"]
+
+    return json_encoder(
+        obj_dict,
+    )
+
+
+def _dump_dict(
+    *,
+    obj: Any,
+) -> Dict[str, Any]:
+    """
+    Dump a dict to a dict, using the same parameters as jsonable_encoder
+    """
+    encoded_dict = {}
+    allowed_keys = set(obj.keys())
+    for key, value in obj.items():
+        if key in allowed_keys:
+            encoded_key = json_encoder(
+                key,
+            )
+            encoded_value = json_encoder(
+                value,
+            )
+            encoded_dict[encoded_key] = encoded_value
+    return encoded_dict
+
+
+def _dump_sequence(
+    *,
+    obj: Any,
+) -> List[Any]:
+    """
+    Dump a sequence to a list, using the same parameters as jsonable_encoder
+    """
+    encoded_list = []
+    for item in obj:
+        encoded_list.append(
+            json_encoder(
+                item,
+            ),
+        )
+    return encoded_list
+
+
+def _dump_other(
+    *,
+    obj: Any,
+) -> Any:
+    """
+    Dump an object to a hashable object, using the same parameters as jsonable_encoder
+    """
+    try:
+        data = dict(obj)
+    except Exception:
+        return repr(obj)
+
+    return json_encoder(
+        data,
+    )
+
+
+def iso_format(o: Union[datetime.date, datetime.time]) -> str:
+    """
+    ISO format for date and time
+    """
+    return o.isoformat()
+
+
+def decimal_encoder(dec_value: Decimal) -> Union[int, float]:
+    """
+    Encodes a Decimal as int of there's no exponent, otherwise float
+
+    This is useful when we use ConstrainedDecimal to represent Numeric(x,0)
+    where an integer (but not int typed) is used. Encoding this as a float
+    results in failed round-tripping between encode and parse.
+
+    >>> decimal_encoder(Decimal("1.0"))
+    1.0
+
+    >>> decimal_encoder(Decimal("1"))
+    1
+    """
+    if dec_value.as_tuple().exponent >= 0:  # type: ignore[operator]
+        return int(dec_value)
+    else:
+        return float(dec_value)
+
+
+ENCODERS_BY_TYPE: Dict[Type[Any], Callable[[Any], Any]] = {
+    bytes: lambda o: o.decode(),
+    datetime.date: iso_format,
+    datetime.datetime: iso_format,
+    datetime.time: iso_format,
+    datetime.timedelta: lambda td: td.total_seconds(),
+    Decimal: decimal_encoder,
+    Enum: lambda o: o.value,
+    frozenset: list,
+    deque: list,
+    GeneratorType: list,
+    Path: str,
+    Pattern: lambda o: o.pattern,
+    SecretBytes: str,
+    SecretStr: str,
+    set: list,
+    UUID: str,
+}
+
+
+# Generates a mapping of encoders to a tuple of classes that they can encode
+def generate_encoders_by_class_tuples(
+    type_encoder_map: Dict[Any, Callable[[Any], Any]],
+) -> Dict[Callable[[Any], Any], Tuple[Any, ...]]:
+    encoders: Dict[Callable[[Any], Any], Tuple[Any, ...]] = defaultdict(tuple)
+    for type_, encoder in type_encoder_map.items():
+        encoders[encoder] += (type_,)
+    return encoders
+
+
+# Mapping of encoders to a tuple of classes that they can encode
+encoders_by_class_tuples = generate_encoders_by_class_tuples(ENCODERS_BY_TYPE)