#!/usr/bin/env python3

"""Update version in pyproject.toml and __init__.py files."""

import sys

import re

from pathlib import Path

def update_version(new_version: str) -> None:

"""Update version in project files."""

# Update pyproject.toml

pyproject_path = Path("pyproject.toml")

content = pyproject_path.read_text()

# Only update the version field in [project] section

content = re.sub(

r'^version = "[^"]*"',

f'version = "{new_version}"',

content,

count=1,

flags=re.MULTILINE

)

pyproject_path.write_text(content)

print(f"Updated pyproject.toml to version {new_version}")

# Update _version.py

version_path = Path("src/claude_agent_sdk/_version.py")

content = version_path.read_text()

# Only update __version__ assignment

content = re.sub(

r'^__version__ = "[^"]*"',

f'__version__ = "{new_version}"',

content,

count=1,

flags=re.MULTILINE

)

version_path.write_text(content)

print(f"Updated _version.py to version {new_version}")

if __name__ == "__main__":

if len(sys.argv) != 2:

print("Usage: python scripts/update_version.py <version>")

sys.exit(1)

update_version(sys.argv[1])