dc-python-sdk 1.5.20__tar.gz → 1.5.21__tar.gz

{dc_python_sdk-1.5.20/src/dc_python_sdk.egg-info → dc_python_sdk-1.5.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dc-python-sdk
-Version: 1.5.20
+Version: 1.5.21
 Summary: Data Connector Python SDK
 Home-page: https://github.com/data-connector/dc-python-sdk
 Author: DataConnector

{dc_python_sdk-1.5.20 → dc_python_sdk-1.5.21}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dc-python-sdk"
-version = "1.5.20"
+version = "1.5.21"
 description = "Data Connector Python SDK"
 readme = "README.md"
 requires-python = ">=3.6"

{dc_python_sdk-1.5.20 → dc_python_sdk-1.5.21}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = dc-python-sdk
-version = 1.5.20
+version = 1.5.21
 author = DataConnector
 author_email = josh@dataconnector.com
 description = A small example package

{dc_python_sdk-1.5.20 → dc_python_sdk-1.5.21/src/dc_python_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dc-python-sdk
-Version: 1.5.20
+Version: 1.5.21
 Summary: Data Connector Python SDK
 Home-page: https://github.com/data-connector/dc-python-sdk
 Author: DataConnector

{dc_python_sdk-1.5.20 → dc_python_sdk-1.5.21}/src/dc_sdk/cli.py

@@ -1,6 +1,13 @@
 import sys
 from importlib.metadata import version
 
+def get_arg(name, default=None):
+    if name in sys.argv:
+        idx = sys.argv.index(name)
+        if idx + 1 < len(sys.argv):
+            return sys.argv[idx + 1]
+    return default
+
 def main():
     if len(sys.argv) < 2:
         print("Usage: dc-sdk [http|ai|ai-http]")
@@ -8,13 +15,15 @@ def main():
 
     command = sys.argv[1]
 
+    port = int(get_arg("--port", 5000))
+
     print("version: ", version("dc-python-sdk"))
 
     print(f"[DC SDK] Command: {command}")
 
     if command == "http":
         from dc_sdk.src.server import start_server
-        start_server()
+        start_server(port)
 
     elif command == "ai":
         from dc_sdk.src.ai import start_ai

{dc_python_sdk-1.5.20 → dc_python_sdk-1.5.21}/src/dc_sdk/src/ai_http.py

@@ -19,6 +19,7 @@ app = FastAPI()
 
 client = None
 workspace = os.getenv("WORKSPACE", "/workspace")
+port_for_connector = 5000
 
 # -----------------------------
 # MODELS
@@ -72,10 +73,9 @@ def get_free_port():
 
 def start_connector():
     global workspace
-    port = int(os.getenv("PORT", 6060))
-
+    global port_for_connector
     proc = subprocess.Popen(
-        ["dc-sdk", "http"],
+        ["dc-sdk", "http", "--port", str(port_for_connector)],
         cwd=workspace,
         stdout=subprocess.DEVNULL,
         stderr=subprocess.DEVNULL
@@ -84,15 +84,15 @@ def start_connector():
     # wait for health
     for _ in range(50):
         try:
-            res = requests.get(f"http://localhost:{port}/health")
+            res = requests.get(f"http://localhost:{port_for_connector}/health")
             if res.status_code == 200:
-                logger.info("Connector ready on port %s", port)
-                return {"process": proc, "port": port}
+                logger.info("Connector ready on port %s", port_for_connector)
+                return {"process": proc, "port": port_for_connector}
         except Exception:
-            logger.debug("Health check not ready yet on port %s", port)
+            logger.debug("Health check not ready yet on port %s", port_for_connector)
         time.sleep(0.1)
 
-    logger.error("Connector failed to become healthy on port %s within timeout", port)
+    logger.error("Connector failed to become healthy on port %s within timeout", port_for_connector)
     proc.terminate()
     raise Exception("Failed to start connector")
 
@@ -704,9 +704,12 @@ def clone_repo():
     subprocess.run(["git", "-C", workspace, "checkout", branch], check=True)
     subprocess.run(["git", "-C", workspace, "pull"], check=True)
 
-def start_ai_http():
+def start_ai_http(connector_port=5000):
     global client
     global workspace
+    global port_for_connector
+
+    port_for_connector = connector_port
 
     api_key = os.getenv("OPENAI_API_KEY")
     ai_port = int(os.getenv("AI_PORT", 5050))
@@ -721,9 +724,13 @@ def start_ai_http():
 
     # ✅ Write code-server config
     with open("/root/.config/code-server/config.yaml", "w") as f:
-        f.write(f"""bind-addr: 0.0.0.0:{code_port}
-auth: none
-""")
+        f.write(
+            f"bind-addr: 0.0.0.0:{code_port}\n"
+            "auth: none\n"
+        )
+
+    with open("/root/.config/code-server/config.yaml", "r") as f:
+        logger.info("code-server config:\n%s", f.read())
 
     # ✅ Ensure workspace + vscode dir exists
     os.makedirs(workspace, exist_ok=True)