Fix type hints

langchain-ai · Oct 9, 2024 · b6fd11b · b6fd11b
1 parent f3d6aa2
commit b6fd11b
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/python/langsmith/client.py b/python/langsmith/client.py
@@ -936,12 +936,12 @@ def request_with_retries(
                     debug_crash_dump_file = ls_utils.get_env_var(
                         "LANGSMITH_DEBUG_CRASH_DUMP"
                     )
-                    if debug_crash_dump_file is not None:
-                        request_data: (
-                            requests.Request | requests.PreparedRequest | dict
-                        ) = (copy.deepcopy(e.request) if e.request else {})
-                        if "x-api-key" in request_data.get("headers", {}):
-                            request_data["headers"]["x-api-key"] = masked_api_key
+                    if debug_crash_dump_file is not None and e.request is not None:
+                        request_data: requests.Request | requests.PreparedRequest = (
+                            copy.deepcopy(e.request)
+                        )
+                        if "x-api-key" in request_data.headers:
+                            request_data.headers["x-api-key"] = masked_api_key
                         with gzip.open(debug_crash_dump_file, "ab") as f:
                             json_data = json.dumps(request_data).encode("utf-8")
                             f.write(json_data + b"\n")