Upload 2 files

deepboreai_sdk/cli.py

@@ -0,0 +1,30 @@
+import typer
+import json
+from deepboreai_sdk.sdk import DeepBoreAI
+
+app = typer.Typer()
+client = DeepBoreAI()
+
+@app.command()
+def ingest(file: str):
+    with open(file, 'r') as f:
+        data = json.load(f)
+    result = client.post_telemetry(data)
+    typer.echo(result)
+
+@app.command()
+def export(out: str = 'drilling_data.csv'):
+    path = client.export_csv(out)
+    typer.echo(f"Exported to {path}")
+
+@app.command()
+def stream():
+    def callback(data):
+        typer.echo(f"Live Alert: {data}")
+    typer.echo("Listening to live data stream...")
+    client.watch_live(callback)
+    input("Press Enter to stop...
+")
+
+if __name__ == "__main__":
+    app()

deepboreai_sdk/sdk.py

@@ -0,0 +1,37 @@
+import requests
+import websocket
+import threading
+import json
+
+class DeepBoreAI:
+    def __init__(self, host='http://localhost:8000'):
+        self.host = host
+
+    def post_telemetry(self, data: dict):
+        url = f"{self.host}/ingest"
+        response = requests.post(url, json=data)
+        return response.json()
+
+    def get_recent_alerts(self):
+        url = f"{self.host}/history"
+        response = requests.get(url)
+        return response.json()
+
+    def export_csv(self, out_path='drilling_data.csv'):
+        url = f"{self.host}/export"
+        r = requests.get(url)
+        with open(out_path, 'wb') as f:
+            f.write(r.content)
+        return out_path
+
+    def watch_live(self, callback):
+        def on_message(ws, message):
+            data = json.loads(message)
+            callback(data)
+
+        ws = websocket.WebSocketApp(f"{self.host.replace('http', 'ws')}/ws",
+                                    on_message=on_message)
+        thread = threading.Thread(target=ws.run_forever)
+        thread.daemon = True
+        thread.start()
+        return ws