diff --git a/libs/partners/fireworks/Makefile b/libs/partners/fireworks/Makefile index ff61279b622..af7366d5f59 100644 --- a/libs/partners/fireworks/Makefile +++ b/libs/partners/fireworks/Makefile @@ -14,7 +14,7 @@ test tests: uv run --group test pytest --disable-socket --allow-unix-socket $(TEST_FILE) integration_test integration_tests: - uv run --group test --group test_integration pytest $(TEST_FILE) + uv run --group test --group test_integration pytest -n auto $(TEST_FILE) test_watch: uv run --group test ptw --snapshot-update --now . -- -vv $(TEST_FILE) diff --git a/libs/partners/fireworks/pyproject.toml b/libs/partners/fireworks/pyproject.toml index 6b7df1d5cf2..4b1aae74576 100644 --- a/libs/partners/fireworks/pyproject.toml +++ b/libs/partners/fireworks/pyproject.toml @@ -37,6 +37,7 @@ test = [ "pytest-watcher>=0.3.4,<1.0.0", "pytest-asyncio>=0.21.1,<1.0.0", "pytest-socket>=0.7.0,<1.0.0", + "pytest-xdist>=3.8.0,<4.0.0", "langchain-core", "langchain-tests", ] diff --git a/libs/partners/fireworks/tests/integration_tests/test_chat_models.py b/libs/partners/fireworks/tests/integration_tests/test_chat_models.py index 2eb48409ee9..77825dc8664 100644 --- a/libs/partners/fireworks/tests/integration_tests/test_chat_models.py +++ b/libs/partners/fireworks/tests/integration_tests/test_chat_models.py @@ -20,9 +20,7 @@ _MODEL = "accounts/fireworks/models/llama-v3p1-8b-instruct" def test_tool_choice_bool() -> None: """Test that tool choice is respected just passing in True.""" - llm = ChatFireworks( - model="accounts/fireworks/models/llama-v3p1-70b-instruct", temperature=0 - ) + llm = ChatFireworks(model="fireworks/kimi-k2-instruct-0905") class MyTool(BaseModel): name: str diff --git a/libs/partners/fireworks/tests/integration_tests/test_standard.py b/libs/partners/fireworks/tests/integration_tests/test_standard.py index 45a9e69cbae..22e7824af0e 100644 --- a/libs/partners/fireworks/tests/integration_tests/test_standard.py +++ b/libs/partners/fireworks/tests/integration_tests/test_standard.py @@ -18,7 +18,7 @@ class TestFireworksStandard(ChatModelIntegrationTests): @property def chat_model_params(self) -> dict: return { - "model": "accounts/fireworks/models/llama-v3p1-70b-instruct", + "model": "fireworks/kimi-k2-instruct-0905", "temperature": 0, } diff --git a/libs/partners/fireworks/uv.lock b/libs/partners/fireworks/uv.lock index 0183783b3f3..f9c8c927478 100644 --- a/libs/partners/fireworks/uv.lock +++ b/libs/partners/fireworks/uv.lock @@ -1,5 +1,5 @@ version = 1 -revision = 3 +revision = 2 requires-python = ">=3.10.0, <4.0.0" resolution-markers = [ "python_full_version >= '3.13' and platform_python_implementation == 'PyPy'", @@ -345,6 +345,15 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/36/f4/c6e662dade71f56cd2f3735141b265c3c79293c109549c1e6933b0651ffc/exceptiongroup-1.3.0-py3-none-any.whl", hash = "sha256:4d111e6e0c13d0644cad6ddaa7ed0261a0b36971f6d23e7ec9b4b9097da78a10", size = 16674, upload-time = "2025-05-10T17:42:49.33Z" }, ] +[[package]] +name = "execnet" +version = "2.1.2" +source = { registry = "https://pypi.org/simple" } +sdist = { url = "https://files.pythonhosted.org/packages/bf/89/780e11f9588d9e7128a3f87788354c7946a9cbb1401ad38a48c4db9a4f07/execnet-2.1.2.tar.gz", hash = "sha256:63d83bfdd9a23e35b9c6a3261412324f964c2ec8dcd8d3c6916ee9373e0befcd", size = 166622, upload-time = "2025-11-12T09:56:37.75Z" } +wheels = [ + { url = "https://files.pythonhosted.org/packages/ab/84/02fc1827e8cdded4aa65baef11296a9bbe595c474f0d6d758af082d849fd/execnet-2.1.2-py3-none-any.whl", hash = "sha256:67fba928dd5a544b783f6056f449e5e3931a5c378b128bc18501f7ea79e296ec", size = 40708, upload-time = "2025-11-12T09:56:36.333Z" }, +] + [[package]] name = "fireworks-ai" version = "0.15.15" @@ -642,7 +651,7 @@ wheels = [ [[package]] name = "langchain-core" -version = "1.0.0" +version = "1.1.0" source = { editable = "../../core" } dependencies = [ { name = "jsonpatch" }, @@ -726,6 +735,7 @@ test = [ { name = "pytest-mock" }, { name = "pytest-socket" }, { name = "pytest-watcher" }, + { name = "pytest-xdist" }, { name = "syrupy" }, ] typing = [ @@ -755,6 +765,7 @@ test = [ { name = "pytest-mock", specifier = ">=3.10.0,<4.0.0" }, { name = "pytest-socket", specifier = ">=0.7.0,<1.0.0" }, { name = "pytest-watcher", specifier = ">=0.3.4,<1.0.0" }, + { name = "pytest-xdist", specifier = ">=3.8.0,<4.0.0" }, { name = "syrupy", specifier = ">=4.0.2,<5.0.0" }, ] test-integration = [] @@ -766,7 +777,7 @@ typing = [ [[package]] name = "langchain-tests" -version = "1.0.0" +version = "1.0.1" source = { editable = "../../standard-tests" } dependencies = [ { name = "httpx" }, @@ -1741,6 +1752,19 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/5b/3a/c44a76c6bb5e9e896d9707fb1c704a31a0136950dec9514373ced0684d56/pytest_watcher-0.4.3-py3-none-any.whl", hash = "sha256:d59b1e1396f33a65ea4949b713d6884637755d641646960056a90b267c3460f9", size = 11852, upload-time = "2024-08-28T17:37:45.731Z" }, ] +[[package]] +name = "pytest-xdist" +version = "3.8.0" +source = { registry = "https://pypi.org/simple" } +dependencies = [ + { name = "execnet" }, + { name = "pytest" }, +] +sdist = { url = "https://files.pythonhosted.org/packages/78/b4/439b179d1ff526791eb921115fca8e44e596a13efeda518b9d845a619450/pytest_xdist-3.8.0.tar.gz", hash = "sha256:7e578125ec9bc6050861aa93f2d59f1d8d085595d6551c2c90b6f4fad8d3a9f1", size = 88069, upload-time = "2025-07-01T13:30:59.346Z" } +wheels = [ + { url = "https://files.pythonhosted.org/packages/ca/31/d4e37e9e550c2b92a9cbc2e4d0b7420a27224968580b5a447f420847c975/pytest_xdist-3.8.0-py3-none-any.whl", hash = "sha256:202ca578cfeb7370784a8c33d6d05bc6e13b4f25b5053c30a152269fd10f0b88", size = 46396, upload-time = "2025-07-01T13:30:56.632Z" }, +] + [[package]] name = "python-dateutil" version = "2.9.0.post0"