diff --git a/docker_pipeline.py b/docker_pipeline.py index b99bcf2..5eb390e 100644 --- a/docker_pipeline.py +++ b/docker_pipeline.py @@ -25,14 +25,17 @@ class Pipeline(FunctionCallingBlueprint): return os.system(docker_cmd) def __init__(self): - self.valves = self.Valves( - **{ - "pipelines": ["pipeline-testmodel"], # Connect to test pipeline - } - ) - self.tools = self.Tools(self) + super().__init__() # 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. # 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.name = "Docker Tool Pipeline" + self.valves = self.Valves( + **{ + **self.valves.model_dump(), + "pipelines": ["pipeline-testmodel"], # Connect to all pipelines + }, + ) + self.tools = self.Tools(self) \ No newline at end of file