update api

LLM.py

@@ -56,7 +56,7 @@ class base_llm:
 
 
 class openai_llm(base_llm):
-    def __init__(self,model = "gpt4o-0513") -> None:
+    def __init__(self,model = "gpt4o-0513",deployment = "gpt-4o-0806") -> None:
         super().__init__()
         self.model = model
         if "AZURE_OPENAI_ENDPOINT" not in os.environ or os.environ["AZURE_OPENAI_ENDPOINT"] == "":
@@ -68,13 +68,13 @@ class openai_llm(base_llm):
         if api_version == "":
             api_version = None
         self.client = AzureOpenAI(
-            azure_deployment= os.environ.get("AZURE_OPENAI_DEPLOYMENT",None),
+            azure_deployment= deployment,
             azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
             api_key=os.environ["AZURE_OPENAI_KEY"],
             api_version= api_version
             )
         self.async_client = AsyncAzureOpenAI(
-            azure_deployment= os.environ.get("AZURE_OPENAI_DEPLOYMENT",None),
+            azure_deployment= deployment,
             azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
             api_key=os.environ["AZURE_OPENAI_KEY"],
             api_version= api_version

agents.py

@@ -6,12 +6,12 @@ from prompts import *
 from utils import extract
 
 
-def get_llm(model = "gpt-4o-0806"):
-    return openai_llm(model)
+def get_llm(model = "gpt-4o-0806",deployment = "gpt-4o-0806"):
+    return openai_llm(model,deployment)
 
 def get_llms():
-    main_llm = get_llm("gpt4o (08-06)")
-    cheap_llm = get_llm("gpt-4o-mini (0718)")
+    main_llm = get_llm("gpt4o (08-06)","gpt-4o-0806")
+    cheap_llm = get_llm("gpt-4o-mini (0718)","gpt-4o-mini-0718")
     return main_llm,cheap_llm
 
 def judge_idea(i,j,idea0,idea1,topic,llm):