fix: fix types

Chainlit · Dec 17, 2024 · 327ffa3 · 327ffa3
1 parent e09d261
commit 327ffa3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/literalai/exporter.py b/literalai/exporter.py
@@ -149,13 +149,13 @@ def _create_step_from_span(self, span: ReadableSpan) -> Step:
             "id": str(span.context.span_id) if span.context else None,
             "name": span_props.get("name", span.name),
             "type": "llm",
-            "metadata": self._extract_json(span_props.get("metadata", "{}")),
+            "metadata": self._extract_json(str(span_props.get("metadata", "{}"))),
             "startTime": start_time,
             "endTime": end_time,
             "threadId": span_props.get("thread_id"),
             "parentId": span_props.get("parent_id"),
             "rootRunId": span_props.get("root_run_id"),
-            "tags": self._extract_json(span_props.get("tags", "[]")),
+            "tags": self._extract_json(str(span_props.get("tags", "[]"))),
             "input": {
                 "content": (
                     generation_content["messages"]

diff --git a/mypy.ini b/mypy.ini
@@ -8,4 +8,7 @@ ignore_missing_imports = True
 ignore_missing_imports = True
 
 [mypy-langchain_community.*]
-ignore_missing_imports = True
+ignore_missing_imports = True
+
+[mypy-traceloop.*]
+ignore_missing_imports = True