Skip to content

Commit e13a29a

Browse files
authored
Merge branch 'main' into add-ci-test-counts
2 parents cedf053 + ebddc4f commit e13a29a

11 files changed

Lines changed: 1217 additions & 1254 deletions

File tree

.gitignore

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,9 @@ megalinter-reports/
1010
# Benchmarks
1111
.asv/
1212

13+
# LLM Cache Files
14+
.tiktoken_cache
15+
1316
# Byte-compiled / optimized / DLL files
1417
__pycache__/
1518
*.py[cod]

tests/mlmodel_langchain/_mock_external_openai_server.py

Lines changed: 0 additions & 1004 deletions
This file was deleted.

tests/mlmodel_langchain/_test_tools.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,22 +32,22 @@ async def add_exclamation_async(message: str) -> str:
3232
return f"{message}!"
3333

3434

35-
@pytest.fixture(scope="session", params=["sync_tool", "async_tool"])
35+
@pytest.fixture(params=["sync_tool", "async_tool"])
3636
def tool_type(request):
3737
return request.param
3838

3939

40-
@pytest.fixture(scope="session")
40+
@pytest.fixture
4141
def tool_method_name(tool_type):
4242
return "run" if tool_type == "sync_tool" else "arun"
4343

4444

45-
@pytest.fixture(scope="session")
45+
@pytest.fixture
4646
def add_exclamation(tool_type, exercise_agent):
4747
if tool_type == "sync_tool":
4848
return add_exclamation_sync
4949
elif tool_type == "async_tool":
50-
if exercise_agent._called_method in {"invoke", "stream"}:
50+
if exercise_agent._called_method in ("invoke", "stream"):
5151
pytest.skip("Async tools cannot be invoked synchronously.")
5252
return add_exclamation_async
5353
else:

0 commit comments

Comments
 (0)