Fix mqtt startup (#54)

* Fix MQTT startup

* Increase version number

* Update readme

Author: martinloretzzz

docs/content/deployment/index.md

@@ -95,7 +95,7 @@ If an error happens during processing the request, that error is published to th
 
 Start the docker container as explained in the HTTP API section.
 
-To connect the MCP to a tool like Claude Desktop, follow their [guide](https://modelcontextprotocol.io/docs/develop/connect-local-servers). The following line has to be added to the `claude_desktop_config.json`:
+To connect the MCP to a tool like Claude Desktop, follow their [guide](https://modelcontextprotocol.io/docs/develop/connect-local-servers). The following `tirex` line has to be added to the `claude_desktop_config.json` under `mcpServers`:
 ```json
 {
   "mcpServers": {

inference/README.md

@@ -3,6 +3,7 @@
 This docker container runs the TiRex model and provides the following APIs to interact with the model:
 - **HTTP API**
 - **MQTT**
+- **MCP**
 
 ## Using the docker container
 
@@ -73,7 +74,7 @@ If an error happens during processing the request, that error is published to th
 
 Start the docker container as explained in the HTTP API section.
 
-To connect the MCP to a tool like Claude Desktop, follow their [guide](https://modelcontextprotocol.io/docs/develop/connect-local-servers). The following line has to be added to the `claude_desktop_config.json`:
+To connect the MCP to a tool like Claude Desktop, follow their [guide](https://modelcontextprotocol.io/docs/develop/connect-local-servers). The following `tirex` line has to be added to the `claude_desktop_config.json` under `mcpServers`:
 ```json
 {
   "mcpServers": {

inference/app/mqtt_server.py

@@ -2,6 +2,7 @@
 # This software may be used and distributed according to the terms of the NXAI Community License Agreement.
 
 import json
+import time
 
 import paho.mqtt.client as mqtt
 import requests
@@ -22,6 +23,7 @@ def __init__(self, config: Settings):
         self.client.on_message = self.on_message
         self.client.on_connect = self.on_connect
         self.client.on_disconnect = self.on_disconnect
+        self.http_url = f"http://{self.config.http_host}:{self.config.http_port}"
 
     def on_message(self, client, userdata, msg):
         try:
@@ -39,7 +41,7 @@ def on_message(self, client, userdata, msg):
 
     def predict(self, context, prediction_length):
         response = requests.post(
-            f"http://{self.config.http_host}:{self.config.http_port}/forecast/quantiles",
+            f"{self.http_url}/forecast/quantiles",
             json={"context": context, "prediction_length": prediction_length},
         )
 
@@ -52,6 +54,8 @@ def predict(self, context, prediction_length):
 
     def connect(self, keepalive=60):
         try:
+            print(f"MQTT is waiting for the HTTP server at {self.http_url} to load the model and go online")
+            self.wait_for_api()
             print(f"Connecting to MQTT broker at {self.config.mqtt_broker_host}:{self.config.mqtt_broker_port}")
             self.client.connect(self.config.mqtt_broker_host, self.config.mqtt_broker_port, keepalive)
             self.client.loop_forever()
@@ -73,3 +77,14 @@ def on_connect(self, client, userdata, connect_flags, reason_code, properties):
     def on_disconnect(self, client, userdata, disconnect_flags, reason_code, properties):
         if reason_code != 0:
             print(f"Unexpected disconnection from MQTT broker with code: {reason_code}")
+
+    def wait_for_api(self, timeout=300):
+        for _ in range(timeout):
+            try:
+                response = requests.get(f"{self.http_url}/health")
+                if response.status_code == 200:
+                    return
+            except:
+                pass
+            time.sleep(1)
+        raise TimeoutError(f"MQTT can't connect to {self.http_url} in {timeout} seconds!")

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tirex-ts"
-version = "1.1.1"
+version = "1.2.0"
 authors = [
   { name="Andreas Auer", email="andreas.auer@nx-ai.com" },
   { name="Martin Loretz", email="martin.loretz@nx-ai.com" },