import sympy as sp

A = sp.symbols('A')
B = sp.symbols('B')
C = sp.symbols('C')
D = sp.symbols('D')
E = sp.symbols('E')
F = sp.symbols('F')
G = sp.symbols('G')
H = sp.symbols('H')
J = sp.symbols('J')
K = sp.symbols('K')

x = sp.symbols('x')
n = sp.symbols('n')
m = sp.symbols('m')
k = sp.symbols('k')

# Sometimes, functions (llm generated) ask for the constants in their arguments
# list. We provide them here, even through they are not variables.
pi = sp.pi
i = sp.I
e = sp.exp(1)


var_mapping = {
    'A': A,
    'B': B,
    'C': C,
    'D': D,
    'E': e,  # E is not a variable, but a constant
    'F': F,
    'G': G,
    'H': H,
    'J': J,
    'K': K,
    'x': x,
    'k': k,
    'n': n,
    'm': m,
    'e': e,  # e is not a variable, but a constant
    'pi': pi,
    'i': i
}

used_vars = list(var_mapping.keys())
