add root_trace_id to update

Author: jalexanderII

parea/client.py

@@ -1,9 +1,8 @@
-from typing import Any, AsyncIterable, Callable, Dict, Iterable, List, Optional, Union
-
 import asyncio
 import logging
 import os
 import time
+from typing import Any, AsyncIterable, Callable, Dict, Iterable, List, Optional, Union
 
 import httpx
 from attrs import asdict, define, field
@@ -14,7 +13,8 @@
 from parea.cache.cache import Cache
 from parea.constants import PAREA_OS_ENV_EXPERIMENT_UUID
 from parea.experiment.datasets import create_test_cases, create_test_collection
-from parea.helpers import gen_trace_id, serialize_metadata_values, structure_trace_log_from_api, structure_trace_logs_from_api
+from parea.helpers import gen_trace_id, serialize_metadata_values, structure_trace_log_from_api, \
+    structure_trace_logs_from_api
 from parea.parea_logger import parea_logger
 from parea.schemas import EvaluationResult
 from parea.schemas.models import (
@@ -32,16 +32,16 @@
     ListExperimentUUIDsFilters,
     ProjectSchema,
     TestCaseCollection,
-    TraceLog,
     TraceLogFilters,
     TraceLogTree,
     UseDeployedPrompt,
     UseDeployedPromptResponse,
 )
-from parea.utils.trace_utils import get_current_trace_id, get_root_trace_id, logger_all_possible, logger_record_log, trace_data
+from parea.utils.trace_utils import get_current_trace_id, get_root_trace_id, logger_all_possible, trace_data
 
 load_dotenv()
 
+
 logger = logging.getLogger()
 
 COMPLETION_ENDPOINT = "/completion"

parea/cookbook/guidance/tracing_guidance.py

@@ -1,7 +1,7 @@
 import os
 
 from dotenv import load_dotenv
-from guidance import assistant, gen, models, select, user
+from guidance import assistant, gen, models, user
 
 from parea import Parea, trace
 

parea/cookbook/langchain/trace_langchain_RAG_evals.py

@@ -4,7 +4,6 @@
 
 import boto3
 from dotenv import load_dotenv
-
 # LangChain libs
 from langchain.chat_models import ChatOpenAI
 from langchain.document_loaders import RecursiveUrlLoader
@@ -13,7 +12,6 @@
 from langchain.prompts import ChatPromptTemplate
 from langchain.schema.output_parser import StrOutputParser
 from langchain.text_splitter import TokenTextSplitter
-from langchain.vectorstores import Chroma
 from langchain_community.llms.bedrock import Bedrock
 from langchain_community.vectorstores.faiss import FAISS
 

parea/cookbook/parea_llm_proxy/tracing_with_parea_streaming.py

@@ -1,4 +1,3 @@
-import asyncio
 import os
 
 from dotenv import load_dotenv

parea/evals/utils.py

@@ -10,9 +10,8 @@
 from openai import __version__ as openai_version
 
 from parea.parea_logger import parea_logger
-from parea.schemas import EvaluationResult, Log
+from parea.schemas import EvaluationResult
 from parea.schemas.log import Log
-from parea.schemas.models import UpdateLog
 from parea.utils.trace_utils import thread_ids_running_evals, trace_data
 
 seg = pysbd.Segmenter(language="en", clean=False)

parea/experiment/cli.py

@@ -2,7 +2,6 @@
 
 import argparse
 import csv
-import importlib
 import os
 import sys
 import traceback
@@ -18,8 +17,8 @@ def load_from_path(module_path):
         sys.path.insert(0, dir_name)
 
     module_name = os.path.basename(module_path)
-    spec = importlib.util.spec_from_file_location(module_name, module_path)
-    module = importlib.util.module_from_spec(spec)
+    spec = util.spec_from_file_location(module_name, module_path)
+    module = util.module_from_spec(spec)
     spec.loader.exec_module(module)
 
     if spec.name not in sys.modules:

parea/parea_logger.py

@@ -53,9 +53,6 @@ async def arecord_log(self, data: TraceLog) -> None:
             data=asdict(data),
         )
 
-    def write_log(self, data: TraceLog) -> None:
-        data = serialize_metadata_values(data)
-
     def default_log(self, data: TraceLog) -> None:
         if self._client:
             if data.target:

parea/schemas/models.py

@@ -159,6 +159,7 @@ class CacheRequest:
 class UpdateLog:
     trace_id: str
     field_name_to_value_map: Dict[str, Any]
+    root_trace_id: Optional[str] = None
 
 
 @define

parea/utils/trace_integrations/instructor.py

@@ -87,7 +87,8 @@ def __call__(
             instructor_val_errs.set(instructor_val_errs.get() + reasons)
 
             report_instructor_validation_errors()
-            logger_update_record(UpdateLog(trace_id=trace_id, field_name_to_value_map={"scores": trace_data.get()[trace_id].scores}))
+            current_log = trace_data.get()[trace_id]
+            logger_update_record(UpdateLog(trace_id=trace_id, field_name_to_value_map={"scores": current_log.scores}, root_trace_id=current_log.root_trace_id))
 
             raise e
 

parea/wrapper/utils.py

@@ -11,7 +11,7 @@
 from parea.constants import ALL_NON_AZURE_MODELS_INFO, AZURE_MODEL_INFO
 from parea.parea_logger import parea_logger
 from parea.schemas.log import LLMInputs, Message, ModelParams, Role
-from parea.schemas.models import UpdateLog, UpdateTraceScenario
+from parea.schemas.models import UpdateTraceScenario
 from parea.utils.trace_utils import fill_trace_data, get_current_trace_id, log_in_thread, trace_data, trace_insert
 from parea.utils.universal_encoder import json_dumps