fix bq errors

Author: benjibc

eval_protocol/adapters/bigquery.py

@@ -4,8 +4,10 @@
 to EvaluationRow format for use in evaluation pipelines.
 """
 
+from __future__ import annotations
+
 import logging
-from typing import Any, Callable, Dict, Iterator, List, Optional, Union
+from typing import TYPE_CHECKING, Any, Callable, Dict, Iterator, List, Optional, Union
 
 from eval_protocol.models import CompletionParams, EvaluationRow, InputMetadata, Message
 
@@ -20,7 +22,18 @@
     BIGQUERY_AVAILABLE = True
 except ImportError:
     BIGQUERY_AVAILABLE = False
-    logger.warning("Google Cloud BigQuery not installed. Install with: pip install 'eval-protocol[bigquery]'")
+    # Optional dependency: avoid noisy warnings during import
+    logger.debug("Google Cloud BigQuery not installed. Optional feature disabled.")
+
+# Avoid importing BigQuery types at runtime for annotations when not installed
+if TYPE_CHECKING:
+    from google.cloud import bigquery as _bigquery_type
+    QueryParameterType = Union[
+        _bigquery_type.ScalarQueryParameter,
+        _bigquery_type.ArrayQueryParameter,
+    ]
+else:
+    QueryParameterType = Any
 
 # Type alias for transformation function
 TransformFunction = Callable[[Dict[str, Any]], Dict[str, Any]]
@@ -96,7 +109,7 @@ def __init__(
     def get_evaluation_rows(
         self,
         query: str,
-        query_params: Optional[List[Union[bigquery.ScalarQueryParameter, bigquery.ArrayQueryParameter]]] = None,
+        query_params: Optional[List[QueryParameterType]] = None,
         limit: Optional[int] = None,
         offset: int = 0,
         model_name: str = "gpt-3.5-turbo",

pyproject.toml

@@ -111,6 +111,10 @@ adapters = [
     "datasets>=2.0.0",
     "transformers>=4.0.0",
 ]
+bigquery = [
+    "google-cloud-bigquery>=3.0.0",
+    "google-auth>=2.0.0",
+]
 svgbench = [
     "selenium>=4.0.0",
 ]