Spaces:
Running
Running
Update app.py
Browse files
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
|
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",
|