search

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

`artkit.api` does not import `CachedDiffusionModel` and `CachedVisionModel` #24

Closed rgriff23 closed 1 week ago

rgriff23 commented 1 week ago

Describe the bug

artkit.api does not import CachedDiffusionModel and CachedVisionModel.

To Reproduce

Steps to reproduce the behavior:

  1. Run the Connecting to Gen AI Models notebook: https://github.com/BCG-X-Official/artkit/blob/1.0.x/sphinx/source/user_guide/introduction_to_artkit/connecting_to_genai_models.ipynb
  2. The OpenAI "text-to-image" and "vision" sections will fail

Expected behavior

CachedDiffusionModel and CachedVisionModel should be imported with artkit.api

Additional context

The fix is to update the following imports in src/artkit/api.py, which currently do not seem to import anything:

from .model.diffusion import *
from .model.vision import *

to:

from .model.diffusion.base import *
from .model.vision.base import *