94 lines
3.2 KiB
Python
94 lines
3.2 KiB
Python
"""
|
||
为workflow命令提供上下文信息
|
||
|
||
编写工作流实现代码需要的上下文：
|
||
1. IDE Service接口定义
|
||
2. ChatMark接口定义；
|
||
3. 工作流命令列表；
|
||
4. 工作流命令组织、实现规范；
|
||
5. 基础可用的函数信息；
|
||
"""
|
||
|
||
import os
|
||
from typing import List
|
||
|
||
def load_file_in_user_scripts(filename: str) -> str:
|
||
"""
|
||
从用户脚本目录中加载文件内容
|
||
"""
|
||
user_path = os.path.expanduser("~/.chat/scripts")
|
||
file_path = os.path.join(user_path, filename)
|
||
with open(file_path, "r") as f:
|
||
return f.read()
|
||
|
||
def load_local_file(filename: str) -> str:
|
||
"""
|
||
从当前脚本所在目录的相对目录加载文件内容
|
||
"""
|
||
script_dir = os.path.dirname(os.path.abspath(__file__))
|
||
file_path = os.path.join(script_dir, filename)
|
||
with open(file_path, "r") as f:
|
||
return f.read()
|
||
|
||
def load_existing_workflow_defines() -> str:
|
||
""" 从user scripts目录遍历找到所有command.yml文件，并加载其内容 """
|
||
merico_path = os.path.expanduser("~/.chat/scripts/merico")
|
||
community_path = os.path.expanduser("~/.chat/scripts/community")
|
||
custom_path = os.path.expanduser("~/.chat/scripts/custom")
|
||
|
||
root_paths = [merico_path]
|
||
# 遍历community_path、custom_path下直接子目录，将其添加到root_paths作为下一步的根目录
|
||
for path in [community_path, custom_path]:
|
||
for root, dirs, files in os.walk(path):
|
||
if root == path:
|
||
root_paths.extend([os.path.join(root, d) for d in dirs])
|
||
break
|
||
|
||
wrkflow_defines = []
|
||
# 遍历所有根目录，对每个根目录进行递归遍历，找到所有command.yml文件，并加载其内容
|
||
# 将目录名称与command.yml内容拼接成一个字符串，添加到wrkflow_defines列表中
|
||
# 例如：~/.chat/scripts/merico/github/commit/command.yml被找到，那么拼接的字符串为：
|
||
# 工作流命令/github.commit的定义：\n<command.yml内容>\n\n
|
||
for root_path in root_paths:
|
||
for root, dirs, files in os.walk(root_path):
|
||
if "command.yml" in files:
|
||
with open(os.path.join(root, "command.yml"), "r") as f:
|
||
wrkflow_defines.append(f"工作流命令/{root[len(root_path)+1:].replace(os.sep, '.')}的定义：\n{f.read()}\n\n")
|
||
return "\n".join(wrkflow_defines)
|
||
|
||
|
||
CONTEXTS = f"""
|
||
工作流开发需要的上下文信息：
|
||
|
||
|
||
# IDE Service接口定义及使用示例
|
||
IDE Service用于在工作流命令中访问与IDE相关的数据，以及调用IDE提供的功能。
|
||
## IDEService接口定义
|
||
接口定义：
|
||
{load_file_in_user_scripts("lib/ide_service/service.py")}
|
||
涉及类型定义：
|
||
{load_file_in_user_scripts("lib/ide_service/types.py")}
|
||
|
||
## IDEService接口示例
|
||
{load_local_file("ide_service_demo.py")}
|
||
|
||
|
||
# ChatMark接口使用示例
|
||
ChatMark用于在工作流命令中与用户交互，展示信息，获取用户输入。
|
||
{load_file_in_user_scripts("lib/chatmark/chatmark_example/main.py")}
|
||
ChatMark Form组件类似于HTML表单，用于组合多个组件，获取相关设置结果。
|
||
|
||
|
||
# 工作流命令规范
|
||
{load_local_file("workflow_guide.md")}
|
||
|
||
|
||
# 已有工作流命令定义
|
||
{load_existing_workflow_defines()}
|
||
|
||
|
||
# 工作流内部函数定义
|
||
{load_local_file("base_functions_guide.md")}
|
||
|
||
"""
|