Merge pull request #44 from AkulS1008/main

Author: maybe-hello-world

pipelines/measurements/netflex_pipeline.py

@@ -0,0 +1,19 @@
+from netunicorn.base import Pipeline
+from netunicorn.library.tasks.measurements.ookla_speedtest import OoklaSpeedtest
+from netunicorn.library.tasks.data_transfer.send_data import SendData
+from netunicorn.library.tasks.data_transfer.fetch_data import FetchData
+import os 
+
+def netflex_full_loop_pipeline() -> Pipeline:
+    """
+    Run a speedtest using ookla CLI and then providing user-friendly analysis of the results
+    through passing the results to a RAG.
+    :return: pipeline to run
+    """
+    pipeline = (
+        Pipeline()
+        .then(OoklaSpeedtest(name="Ookla CLI Speedtest"))
+        .then(SendData(task_descriptors=[SendData.TaskDescriptor("Ookla CLI Speedtest", "ookla-speedtest", None)], endpoint=os.getenv('RAG_ENDPOINT'), name="send"))
+        .then(FetchData(send_data_task="send", endpoint=os.getenv("RAG_ENDPOINT"), name="fetch"))
+    )
+    return pipeline