mirror of
https://github.com/coleam00/Archon.git
synced 2025-12-24 02:39:17 -05:00
Added import platform to detect the operating system Used platform.system() to check if it's Windows or not Adjusted pip and python paths: Windows: uses 'Scripts' folder and '.exe' extension Mac/Linux: uses 'bin' folder without extension The script now automatically detects the operating system and uses the appropriate paths for each one, making it compatible with both Windows and macOS/Linux systems.
70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
import os
|
|
import json
|
|
import subprocess
|
|
import sys
|
|
import platform
|
|
|
|
def setup_venv():
|
|
# Get the absolute path to the current directory
|
|
base_path = os.path.abspath(os.path.dirname(__file__))
|
|
venv_path = os.path.join(base_path, 'venv')
|
|
venv_created = False
|
|
|
|
# Create virtual environment if it doesn't exist
|
|
if not os.path.exists(venv_path):
|
|
print("Creating virtual environment...")
|
|
subprocess.run([sys.executable, '-m', 'venv', venv_path], check=True)
|
|
print("Virtual environment created successfully!")
|
|
venv_created = True
|
|
else:
|
|
print("Virtual environment already exists.")
|
|
|
|
# Install requirements if we just created the venv
|
|
if venv_created:
|
|
print("\nInstalling requirements...")
|
|
# Determine the correct pip path based on the OS
|
|
if platform.system() == "Windows":
|
|
pip_path = os.path.join(venv_path, 'Scripts', 'pip.exe')
|
|
else: # macOS or Linux
|
|
pip_path = os.path.join(venv_path, 'bin', 'pip')
|
|
|
|
requirements_path = os.path.join(base_path, 'requirements.txt')
|
|
subprocess.run([pip_path, 'install', '-r', requirements_path], check=True)
|
|
print("Requirements installed successfully!")
|
|
|
|
def generate_mcp_config():
|
|
# Get the absolute path to the current directory
|
|
base_path = os.path.abspath(os.path.dirname(__file__))
|
|
|
|
# Determine the correct python path based on the OS
|
|
if platform.system() == "Windows":
|
|
python_path = os.path.join(base_path, 'venv', 'Scripts', 'python.exe')
|
|
else: # macOS or Linux
|
|
python_path = os.path.join(base_path, 'venv', 'bin', 'python')
|
|
|
|
server_script_path = os.path.join(base_path, 'mcp_server.py')
|
|
|
|
# Create the config dictionary
|
|
config = {
|
|
"mcpServers": {
|
|
"archon": {
|
|
"command": python_path,
|
|
"args": [server_script_path]
|
|
}
|
|
}
|
|
}
|
|
|
|
# Write the config to a file
|
|
config_path = os.path.join(base_path, 'mcp-config.json')
|
|
with open(config_path, 'w') as f:
|
|
json.dump(config, f, indent=2)
|
|
|
|
print(f"\nMCP configuration has been written to: {config_path}")
|
|
print(f"\nMCP configuration for Cursor:\n\n{python_path} {server_script_path}")
|
|
print("\nMCP configuration for Windsurf/Claude Desktop:")
|
|
print(json.dumps(config, indent=2))
|
|
|
|
if __name__ == '__main__':
|
|
setup_venv()
|
|
generate_mcp_config()
|