1.调整模板，测试生成岗位-所需技能三元组

src/kg_construction/llm_construct_kg.py

@@ -16,8 +16,16 @@ def get_response_llm(prompt):
 def standard_format_triple(candidate_triple):
     template_triple_standard = """
         ### 指导:
-        给出一个候选三元组，判断该三元组是否为标准格式：(头实体, 关系, 尾实体)，如果该候选三元组是标准格式三元组则输出这个三元组，否则将该候选三元组更改为标准格式三元组并输出，你的回答只需要包含这个标准格式三元组：(头实体, 关系, 尾实体)，严格按照三元组格式，不需要包含其他的内容。
-
+        给出一个候选三元组，判断该三元组是否为标准格式：(头实体, 关系, 尾实体)，如果是则输出：YES，如果不是则输出：NO
+
+        ### 例子1
+        输入：(试模工程师, 需要技能, 精密模具独立试模经验)
+        输出：YES
+        
+        ### 例子2
+        输入：(上下模师傅, 需要技能, [具体需要的技能如：模具设计、数控操作等]])
+        输出：NO
+        
         ### 候选三元组
         {}
 
@@ -87,15 +95,16 @@ def realtion_generation(head, tail):
 def job_duty_skill(job_name, duty):
     template_result = """
         ### 指导:
-        给一个公司招聘的岗位名称 {} 作为头实体，给出：需要技能 作为关系，请根据岗位名称与岗位职责补全三元组，你的回答只需要包含这个补全完整的三元组：(头实体, 关系, 尾实体)，严格按照三元组格式，不需要包含其他的内容。
+        给一个公司招聘的岗位名称 {} 作为头实体，给出：需要技能 作为关系，请根据岗位名称与岗位职责补全三元组，你只需要回答这个补全完整的三元组：(头实体, 关系, 尾实体)，严格按照三元组格式，不需要包含其他的内容。
 
         ### 岗位职责
         {}
 
         ### 回答:
         """
-    template_result_improve = gen(template_result)
-    respond = get_response_llm(template_result_improve.format(job_name, duty))
+    #template_result_improve = gen(template_result)
+    #print(template_result_improve)
+    respond = get_response_llm(template_result.format(job_name, duty))
     return respond
 
 

src/kg_construction/manager.py

@@ -33,8 +33,8 @@ if __name__ == "__main__":
     data_list = readXls(path)
     for data in data_list:
         respond = job_duty_skill(data['岗位名称'], data['岗位职责'])
-        respond_standard = standard_format_triple(respond)
-        print(respond, respond_standard)
+        #respond_standard = standard_format_triple(respond)
+        print(respond)