Adding support for AI model routing with Not Diamond.

[acompa]

Based off of the existing OpenAI integration.

This PR is a WIP for the moment while I update the docs and run integration tests. But feel free to review if you'd like!

[cmpadden]

Looking really good!

Minor change requests:

• Using the templates in GitHub actions
• Updating references to package name as dagster_contrib_notdiamond

Thanks again for taking the time to contribute, and work with the new integrations repo. Let me know if anything has been confusing.

[acompa]

Okay @cmpadden - think I'm ready for a second set of eyes here. The method docstrings and README might have some inconsistencies, fyi!

[cmpadden]

Okay @cmpadden - think I'm ready for a second set of eyes here. The method docstrings and README might have some inconsistencies, fyi!

This is looking really fantastic.

I'm attempting to run the unit tests locally, and am getting a weird setuptools error:

FAILED dagster_contrib_notdiamond_tests/test_resources.py::test_notdiamond_client_with_config - importlib.metadata.PackageNotFoundError: No package metadata was found for dagster-con...

Still looking into it, but if you have any ideas, please share!

---

Edit: looks like including setuptools in pyproject.toml helps:

[build-system]
requires = ["setuptools>=42"]
build-backend = "setuptools.build_meta"

Still a couple of test failures though--looking:

FAILED dagster_contrib_notdiamond_tests/test_resources.py::test_notdiamond_client - AssertionError: expected call not found.
FAILED dagster_contrib_notdiamond_tests/test_resources.py::test_notdiamond_client_with_config - AssertionError: expected call not found.

[cmpadden]

Oh! It looks like the dagster-openai integration uses a mock library, and not the version that we're using here. Not sure why that's the case. Hmph.

from mock import ANY, MagicMock, patch

[acompa]

Oh! It looks like the dagster-openai integration uses a mock library, and not the version that we're using here. Not sure why that's the case. Hmph.

from mock import ANY, MagicMock, patch

Yeah, and actually this library is not independently supported in modern Python versions. So I'm usingunittest.mock here instead.

Issue seems to be caused by my forgetting of a mock call parameter. I've added it back, and all tests pass again locally.

[cmpadden]

Checks are running green--will follow up with you in Slack on next steps. 🤝