From 1792684e8fd198a2c5135ffe5b8ce86f424dcbcc Mon Sep 17 00:00:00 2001 From: Srijan Dubey <87586713+Srijan-D@users.noreply.github.com> Date: Wed, 17 Jul 2024 19:18:32 +0530 Subject: [PATCH] removed deprecated classes from pipelineai.ipynb, added support for LangChain v0.2 for PipelineAI integration (#24333) Description: added support for LangChain v0.2 for PipelineAI integration. Removed deprecated classes and incorporated support for LangChain v0.2 to integrate with PipelineAI. Removed LLMChain and replaced it with Runnable interface. Also added StrOutputParser, that parses LLMResult into the top likely string. Issue: None Dependencies: None. --------- Co-authored-by: Chester Curme --- docs/docs/integrations/llms/pipelineai.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/docs/integrations/llms/pipelineai.ipynb b/docs/docs/integrations/llms/pipelineai.ipynb index 142d72d3e6f72..d20051f9dd581 100644 --- a/docs/docs/integrations/llms/pipelineai.ipynb +++ b/docs/docs/integrations/llms/pipelineai.ipynb @@ -50,8 +50,8 @@ "source": [ "import os\n", "\n", - "from langchain.chains import LLMChain\n", "from langchain_community.llms import PipelineAI\n", + "from langchain_core.output_parsers import StrOutputParser\n", "from langchain_core.prompts import PromptTemplate" ] }, @@ -123,7 +123,7 @@ "metadata": {}, "outputs": [], "source": [ - "llm_chain = LLMChain(prompt=prompt, llm=llm)" + "llm_chain = prompt | llm | StrOutputParser()" ] }, { @@ -142,7 +142,7 @@ "source": [ "question = \"What NFL team won the Super Bowl in the year Justin Beiber was born?\"\n", "\n", - "llm_chain.run(question)" + "llm_chain.invoke(question)" ] } ],