fix typing on opik.py

litellm/integrations/opik/utils.py

@@ -1,11 +1,13 @@
+import configparser
 import os
 import time
-from typing import Optional, Final, Dict, List
-import configparser
+from typing import Dict, Final, List, Optional
+
 from litellm.types.utils import ModelResponse
 
 CONFIG_FILE_PATH_DEFAULT: Final[str] = "~/.opik.config"
 
+
 def create_uuid7():
     ns = time.time_ns()
     last = [0, 0, 0, 0]
@@ -36,6 +38,7 @@ def create_uuid7():
     rand = os.urandom(6)
     return f"{t1:>08x}-{t2:>04x}-{t3:>04x}-{t4:>04x}-{rand.hex()}"
 
+
 def _read_opik_config_file() -> Dict[str, str]:
     config_path = os.path.expanduser(CONFIG_FILE_PATH_DEFAULT)
 
@@ -51,15 +54,15 @@ def _read_opik_config_file() -> Dict[str, str]:
 
     return {}
 
-def _get_env_variable(key: str) -> str:
+
+def _get_env_variable(key: str) -> Optional[str]:
     env_prefix = "opik_"
     return os.getenv((env_prefix + key).upper(), None)
 
+
 def get_opik_config_variable(
-        key: str,
-        user_value: Optional[str] = None,
-        default_value: Optional[str] = None
-    ) -> str:
+    key: str, user_value: Optional[str] = None, default_value: Optional[str] = None
+) -> Optional[str]:
     """
     Get the configuration value of a variable, order priority is:
     1. user provided value
@@ -85,6 +88,7 @@ def get_opik_config_variable(
     # Return default value if it is not None
     return default_value
 
+
 def create_usage_object(usage):
     usage_dict = {}
 
@@ -96,10 +100,12 @@ def create_usage_object(usage):
         usage_dict["total_tokens"] = usage.total_tokens
     return usage_dict
 
+
 def _remove_nulls(x):
     x_ = {k: v for k, v in x.items() if v is not None}
     return x_
 
+
 def get_traces_and_spans_from_payload(payload: List):
     traces = [_remove_nulls(x) for x in payload if "type" not in x]
     spans = [_remove_nulls(x) for x in payload if "type" in x]