diff --git a/docker_pipeline.py b/docker_pipeline.py index 7d8654f..dc1daf9 100644 --- a/docker_pipeline.py +++ b/docker_pipeline.py @@ -78,11 +78,13 @@ class Pipeline(FunctionCallingBlueprint): return "Invalid equation" def __init__(self): + super().__init__() self.valves = self.Valves( **{ "pipelines": ["pipeline-testmodel"], # Connect to all pipelines } ) + self.tools = self.Tools(self) # Optionally, you can set the id and name of the pipeline. # 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.