import numpy as np
from qiskit import QuantumCircuit, generate_preset_pass_manager
from qiskit import __version__ as qiskit_version
from qiskit.circuit import ParameterVector
from qiskit.circuit.library import QAOAAnsatz, qaoa_ansatz
from qiskit.quantum_info import SparsePauliOp
from qiskit_ibm_runtime import QiskitRuntimeService, Sampler
from qiskit_ibm_runtime import __version__ as runtime_version

print(f"{qiskit_version=} {runtime_version=}")

op = SparsePauliOp(["ZIZI", "IZIZ", "ZIIZ"])
qaoa = qaoa_ansatz(op, reps=1)
x = ParameterVector("γ", 1)
qaoa = qaoa.assign_parameters({qaoa.parameters[1]: x[0]})
qc = QuantumCircuit(4)
qc.append(qaoa, range(4))
qc.measure_all()
# print(qc)

for name, iqp in [("primitives", "conventional IQP"), ("notebooks-internal", "new IQP")]:
    print(iqp)
    service = QiskitRuntimeService(name=name)
    backend = service.backend("test_eagle_us-east")
    pm = generate_preset_pass_manager(optimization_level=2, backend=backend)
    t_qc = pm.run(qc)

    sampler = Sampler(backend)
    params = np.ones(t_qc.num_parameters)
    try:
        job = sampler.run([(t_qc, params)])
        print(job)
        result = job.result()
        print(result)
    except Exception as ex:
        print(ex.message)
    print()