bstraehle commited on
Commit
8a597b5
1 Parent(s): d1c6ae1

Update agents.py

Browse files
Files changed (1) hide show
  1. agents.py +13 -3
agents.py CHANGED
@@ -2,12 +2,22 @@
2
 
3
  from openai import OpenAI
4
 
5
- from tools import escalate_to_human_json
6
- from tools import transfer_to_sales_agent_json, transfer_to_issues_repairs_agent_json, transfer_to_triage_agent_json
7
- from tools import execute_order_json, execute_refund_json, look_up_item_json
8
 
9
  MODEL = "gpt-4o-mini"
10
 
 
 
 
 
 
 
 
 
 
 
11
  def create_triage_agent(client):
12
  return client.beta.assistants.create(
13
  name="Triage Agent",
 
2
 
3
  from openai import OpenAI
4
 
5
+ from tools import escalate_to_human
6
+ from tools import transfer_to_sales_agent, transfer_to_issues_repairs_agent, transfer_to_triage_agent
7
+ from tools import execute_order, execute_refund, look_up_item
8
 
9
  MODEL = "gpt-4o-mini"
10
 
11
+ escalate_to_human_json = function_to_schema(escalate_to_human)
12
+
13
+ transfer_to_sales_agent_json = function_to_schema(transfer_to_sales_agent)
14
+ transfer_to_issues_repairs_agent_json = function_to_schema(transfer_to_issues_repairs_agent)
15
+ transfer_to_triage_agent_json = function_to_schema(transfer_to_triage_agent)
16
+
17
+ execute_order_json = function_to_schema(execute_order)
18
+ execute_refund_json = function_to_schema(execute_refund)
19
+ look_up_item_json = function_to_schema(look_up_item)
20
+
21
  def create_triage_agent(client):
22
  return client.beta.assistants.create(
23
  name="Triage Agent",