Still trying to package python snap - some progress but code doesn't seem to be there

name: aiyy
base: core22
version: '0.2'
summary: CLI assistant for technical information
description: |
  Get technical information on the command line with ease using AIY. Simply type in your question as the prompt and 
  receive a beautifully formatted and helpful response. This snap also has options for resetting the API key and 
  toggling warnings.
architectures:
  - build-on: [amd64]
grade: devel
confinement: devmode

apps:
  aiyy:
    command: bin/aiyy

parts:
  aiyy:
    plugin: python
    source: https://github.com/visioninit/aiy.git

from setuptools import setup, find_packages

setup(
    name='aiy',
    version='0.2.0',
    description='Ask question from bash shell and get tailored documentation response',
    author='VisionInit',
    author_email='contact@aimodels.org',
    url='https://github.com/visioninit/aiy',
    packages=find_packages(),
    install_requires=[
        'openai==0.26.4',
        'appdirs==1.4.4',
        'rich==13.3.1',
    ],
    entry_points={
        'console_scripts': [
            'aiyy=aiy:main',
        ],
    },
)
Traceback (most recent call last):
  File "/snap/aiyy/x5/bin/aiyy", line 5, in <module>
    from aiy import main
ModuleNotFoundError: No module named 'aiy'

I can’t figure out where the rest of the code is supposed to be. Is it supposed to be in bin? The example used in the documentation for yt-dlp has a setup.py but it’s quite complex… do I need to do something with a wheel? Or is the code supposed to be packaged with the snap?