refactor: replace name TEACHER/STUDENT with SYNTHESIZER/TRAINEE

Author: ChenZiHong-Gavin

.env.example

@@ -1,6 +1,6 @@
-TEACHER_MODEL=
-TEACHER_BASE_URL=
-TEACHER_API_KEY=
-STUDENT_MODEL=
-STUDENT_BASE_URL=
-STUDENT_API_KEY=
+SYNTHESIZER_MODEL=
+SYNTHESIZER_BASE_URL=
+SYNTHESIZER_API_KEY=
+TRAINEE_MODEL=
+TRAINEE_BASE_URL=
+TRAINEE_API_KEY=

baselines/EntiGraph/entigraph.py

@@ -95,7 +95,7 @@ def _post_process_synthetic_data(data):
 
 async def generate_synthetic_data_for_document(input_file, data_type):
     random.seed(42)
-    model_name = os.getenv("TEACHER_MODEL")
+    model_name = os.getenv("SYNTHESIZER_MODEL")
     task = BaselineTask(input_file, data_type)
 
     max_concurrent = 1000

baselines/EntiGraph/inference/devapi.py

@@ -10,10 +10,10 @@ async def gptqa(prompt: str,
           json_format: bool = False,
           temp: float = 1.0):
     client = AsyncOpenAI(
-        api_key=os.getenv("TEACHER_API_KEY"),
-        base_url=os.getenv("TEACHER_BASE_URL")
+        api_key=os.getenv("SYNTHESIZER_API_KEY"),
+        base_url=os.getenv("SYNTHESIZER_BASE_URL")
     )
-    openai_model_name = openai_model_name or os.getenv("TEACHER_MODEL")
+    openai_model_name = openai_model_name or os.getenv("SYNTHESIZER_MODEL")
 
     if json_format:
         completion = await client.chat.completions.create(

baselines/Genie/genie.py

@@ -112,9 +112,9 @@ async def process_chunk(content: str):
     load_dotenv()
 
     llm_client = OpenAIModel(
-        model_name=os.getenv("TEACHER_MODEL"),
-        api_key=os.getenv("TEACHER_API_KEY"),
-        base_url=os.getenv("TEACHER_BASE_URL")
+        model_name=os.getenv("SYNTHESIZER_MODEL"),
+        api_key=os.getenv("SYNTHESIZER_API_KEY"),
+        base_url=os.getenv("SYNTHESIZER_BASE_URL")
     )
 
     genie = Genie(llm_client=llm_client)

baselines/LongForm/longform.py

@@ -77,9 +77,9 @@ async def process_chunk(content: str):
     load_dotenv()
 
     llm_client = OpenAIModel(
-        model_name=os.getenv("TEACHER_MODEL"),
-        api_key=os.getenv("TEACHER_API_KEY"),
-        base_url=os.getenv("TEACHER_BASE_URL")
+        model_name=os.getenv("SYNTHESIZER_MODEL"),
+        api_key=os.getenv("SYNTHESIZER_API_KEY"),
+        base_url=os.getenv("SYNTHESIZER_BASE_URL")
     )
 
     longform = LongForm(llm_client=llm_client)

baselines/SELF-QA/self-qa.py

@@ -131,9 +131,9 @@ async def process_chunk(content: str):
     load_dotenv()
 
     llm_client = OpenAIModel(
-        model_name=os.getenv("TEACHER_MODEL"),
-        api_key=os.getenv("TEACHER_API_KEY"),
-        base_url=os.getenv("TEACHER_BASE_URL")
+        model_name=os.getenv("SYNTHESIZER_MODEL"),
+        api_key=os.getenv("SYNTHESIZER_API_KEY"),
+        base_url=os.getenv("SYNTHESIZER_BASE_URL")
     )
 
     self_qa = SelfQA(llm_client=llm_client)

baselines/Wrap/wrap.py

@@ -100,9 +100,9 @@ async def process_chunk(content: str):
     load_dotenv()
 
     llm_client = OpenAIModel(
-        model_name=os.getenv("TEACHER_MODEL"),
-        api_key=os.getenv("TEACHER_API_KEY"),
-        base_url=os.getenv("TEACHER_BASE_URL")
+        model_name=os.getenv("SYNTHESIZER_MODEL"),
+        api_key=os.getenv("SYNTHESIZER_API_KEY"),
+        base_url=os.getenv("SYNTHESIZER_BASE_URL")
     )
 
     wrap = Wrap(llm_client=llm_client)

generate.py

@@ -44,14 +44,14 @@ def save_config(global_config):
         raise ValueError(f"Invalid data type: {config['data_type']}")
 
     synthesizer_llm_client = OpenAIModel(
-        model_name=os.getenv("TEACHER_MODEL"),
-        api_key=os.getenv("TEACHER_API_KEY"),
-        base_url=os.getenv("TEACHER_BASE_URL")
+        model_name=os.getenv("SYNTHESIZER_MODEL"),
+        api_key=os.getenv("SYNTHESIZER_API_KEY"),
+        base_url=os.getenv("SYNTHESIZER_BASE_URL")
     )
     training_llm_client = OpenAIModel(
-        model_name=os.getenv("STUDENT_MODEL"),
-        api_key=os.getenv("STUDENT_API_KEY"),
-        base_url=os.getenv("STUDENT_BASE_URL")
+        model_name=os.getenv("TRAINEE_MODEL"),
+        api_key=os.getenv("TRAINEE_API_KEY"),
+        base_url=os.getenv("TRAINEE_BASE_URL")
     )
 
     traverse_strategy = TraverseStrategy(

webui/app.py

@@ -38,15 +38,15 @@ def init_graph_gen(config: dict, env: dict) -> GraphGen:
 
     # Set up LLM clients
     graph_gen.synthesizer_llm_client = OpenAIModel(
-        model_name=env.get("TEACHER_MODEL", ""),
-        base_url=env.get("TEACHER_BASE_URL", ""),
-        api_key=env.get("TEACHER_API_KEY", "")
+        model_name=env.get("SYNTHESIZER_MODEL", ""),
+        base_url=env.get("SYNTHESIZER_BASE_URL", ""),
+        api_key=env.get("SYNTHESIZER_API_KEY", "")
     )
 
     graph_gen.training_llm_client = OpenAIModel(
-        model_name=env.get("STUDENT_MODEL", ""),
-        base_url=env.get("STUDENT_BASE_URL", ""),
-        api_key=env.get("STUDENT_API_KEY", "")
+        model_name=env.get("TRAINEE_MODEL", ""),
+        base_url=env.get("TRAINEE_BASE_URL", ""),
+        api_key=env.get("TRAINEE_API_KEY", "")
     )
 
     # Set up tokenizer
@@ -116,12 +116,12 @@ def run_graphgen(
     save_config(config)
 
     env = {
-        "TEACHER_MODEL": teacher_model,
-        "TEACHER_BASE_URL": teacher_base_url,
-        "TEACHER_API_KEY": teacher_api_key,
-        "STUDENT_MODEL": student_model,
-        "STUDENT_BASE_URL": student_base_url,
-        "STUDENT_API_KEY": student_api_key
+        "SYNTHESIZER_MODEL": teacher_model,
+        "SYNTHESIZER_BASE_URL": teacher_base_url,
+        "SYNTHESIZER_API_KEY": teacher_api_key,
+        "TRAINEE_MODEL": student_model,
+        "TRAINEE_BASE_URL": student_base_url,
+        "TRAINEE_API_KEY": student_api_key
     }
     save_env(env)