Included bin/server script

bin/server

@@ -0,0 +1,12 @@
+#!/usr/bin/env python3
+
+from embedding_server import server
+
+if __name__ == "__main__":
+    import uvicorn
+
+    uvicorn.run(
+        server.app,
+        host="0.0.0.0",
+        port=8080,
+    )

embedding_server/server.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python3
-
 import logging
 import time
 from typing import Optional
@@ -51,13 +49,3 @@ async def root(body: EmbeddingBody, Authorization: Optional[str] = Header(None))
         "object": "list",
         "usage": {"prompt_tokens": 0, "total_tokens": 0},
     }
-
-
-if __name__ == "__main__":
-    import uvicorn
-
-    uvicorn.run(
-        app,
-        host="0.0.0.0",
-        port=8080,
-    )

setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-VERSION = "0.0.1"
+VERSION = "0.0.2"
 DESCRIPTION = "Embedding Server"
 LONG_DESCRIPTION = "Drop in replacement for OpenAI's embedding API. Can be used with official OpenAPI libraries"
 
@@ -23,9 +23,6 @@
         "Operating System :: MacOS :: MacOS X",
         "Operating System :: Microsoft :: Windows",
     ],
-    entry_points={
-        "console_scripts": [
-            "embedding-server = embedding_server.server:main",
-        ],
-    },
+    package_data={'embedding-server': ['bin/server']},
+    scripts=['embedding_server/bin/server']
 )