Fix bug with custom header in request (#1000)

* Custom header bug * fix logic
Qiskit · Aug 10, 2023 · f646b56 · f646b56
1 parent 8bcf622
commit f646b56
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/qiskit_ibm_runtime/api/session.py b/qiskit_ibm_runtime/api/session.py
@@ -283,9 +283,19 @@ def request(  # type: ignore[override]
                 if str(caller) in frame_path:
                     caller_str = str(caller) + frame_path.split(str(caller), 1)[-1]
                     sanitized_caller_str = caller_str.replace("/", "~")
-                    headers.update(
-                        {"X-Qx-Client-Application": f"{CLIENT_APPLICATION}/{sanitized_caller_str}"}
-                    )
+                    if self.custom_header:
+                        headers.update(
+                            {
+                                "X-Qx-Client-Application": f"{CLIENT_APPLICATION}/"
+                                f"{sanitized_caller_str}/{self.custom_header}"
+                            }
+                        )
+                    else:
+                        headers.update(
+                            {
+                                "X-Qx-Client-Application": f"{CLIENT_APPLICATION}/{sanitized_caller_str}"
+                            }
+                        )
                     found_caller = True
                     break  # break out of the inner loop
             if found_caller: