ai-code-switcher 0.1.0__tar.gz

ai_code_switcher-0.1.0/PKG-INFO

@@ -0,0 +1,7 @@
+Metadata-Version: 2.4
+Name: ai-code-switcher
+Version: 0.1.0
+Summary: Switch AI coding tool profiles and launch the correct CLI
+License-Expression: MIT
+Requires-Python: >=3.8
+Requires-Dist: pyyaml>=5.0

ai_code_switcher-0.1.0/README.md

@@ -0,0 +1,98 @@
+# code-ai
+
+一个用于切换 AI 编码工具配置文件并启动相应 CLI 的工具。
+
+## 功能特性
+
+- 管理多个 AI 编码工具配置文件（Claude、Codex、Gemini）
+- 快速切换不同的配置文件
+- 统一的命令行接口
+- 支持一键升级所有 AI CLI 工具
+
+## 安装
+
+```bash
+pip install -e .
+```
+
+## 使用方法
+
+### 列出所有配置文件
+
+```bash
+code-ai list
+```
+
+### 添加新配置文件
+
+```bash
+code-ai add
+```
+
+### 使用指定配置文件启动
+
+```bash
+# 使用 fox-gemini 配置启动 Gemini CLI
+code-ai fox-gemini
+
+# 使用 4399 配置启动 Claude CLI
+code-ai 4399
+
+# 传递额外参数
+code-ai fox-claude -p "hi"
+```
+
+### 删除配置文件
+
+```bash
+code-ai remove <profile-name>
+```
+
+### 升级 AI CLI 工具
+
+```bash
+code-ai upgrade
+```
+
+该命令会通过 npm 升级以下工具：
+- @anthropic-ai/claude-code
+- @openai/codex
+- @google/gemini-cli
+
+### 查看版本
+
+```bash
+code-ai --version
+```
+
+### 查看帮助
+
+```bash
+code-ai --help
+```
+
+## 配置
+
+配置文件存储在用户目录下，包含各个配置文件的设置信息。
+
+## 开发
+
+### 项目结构
+
+```
+src/code_ai/
+├── __init__.py      # 包初始化
+├── cli.py           # 命令行入口
+├── config.py        # 配置管理
+├── launcher.py      # 启动器
+└── profiles.py      # 配置文件管理
+```
+
+### 依赖
+
+- Python >= 3.8
+- pyyaml >= 5.0
+
+## 许可证
+
+MIT

ai_code_switcher-0.1.0/pyproject.toml

