bstraehle commited on
Commit
0c8e3ae
·
verified ·
1 Parent(s): 4e432fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -6
app.py CHANGED
@@ -6,7 +6,12 @@ from utils import function_to_schema, show_json
6
 
7
  # Tools
8
 
9
- sales_agent, issues_repairs_agent, triage_agent = None, None, None
 
 
 
 
 
10
 
11
  def transfer_to_sales_agent():
12
  """Use for anything sales or buying related."""
@@ -16,11 +21,6 @@ def transfer_to_issues_repairs_agent():
16
  """Use for issues, repairs, or refunds."""
17
  set_current_agent(issues_repairs_agent)
18
 
19
- def transfer_to_triage_agent():
20
- """Call this if the user brings up a topic outside of your purview,
21
- including escalating to human."""
22
- set_current_agent(triage_agent)
23
-
24
  def escalate_to_human(summary):
25
  """Only call this if explicitly asked to."""
26
  print("Escalating to human agent...")
@@ -182,15 +182,19 @@ def extract_content_values(data):
182
  current_agent, current_thread = None, None
183
 
184
  def set_current_agent(agent):
 
185
  current_agent = agent
186
 
187
  def set_current_thread(thread):
 
188
  current_thread = thread
189
 
190
  def get_current_agent():
 
191
  return current_agent
192
 
193
  def get_current_thread():
 
194
  return current_thread
195
 
196
  #
@@ -198,14 +202,20 @@ def get_current_thread():
198
  _client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
199
 
200
  triage_agent = create_triage_agent(_client)
 
201
  sales_agent = create_sales_agent(_client)
 
202
  issues_repairs_agent = create_issues_repairs_agent(_client)
 
203
 
204
  set_current_agent(triage_agent)
205
 
206
  triage_thread = create_thread(_client)
 
207
  sales_thread = create_thread(_client)
 
208
  issues_repairs_thread = create_thread(_client)
 
209
 
210
  set_current_thread(triage_thread)
211
 
 
6
 
7
  # Tools
8
 
9
+ triage_agent, sales_agent, issues_repairs_agent = None, None, None
10
+
11
+ def transfer_to_triage_agent():
12
+ """Call this if the user brings up a topic outside of your purview,
13
+ including escalating to human."""
14
+ set_current_agent(triage_agent)
15
 
16
  def transfer_to_sales_agent():
17
  """Use for anything sales or buying related."""
 
21
  """Use for issues, repairs, or refunds."""
22
  set_current_agent(issues_repairs_agent)
23
 
 
 
 
 
 
24
  def escalate_to_human(summary):
25
  """Only call this if explicitly asked to."""
26
  print("Escalating to human agent...")
 
182
  current_agent, current_thread = None, None
183
 
184
  def set_current_agent(agent):
185
+ global current_agent
186
  current_agent = agent
187
 
188
  def set_current_thread(thread):
189
+ global current_thread
190
  current_thread = thread
191
 
192
  def get_current_agent():
193
+ global current_agent
194
  return current_agent
195
 
196
  def get_current_thread():
197
+ global current_thread
198
  return current_thread
199
 
200
  #
 
202
  _client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
203
 
204
  triage_agent = create_triage_agent(_client)
205
+ show_json("triage_agent", triage_agent)
206
  sales_agent = create_sales_agent(_client)
207
+ show_json("sales_agent", sales_agent)
208
  issues_repairs_agent = create_issues_repairs_agent(_client)
209
+ show_json("issues_repairs_agent", issues_repairs_agent)
210
 
211
  set_current_agent(triage_agent)
212
 
213
  triage_thread = create_thread(_client)
214
+ show_json("triage_thread", triage_thread)
215
  sales_thread = create_thread(_client)
216
+ show_json("sales_thread", sales_thread)
217
  issues_repairs_thread = create_thread(_client)
218
+ show_json("issues_repairs_thread", issues_repairs_thread)
219
 
220
  set_current_thread(triage_thread)
221