diff --git a/libs/core/langchain_core/runnables/passthrough.py b/libs/core/langchain_core/runnables/passthrough.py index 37d97fb968..4e04b37a0d 100644 --- a/libs/core/langchain_core/runnables/passthrough.py +++ b/libs/core/langchain_core/runnables/passthrough.py @@ -62,7 +62,11 @@ class RunnablePassthrough(RunnableSerializable[Other, Other]): .. code-block:: python - from langchain_core.runnables import RunnablePassthrough, RunnableParallel + from langchain_core.runnables import ( + RunnableLambda, + RunnableParallel, + RunnablePassthrough, + ) runnable = RunnableParallel( origin=RunnablePassthrough(), @@ -72,7 +76,7 @@ class RunnablePassthrough(RunnableSerializable[Other, Other]): runnable.invoke(1) # {'origin': 1, 'modified': 2} - def fake_llm(prompt: str) -> str: # Fake LLM for the example + def fake_llm(prompt: str) -> str: # Fake LLM for the example return "completion" chain = RunnableLambda(fake_llm) | { @@ -89,7 +93,7 @@ class RunnablePassthrough(RunnableSerializable[Other, Other]): from langchain_core.runnables import RunnablePassthrough, RunnableParallel - def fake_llm(prompt: str) -> str: # Fake LLM for the example + def fake_llm(prompt: str) -> str: # Fake LLM for the example return "completion" runnable = {