@@ -164,8 +164,17 @@ <h1>Source code for agent_deployer.agent_deployer</h1><div class="highlight"><pr
164164< span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> mlrun.datastore.datastore_profile</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="p "> (</ span >
165165 < span class ="n "> DatastoreProfileV3io</ span > < span class ="p "> ,</ span >
166166 < span class ="n "> DatastoreProfileKafkaStream</ span > < span class ="p "> ,</ span >
167- < span class ="n "> DatastoreProfileTDEngine</ span > < span class ="p "> ,</ span >
168167< span class ="p "> )</ span >
168+
169+ < span class ="c1 "> # TimescaleDB support (mlrun >= 1.11), fallback to TDEngine for older versions</ span >
170+ < span class ="k "> try</ span > < span class ="p "> :</ span >
171+ < span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> mlrun.datastore.datastore_profile</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> DatastoreProfilePostgreSQL</ span >
172+
173+ < span class ="n "> _USE_TIMESCALEDB</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span >
174+ < span class ="k "> except</ span > < span class ="ne "> ImportError</ span > < span class ="p "> :</ span >
175+ < span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> mlrun.datastore.datastore_profile</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> DatastoreProfileTDEngine</ span >
176+
177+ < span class ="n "> _USE_TIMESCALEDB</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span >
169178< span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> mlrun.utils</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> logger</ span >
170179
171180
@@ -233,13 +242,24 @@ <h1>Source code for agent_deployer.agent_deployer</h1><div class="highlight"><pr
233242 < span class ="p "> )</ span >
234243 < span class ="k "> if</ span > < span class ="n "> mlconf</ span > < span class ="o "> .</ span > < span class ="n "> is_ce_mode</ span > < span class ="p "> ():</ span >
235244 < span class ="n "> mlrun_namespace</ span > < span class ="o "> =</ span > < span class ="n "> os</ span > < span class ="o "> .</ span > < span class ="n "> environ</ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class ="s2 "> "MLRUN_NAMESPACE"</ span > < span class ="p "> ,</ span > < span class ="s2 "> "mlrun"</ span > < span class ="p "> )</ span >
236- < span class ="n "> tsdb_profile</ span > < span class ="o "> =</ span > < span class ="n "> DatastoreProfileTDEngine</ span > < span class ="p "> (</ span >
237- < span class ="n "> name</ span > < span class ="o "> =</ span > < span class ="s2 "> "tdengine-tsdb-profile"</ span > < span class ="p "> ,</ span >
238- < span class ="n "> user</ span > < span class ="o "> =</ span > < span class ="s2 "> "root"</ span > < span class ="p "> ,</ span >
239- < span class ="n "> password</ span > < span class ="o "> =</ span > < span class ="s2 "> "taosdata"</ span > < span class ="p "> ,</ span >
240- < span class ="n "> host</ span > < span class ="o "> =</ span > < span class ="sa "> f</ span > < span class ="s2 "> "tdengine-tsdb.</ span > < span class ="si "> {</ span > < span class ="n "> mlrun_namespace</ span > < span class ="si "> }</ span > < span class ="s2 "> .svc.cluster.local"</ span > < span class ="p "> ,</ span >
241- < span class ="n "> port</ span > < span class ="o "> =</ span > < span class ="s2 "> "6041"</ span > < span class ="p "> ,</ span >
242- < span class ="p "> )</ span >
245+ < span class ="k "> if</ span > < span class ="n "> _USE_TIMESCALEDB</ span > < span class ="p "> :</ span >
246+ < span class ="n "> tsdb_profile</ span > < span class ="o "> =</ span > < span class ="n "> DatastoreProfilePostgreSQL</ span > < span class ="p "> (</ span >
247+ < span class ="n "> name</ span > < span class ="o "> =</ span > < span class ="s2 "> "timescaledb-tsdb-profile"</ span > < span class ="p "> ,</ span >
248+ < span class ="n "> user</ span > < span class ="o "> =</ span > < span class ="s2 "> "postgres"</ span > < span class ="p "> ,</ span >
249+ < span class ="n "> password</ span > < span class ="o "> =</ span > < span class ="s2 "> "postgres"</ span > < span class ="p "> ,</ span >
250+ < span class ="n "> host</ span > < span class ="o "> =</ span > < span class ="sa "> f</ span > < span class ="s2 "> "timescaledb.</ span > < span class ="si "> {</ span > < span class ="n "> mlrun_namespace</ span > < span class ="si "> }</ span > < span class ="s2 "> .svc.cluster.local"</ span > < span class ="p "> ,</ span >
251+ < span class ="n "> port</ span > < span class ="o "> =</ span > < span class ="s2 "> "5432"</ span > < span class ="p "> ,</ span >
252+ < span class ="n "> database</ span > < span class ="o "> =</ span > < span class ="s2 "> "postgres"</ span > < span class ="p "> ,</ span >
253+ < span class ="p "> )</ span >
254+ < span class ="k "> else</ span > < span class ="p "> :</ span >
255+ < span class ="c1 "> # Fallback for older mlrun versions</ span >
256+ < span class ="n "> tsdb_profile</ span > < span class ="o "> =</ span > < span class ="n "> DatastoreProfileTDEngine</ span > < span class ="p "> (</ span >
257+ < span class ="n "> name</ span > < span class ="o "> =</ span > < span class ="s2 "> "tdengine-tsdb-profile"</ span > < span class ="p "> ,</ span >
258+ < span class ="n "> user</ span > < span class ="o "> =</ span > < span class ="s2 "> "root"</ span > < span class ="p "> ,</ span >
259+ < span class ="n "> password</ span > < span class ="o "> =</ span > < span class ="s2 "> "taosdata"</ span > < span class ="p "> ,</ span >
260+ < span class ="n "> host</ span > < span class ="o "> =</ span > < span class ="sa "> f</ span > < span class ="s2 "> "tdengine-tsdb.</ span > < span class ="si "> {</ span > < span class ="n "> mlrun_namespace</ span > < span class ="si "> }</ span > < span class ="s2 "> .svc.cluster.local"</ span > < span class ="p "> ,</ span >
261+ < span class ="n "> port</ span > < span class ="o "> =</ span > < span class ="s2 "> "6041"</ span > < span class ="p "> ,</ span >
262+ < span class ="p "> )</ span >
243263
244264 < span class ="n "> stream_profile</ span > < span class ="o "> =</ span > < span class ="n "> DatastoreProfileKafkaStream</ span > < span class ="p "> (</ span >
245265 < span class ="n "> name</ span > < span class ="o "> =</ span > < span class ="s2 "> "kafka-stream-profile"</ span > < span class ="p "> ,</ span >
0 commit comments