From 04518b74fe6658fe5d1779c48d035689532d9542 Mon Sep 17 00:00:00 2001
From: Josh Knapp <jknapp85@gmail.com>
Date: Tue, 11 Feb 2025 09:35:06 -0800
Subject: [PATCH] trying again with the pipeline

---
 docker_pipeline.py | 77 ++++++++--------------------------------------
 1 file changed, 12 insertions(+), 65 deletions(-)

diff --git a/docker_pipeline.py b/docker_pipeline.py
index f8f7f38..b99bcf2 100644
--- a/docker_pipeline.py
+++ b/docker_pipeline.py
@@ -1,7 +1,6 @@
 import os
 import requests
 from typing import Literal, List, Optional
-from datetime import datetime
 
 
 from blueprints.function_calling_blueprint import Pipeline as FunctionCallingBlueprint
@@ -12,75 +11,23 @@ class Pipeline(FunctionCallingBlueprint):
         def __init__(self, pipeline) -> None:
             self.pipeline = pipeline
 
-        def get_current_time(
-            self,
-        ) -> str:
+        def call_docker(
+            self,image: str,command: Optional[str] = None):
             """
-            Get the current time.
-
-            :return: The current time.
+            Run a Docker container tool with the specified image and an optional command.
+            :param image: The Docker image name to run
+            :param command: Optional command to run in the container (default: None)
+            :return: The output from running the Docker container
             """
-
-            now = datetime.now()
-            current_time = now.strftime("%H:%M:%S")
-            return f"Current Time = {current_time}"
-
-        def get_current_weather(
-            self,
-            location: str,
-            unit: Literal["metric", "fahrenheit"] = "fahrenheit",
-        ) -> str:
-            """
-            Get the current weather for a location. If the location is not found, return an empty string.
-
-            :param location: The location to get the weather for.
-            :param unit: The unit to get the weather in. Default is fahrenheit.
-            :return: The current weather for the location.
-            """
-
-            # https://openweathermap.org/api
-
-            if self.pipeline.valves.OPENWEATHERMAP_API_KEY == "":
-                return "OpenWeatherMap API Key not set, ask the user to set it up."
-            else:
-                units = "imperial" if unit == "fahrenheit" else "metric"
-                params = {
-                    "q": location,
-                    "appid": self.pipeline.valves.OPENWEATHERMAP_API_KEY,
-                    "units": units,
-                }
-
-                response = requests.get(
-                    "http://api.openweathermap.org/data/2.5/weather", params=params
-                )
-                response.raise_for_status()  # Raises an HTTPError for bad responses
-                data = response.json()
-
-                weather_description = data["weather"][0]["description"]
-                temperature = data["main"]["temp"]
-
-                return f"{location}: {weather_description.capitalize()}, {temperature}°{unit.capitalize()[0]}"
-
-        def calculator(self, equation: str) -> str:
-            """
-            Calculate the result of an equation.
-
-            :param equation: The equation to calculate.
-            """
-
-            # Avoid using eval in production code
-            # https://nedbatchelder.com/blog/201206/eval_really_is_dangerous.html
-            try:
-                result = eval(equation)
-                return f"{equation} = {result}"
-            except Exception as e:
-                print(e)
-                return "Invalid equation"
+            docker_cmd = f"docker run {image}"
+            if command:
+                docker_cmd += f" {command}"
+            return os.system(docker_cmd)
 
     def __init__(self):
         self.valves = self.Valves(
             **{
-                "pipelines": ["pipeline-testmodel"],  # Connect to all pipelines
+                "pipelines": ["pipeline-testmodel"],  # Connect to test pipeline
             }
         )
         self.tools = self.Tools(self)
@@ -88,4 +35,4 @@ class Pipeline(FunctionCallingBlueprint):
         # Best practice is to not specify the id so that it can be automatically inferred from the filename, so that users can install multiple versions of the same pipeline.
         # The identifier must be unique across all pipelines.
         # The identifier must be an alphanumeric string that can include underscores or hyphens. It cannot contain spaces, special characters, slashes, or backslashes.
-        # self.id = "my_tools_pipeline"
\ No newline at end of file
+        # self.id = "my_tools_pipeline"