Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from langchain_core.runnables import RunnableLambda, RunnableParallel
- import grandalf
- equation_data = {"a": 1, "b": 2, "c": 5}
- D = RunnableLambda(lambda data: data["b"]**2 - 4 * data["a"] * data["c"])
- x = RunnableLambda(lambda data: -data["b"]/(2 * data["a"]))
- x1 = RunnableLambda(lambda data: (-data["b"] + D.invoke(data)**0.5)/(2 * data["a"]))
- x2 = RunnableLambda(lambda data: (-data["b"] - D.invoke(data)**0.5)/(2 * data["a"]))
- roots = RunnableParallel({"Корень 1:": x1, "Корень 2:": x2})
- chain = RunnableLambda(lambda data: x.invoke(data) if D.invoke(data) == 0 else("Корней нет" if D.invoke(data) < 0 else roots.invoke(data)))
- chain.get_graph().print_ascii()
- result = chain.invoke(equation_data)
- print(result)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement