Skip to content

Track frameworks using the NAT registry #1174

@dagardner-nv

Description

@dagardner-nv

Is this a new feature, an improvement, or a change to existing functionality?

New Feature

How would you describe the priority of this feature request

High

Please provide a clear description of problem this feature solves

Currently adding a new framework requires modifying src/nat/builder/framework_enum.py which creates two problems:

  1. Frameworks added by plugins are codified within the core nvidia-nat package
  2. Third-party developers are not able to add a new framework in an external repo without also submitting a PR to modify LLMFrameworkEnum

Describe your ideal solution

Track frameworks using the component registry allowing developers to register a new framework with something like:

class CrewAIConfig(FrameWorkBaseConfig, name="crewai"):
    ...

@register_framework(config_type=CrewAIConfig)
async def crew_ai():
    import crewai

Additional context

No response

Code of Conduct

  • I agree to follow this project's Code of Conduct
  • I have searched the open feature requests and have found no duplicates for this feature request

Metadata

Metadata

Assignees

Labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions