81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
import sys
|
||
import os
|
||
import argparse
|
||
|
||
from autoterminal.config.loader import ConfigLoader
|
||
from autoterminal.config.manager import ConfigManager
|
||
from autoterminal.llm.client import LLMClient
|
||
from autoterminal.utils.helpers import clean_command
|
||
|
||
def main():
|
||
"""主程序入口"""
|
||
# 解析命令行参数
|
||
parser = argparse.ArgumentParser(description='AutoTerminal - 智能终端工具')
|
||
parser.add_argument('user_input', nargs='*', help='用户输入的自然语言命令')
|
||
parser.add_argument('--api-key', help='API密钥')
|
||
parser.add_argument('--base-url', help='Base URL')
|
||
parser.add_argument('--model', help='模型名称')
|
||
|
||
args = parser.parse_args()
|
||
|
||
# 合并用户输入
|
||
user_input = ' '.join(args.user_input).strip()
|
||
|
||
# 加载配置
|
||
config_loader = ConfigLoader()
|
||
config = config_loader.get_config()
|
||
|
||
# 命令行参数优先级最高
|
||
if args.api_key:
|
||
config['api_key'] = args.api_key
|
||
if args.base_url:
|
||
config['base_url'] = args.base_url
|
||
if args.model:
|
||
config['model'] = args.model
|
||
|
||
# 如果配置不完整，使用配置管理器初始化
|
||
config_manager = ConfigManager()
|
||
if not all([config.get('api_key'), config.get('base_url'), config.get('model')]):
|
||
config = config_manager.get_or_create_config()
|
||
if not config:
|
||
print("错误: 缺少必要的配置参数，请通过命令行参数或配置文件提供API密钥、Base URL和模型名称。")
|
||
return 1
|
||
# 如果有命令行参数输入，直接处理
|
||
if user_input:
|
||
# 初始化LLM客户端
|
||
try:
|
||
llm_client = LLMClient(config)
|
||
except Exception as e:
|
||
print(f"LLM客户端初始化失败: {e}")
|
||
return 1
|
||
|
||
# 调用LLM生成命令
|
||
try:
|
||
generated_command = llm_client.generate_command(user_input)
|
||
cleaned_command = clean_command(generated_command)
|
||
|
||
# 优化输出格式
|
||
print(f"\033[1;32m$\033[0m {cleaned_command}")
|
||
print("\033[1;37mPress Enter to execute...\033[0m")
|
||
|
||
# 等待用户回车确认执行
|
||
input()
|
||
|
||
# 在用户的环境中执行命令
|
||
os.system(cleaned_command)
|
||
|
||
except Exception as e:
|
||
print(f"命令生成失败: {e}")
|
||
return 1
|
||
|
||
return 0
|
||
else:
|
||
print("错误: 请提供要执行的命令，例如: python main.py \"查看当前目录\"")
|
||
return 1
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|