This works

cross_py_builder/agent_manager.py

@@ -13,8 +13,8 @@ import requests
 from packaging.version import Version
 from tabulate import tabulate
 
-from build_agent import build_agent_version
-from zeroconf_server import ZeroconfServer
+from .build_agent import build_agent_version
+from .zeroconf_server import ZeroconfServer
 
 DEFAULT_PORT = 9001
 TIMEOUT = 10
@@ -32,9 +32,9 @@ def find_server_ips():
         ZeroconfServer.stop()
 
     # get known hosts
-    with open("../known_hosts", "r") as file:
-        lines = file.readlines()
-    hostnames.extend(lines)
+    # with open("../known_hosts", "r") as file:
+    #     lines = file.readlines()
+    #     hostnames.extend(lines)
 
     return hostnames
 

cross_py_builder/build_agent.py

@@ -15,7 +15,7 @@ import glob
 import shutil
 import uuid
 import platform
-from zeroconf_server import ZeroconfServer
+from .zeroconf_server import ZeroconfServer
 
 APP_NAME = "cross_py_builder"
 build_agent_version = "0.1.34"
@@ -394,8 +394,7 @@ def delete_cache():
     return jsonify(job_cache()), 200
 
 
-if __name__ == "__main__":
-
+def main():
     print(f"===== {APP_NAME} Build Agent (v{build_agent_version}) =====")
     ZeroconfServer.configure("_crosspybuilder._tcp.local.", socket.gethostname(), app_port)
     try:
@@ -405,3 +404,6 @@ if __name__ == "__main__":
         pass
     finally:
         ZeroconfServer.stop()
+
+if __name__ == "__main__":
+    main()

setup.py

@@ -8,7 +8,7 @@ setup(
     entry_points={
         "console_scripts": [
             "cross-py-agent=cross_py_builder.build_agent:main",
-            "cross_py_builder=cross_py_builder.agent_manager:main",
+            "cross-py-builder=cross_py_builder.agent_manager:main",
         ],
     },
     author="Brett Williams",