chore: use json.dumps instead of str in sanitate attrs (#43)

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "greptimeai"
-version = "0.1.2"
+version = "0.1.3"
 description = "Observability tool for LLM application"
 authors = [
     { name = "Greptime", email = "[email protected]" },

src/greptimeai/langchain/__init__.py

@@ -1,3 +1,4 @@
+import json
 import logging
 import time
 from typing import Any, Dict, Iterable, List, Optional, Sequence, Tuple, Union
@@ -21,9 +22,6 @@
 _MODEL_NAME_LABEL = "model"
 
 _INSTRUMENT_LIB_NAME = "greptime-llm"
-_INSTRUMENT_LIB_VERSION = (
-    "0.1.2"
-)
 
 _GREPTIME_HOST_ENV_NAME = "GREPTIMEAI_HOST"
 _GREPTIME_DATABASE_ENV_NAME = "GREPTIMEAI_DATABASE"
@@ -78,7 +76,7 @@ def _sanitate_list(lst: list) -> list:
             if _is_valid_otel_attributes_value_type(item):
                 result.append(item)
             else:
-                result.append(str(item))
+                result.append(json.dumps(item))
         return result
 
     for key, val in attrs.items():
@@ -87,7 +85,7 @@ def _sanitate_list(lst: list) -> list:
         elif isinstance(val, list):
             result[key] = _sanitate_list(val)
         else:
-            result[key] = str(val)
+            result[key] = json.dumps(val)
 
     return result
 
@@ -168,7 +166,7 @@ def _parse_generations(
     parse LLMResult.generations[0] to structured fields
     """
     if gens and len(gens) > 0:
-        return filter(None, [_parse_generation(gen) for gen in gens if gen])
+        return list(filter(None, [_parse_generation(gen) for gen in gens if gen]))
 
     return None
 

src/greptimeai/langchain/callback.py

@@ -34,7 +34,6 @@
     _GREPTIME_PASSWORD_ENV_NAME,
     _GREPTIME_USERNAME_ENV_NAME,
     _INSTRUMENT_LIB_NAME,
-    _INSTRUMENT_LIB_VERSION,
     _MODEL_NAME_LABEL,
     _SPAN_NAME_AGENT,
     _SPAN_NAME_CHAIN,
@@ -54,6 +53,7 @@
     _TimeTable,
     _TraceTable,
 )
+from greptimeai.version import __version__
 
 
 class _Collector:
@@ -168,12 +168,10 @@ def _setup_otel_metrics(self):
         """
         self._tracer = trace.get_tracer(
             instrumenting_module_name=_INSTRUMENT_LIB_NAME,
-            instrumenting_library_version=_INSTRUMENT_LIB_VERSION,
+            instrumenting_library_version=__version__,
         )
 
-        meter = metrics.get_meter(
-            name=_INSTRUMENT_LIB_NAME, version=_INSTRUMENT_LIB_VERSION
-        )
+        meter = metrics.get_meter(name=_INSTRUMENT_LIB_NAME, version=__version__)
 
         self._prompt_tokens_count = meter.create_counter(
             "llm_prompt_tokens",

src/greptimeai/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.3"