Fix error str (#23)

* Update langchain_callback.py * Update langchain_callback.py
newrelic · Sep 10, 2023 · 4c95c99 · 4c95c99
1 parent 4de0227
commit 4c95c99
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/nr_openai_observability/langchain_callback.py b/src/nr_openai_observability/langchain_callback.py
@@ -139,7 +139,7 @@ def on_chain_error(
         self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
     ) -> Any:
         """Run when chain errors."""
-        tags = {error: str(error)}
+        tags = {"error": str(error)}
         span = self.spans_stack.pop()
         assert span["attributes"]["name"] == "LlmChain"
         self.finish_and_record_span(span, tags)
@@ -173,9 +173,7 @@ def on_tool_error(
         self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
     ) -> Any:
         """Run when tool errors."""
-        tags = {
-            "error": error,
-        }
+        tags = {"error": str(error)}
         span = self.spans_stack.pop()
         assert span["attributes"]["name"] == "LlmTool"
         tool_name = kwargs.get("name")