Update setup.py

codelion · Oct 24, 2024 · 4682976 · 4682976
1 parent 0fdf9d1
commit 4682976
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -4,6 +4,7 @@
     name="optillm",
     version="0.0.1",
     packages=find_packages(),
+    py_modules=['optillm'],
     install_requires=[
         "numpy",
         "networkx",
@@ -27,6 +28,11 @@
         "ipython",
         "ipykernel",
     ],
+    entry_points={
+        'console_scripts': [
+            'optillm=optillm:main',  # Points directly to the main function in optillm.py
+        ],
+    },
     author="codelion",
     author_email="[email protected]",
     description="An optimizing inference proxy for LLMs.",