aisuite icon indicating copy to clipboard operation
aisuite copied to clipboard

Published 20 hours ago • verified publisher icon andrewyng

Add CentML Provider

Open anurlybayev opened this issue 11 months ago • 3 comments

Add CentML Provider Support

Description

Added support for CentML, to expand aisuite's provider options.

Changes Made

  • Added CentML provider implementation in providers/centml_provider.py
  • Added necessary environment variables for CentML API authentication

Testing

  • Tested provider initialization
  • Tested chat completion functionality
  • All existing tests pass

Environment Variables Added

CENTML_API_KEY=

Related Documentation

  • Added CentML to the list of supported providers in README.md

Checklist

  • [x] Code follows project style guidelines
  • [x] Updated documentation
  • [x] Pre-commit hooks pass

anurlybayev avatar Dec 03 '24 00:12 anurlybayev

@ksolo can you please review?

anurlybayev avatar Dec 05 '24 21:12 anurlybayev

@ksolo @rohitprasad15 I rebased my branch from main. Can you please review?

anurlybayev avatar Jan 04 '25 23:01 anurlybayev

Thanks for the PR. Please re-test after the addition of pydantic checks in the codebase. If possible, please add tool calling support as well. If the model uses OpenAI message format, then you can use Mistral or Together as an example on how to use an existing base class to do message conversion. Please tag me after changes are done.

rohitprasad15 avatar Jan 24 '25 06:01 rohitprasad15