try vscode plugin

Html/apps/utils/chapterUtil.py

@@ -0,0 +1,159 @@
+from typing import Callable, Any
+from AlgoriAgent.projects.algoriAgent.tools.judge_tools import judge
+
+CHAPTER_FINISH = 0
+CHAPTER_FOCUS = 1
+CHAPTER_LATTER = 2
+CHAPTER_FAILED = 3
+
+def tool_name_to_tool(tool_name_list):
+    tools = []
+    for tool_name in tool_name_list:
+        if tool_name == "":
+            continue
+        if tool_name == "judge":
+            tools.append(judge)
+    return tools
+
+def tool_name_to_tool_with_args(tool_name_list, tool_args_list) -> list[tuple[Callable, dict]]:
+    tools = []
+    for tool_name, tool_args in zip(tool_name_list, tool_args_list):
+        if tool_name == "":
+            continue
+        if tool_name == "judge":
+            tools.append((judge, {"name": tool_args[0]}))
+    return tools
+    
+class  Chapter():
+    def __init__(self, No = 1, focus = 2,title="", chapter = "", chapter_prompt = "", 
+                 score_prompt = "", append_tools : list[Callable[..., Any], dict] = None) -> None:
+        '''
+        No : 序号
+        focus : 0: Finished, 1: Focus, 2: Latter, 3: Failed
+        title : 子章节标题
+        chapter : 教案子章节内容
+        chapter_prompt : 教案的提示
+        score_prompt : 评测标准
+        append_tools : 追加工具函数
+        '''
+        self.No = No
+        self.focus = focus
+        self.chapter = chapter
+        self.chapter_prompt = chapter_prompt
+        self.score_prompt = score_prompt
+        self.append_tools = append_tools
+        self.memory = None
+        self.title =title
+
+
+    def GetChapter(self):
+        if self.focus == CHAPTER_FINISH: f = "(Finished Chapter)"
+        elif self.focus == CHAPTER_FOCUS: f = "(Your Now Chapter)"
+        elif self.focus == CHAPTER_LATTER: f = "(Ignore for Latter Chapter)"
+        elif self.focus == CHAPTER_FAILED: f = "(Failed Chapter)"
+        # des = f"{f} Chapter {self.No}: {self.chapter if self.focus else 'Hidden for further study.'}"
+        des = f"{f} Chapter {self.No}: "
+        if self.focus == CHAPTER_FINISH or self.focus == CHAPTER_FAILED: des += 'Hidden due to finished.'
+        if self.focus == CHAPTER_LATTER: des += 'Hidden for further study.'
+        if self.focus == CHAPTER_FOCUS:
+            des+=self.chapter
+            des+='\n'
+            des+='Here are some concrete instructions:\n'
+            des+=self.chapter_prompt+'\n'
+        
+        return des
+    
+
+
+    def Finished(self, successful: bool = True):
+        if successful: self.focus = 0
+        else: self.focus = 3
+    
+    def Focus(self):
+        self.focus = 1
+def load_chapters(markdown, markdown_prompt, score_prompt):
+    markdown_list = markdown.split("\n")
+    markdown_prompt_list = markdown_prompt.split("\n")
+    score_prompt_list = score_prompt.split("\n")
+    # 获取 H1 标题
+    title = ""
+    for line in markdown_list:
+        if line.startswith("# "):
+            title = line[2:]
+            break
+    
+    # 对于 H3 标题 构造每一个 Chapter
+    # 首先找到所有的 H3 标题 
+    chapter_dict = {}
+    chapter_sequence = []
+    for line in markdown_list:
+        if line.startswith("### "):
+            chapter_name = line[4:]
+            chapter_dict[chapter_name] = {}
+            chapter_sequence.append(chapter_name)
+    # 将 H3 标题 和 其对应的内容 构造成一个 Chapter
+
+    h3_name = ""
+    content = ""
+    for i in range(len(markdown_list)):
+        if markdown_list[i].startswith("### "): 
+            if(h3_name != ""):
+                chapter_dict[h3_name]["markdown"] = content
+            h3_name = markdown_list[i][4:]
+            content = ""
+            continue
+        if h3_name != "":
+            content += markdown_list[i]+"\n"
+    if(h3_name != ""):
+        chapter_dict[h3_name]["markdown"] = content
+    
+
+    h3_name = ""
+    content = ""
+    require_tools = []
+    require_tools_args = []
+    for i in range(len(markdown_prompt_list)):
+        if markdown_prompt_list[i].startswith("### "): 
+            if(h3_name != ""):
+                chapter_dict[h3_name]["markdown_prompt"] = content
+                chapter_dict[h3_name]["require_tools"] = tool_name_to_tool_with_args(require_tools, require_tools_args)
+            h3_name = markdown_prompt_list[i][4:]
+            content = ""
+            require_tools=[]
+            require_tools_args = []
+            continue
+
+        if h3_name != "":
+            if markdown_prompt_list[i].startswith("_require_tools"):
+                require_tools.append(markdown_prompt_list[i].split("=")[1].strip().split(",")[0])
+                require_tools_args.append((markdown_prompt_list[i].split("=")[1].strip().split(",")[1:]))
+                continue
+            content += markdown_prompt_list[i]+"\n"
+    if(h3_name != ""):
+        chapter_dict[h3_name]["markdown_prompt"] = content
+        chapter_dict[h3_name]["require_tools"] = tool_name_to_tool_with_args(require_tools, require_tools_args)
+        
+    
+    h3_name = ""
+    content = ""
+    for i in range(len(score_prompt_list)):
+        if score_prompt_list[i].startswith("### "): 
+            if(h3_name != ""):
+                chapter_dict[h3_name]["score_prompt"] = content
+            h3_name = score_prompt_list[i][4:]
+            content = ""
+            continue
+
+        if h3_name != "":
+            content += score_prompt_list[i]+"\n"
+    if(h3_name != ""):
+        chapter_dict[h3_name]["score_prompt"] = content
+
+
+    chapter_chain = []
+    No = 1
+    print (chapter_dict)
+    for chapter_name in chapter_sequence:
+        chapter_chain.append(Chapter(  No, CHAPTER_LATTER, chapter_name, chapter_dict[chapter_name]["markdown"], chapter_dict[chapter_name]["markdown_prompt"], chapter_dict[chapter_name]["score_prompt"],chapter_dict[chapter_name]["require_tools"]))
+        No+=1
+    return chapter_chain