Skip to content

Commit f36eafd

Browse files
committed
Test w/ latest langchain
1 parent a0a9dd7 commit f36eafd

File tree

2 files changed

+7
-8
lines changed

2 files changed

+7
-8
lines changed

tests/mlmodel_langchain/test_tool.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def events_sans_content(event):
8080
"run_id": None,
8181
"output": "Python Agent",
8282
"name": "_single_arg_tool",
83-
"description": "_single_arg_tool(query: str) - A test tool that returns query string",
83+
"description": "A test tool that returns query string",
8484
"span_id": None,
8585
"trace_id": "trace-id",
8686
"input": "{'query': 'Python Agent'}",
@@ -178,7 +178,7 @@ def test_langchain_single_arg_tool_async_no_content(set_trace_info, single_arg_t
178178
"run_id": None,
179179
"output": "81",
180180
"name": "_multi_arg_tool",
181-
"description": "_multi_arg_tool(first_num: int, second_num: int) - A test tool that adds two integers together",
181+
"description": "A test tool that adds two integers together",
182182
"span_id": None,
183183
"trace_id": "trace-id",
184184
"input": "{'first_num': 53, 'second_num': 28}",
@@ -250,7 +250,7 @@ def test_langchain_multi_arg_tool_async(set_trace_info, multi_arg_tool, loop):
250250
"id": None, # UUID that varies with each run
251251
"run_id": None, # No run ID created on error
252252
"name": "_multi_arg_tool",
253-
"description": "_multi_arg_tool(first_num: int, second_num: int) - A test tool that adds two integers together",
253+
"description": "A test tool that adds two integers together",
254254
"span_id": None,
255255
"trace_id": "trace-id",
256256
"input": "{'first_num': 53}",

tox.ini

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -150,8 +150,7 @@ envlist =
150150
python-logger_loguru-{py37,py38,py39,py310,py311,py312,pypy310}-logurulatest,
151151
python-logger_loguru-py39-loguru{06,05},
152152
python-logger_structlog-{py37,py38,py39,py310,py311,py312,pypy310}-structloglatest,
153-
; langchain dependency faiss-cpu isn't compatible with 3.12 yet.
154-
python-mlmodel_langchain-{py38,py39,py310,py311},
153+
python-mlmodel_langchain-{py38,py39,py310,py311,py312},
155154
python-mlmodel_openai-openai0-{py37,py38,py39,py310,py311,py312},
156155
python-mlmodel_openai-openai107-py312,
157156
python-mlmodel_openai-openailatest-{py37,py38,py39,py310,py311,py312},
@@ -373,16 +372,16 @@ deps =
373372
; Required for openai testing
374373
mlmodel_openai: protobuf
375374
; Pinning to 0.1.16 while adding support for with_structured_output in chain tests
376-
mlmodel_langchain: langchain<0.1.17
375+
mlmodel_langchain: langchain
377376
mlmodel_langchain: langchain-community
378-
mlmodel_langchain: openai[datalib]
377+
mlmodel_langchain: langchain-core
378+
mlmodel_langchain: langchain-openai
379379
; Required for langchain testing
380380
mlmodel_langchain: pypdf
381381
mlmodel_langchain: tiktoken
382382
mlmodel_langchain: faiss-cpu
383383
mlmodel_langchain: mock
384384
mlmodel_langchain: asyncio
385-
mlmodel_langchain: langchain-openai
386385
logger_loguru-logurulatest: loguru
387386
logger_loguru-loguru06: loguru<0.7
388387
logger_loguru-loguru05: loguru<0.6

0 commit comments

Comments
 (0)