Skip to content

Commit c13f281

Browse files
committed
chore: Revert back langchain imports
1 parent 4f8147a commit c13f281

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

portkey_ai/llms/langchain/chat.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717
try:
1818
from langchain.callbacks.manager import CallbackManagerForLLMRun
19-
from langchain_community.chat_models.base import SimpleChatModel
19+
from langchain.chat_models.base import SimpleChatModel
2020
from langchain.pydantic_v1 import Field, PrivateAttr
2121
from langchain.schema.messages import (
2222
AIMessage,

tests/test_llm_langchain.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,9 @@
77

88
from tests.utils import read_json_file
99
from portkey_ai.langchain import LangchainCallbackHandler
10-
from langchain_community.chat_models import ChatOpenAI
10+
from langchain.chat_models import ChatOpenAI
1111
from langchain_core.prompts import ChatPromptTemplate
12-
from langchain_classic.chains import LLMChain
12+
from langchain.chains import LLMChain
1313

1414
api_key = os.environ.get("PORTKEY_API_KEY")
1515

0 commit comments

Comments
 (0)