@@ -0,0 +1,19 @@
+[build-system]
+requires = ["setuptools>=64"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "ai-code-switcher"
+version = "0.1.0"
+description = "Switch AI coding tool profiles and launch the correct CLI"
+license = "MIT"
+requires-python = ">=3.8"
+dependencies = [
+    "pyyaml>=5.0",
+]
+
+[project.scripts]
+code-ai = "code_ai.cli:main"
+
+[tool.setuptools.packages.find]
+where = ["src"]

ai_code_switcher-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

ai_code_switcher-0.1.0/src/ai_code_switcher.egg-info/PKG-INFO

@@ -0,0 +1,7 @@
+Metadata-Version: 2.4
+Name: ai-code-switcher
+Version: 0.1.0
+Summary: Switch AI coding tool profiles and launch the correct CLI
+License-Expression: MIT
+Requires-Python: >=3.8
+Requires-Dist: pyyaml>=5.0

ai_code_switcher-0.1.0/src/ai_code_switcher.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+README.md
+pyproject.toml
+src/ai_code_switcher.egg-info/PKG-INFO
+src/ai_code_switcher.egg-info/SOURCES.txt
+src/ai_code_switcher.egg-info/dependency_links.txt
+src/ai_code_switcher.egg-info/entry_points.txt
+src/ai_code_switcher.egg-info/requires.txt
+src/ai_code_switcher.egg-info/top_level.txt
+src/code_ai/__init__.py
+src/code_ai/cli.py
+src/code_ai/config.py
+src/code_ai/launcher.py
+src/code_ai/profiles.py

ai_code_switcher-0.1.0/src/ai_code_switcher.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

ai_code_switcher-0.1.0/src/ai_code_switcher.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+code-ai = code_ai.cli:main

ai_code_switcher-0.1.0/src/ai_code_switcher.egg-info/requires.txt

@@ -0,0 +1 @@
+pyyaml>=5.0

ai_code_switcher-0.1.0/src/ai_code_switcher.egg-info/top_level.txt

@@ -0,0 +1 @@
+code_ai

ai_code_switcher-0.1.0/src/code_ai/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

ai_code_switcher-0.1.0/src/code_ai/cli.py

@@ -0,0 +1,77 @@
+import sys
+import subprocess
+
+from .config import load_config, save_config
+from .profiles import list_profiles, add_profile, remove_profile
+from .launcher import launch
+
+
+USAGE = """\
+usage: code-ai <profile|command> [args...]
+
+commands:
+  list              List all profiles
+  add               Add a new profile interactively
+  upgrade           Upgrade claude, codex, and gemini CLI via npm
+  remove <name>     Remove a profile
+  --version         Show version
+  --help            Show this help
+
+examples:
+  code-ai fox-gemini          Launch Gemini CLI with fox profile
+  code-ai 4399                Launch Claude CLI with 4399 profile
+  code-ai fox-claude -p "hi"  Pass extra args to Claude CLI\
+"""
+
+
+def main():
+    args = sys.argv[1:]
+
+    if not args or args[0] in ("-h", "--help"):
+        print(USAGE)
+        return
+
+    if args[0] == "--version":
+        from . import __version__
+        print(f"code-ai {__version__}")
+        return
+
+    cmd = args[0]
+    config = load_config()
+
+    if cmd == "list":
+        list_profiles(config)
+    elif cmd == "add":
+        config = add_profile(config)
+        save_config(config)
+        print("Profile added.")
+    elif cmd == "upgrade":
+        upgrade()
+    elif cmd == "remove":
+        if len(args) < 2:
+            print("Usage: code-ai remove <name>")
+            sys.exit(1)
+        config = remove_profile(config, args[1])
+        save_config(config)
+    else:
+        profiles = config.get("profiles", {})
+        if cmd not in profiles:
+            print(f"Unknown profile or command: '{cmd}'")
+            print("Run 'code-ai list' to see available profiles.")
+            sys.exit(1)
+        launch(profiles[cmd], args[1:])
+
+
+UPGRADE_PACKAGES = [
+    "@anthropic-ai/claude-code",
+    "@openai/codex",
+    "@google/gemini-cli",
+]
+
+
+def upgrade():
+    print("Upgrading claude, codex, gemini CLI...")
+    result = subprocess.run(
+        ["npm", "install", "-g"] + UPGRADE_PACKAGES,
+    )
+    sys.exit(result.returncode)

ai_code_switcher-0.1.0/src/code_ai/config.py

@@ -0,0 +1,56 @@
+import os
+import sys
+from pathlib import Path
+
+import yaml
+
+CONFIG_DIR = Path.home() / ".code-ai"
+CONFIG_FILE = CONFIG_DIR / "config.yaml"
+
+DEFAULT_PROFILES = {
+    "fox-claude": {
+        "type": "claude",
+        "base_url": "https://code.newcli.com/claude/aws",
+        "token": "sk-ant-oat01-j6FU43eI3djEeLFWcw_2mtngnBqwSEsOXMYkJtidPDxeCiqIeaTtHg6p12wvi7HaYIsp7VsK5kVbdwYlBW_mCBnSYmZ-qAA",
+    },
+    "fox-gemini": {
+        "type": "gemini",
+        "base_url": "https://code.newcli.com/gemini",
+        "api_key": "sk-ant-oat01-j6FU43eI3djEeLFWcw_2mtngnBqwSEsOXMYkJtidPDxeCiqIeaTtHg6p12wvi7HaYIsp7VsK5kVbdwYlBW_mCBnSYmZ-qAA",
+    },
+    "fox-codex": {
+        "type": "codex",
+        "base_url": "https://code.newcli.com/codex/v1",
+        "api_key": "sk-ant-oat01-j6FU43eI3djEeLFWcw_2mtngnBqwSEsOXMYkJtidPDxeCiqIeaTtHg6p12wvi7HaYIsp7VsK5kVbdwYlBW_mCBnSYmZ-qAA",
+    },
+    "4399": {
+        "type": "claude",
+        "base_url": "https://4399code.com/claudecode",
+        "token": "sk-ant-oat01-ozUCp9rqL9sB46GoG5ISB1QPmlQPKZyk",
+    },
+    "2233": {
+        "type": "claude",
+        "base_url": "https://aicoding.2233.ai",
+        "token": "sk-Xn6481958ee796a17126b676467f0e06d5e6f39ac14Uusba",
+    },
+}
+
+
+def load_config():
+    if not CONFIG_FILE.exists():
+        init_config()
+    with open(CONFIG_FILE, "r", encoding="utf-8") as f:
+        data = yaml.safe_load(f)
+    if not data or "profiles" not in data:
+        data = {"profiles": {}}
+    return data
+
+
+def save_config(data):
+    os.makedirs(CONFIG_DIR, exist_ok=True)
+    with open(CONFIG_FILE, "w", encoding="utf-8") as f:
+        yaml.dump(data, f, default_flow_style=False, sort_keys=False, allow_unicode=True)
+
+
+def init_config():
+    save_config({"profiles": DEFAULT_PROFILES})

ai_code_switcher-0.1.0/src/code_ai/launcher.py

@@ -0,0 +1,48 @@
+import os
+import sys
+import shutil
+import subprocess
+
+ENV_MAP = {
+    "claude": {
+        "env": {"ANTHROPIC_BASE_URL": "base_url", "ANTHROPIC_AUTH_TOKEN": "token"},
+        "cmd": "claude",
+    },
+    "gemini": {
+        "env": {"GOOGLE_GEMINI_BASE_URL": "base_url", "GEMINI_API_KEY": "api_key"},
+        "cmd": "gemini",
+    },
+    "codex": {
+        "env": {"OPENAI_BASE_URL": "base_url", "OPENAI_API_KEY": "api_key"},
+        "cmd": "codex",
+    },
+}
+
+
+def launch(profile, extra_args):
+    ptype = profile["type"]
+    if ptype not in ENV_MAP:
+        print(f"Error: unknown profile type '{ptype}'.")
+        sys.exit(1)
+
+    spec = ENV_MAP[ptype]
+    cmd = spec["cmd"]
+
+    if not shutil.which(cmd):
+        print(f"Error: '{cmd}' not found in PATH. Install it first.")
+        sys.exit(1)
+
+    env = os.environ.copy()
+    for env_var, config_key in spec["env"].items():
+        value = profile.get(config_key)
+        if value:
+            env[env_var] = value
+
+    full_cmd = [cmd] + extra_args
+
+    if sys.platform == "win32":
+        result = subprocess.run(full_cmd, env=env)
+        sys.exit(result.returncode)
+    else:
+        os.environ.update(env)
+        os.execvp(cmd, full_cmd)

ai_code_switcher-0.1.0/src/code_ai/profiles.py

@@ -0,0 +1,61 @@
+import sys
+
+
+VALID_TYPES = ("claude", "gemini", "codex")
+
+
+def list_profiles(config):
+    profiles = config.get("profiles", {})
+    if not profiles:
+        print("No profiles configured.")
+        return
+    print(f"{'Name':<20} {'Type':<10} {'Base URL'}")
+    print("-" * 60)
+    for name, p in profiles.items():
+        print(f"{name:<20} {p['type']:<10} {p.get('base_url', '')}")
+
+
+def add_profile(config):
+    name = input("Profile name: ").strip()
+    if not name:
+        print("Error: name cannot be empty.")
+        sys.exit(1)
+    if name in config.get("profiles", {}):
+        print(f"Error: profile '{name}' already exists.")
+        sys.exit(1)
+
+    ptype = input(f"Type ({'/'.join(VALID_TYPES)}): ").strip().lower()
+    if ptype not in VALID_TYPES:
+        print(f"Error: type must be one of {VALID_TYPES}.")
+        sys.exit(1)
+
+    base_url = input("Base URL: ").strip()
+    if not base_url:
+        print("Error: base URL cannot be empty.")
+        sys.exit(1)
+
+    if ptype == "claude":
+        token = input("Auth token: ").strip()
+        config.setdefault("profiles", {})[name] = {
+            "type": ptype,
+            "base_url": base_url,
+            "token": token,
+        }
+    else:
+        api_key = input("API key: ").strip()
+        config.setdefault("profiles", {})[name] = {
+            "type": ptype,
+            "base_url": base_url,
+            "api_key": api_key,
+        }
+
+    return config
+
+
+def remove_profile(config, name):
+    if name not in config.get("profiles", {}):
+        print(f"Error: profile '{name}' not found.")
+        sys.exit(1)
+    del config["profiles"][name]
+    print(f"Removed profile '{name}'.")
+    return config