43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
|
# 在 ask/command.py 中
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
from devchat.llm import chat
|
|||
|
|
from lib.ide_service import IDEService
|
|||
|
|
|
|||
|
|
ROOT_WORKFLOW_DIR = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|||
|
|
sys.path.append(ROOT_WORKFLOW_DIR)
|
|||
|
|
|
|||
|
|
from chatflow.util.contexts import CONTEXTS
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
PROMPT = f"""
|
|||
|
|
{CONTEXTS.replace("{", "{{").replace("}", "}}")}
|
|||
|
|
""" + """
|
|||
|
|
当前选中代码：{selected_code}
|
|||
|
|
当前打开文件路径：{file_path}
|
|||
|
|
用户要求或问题：{question}
|
|||
|
|
"""
|
|||
|
|
@chat(prompt=PROMPT, stream_out=True)
|
|||
|
|
def ask(question, selected_code, file_path):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_selected_code():
|
|||
|
|
"""Retrieves the selected lines of code from the user's selection."""
|
|||
|
|
selected_data = IDEService().get_selected_range().dict()
|
|||
|
|
return selected_data
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main(question):
|
|||
|
|
selected_text = get_selected_code()
|
|||
|
|
file_path = selected_text.get("abspath", "")
|
|||
|
|
code_text = selected_text.get("text", "")
|
|||
|
|
|
|||
|
|
ask(question=question, selected_code=code_text, file_path=file_path)
|
|||
|
|
sys.exit(0)
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main(sys.argv[1])
|