fix: MarkdownFileCallbackHandler to aboid null exception

nobu007 · Sep 17, 2024 · f03d3f9 · f03d3f9
1 parent ad69b29
commit f03d3f9
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/codeinterpreterapi/callbacks/markdown/callbacks.py b/src/codeinterpreterapi/callbacks/markdown/callbacks.py
@@ -32,10 +32,11 @@ def on_llm_end(self, response: LLMResult, **kwargs: Any) -> None:
 
     def on_chain_start(self, serialized: Dict[str, Any], inputs: Dict[str, Any], **kwargs: Any) -> None:
         self.step_count += 1
-        chain_name = serialized.get("name", "Unknown Chain")
-        self._write_header(f"Chain Start - {chain_name}")
-        self._write_serialized(serialized)
-        self._write_inputs(inputs)
+        if serialized:
+            chain_name = serialized.get("name", "Unknown Chain")
+            self._write_header(f"Chain Start - {chain_name}")
+            self._write_serialized(serialized)
+            self._write_inputs(inputs)
 
     def on_chain_end(self, outputs: Dict[str, Any], **kwargs: Any) -> None:
         self._write_header("Chain End")