Skip to content

Commit fa938cd

Browse files
committed
Update test_serialization.py to work with langchain_core.
1 parent ac8e8fc commit fa938cd

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

tests/unitary/with_extras/langchain/test_serialization.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def setUp(self) -> None:
7575
"prompt": {
7676
"lc": 1,
7777
"type": "constructor",
78-
"id": ["langchain", "prompts", "prompt", "PromptTemplate"],
78+
"id": ["langchain_core", "prompts", "prompt", "PromptTemplate"],
7979
"kwargs": {
8080
"input_variables": ["subject"],
8181
"template": "Tell me a joke about {subject}",
@@ -118,21 +118,20 @@ def setUp(self) -> None:
118118
EXPECTED_RUNNABLE_SEQUENCE = {
119119
"lc": 1,
120120
"type": "constructor",
121-
"id": ["langchain", "schema", "runnable", "RunnableSequence"],
121+
"id": ["langchain_core", "runnables", "RunnableSequence"],
122122
"kwargs": {
123123
"first": {
124124
"lc": 1,
125125
"type": "constructor",
126-
"id": ["langchain", "schema", "runnable", "RunnableParallel"],
126+
"id": ["langchain_core", "runnables", "RunnableParallel"],
127127
"kwargs": {
128128
"steps": {
129129
"text": {
130130
"lc": 1,
131131
"type": "constructor",
132132
"id": [
133-
"langchain",
134-
"schema",
135-
"runnable",
133+
"langchain_core",
134+
"runnables",
136135
"RunnablePassthrough",
137136
],
138137
"kwargs": {"func": None, "afunc": None, "input_type": None},
@@ -145,7 +144,7 @@ def setUp(self) -> None:
145144
{
146145
"lc": 1,
147146
"type": "constructor",
148-
"id": ["langchain", "prompts", "prompt", "PromptTemplate"],
147+
"id": ["langchain_core", "prompts", "prompt", "PromptTemplate"],
149148
"kwargs": {
150149
"input_variables": ["subject"],
151150
"template": "Tell me a joke about {subject}",

0 commit comments

Comments
 (0)