Dont serialize result (#81)

* Don't serialize result * lint
AbanteAI · May 27, 2024 · 7ab7f9f · 7ab7f9f
1 parent 880f105
commit 7ab7f9f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,7 +12,7 @@ packages=["spice"]
 
 [project]
 name = "spiceai"
-version = "0.3.8"
+version = "0.3.9"
 license = {text = "Apache-2.0"}
 description = "A Python library for building AI-powered applications."
 readme = "README.md"

diff --git a/spice/spice.py b/spice/spice.py
@@ -288,6 +288,7 @@ def _log_response(self, response: SpiceResponse, name: Optional[str] = None):
             full_name = f"{base_name}_{self._cur_logged_names[base_name]}.json"
             self._cur_logged_names[base_name] += 1
             response_dict = dataclasses.asdict(response)
+            response_dict.pop("_result", "")  # May not be serializable
             response_json = json.dumps(response_dict, cls=MessagesEncoder)
 
             logging_dir = self.logging_dir / self._cur_run