OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-06-26 18:15:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fc pipeline

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-31 23:04:23 -07:00
parent 6b42931cb5
commit ca98e4dea3
1 changed files with 30 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
examples/function_calling_filter_pipeline.py
View File

@ -41,12 +41,14 @@ class Pipeline:
TASK_MODEL: str TASK_MODEL: str
TEMPLATE: str TEMPLATE: str
OPENWEATHERMAP_API_KEY: str = ""
# Initialize valves # Initialize valves
self.valves = Valves( self.valves = Valves(
**{ **{
"pipelines": ["*"], # Connect to all pipelines "pipelines": ["*"], # Connect to all pipelines
"OPENAI_API_BASE_URL": "https://api.openai.com/v1", "OPENAI_API_BASE_URL": "https://api.openai.com/v1",
"OPENAI_API_KEY": os.getenv("OPENAI_API_KEY"), "OPENAI_API_KEY": os.getenv("OPENAI_API_KEY", "YOUR_OPENAI_API_KEY"),
"TASK_MODEL": "gpt-3.5-turbo", "TASK_MODEL": "gpt-3.5-turbo",
"TEMPLATE": """Use the following context as your learned knowledge, inside <context></context> XML tags. "TEMPLATE": """Use the following context as your learned knowledge, inside <context></context> XML tags.
<context> <context>
@ -62,6 +64,9 @@ And answer according to the language of the user's question.""",
) )
class Functions: class Functions:
def __init__(self, pipeline) -> None:
self.pipeline = pipeline
def get_current_weather( def get_current_weather(
self, self,
location: str, location: str,
@ -74,8 +79,29 @@ And answer according to the language of the user's question.""",
:param unit: The unit to get the weather in. Default is fahrenheit. :param unit: The unit to get the weather in. Default is fahrenheit.
:return: The current weather for the location. :return: The current weather for the location.
""" """
print(location, unit)
return f"{location}: Sunny" # 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: def calculator(self, equation: str) -> str:
""" """
@ -93,7 +119,7 @@ And answer according to the language of the user's question.""",
print(e) print(e)
return "Invalid equation" return "Invalid equation"
self.functions = Functions() self.functions = Functions(self)
async def on_startup(self): async def on_startup(self):
# This function is called when the server is started. # This function is called when the server is started.