OpenWebUI-Tools/Pipelines
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

trying again with the pipeline

Browse Source
This commit is contained in:
Josh Knapp 2025-02-11 09:35:06 -08:00
parent cc087dbf91
commit 04518b74fe
1 changed files with 12 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
docker_pipeline.py
View File

@ -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"
# self.id = "my_tools_pipeline"