Skip to content

Commit 67de093

Browse files
yeesiancopybara-github
authored andcommitted
chore: Add kokoro config files for langchain template unit tests
PiperOrigin-RevId: 624975886
1 parent f05924d commit 67de093

File tree

5 files changed

+57
-1
lines changed

5 files changed

+57
-1
lines changed
+13
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
# Format: //devtools/kokoro/config/proto/build.proto
2+
3+
# Run unit tests for LangChain on Python 3.10
4+
env_vars: {
5+
key: "NOX_SESSION"
6+
value: "unit_langchain-3.10"
7+
}
8+
9+
# Run unit tests in parallel, splitting up by file
10+
env_vars: {
11+
key: "PYTEST_ADDOPTS"
12+
value: "-n=auto --dist=loadscope"
13+
}
+13
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
# Format: //devtools/kokoro/config/proto/build.proto
2+
3+
# Run unit tests for LangChain on Python 3.11
4+
env_vars: {
5+
key: "NOX_SESSION"
6+
value: "unit_langchain-3.11"
7+
}
8+
9+
# Run unit tests in parallel, splitting up by file
10+
env_vars: {
11+
key: "PYTEST_ADDOPTS"
12+
value: "-n=auto --dist=loadscope"
13+
}
+13
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
# Format: //devtools/kokoro/config/proto/build.proto
2+
3+
# Run unit tests for LangChain on Python 3.8
4+
env_vars: {
5+
key: "NOX_SESSION"
6+
value: "unit_langchain-3.8"
7+
}
8+
9+
# Run unit tests in parallel, splitting up by file
10+
env_vars: {
11+
key: "PYTEST_ADDOPTS"
12+
value: "-n=auto --dist=loadscope"
13+
}
+13
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
# Format: //devtools/kokoro/config/proto/build.proto
2+
3+
# Run unit tests for LangChain on Python 3.9
4+
env_vars: {
5+
key: "NOX_SESSION"
6+
value: "unit_langchain-3.9"
7+
}
8+
9+
# Run unit tests in parallel, splitting up by file
10+
env_vars: {
11+
key: "PYTEST_ADDOPTS"
12+
value: "-n=auto --dist=loadscope"
13+
}

setup.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -156,6 +156,10 @@
156156
"langchain-google-vertexai < 0.2",
157157
]
158158

159+
langchain_testing_extra_require = langchain_extra_require + [
160+
"pytest-xdist",
161+
]
162+
159163
full_extra_require = list(
160164
set(
161165
tensorboard_extra_require
@@ -260,7 +264,7 @@
260264
"reasoningengine": reasoning_engine_extra_require,
261265
"rapid_evaluation": rapid_evaluation_extra_require,
262266
"langchain": langchain_extra_require,
263-
"langchain_testing": langchain_extra_require,
267+
"langchain_testing": langchain_testing_extra_require,
264268
},
265269
python_requires=">=3.8",
266270
classifiers=[

0 commit comments

Comments
 (0)