defaults for each contextvar (#1586)

* defaults for each contextvar * one more * update api doc * relock * lock and api dump --------- Co-authored-by: Josh Reini <[email protected]>
truera · Oct 25, 2024 · 241e59a · 241e59a
1 parent bfb8ad0
commit 241e59a
Show file tree

Hide file tree

Showing 20 changed files with 6,316 additions and 6,082 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/src/apps/langchain/poetry.lock b/src/apps/langchain/poetry.lock
diff --git a/src/apps/llamaindex/poetry.lock b/src/apps/llamaindex/poetry.lock
diff --git a/src/apps/nemo/poetry.lock b/src/apps/nemo/poetry.lock
diff --git a/src/benchmark/poetry.lock b/src/benchmark/poetry.lock
diff --git a/src/core/poetry.lock b/src/core/poetry.lock
diff --git a/src/core/trulens/core/app.py b/src/core/trulens/core/app.py
@@ -437,7 +437,7 @@ def __init__(
             kwargs["connector"] = connector
         kwargs["feedbacks"] = feedbacks
         kwargs["recording_contexts"] = contextvars.ContextVar(
-            "recording_contexts"
+            "recording_contexts", default=None
         )
 
         super().__init__(**kwargs)

diff --git a/src/core/trulens/core/feedback/endpoint.py b/src/core/trulens/core/feedback/endpoint.py
@@ -208,7 +208,7 @@ class EndpointSetup:
     """Name of variable that stores the callback noted above."""
 
     _context_endpoints: ClassVar[contextvars.ContextVar] = (
-        contextvars.ContextVar("endpoints")
+        contextvars.ContextVar("endpoints", default={})
     )
     _context_endpoints.set({})
 

diff --git a/src/core/trulens/core/instruments.py b/src/core/trulens/core/instruments.py
@@ -62,11 +62,13 @@ class WithInstrumentCallbacks:
     Needs to be mixed into [App][trulens.core.app.App].
     """
 
-    _context_contexts = contextvars.ContextVar("context_contexts")
+    _context_contexts = contextvars.ContextVar(
+        "context_contexts", default=set()
+    )
     """ContextVars for storing collections of RecordingContext ."""
     _context_contexts.set(set())
 
-    _stack_contexts = contextvars.ContextVar("stack_contexts")
+    _stack_contexts = contextvars.ContextVar("stack_contexts", default={})
     """ContextVars for storing call stacks."""
     _stack_contexts.set({})
 

diff --git a/src/core/trulens/core/utils/serial.py b/src/core/trulens/core/utils/serial.py
@@ -547,7 +547,7 @@ class SerialModel(pydantic.BaseModel):
     """
 
     formatted_objects: ClassVar[ContextVar[Set[int]]] = ContextVar(
-        "formatted_objects"
+        "formatted_objects", default=set()
     )
 
     def __rich_repr__(self) -> rich.repr.Result:

diff --git a/src/dashboard/poetry.lock b/src/dashboard/poetry.lock
diff --git a/src/feedback/poetry.lock b/src/feedback/poetry.lock
diff --git a/src/providers/bedrock/poetry.lock b/src/providers/bedrock/poetry.lock
diff --git a/src/providers/cortex/poetry.lock b/src/providers/cortex/poetry.lock
diff --git a/src/providers/huggingface/poetry.lock b/src/providers/huggingface/poetry.lock
diff --git a/src/providers/langchain/poetry.lock b/src/providers/langchain/poetry.lock
diff --git a/src/providers/litellm/poetry.lock b/src/providers/litellm/poetry.lock
diff --git a/src/providers/openai/poetry.lock b/src/providers/openai/poetry.lock
diff --git a/src/trulens_eval/poetry.lock b/src/trulens_eval/poetry.lock
diff --git a/tests/unit/static/golden/api.trulens.3.11.yaml b/tests/unit/static/golden/api.trulens.3.11.yaml
@@ -4,7 +4,7 @@ trulens:
   lows: {}
 trulens.benchmark:
   __class__: builtins.module
-  __version__: 1.0.1
+  __version__: 1.1.0
   highs: {}
   lows:
     __version__: builtins.str
@@ -231,6 +231,7 @@ trulens.core.database.base.DB:
     get_app: builtins.function
     get_apps: builtins.function
     get_datasets: builtins.function
+    get_db_revision: builtins.function
     get_feedback: builtins.function
     get_feedback_count_by_status: builtins.function
     get_feedback_defs: builtins.function
@@ -1628,7 +1629,7 @@ trulens.core.utils.trulens.Other:
   attributes: {}
 trulens.dashboard:
   __class__: builtins.module
-  __version__: 1.0.2a0
+  __version__: 1.1.0
   highs:
     run_dashboard: builtins.function
     stop_dashboard: builtins.function
@@ -1902,7 +1903,7 @@ trulens.dashboard.ux.styles.ResultCategoryType:
     value: enum.property
 trulens.feedback:
   __class__: builtins.module
-  __version__: 1.0.1
+  __version__: 1.1.0
   highs:
     Embeddings: pydantic._internal._model_construction.ModelMetaclass
     GroundTruthAggregator: pydantic._internal._model_construction.ModelMetaclass