This commit is contained in:
Timothy J. Baek 2024-06-24 10:37:57 -07:00
parent f54a66b86b
commit 6bad71adca
2 changed files with 16 additions and 4 deletions

View File

@ -116,6 +116,13 @@ async def get_pipe_models():
else:
function_module = app.state.FUNCTIONS[pipe.id]
if hasattr(function_module, "valves") and hasattr(function_module, "Valves"):
print(f"Getting valves for {pipe.id}")
valves = Functions.get_function_valves_by_id(pipe.id)
function_module.valves = function_module.Valves(
**(valves if valves else {})
)
# Check if function is a manifold
if hasattr(function_module, "type"):
if function_module.type == "manifold":

View File

@ -16,12 +16,16 @@ def extract_frontmatter(file_path):
try:
with open(file_path, "r", encoding="utf-8") as file:
first_line = file.readline()
if first_line.strip() != '"""':
# The file doesn't start with triple quotes
return {}
frontmatter_started = True
for line in file:
if '"""' in line:
if not frontmatter_started:
frontmatter_started = True
continue # skip the line with the opening triple quotes
else:
if frontmatter_started:
frontmatter_ended = True
break
@ -30,6 +34,7 @@ def extract_frontmatter(file_path):
if match:
key, value = match.groups()
frontmatter[key.strip()] = value.strip()
except FileNotFoundError:
print(f"Error: The file {file_path} does not exist.")
return {}