Fix PromptTemplates import. (#604)

Co-authored-by: Josh Reini <[email protected]>
truera · Dec 1, 2023 · e82468d · e82468d
1 parent f04d915
commit e82468d
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 9 deletions.
diff --git a/docs/trulens_eval/langchain_instrumentation.ipynb b/docs/trulens_eval/langchain_instrumentation.ipynb
@@ -39,7 +39,8 @@
     "# required imports\n",
     "from langchain.chains import LLMChain\n",
     "from langchain.llms import OpenAI\n",
-    "from langchain.prompts.chat import HumanMessagePromptTemplate, ChatPromptTemplate, PromptTemplate\n",
+    "from langchain.prompts import PromptTemplate\n",
+    "from langchain.prompts.chat import HumanMessagePromptTemplate, ChatPromptTemplate\n",
     "from trulens_eval import TruChain\n",
     "\n",
     "# typical langchain setup\n",

diff --git a/trulens_eval/examples/expositional/models/google_vertex_quickstart.ipynb b/trulens_eval/examples/expositional/models/google_vertex_quickstart.ipynb
@@ -77,8 +77,8 @@
     "# ! pip install langchain>=0.0.170\n",
     "from langchain.chains import LLMChain\n",
     "from langchain.llms import VertexAI\n",
-    "from langchain.prompts.chat import ChatPromptTemplate, PromptTemplate\n",
-    "from langchain.prompts.chat import HumanMessagePromptTemplate"
+    "from langchain.prompts import PromptTemplate\n",
+    "from langchain.prompts.chat import HumanMessagePromptTemplate, ChatPromptTemplate"
    ]
   },
   {

diff --git a/trulens_eval/examples/expositional/models/litellm_quickstart.ipynb b/trulens_eval/examples/expositional/models/litellm_quickstart.ipynb
@@ -70,8 +70,8 @@
     "# ! pip install langchain>=0.0.170\n",
     "from langchain.chains import LLMChain\n",
     "from langchain.llms import OpenAI\n",
-    "from langchain.prompts.chat import ChatPromptTemplate, PromptTemplate\n",
-    "from langchain.prompts.chat import HumanMessagePromptTemplate"
+    "from langchain.prompts import PromptTemplate\n",
+    "from langchain.prompts.chat import HumanMessagePromptTemplate, ChatPromptTemplate"
    ]
   },
   {

diff --git a/trulens_eval/examples/expositional/use_cases/pii_detection.ipynb b/trulens_eval/examples/expositional/use_cases/pii_detection.ipynb
@@ -65,8 +65,8 @@
     "# ! pip install langchain>=0.0.170\n",
     "from langchain.chains import LLMChain\n",
     "from langchain.llms import OpenAI\n",
-    "from langchain.prompts.chat import ChatPromptTemplate, PromptTemplate\n",
-    "from langchain.prompts.chat import HumanMessagePromptTemplate"
+    "from langchain.prompts import PromptTemplate\n",
+    "from langchain.prompts.chat import HumanMessagePromptTemplate, ChatPromptTemplate"
    ]
   },
   {

diff --git a/trulens_eval/generated_files/all_tools.py b/trulens_eval/generated_files/all_tools.py
@@ -31,9 +31,9 @@
 # ! pip install langchain>=0.0.170
 from langchain.chains import LLMChain
 from langchain.llms import OpenAI
+from langchain.prompts import PromptTemplate
 from langchain.prompts.chat import ChatPromptTemplate
 from langchain.prompts.chat import HumanMessagePromptTemplate
-from langchain.prompts.chat import PromptTemplate
 
 # ### Create Simple LLM Application
 #

diff --git a/trulens_eval/trulens_eval/tru_chain.py b/trulens_eval/trulens_eval/tru_chain.py
@@ -117,9 +117,9 @@ class TruChain(App):
          # Code snippet taken from langchain 0.0.281 (API subject to change with new versions)
         from langchain.chains import LLMChain
         from langchain.llms import OpenAI
+        from langchain.prompts import PromptTemplate
         from langchain.prompts.chat import ChatPromptTemplate
         from langchain.prompts.chat import HumanMessagePromptTemplate
-        from langchain.prompts.chat import PromptTemplate
 
         full_prompt = HumanMessagePromptTemplate(
             prompt=PromptTemplate(