pipelines/examples/function_calling_scaffold.py
2024-06-01 11:29:08 -07:00

29 lines
863 B
Python

from blueprints.function_calling_blueprint import Pipeline as FunctionCallingBlueprint
class Pipeline(FunctionCallingBlueprint):
class Valves(FunctionCallingBlueprint.Valves):
# Add your custom parameters here
pass
class Tools:
def __init__(self, pipeline) -> None:
self.pipeline = pipeline
# Add your custom tools here
# Please refer to function_calling_filter_pipeline.py for an example
# Pure Python code can be added here
pass
def __init__(self):
super().__init__()
self.id = "my_tools_pipeline"
self.name = "My Tools Pipeline"
self.valves = self.Valves(
**{
**self.valves.model_dump(),
"pipelines": ["*"], # Connect to all pipelines
},
)
self.tools = self.Tools(self)