fix(graphgen): change model name

Author: ChenZiHong-Gavin

graphgen/operators/judge.py

@@ -7,15 +7,15 @@
 
 
 async def judge_statement( # pylint: disable=too-many-statements
-        student_llm_client: OpenAIModel,
+        training_llm_client: OpenAIModel,
         graph_storage: NetworkXStorage,
         rephrase_storage: JsonKVStorage,
         re_judge: bool = False,
         max_concurrent: int = 1000) -> NetworkXStorage:
     """
     Get all edges and nodes and judge them
 
-    :param student_llm_client: judge the statements to get comprehension loss
+    :param training_llm_client: judge the statements to get comprehension loss
     :param graph_storage: graph storage instance
     :param rephrase_storage: rephrase storage instance
     :param re_judge: re-judge the relations
@@ -46,7 +46,7 @@ async def _judge_single_relation(
                 judgements = []
                 gts = [gt for _, gt in descriptions]
                 for description, gt in descriptions:
-                    judgement = await student_llm_client.generate_topk_per_token(
+                    judgement = await training_llm_client.generate_topk_per_token(
                         STATEMENT_JUDGEMENT_PROMPT['TEMPLATE'].format(statement=description)
                     )
                     judgements.append(judgement[0].top_candidates)
@@ -94,7 +94,7 @@ async def _judge_single_entity(
                 judgements = []
                 gts = [gt for _, gt in descriptions]
                 for description, gt in descriptions:
-                    judgement = await student_llm_client.generate_topk_per_token(
+                    judgement = await training_llm_client.generate_topk_per_token(
                         STATEMENT_JUDGEMENT_PROMPT['TEMPLATE'].format(statement=description)
                     )
                     judgements.append(judgement[0].top_candidates)

graphgen/operators/quiz.py

@@ -8,15 +8,15 @@
 
 
 async def quiz(
-        teacher_llm_client: OpenAIModel,
+        synth_llm_client: OpenAIModel,
         graph_storage: NetworkXStorage,
         rephrase_storage: JsonKVStorage,
         max_samples: int = 1,
         max_concurrent: int = 1000) -> JsonKVStorage:
     """
     Get all edges and quiz them
 
-    :param teacher_llm_client: generate statements
+    :param synth_llm_client: generate statements
     :param graph_storage: graph storage instance
     :param rephrase_storage: rephrase storage instance
     :param max_samples: max samples for each edge
@@ -38,7 +38,7 @@ async def _process_single_quiz(
                 if descriptions:
                     return None
 
-                new_description = await teacher_llm_client.generate_answer(
+                new_description = await synth_llm_client.generate_answer(
                     prompt,
                     temperature=1
                 )