|
3 | 3 | import os
|
4 | 4 | import types
|
5 | 5 | from typing import TYPE_CHECKING, Any
|
| 6 | +from uuid import UUID |
6 | 7 |
|
7 | 8 | from langchain_core.documents import Document
|
8 | 9 | from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage
|
@@ -60,7 +61,7 @@ def __init__(
|
60 | 61 | self.global_vars = global_vars or {}
|
61 | 62 |
|
62 | 63 | config = self._get_config(self.global_vars)
|
63 |
| - self._ready: bool = self._setup_opik(config, trace_id) if config else False |
| 64 | + self._ready: bool = bool(config and self._setup_opik(config, trace_id)) |
64 | 65 | self._distributed_headers = None
|
65 | 66 |
|
66 | 67 | @property
|
@@ -235,14 +236,14 @@ def _convert_to_opik_type(self, value):
|
235 | 236 |
|
236 | 237 | @staticmethod
|
237 | 238 | def _get_config(global_vars) -> dict:
|
238 |
| - if global_vars: |
239 |
| - host = global_vars.get("OPIK_URL_OVERRIDE", None) |
240 |
| - api_key = global_vars.get("OPIK_API_KEY", None) |
241 |
| - workspace = global_vars.get("OPIK_WORKSPACE", None) |
| 239 | + if global_vars is not None: |
| 240 | + host = global_vars.get("OPIK_URL_OVERRIDE") |
| 241 | + api_key = global_vars.get("OPIK_API_KEY") |
| 242 | + workspace = global_vars.get("OPIK_WORKSPACE") |
242 | 243 | else:
|
243 |
| - host = os.getenv("OPIK_URL_OVERRIDE", None) |
244 |
| - api_key = os.getenv("OPIK_API_KEY", None) |
245 |
| - workspace = os.getenv("OPIK_WORKSPACE", None) |
| 244 | + host = os.getenv("OPIK_URL_OVERRIDE") |
| 245 | + api_key = os.getenv("OPIK_API_KEY") |
| 246 | + workspace = os.getenv("OPIK_WORKSPACE") |
246 | 247 |
|
247 | 248 | # API Key is mandatory for Opik Cloud and URL is mandatory for Open-Source Opik Server
|
248 | 249 | if host or api_key:
|
|
0 commit comments