Rahul-8799 commited on
Commit
59316be
·
verified ·
1 Parent(s): 7e9ce5e

Create agents/product_manager.py

Browse files
Files changed (1) hide show
  1. agents/product_manager.py +16 -0
agents/product_manager.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain_core.messages import AIMessage
2
+ from utils.inference import call_model
3
+
4
+ def run(state):
5
+ user_request = state["messages"][-1].content
6
+ prompt = f"""You are a product manager. Convert the user’s request into clear, detailed product requirements. Be concise, and list out the main goals, features, and any user expectations.
7
+
8
+ User Request:
9
+ {user_request}
10
+
11
+ Respond with clear bullet points."""
12
+ output = call_model(prompt)
13
+ return {
14
+ "messages": state["messages"] + [AIMessage(content=output)],
15
+ "product_requirements": output
16
+ }