From 204093523ca9443a2407524ddbced39b967705a9 Mon Sep 17 00:00:00 2001 From: Joschka Braun Date: Tue, 9 Jul 2024 23:49:18 +0200 Subject: [PATCH] fix: target being zero --- cookbook/openai/simple_experiment_with_openai.py | 4 +++- parea/utils/trace_utils.py | 2 +- pyproject.toml | 2 +- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/cookbook/openai/simple_experiment_with_openai.py b/cookbook/openai/simple_experiment_with_openai.py index ad08535f..f649f491 100644 --- a/cookbook/openai/simple_experiment_with_openai.py +++ b/cookbook/openai/simple_experiment_with_openai.py @@ -1,3 +1,5 @@ +from typing import Union + import os from dotenv import load_dotenv @@ -22,7 +24,7 @@ def eval_func(log: Log) -> float: @trace(eval_funcs=[eval_func]) -def func(topic: str) -> dict[str, str | None]: +def func(topic: str) -> dict[str, Union[str, None]]: return { "data": ( client.chat.completions.create( diff --git a/parea/utils/trace_utils.py b/parea/utils/trace_utils.py index 23e72895..0d8af82b 100644 --- a/parea/utils/trace_utils.py +++ b/parea/utils/trace_utils.py @@ -214,7 +214,7 @@ def init_trace(func_name, _parea_target_field, args, kwargs, func) -> Tuple[str, end_user_identifier=end_user_identifier, session_id=session_id, metadata=metadata, - target=_parea_target_field or parent_target, + target=_parea_target_field if _parea_target_field is not None else parent_target, tags=tags, inputs={} if log_omit_inputs else inputs, experiment_uuid=os.environ.get(PAREA_OS_ENV_EXPERIMENT_UUID, None), diff --git a/pyproject.toml b/pyproject.toml index 5f29f901..dbc7dc9f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api" [tool.poetry] name = "parea-ai" packages = [{ include = "parea" }] -version = "0.2.185" +version = "0.2.186" description = "Parea python sdk" readme = "README.md" authors = ["joel-parea-ai "]