Add gpt 4o mini and claude 3.5

run.py

@@ -1,10 +1,10 @@
 # Copyright Sierra
 
-import os
-import json
-import random
 import argparse
+import json
 import multiprocessing
+import os
+import random
 from concurrent.futures import ThreadPoolExecutor
 from datetime import datetime
 
@@ -188,7 +188,9 @@ def main():
             "gpt-3.5-turbo-1106",
             "gpt-3.5-turbo-0125",
             "gpt-4o",
+            "gpt-4o-mini",
             # anthropic api models
+            "claude-3-5-sonnet-20240620",
             "claude-3-opus-20240229",
             "claude-3-sonnet-20240229",
             "claude-3-haiku-20240307",

tau_bench/agents/gpt_function_calling_agent.py

@@ -4,14 +4,13 @@
 from typing import Dict, List
 
 from openai import OpenAI
-from tenacity import retry, stop_after_attempt, wait_random_exponential
-
 from tau_bench.agents.base import BaseAgent
 from tau_bench.agents.utils import (
     message_to_action,
     message_to_dict,
     pretty_print_conversation,
 )
+from tenacity import retry, stop_after_attempt, wait_random_exponential
 
 client = None
 
@@ -44,6 +43,7 @@ def chat_completion_request(
 
 
 prompt_price_per_million = {
+    "gpt-4o-mini": 0.15,
     "gpt-4o": 5,
     "gpt-4-turbo": 10,
     "gpt-4-32k-0613": 60,
@@ -52,6 +52,7 @@ def chat_completion_request(
     "meta-llama/Meta-Llama-3-70B-Instruct": 1.0,
 }
 completion_price_per_million = {
+    "gpt-4o-mini": 0.60,
     "gpt-4o": 15,
     "gpt-4-turbo": 30,
     "gpt-4-32k-0613": 120,