Update to recent versions

esbmc · Oct 10, 2023 · 96515ca · 96515ca
1 parent 3db33c3
commit 96515ca
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/esbmc_ai_lib/ai_models.py b/esbmc_ai_lib/ai_models.py
@@ -5,10 +5,11 @@
 from enum import Enum
 from typing_extensions import override
 
-from langchain import HuggingFaceTextGenInference, PromptTemplate
 from langchain.base_language import BaseLanguageModel
 from langchain.chat_models import ChatOpenAI
+from langchain.llms import HuggingFaceTextGenInference
 
+from langchain.prompts import PromptTemplate
 from langchain.prompts.chat import ChatPromptValue
 from langchain.schema import (
     BaseMessage,