Rahul-8799 commited on
Commit
2166c8f
·
verified ·
1 Parent(s): 0134d89

Create software_architect.py

Browse files
Files changed (1) hide show
  1. agents/software_architect.py +19 -0
agents/software_architect.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain_core.messages import AIMessage
2
+ from utils.inference import call_model
3
+
4
+ def run(state):
5
+ plan = state["project_plan"]
6
+ prompt = f"""You are a software architect. Based on the following project plan, describe the architecture for the front-end web app to be built. Include:
7
+
8
+ - Component structure
9
+ - Layout strategy
10
+ - Any libraries or standards
11
+ - HTML/CSS structuring notes
12
+
13
+ Project Plan:
14
+ {plan}"""
15
+ output = call_model(prompt)
16
+ return {
17
+ "messages": state["messages"] + [AIMessage(content=output)],
18
+ "architecture": output
19
+ }