Spaces:
Sleeping
Sleeping
File size: 1,827 Bytes
0242ab2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 | from langchain_openai import ChatOpenAI
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.output_parsers import JsonOutputParser
from core.config import settings
class WorkflowValidatorAgent:
def __init__(self):
self.llm = ChatOpenAI(
api_key=settings.MISTRAL_API_KEY,
base_url="https://api.mistral.ai/v1",
model="mistral-medium-latest",
temperature=0
)
self.prompt = ChatPromptTemplate.from_messages(
[
(
"system",
"""
You are a workflow validation agent.
Validate the workflow JSON.
Checks:
1. workflow_name exists.
2. execution_type exists and is:
- instant
- scheduled
3. steps exists and contains at least
one action.
4. Allowed actions:
- goto
- click
- fill
- wait
- search
- extract
5. If execution_type is scheduled,
schedule must exist.
6. If action is goto,
url must exist.
7. If action is search,
query must exist.
8. If action is fill,
selector and value must exist.
9. If action is click,
selector must exist.
10. If action is wait,
duration must exist.
11. If action is extract and limit
is missing, automatically set:
{{
"limit": 5
}}
12. Fix minor issues if possible.
Return ONLY JSON.
Format:
{{
"valid": true,
"message": "",
"workflow": {{}}
}}
"""
),
("user", "{workflow}")
]
)
self.chain = (
self.prompt
| self.llm
| JsonOutputParser()
)
def validate(
self,
workflow: dict
):
return self.chain.invoke(
{
"workflow": workflow
}
) |