CultriX commited on
Commit
fdd250f
·
verified ·
1 Parent(s): c3cdef4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -68,7 +68,7 @@ async def async_stream_task(task_message, api_key, primary_system_message, criti
68
  documentation_agent = AssistantAgent(
69
  "documentation",
70
  model_client=model_client,
71
- system_message="You are a documentation assistant. Write a short and concise '--help' message for the provided code.",
72
  )
73
  doc_task = f"Generate a '--help' message for the following code:\n\n{final_output}"
74
  async for doc_message in documentation_agent.run_stream(task=doc_task):
@@ -111,7 +111,8 @@ iface = gr.Interface(
111
  inputs=[
112
  gr.Textbox(label="OpenAI API Key", type="password", placeholder="Enter your OpenAI API Key"),
113
  gr.Textbox(label="Primary Agent System Message", placeholder="Enter the system message for the primary agent", value="You are a creative assistant focused on producing high-quality code."),
114
- gr.Textbox(label="Critic Agent System Message", placeholder="Enter the system message for the critic agent", value="Critic. You are a helpful assistant highly skilled in evaluating the quality of a given code or response. Provide constructive feedback and respond with 'APPROVE' once the feedback is addressed."),
 
115
  gr.Textbox(label="Task Message", placeholder="Enter your task message"),
116
  ],
117
  outputs=[
 
68
  documentation_agent = AssistantAgent(
69
  "documentation",
70
  model_client=model_client,
71
+ system_message=documentation_system_message,
72
  )
73
  doc_task = f"Generate a '--help' message for the following code:\n\n{final_output}"
74
  async for doc_message in documentation_agent.run_stream(task=doc_task):
 
111
  inputs=[
112
  gr.Textbox(label="OpenAI API Key", type="password", placeholder="Enter your OpenAI API Key"),
113
  gr.Textbox(label="Primary Agent System Message", placeholder="Enter the system message for the primary agent", value="You are a creative assistant focused on producing high-quality code."),
114
+ gr.Textbox(label="Critic Agent System Message", placeholder="Enter the system message for the critic agent (requires APPROVAL tag!)", value="Critic. You are a helpful assistant highly skilled in evaluating the quality of a given code or response. Provide constructive feedback and respond with 'APPROVE' once the feedback is addressed."),
115
+ gr.Textbox(label="Documentation Agent System Message", placeholder="Enter the system message for the documentation agent", value="You are a documentation assistant. Write a short and concise '--help' message for the provided code."),
116
  gr.Textbox(label="Task Message", placeholder="Enter your task message"),
117
  ],
118
  outputs=[