fix

LLM.py

@@ -56,27 +56,23 @@ class base_llm:
 
 
 class openai_llm(base_llm):
-    def __init__(self,model = "gpt4o-0513",deployment = "gpt-4o-0806") -> None:
+    def __init__(self,model = None,deployment = None,endpoint=None,api_key = None) -> None:
         super().__init__()
         self.model = model
-        if "AZURE_OPENAI_ENDPOINT" not in os.environ or os.environ["AZURE_OPENAI_ENDPOINT"] == "":
-            raise ValueError("AZURE_OPENAI_ENDPOINT is not set")
-        if "AZURE_OPENAI_KEY" not in os.environ or os.environ["AZURE_OPENAI_KEY"] == "":
-            raise ValueError("AZURE_OPENAI_KEY is not set")
         
-        api_version = os.environ.get("AZURE_OPENAI_API_VERSION",None)
+        api_version= "2024-02-15-preview"
         if api_version == "":
             api_version = None
         self.client = AzureOpenAI(
             azure_deployment= deployment,
-            azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
-            api_key=os.environ["AZURE_OPENAI_KEY"],
+            azure_endpoint=endpoint,
+            api_key=api_key,
             api_version= api_version
             )
         self.async_client = AsyncAzureOpenAI(
             azure_deployment= deployment,
-            azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
-            api_key=os.environ["AZURE_OPENAI_KEY"],
+            azure_endpoint=endpoint,
+            api_key=api_key,
             api_version= api_version
             )
 
@@ -114,8 +110,8 @@ class openai_llm(base_llm):
             client = AzureOpenAI(
             azure_endpoint=os.environ.get("EMBEDDING_API_ENDPOINT",None),
             api_key=os.environ.get("EMBEDDING_API_KEY",None),
-            api_version= os.environ.get("AZURE_OPENAI_API_VERSION",None),
-            azure_deployment="embedding-3-large"
+            api_version= "2024-02-15-preview",
+            azure_deployment="text-embedding-3-large"
             )
         else:
             client = self.client
@@ -143,8 +139,8 @@ class openai_llm(base_llm):
             async_client = AsyncAzureOpenAI(
             azure_endpoint=os.environ.get("EMBEDDING_API_ENDPOINT",None),
             api_key=os.environ.get("EMBEDDING_API_KEY",None),
-            api_version= os.environ.get("AZURE_OPENAI_API_VERSION",None),
-            azure_deployment="embedding-3-large"
+            api_version= "2024-02-15-preview",
+            azure_deployment="text-embedding-3-large"
             )
         else:
             async_client = self.async_client

agents.py

@@ -4,14 +4,14 @@ from searcher import Result,SementicSearcher
 from LLM import openai_llm
 from prompts import *
 from utils import extract
+import os
 
-
-def get_llm(model = "gpt-4o-0806",deployment = "gpt-4o-0806"):
-    return openai_llm(model,deployment)
+def get_llm(model,deployment ,endpoint,api_key):
+    return openai_llm(model,deployment,endpoint,api_key)
 
 def get_llms():
-    main_llm = get_llm("gpt4o (08-06)","gpt-4o-0806")
-    cheap_llm = get_llm("gpt-4o-mini (0718)","gpt-4o-mini-0718")
+    main_llm = get_llm("gpt4o (05-13)","gpt-4o-0513",os.environ.get("AZURE_OPENAI_API_ENDPOINT",None),os.environ.get("AZURE_OPENAI_API",None))
+    cheap_llm = get_llm("gpt-4o-mini (0718)","gpt-4o-mini-0718",os.environ.get("CHEAP_AZURE_OPENAI_API_ENDPOINT",None),os.environ.get("CHEAP_AZURE_OPENAI_API",None))
     return main_llm,cheap_llm
 
 def judge_idea(i,j,idea0,idea1,topic,llm):