cyberandy commited on
Commit
b35b74b
·
verified ·
1 Parent(s): 35bfe8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -94,20 +94,20 @@ class AIShoppingAnalyzer:
94
  """Set up the team of agents"""
95
  user_proxy = UserProxyAgent(
96
  name="user_proxy",
97
- description="An e-commerce site owner looking for AI shopping analysis"
98
- )
99
-
100
- selector_prompt = (
101
- "You are coordinating an e-commerce analysis system. Select the next role from these participants:\n"
102
- "- The websurfer_agent searches products and analyzes website structure\n"
103
- "- The assistant_agent evaluates findings and makes recommendations\n"
104
- "- The user_proxy provides input when needed\n\n"
105
- "Return only the role name."
106
  )
107
 
108
  return SelectorGroupChat(
109
  participants=[websurfer_agent, assistant_agent, user_proxy],
110
- selector_prompt=selector_prompt,
 
 
 
 
 
 
 
 
111
  model_client=self.model_client,
112
  termination_condition=self.termination
113
  )
@@ -344,7 +344,7 @@ def create_gradio_interface() -> gr.Blocks:
344
  analyze_button = gr.Button(
345
  "Analyze Site",
346
  size="lg",
347
- variant="secondary"
348
  )
349
 
350
  # Right column for output
 
94
  """Set up the team of agents"""
95
  user_proxy = UserProxyAgent(
96
  name="user_proxy",
97
+ description="A user looking for product recommendations"
 
 
 
 
 
 
 
 
98
  )
99
 
100
  return SelectorGroupChat(
101
  participants=[websurfer_agent, assistant_agent, user_proxy],
102
+ selector_prompt="""You are coordinating a shopping assistance system. The following roles are available:
103
+ {roles}
104
+
105
+ Given the conversation history {history}, select the next role from {participants}.
106
+ - The websurfer_agent searches products and analyzes website structure
107
+ - The assistant_agent analyzes findings and makes recommendations
108
+ - The user_proxy provides input when needed
109
+
110
+ Return only the role name.""",
111
  model_client=self.model_client,
112
  termination_condition=self.termination
113
  )
 
344
  analyze_button = gr.Button(
345
  "Analyze Site",
346
  size="lg",
347
+ variant="primary"
348
  )
349
 
350
  # Right column for output