jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: update to validate prompt endpoint

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-04 22:07:40 -03:00
commit 41f05b2e85
3 changed files with 13 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/api/base.py
View file

@ -27,10 +27,3 @@ class CodeValidationResponse(BaseModel):
class PromptValidationResponse(BaseModel):
input_variables: list
valid: bool
def validate_prompt(template):
# Extract the input variables from template
input_variables = extract_input_variables_from_prompt(template)
return input_variables, len(input_variables) > 0

7
src/backend/langflow/api/validate.py
View file

@ -5,8 +5,8 @@ from langflow.api.base import (
CodeValidationResponse,
Prompt,
PromptValidationResponse,
validate_prompt,
)
from langflow.graph.utils import extract_input_variables_from_prompt
from langflow.utils.logger import logger
from langflow.utils.validate import validate_code
@ -29,7 +29,8 @@ def post_validate_code(code: Code):
@router.post("/prompt", status_code=200, response_model=PromptValidationResponse)
def post_validate_prompt(prompt: Prompt):
try:
input_variables, valid = validate_prompt(prompt.template)
return PromptValidationResponse(input_variables=input_variables, valid=valid)
input_variables = extract_input_variables_from_prompt(prompt.template)
return PromptValidationResponse(input_variables=input_variables)
except Exception as e:
logger.exception(e)
return HTTPException(status_code=500, detail=str(e))