NithikaShree commited on
Commit
3687e6b
·
verified ·
1 Parent(s): b7a0496

Create reviewer.py

Browse files
Files changed (1) hide show
  1. reviewer.py +16 -0
reviewer.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from llm import llm
2
+ from state import AgentState
3
+
4
+ PROMPT = """
5
+ Review the following HTML code.
6
+ Fix errors if any.
7
+ Improve performance, accessibility, and responsiveness.
8
+ Return the corrected FULL HTML code.
9
+
10
+ HTML:
11
+ {}
12
+ """
13
+
14
+ def reviewer(state: AgentState):
15
+ result = llm.invoke(PROMPT.format(state["code"]))
16
+ return {"code": result.content}