From cf20d657ec122ed8057bf029dd87a07a7efa4e38 Mon Sep 17 00:00:00 2001
From: xuchuanyin <xuchuanyin@apache.org>
Date: Tue, 1 Apr 2025 11:39:21 +0800
Subject: [PATCH] fix bugs in assignment

1. assignment should use prompt file for assignment not generation_1
2. topic file is input for assignment not output, fix typos
---
 topicgpt_python/assignment.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/topicgpt_python/assignment.py b/topicgpt_python/assignment.py
index 46373de..22b632e 100644
--- a/topicgpt_python/assignment.py
+++ b/topicgpt_python/assignment.py
@@ -282,7 +282,7 @@ def assign_topics(api, model, data, prompt_file, out_file, topic_file, verbose):
     parser.add_argument(
         "--prompt_file",
         type=str,
-        default="prompt/generation_1.txt",
+        default="prompt/assignment.txt",
         help="File to read prompts from",
     )
     parser.add_argument(
@@ -295,7 +295,7 @@ def assign_topics(api, model, data, prompt_file, out_file, topic_file, verbose):
         "--topic_file",
         type=str,
         default="data/output/generation_1.md",
-        help="File to write topics to",
+        help="File to read topics from",
     )
 
     parser.add_argument(