Python更新源代码wrt函数的输出

Question

例如，我的第一个模块是

from sympy import *
x,y=symbols('x y')

def A():
    equation=2*x+y
    return equation

print(A())

输出

2*x + y

我的第二个模块是

def B(x,y):
    equation=2*x + y
    return equation

我必须每次从终端复制第一个模块的输出到第二个模块的输出。有没有办法将此输出自动传递给B（x，y）的源代码？

感谢您的关注