BCG-X-Official / artkit

Automated prompt-based testing and evaluation of Gen AI applications
https://bcg-x-official.github.io/artkit/_generated/home.html
Apache License 2.0
76 stars 6 forks source link

Expose ChatbotModel as part of default artkit.api export #39

Open seanggani opened 4 days ago

seanggani commented 4 days ago

Is your feature request related to a problem? Please describe.

Previously, classes in model.llm.base like ChatModel where reachable through artkit.api and its been popular to use ak.ChatModel for type notations in notebooks. However, this is no longer possible as from .model.llm.base import * is missing from api.py

Describe the solution you'd like

Selectively include classes from .base modules such as ChatModel : from .model.llm.base import ChatModel

Describe alternatives you've considered

Alternative is to import all packages from base, which might be too much