bstraehle commited on
Commit
39bf620
·
verified ·
1 Parent(s): 348cc85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -13
app.py CHANGED
@@ -17,7 +17,7 @@ from utils import show_json
17
 
18
  def create_thread(client):
19
  thread = client.beta.threads.create()
20
- show_json("thread", thread)
21
  return thread
22
 
23
  def create_message(client, thread, msg):
@@ -26,7 +26,7 @@ def create_message(client, thread, msg):
26
  thread_id=thread.id,
27
  content=msg,
28
  )
29
- show_json("message", message)
30
  return message
31
 
32
  def create_run(client, assistant, thread):
@@ -34,7 +34,7 @@ def create_run(client, assistant, thread):
34
  assistant_id=assistant.id,
35
  thread_id=thread.id,
36
  )
37
- show_json("run", run)
38
  return run
39
 
40
  def wait_on_run(client, thread, run):
@@ -44,7 +44,7 @@ def wait_on_run(client, thread, run):
44
  run_id=run.id,
45
  )
46
  time.sleep(0.25)
47
- show_json("run", run)
48
  return run
49
 
50
  def list_run_steps(client, thread, run):
@@ -55,7 +55,6 @@ def list_run_steps(client, thread, run):
55
  )
56
  for step in run_steps.data:
57
  step_details = step.step_details
58
- #print(json.dumps(show_json("step_details", step_details), indent=4))
59
  show_json("step_details", step_details)
60
  return run_steps
61
 
@@ -63,7 +62,7 @@ def list_messages(client, thread):
63
  messages = client.beta.threads.messages.list(
64
  thread_id=thread.id
65
  )
66
- show_json("messages", messages)
67
  return messages
68
 
69
  def extract_content_values(data):
@@ -80,18 +79,12 @@ _client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
80
 
81
  ###
82
  triage_agent = create_triage_agent(_client)
83
- show_json("triage_agent", triage_agent)
84
  sales_agent = create_sales_agent(_client)
85
- show_json("sales_agent", sales_agent)
86
  issues_repairs_agent = create_issues_repairs_agent(_client)
87
- show_json("issues_repairs_agent", issues_repairs_agent)
88
 
89
  triage_thread = create_thread(_client)
90
- show_json("triage_thread", triage_thread)
91
  sales_thread = create_thread(_client)
92
- show_json("sales_thread", sales_thread)
93
  issues_repairs_thread = create_thread(_client)
94
- show_json("issues_repairs_thread", issues_repairs_thread)
95
 
96
  _assistant = triage_agent
97
  _thread = triage_thread
@@ -123,7 +116,7 @@ gr.ChatInterface(
123
  chatbot=gr.Chatbot(height=300),
124
  textbox=gr.Textbox(placeholder="Question", container=False, scale=7),
125
  title="Multi-Agent Orchestration",
126
- description="Demo with three agents using hand-off pattern: triage agent, sales agent, and issues & repairs agent",
127
  retry_btn=None,
128
  undo_btn=None,
129
  clear_btn="Clear",
 
17
 
18
  def create_thread(client):
19
  thread = client.beta.threads.create()
20
+ #show_json("thread", thread)
21
  return thread
22
 
23
  def create_message(client, thread, msg):
 
26
  thread_id=thread.id,
27
  content=msg,
28
  )
29
+ #show_json("message", message)
30
  return message
31
 
32
  def create_run(client, assistant, thread):
 
34
  assistant_id=assistant.id,
35
  thread_id=thread.id,
36
  )
37
+ #show_json("run", run)
38
  return run
39
 
40
  def wait_on_run(client, thread, run):
 
44
  run_id=run.id,
45
  )
46
  time.sleep(0.25)
47
+ #show_json("run", run)
48
  return run
49
 
50
  def list_run_steps(client, thread, run):
 
55
  )
56
  for step in run_steps.data:
57
  step_details = step.step_details
 
58
  show_json("step_details", step_details)
59
  return run_steps
60
 
 
62
  messages = client.beta.threads.messages.list(
63
  thread_id=thread.id
64
  )
65
+ #show_json("messages", messages)
66
  return messages
67
 
68
  def extract_content_values(data):
 
79
 
80
  ###
81
  triage_agent = create_triage_agent(_client)
 
82
  sales_agent = create_sales_agent(_client)
 
83
  issues_repairs_agent = create_issues_repairs_agent(_client)
 
84
 
85
  triage_thread = create_thread(_client)
 
86
  sales_thread = create_thread(_client)
 
87
  issues_repairs_thread = create_thread(_client)
 
88
 
89
  _assistant = triage_agent
90
  _thread = triage_thread
 
116
  chatbot=gr.Chatbot(height=300),
117
  textbox=gr.Textbox(placeholder="Question", container=False, scale=7),
118
  title="Multi-Agent Orchestration",
119
+ description="Demo using hand-off pattern: triage agent, sales agent, and issues & repairs agent",
120
  retry_btn=None,
121
  undo_btn=None,
122
  clear_btn="Clear",