|
@@ -100,8 +100,8 @@ class HierarchyClassifier(IHierarchyClassifier):
|
|
|
level1_title=level1_item["title"],
|
|
level1_title=level1_item["title"],
|
|
|
level2_titles=level2_titles
|
|
level2_titles=level2_titles
|
|
|
)
|
|
)
|
|
|
- with open('temp/document_temp/prompt.txt', "w", encoding="utf-8") as f:
|
|
|
|
|
- f.write(prompt["user"])
|
|
|
|
|
|
|
+ # with open('temp/document_temp/prompt.txt', "w", encoding="utf-8") as f:
|
|
|
|
|
+ # f.write(prompt["user"])
|
|
|
# 构建消息列表
|
|
# 构建消息列表
|
|
|
messages = [
|
|
messages = [
|
|
|
{"role": "system", "content": prompt["system"]},
|
|
{"role": "system", "content": prompt["system"]},
|
|
@@ -112,8 +112,8 @@ class HierarchyClassifier(IHierarchyClassifier):
|
|
|
|
|
|
|
|
# 批量异步调用LLM API
|
|
# 批量异步调用LLM API
|
|
|
llm_results = await self.llm_client.batch_call_async(llm_requests)
|
|
llm_results = await self.llm_client.batch_call_async(llm_requests)
|
|
|
- with open('temp/document_temp/llm_results.json', "w", encoding="utf-8") as f:
|
|
|
|
|
- json.dump(llm_results, f, ensure_ascii=False, indent=4)
|
|
|
|
|
|
|
+ # with open('temp/document_temp/llm_results.json', "w", encoding="utf-8") as f:
|
|
|
|
|
+ # json.dump(llm_results, f, ensure_ascii=False, indent=4)
|
|
|
# 处理分类结果
|
|
# 处理分类结果
|
|
|
classified_items = []
|
|
classified_items = []
|
|
|
category_stats = Counter()
|
|
category_stats = Counter()
|