PyPI - klaude-code - Versions diffs - 2.10.2__py3-none-any.whl → 2.10.4__py3-none-any.whl - Mend

klaude-code 2.10.2py3-none-any.whl → 2.10.4py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (63) hide show

klaude_code/auth/AGENTS.md +4 -24
klaude_code/auth/__init__.py +1 -17
klaude_code/cli/auth_cmd.py +3 -53
klaude_code/cli/list_model.py +0 -50
klaude_code/config/assets/builtin_config.yaml +7 -35
klaude_code/config/config.py +5 -42
klaude_code/const.py +5 -2
klaude_code/core/agent_profile.py +2 -10
klaude_code/core/backtrack/__init__.py +3 -0
klaude_code/core/backtrack/manager.py +48 -0
klaude_code/core/memory.py +25 -9
klaude_code/core/task.py +53 -7
klaude_code/core/tool/__init__.py +2 -0
klaude_code/core/tool/backtrack/__init__.py +3 -0
klaude_code/core/tool/backtrack/backtrack_tool.md +17 -0
klaude_code/core/tool/backtrack/backtrack_tool.py +65 -0
klaude_code/core/tool/context.py +5 -0
klaude_code/core/turn.py +3 -0
klaude_code/llm/anthropic/input.py +28 -4
klaude_code/llm/input_common.py +70 -1
klaude_code/llm/openai_compatible/input.py +5 -2
klaude_code/llm/openrouter/input.py +5 -2
klaude_code/llm/registry.py +0 -1
klaude_code/protocol/events.py +10 -0
klaude_code/protocol/llm_param.py +0 -1
klaude_code/protocol/message.py +10 -1
klaude_code/protocol/tools.py +1 -0
klaude_code/session/session.py +111 -2
klaude_code/session/store.py +2 -0
klaude_code/skill/assets/executing-plans/SKILL.md +84 -0
klaude_code/skill/assets/writing-plans/SKILL.md +116 -0
klaude_code/tui/commands.py +15 -0
klaude_code/tui/components/developer.py +1 -1
klaude_code/tui/components/errors.py +2 -4
klaude_code/tui/components/metadata.py +5 -10
klaude_code/tui/components/rich/markdown.py +5 -1
klaude_code/tui/components/rich/status.py +7 -76
klaude_code/tui/components/rich/theme.py +12 -2
klaude_code/tui/components/tools.py +31 -18
klaude_code/tui/components/user_input.py +1 -1
klaude_code/tui/display.py +4 -0
klaude_code/tui/input/completers.py +51 -17
klaude_code/tui/input/images.py +127 -0
klaude_code/tui/input/prompt_toolkit.py +16 -2
klaude_code/tui/machine.py +26 -8
klaude_code/tui/renderer.py +97 -0
klaude_code/tui/runner.py +7 -2
klaude_code/tui/terminal/image.py +28 -12
klaude_code/ui/terminal/title.py +8 -3
{klaude_code-2.10.2.dist-info → klaude_code-2.10.4.dist-info}/METADATA +1 -1
{klaude_code-2.10.2.dist-info → klaude_code-2.10.4.dist-info}/RECORD +53 -56
klaude_code/auth/antigravity/__init__.py +0 -20
klaude_code/auth/antigravity/exceptions.py +0 -17
klaude_code/auth/antigravity/oauth.py +0 -315
klaude_code/auth/antigravity/pkce.py +0 -25
klaude_code/auth/antigravity/token_manager.py +0 -27
klaude_code/core/prompts/prompt-antigravity.md +0 -80
klaude_code/llm/antigravity/__init__.py +0 -3
klaude_code/llm/antigravity/client.py +0 -558
klaude_code/llm/antigravity/input.py +0 -268
klaude_code/skill/assets/create-plan/SKILL.md +0 -74
{klaude_code-2.10.2.dist-info → klaude_code-2.10.4.dist-info}/WHEEL +0 -0
{klaude_code-2.10.2.dist-info → klaude_code-2.10.4.dist-info}/entry_points.txt +0 -0

klaude_code/auth/antigravity/oauth.py DELETED Viewed

@@ -1,315 +0,0 @@
-"""OAuth PKCE flow for Antigravity authentication."""
-import base64
-import json
-import secrets
-import time
-import webbrowser
-from http.server import BaseHTTPRequestHandler, HTTPServer
-from threading import Thread
-from typing import Any, cast
-from urllib.parse import parse_qs, urlencode, urlparse
-import httpx
-from klaude_code.auth.antigravity.exceptions import (
-    AntigravityNotLoggedInError,
-    AntigravityOAuthError,
-    AntigravityTokenExpiredError,
-)
-from klaude_code.auth.antigravity.pkce import generate_pkce
-from klaude_code.auth.antigravity.token_manager import AntigravityAuthState, AntigravityTokenManager
-# OAuth configuration (decoded from base64 for compatibility with pi implementation)
-CLIENT_ID = base64.b64decode(
-    "MTA3MTAwNjA2MDU5MS10bWhzc2luMmgyMWxjcmUyMzV2dG9sb2poNGc0MDNlcC5hcHBzLmdvb2dsZXVzZXJjb250ZW50LmNvbQ=="
-).decode()
-CLIENT_SECRET = base64.b64decode("R09DU1BYLUs1OEZXUjQ4NkxkTEoxbUxCOHNYQzR6NnFEQWY=").decode()
-REDIRECT_URI = "http://localhost:51121/oauth-callback"
-REDIRECT_PORT = 51121
-# Google OAuth endpoints
-AUTH_URL = "https://accounts.google.com/o/oauth2/v2/auth"
-TOKEN_URL = "https://oauth2.googleapis.com/token"
-# Antigravity requires additional scopes
-SCOPES = [
-    "https://www.googleapis.com/auth/cloud-platform",
-    "https://www.googleapis.com/auth/userinfo.email",
-    "https://www.googleapis.com/auth/userinfo.profile",
-    "https://www.googleapis.com/auth/cclog",
-    "https://www.googleapis.com/auth/experimentsandconfigs",
-]
-# Fallback project ID when discovery fails
-DEFAULT_PROJECT_ID = "rising-fact-p41fc"
-# Cloud Code Assist endpoint
-CLOUDCODE_ENDPOINT = "https://cloudcode-pa.googleapis.com"
-class OAuthCallbackHandler(BaseHTTPRequestHandler):
-    """HTTP request handler for OAuth callback."""
-    code: str | None = None
-    state: str | None = None
-    error: str | None = None
-    def log_message(self, format: str, *args: Any) -> None:
-        """Suppress HTTP server logs."""
-        pass
-    def do_GET(self) -> None:
-        """Handle GET request from OAuth callback."""
-        parsed = urlparse(self.path)
-        params = parse_qs(parsed.query)
-        OAuthCallbackHandler.code = params.get("code", [None])[0]
-        OAuthCallbackHandler.state = params.get("state", [None])[0]
-        OAuthCallbackHandler.error = params.get("error", [None])[0]
-        self.send_response(200)
-        self.send_header("Content-Type", "text/html")
-        self.end_headers()
-        if OAuthCallbackHandler.error:
-            html = f"""
-            <html><body style="font-family: sans-serif; text-align: center; padding: 50px;">
-            <h1>Authentication Failed</h1>
-            <p>Error: {OAuthCallbackHandler.error}</p>
-            <p>Please close this window and try again.</p>
-            </body></html>
-            """
-        else:
-            html = """
-            <html><body style="font-family: sans-serif; text-align: center; padding: 50px;">
-            <h1>Authentication Successful!</h1>
-            <p>You can close this window now.</p>
-            <script>setTimeout(function() { window.close(); }, 2000);</script>
-            </body></html>
-            """
-        self.wfile.write(html.encode())
-def _discover_project(access_token: str) -> str:
-    """Discover or provision a project for the user."""
-    headers = {
-        "Authorization": f"Bearer {access_token}",
-        "Content-Type": "application/json",
-        "User-Agent": "google-api-nodejs-client/9.15.1",
-        "X-Goog-Api-Client": "google-cloud-sdk vscode_cloudshelleditor/0.1",
-        "Client-Metadata": json.dumps(
-            {
-                "ideType": "IDE_UNSPECIFIED",
-                "platform": "PLATFORM_UNSPECIFIED",
-                "pluginType": "GEMINI",
-            }
-        ),
-    }
-    try:
-        with httpx.Client() as client:
-            response = client.post(
-                f"{CLOUDCODE_ENDPOINT}/v1internal:loadCodeAssist",
-                headers=headers,
-                json={
-                    "metadata": {
-                        "ideType": "IDE_UNSPECIFIED",
-                        "platform": "PLATFORM_UNSPECIFIED",
-                        "pluginType": "GEMINI",
-                    },
-                },
-                timeout=30,
-            )
-            if response.status_code == 200:
-                data: dict[str, Any] = response.json()
-                project = data.get("cloudaicompanionProject")
-                if isinstance(project, str) and project:
-                    return project
-                if isinstance(project, dict):
-                    project_dict = cast(dict[str, Any], project)
-                    project_id = project_dict.get("id")
-                    if isinstance(project_id, str) and project_id:
-                        return project_id
-    except Exception:
-        pass
-    return DEFAULT_PROJECT_ID
-def _get_user_email(access_token: str) -> str | None:
-    """Get user email from the access token."""
-    try:
-        with httpx.Client() as client:
-            response = client.get(
-                "https://www.googleapis.com/oauth2/v1/userinfo?alt=json",
-                headers={"Authorization": f"Bearer {access_token}"},
-                timeout=10,
-            )
-            if response.status_code == 200:
-                data = response.json()
-                return data.get("email")
-    except Exception:
-        pass
-    return None
-class AntigravityOAuth:
-    """Handle OAuth PKCE flow for Antigravity authentication."""
-    def __init__(self, token_manager: AntigravityTokenManager | None = None):
-        self.token_manager = token_manager or AntigravityTokenManager()
-    def login(self) -> AntigravityAuthState:
-        """Run the complete OAuth login flow."""
-        verifier, challenge = generate_pkce()
-        state = secrets.token_urlsafe(32)
-        # Build authorization URL
-        auth_params = {
-            "client_id": CLIENT_ID,
-            "response_type": "code",
-            "redirect_uri": REDIRECT_URI,
-            "scope": " ".join(SCOPES),
-            "code_challenge": challenge,
-            "code_challenge_method": "S256",
-            "state": state,
-            "access_type": "offline",
-            "prompt": "consent",
-        }
-        auth_url = f"{AUTH_URL}?{urlencode(auth_params)}"
-        # Reset callback handler state
-        OAuthCallbackHandler.code = None
-        OAuthCallbackHandler.state = None
-        OAuthCallbackHandler.error = None
-        # Start callback server
-        server = HTTPServer(("localhost", REDIRECT_PORT), OAuthCallbackHandler)
-        server_thread = Thread(target=server.handle_request)
-        server_thread.start()
-        # Open browser for user to authenticate
-        webbrowser.open(auth_url)
-        # Wait for callback
-        server_thread.join(timeout=300)  # 5 minute timeout
-        server.server_close()
-        # Check for errors
-        if OAuthCallbackHandler.error:
-            raise AntigravityOAuthError(f"OAuth error: {OAuthCallbackHandler.error}")
-        if not OAuthCallbackHandler.code:
-            raise AntigravityOAuthError("No authorization code received")
-        if OAuthCallbackHandler.state is None or OAuthCallbackHandler.state != state:
-            raise AntigravityOAuthError("OAuth state mismatch")
-        # Exchange code for tokens
-        auth_state = self._exchange_code(OAuthCallbackHandler.code, verifier)
-        # Save tokens
-        self.token_manager.save(auth_state)
-        return auth_state
-    def _exchange_code(self, code: str, verifier: str) -> AntigravityAuthState:
-        """Exchange authorization code for tokens."""
-        data = {
-            "client_id": CLIENT_ID,
-            "client_secret": CLIENT_SECRET,
-            "code": code,
-            "grant_type": "authorization_code",
-            "redirect_uri": REDIRECT_URI,
-            "code_verifier": verifier,
-        }
-        with httpx.Client() as client:
-            response = client.post(TOKEN_URL, data=data, timeout=30)
-        if response.status_code != 200:
-            raise AntigravityOAuthError(f"Token exchange failed: {response.text}")
-        tokens = response.json()
-        access_token = tokens["access_token"]
-        refresh_token = tokens.get("refresh_token")
-        expires_in = tokens.get("expires_in", 3600)
-        if not refresh_token:
-            raise AntigravityOAuthError("No refresh token received. Please try again.")
-        # Get user email
-        email = _get_user_email(access_token)
-        # Discover project
-        project_id = _discover_project(access_token)
-        # Calculate expiry time with 5 minute buffer
-        expires_at = int(time.time()) + expires_in - 300
-        return AntigravityAuthState(
-            access_token=access_token,
-            refresh_token=refresh_token,
-            expires_at=expires_at,
-            project_id=project_id,
-            email=email,
-        )
-    def refresh(self) -> AntigravityAuthState:
-        """Refresh the access token using refresh token with file locking.
-        Uses file locking to prevent multiple instances from refreshing simultaneously.
-        If another instance has already refreshed, returns the updated state.
-        """
-        def do_refresh(current_state: AntigravityAuthState) -> AntigravityAuthState:
-            data = {
-                "client_id": CLIENT_ID,
-                "client_secret": CLIENT_SECRET,
-                "refresh_token": current_state.refresh_token,
-                "grant_type": "refresh_token",
-            }
-            with httpx.Client() as client:
-                response = client.post(TOKEN_URL, data=data, timeout=30)
-            if response.status_code != 200:
-                raise AntigravityTokenExpiredError(f"Token refresh failed: {response.text}")
-            tokens = response.json()
-            access_token = tokens["access_token"]
-            refresh_token = tokens.get("refresh_token", current_state.refresh_token)
-            expires_in = tokens.get("expires_in", 3600)
-            # Calculate expiry time with 5 minute buffer
-            expires_at = int(time.time()) + expires_in - 300
-            return AntigravityAuthState(
-                access_token=access_token,
-                refresh_token=refresh_token,
-                expires_at=expires_at,
-                project_id=current_state.project_id,
-                email=current_state.email,
-            )
-        try:
-            return self.token_manager.refresh_with_lock(do_refresh)
-        except ValueError as e:
-            raise AntigravityNotLoggedInError(str(e)) from e
-    def ensure_valid_token(self) -> tuple[str, str]:
-        """Ensure we have a valid access token, refreshing if needed.
-        Returns:
-            Tuple of (access_token, project_id).
-        """
-        state = self.token_manager.get_state()
-        if state is None:
-            raise AntigravityNotLoggedInError("Not logged in to Antigravity. Run 'klaude login antigravity' first.")
-        if state.is_expired():
-            state = self.refresh()
-        return state.access_token, state.project_id

klaude_code/auth/antigravity/pkce.py DELETED Viewed

@@ -1,25 +0,0 @@
-"""PKCE utilities for Antigravity OAuth."""
-import base64
-import hashlib
-import secrets
-def base64url_encode(data: bytes) -> str:
-    """Encode bytes as base64url string without padding."""
-    return base64.urlsafe_b64encode(data).rstrip(b"=").decode("ascii")
-def generate_pkce() -> tuple[str, str]:
-    """Generate PKCE code verifier and challenge.
-    Returns:
-        Tuple of (verifier, challenge).
-    """
-    verifier_bytes = secrets.token_bytes(32)
-    verifier = base64url_encode(verifier_bytes)
-    challenge_hash = hashlib.sha256(verifier.encode("ascii")).digest()
-    challenge = base64url_encode(challenge_hash)
-    return verifier, challenge

klaude_code/auth/antigravity/token_manager.py DELETED Viewed

@@ -1,27 +0,0 @@
-"""Token storage and management for Antigravity authentication."""
-from pathlib import Path
-from typing import Any
-from klaude_code.auth.base import BaseAuthState, BaseTokenManager
-class AntigravityAuthState(BaseAuthState):
-    """Stored authentication state for Antigravity."""
-    project_id: str
-    email: str | None = None
-class AntigravityTokenManager(BaseTokenManager[AntigravityAuthState]):
-    """Manage Antigravity OAuth tokens."""
-    def __init__(self, auth_file: Path | None = None):
-        super().__init__(auth_file)
-    @property
-    def storage_key(self) -> str:
-        return "antigravity"
-    def _create_state(self, data: dict[str, Any]) -> AntigravityAuthState:
-        return AntigravityAuthState.model_validate(data)

klaude_code/core/prompts/prompt-antigravity.md DELETED Viewed

@@ -1,80 +0,0 @@
-<identity>
-You are Antigravity, a powerful agentic AI coding assistant designed by the Google DeepMind team working on Advanced Agentic Coding.
-You are pair programming with a USER to solve their coding task. The task may require creating a new codebase, modifying or debugging an existing codebase, or simply answering a question.
-The USER will send you requests, which you must always prioritize addressing. Along with each USER request, we will attach additional metadata about their current state, such as what files they have open and where their cursor is.
-This information may or may not be relevant to the coding task, it is up for you to decide.
-</identity>
-<tool_calling>
-Call tools as you normally would. The following list provides additional guidance to help you avoid errors:
-  - **Absolute paths only**. When using tools that accept file path arguments, ALWAYS use the absolute file path.
-</tool_calling>
-<web_application_development>
-## Technology Stack
-Your web applications should be built using the following technologies:
-1. **Core**: Use HTML for structure and JavaScript for logic.
-2. **Styling (CSS)**: Use Vanilla CSS for maximum flexibility and control. Avoid using TailwindCSS unless the USER explicitly requests it; in this case, first confirm which TailwindCSS version to use.
-3. **Web App**: If the USER specifies that they want a more complex web app, use a framework like Next.js or Vite. Only do this if the USER explicitly requests a web app.
-4. **New Project Creation**: If you need to use a framework for a new app, use `npx` with the appropriate script, but there are some rules to follow:
-   - Use `npx -y` to automatically install the script and its dependencies
-   - You MUST run the command with `--help` flag to see all available options first
-   - Initialize the app in the current directory with `./` (example: `npx -y create-vite-app@latest ./`)
-   - You should run in non-interactive mode so that the user doesn't need to input anything
-5. **Running Locally**: When running locally, use `npm run dev` or equivalent dev server. Only build the production bundle if the USER explicitly requests it or you are validating the code for correctness.
-# Design Aesthetics
-1. **Use Rich Aesthetics**: The USER should be wowed at first glance by the design. Use best practices in modern web design (e.g. vibrant colors, dark modes, glassmorphism, and dynamic animations) to create a stunning first impression. Failure to do this is UNACCEPTABLE.
-2. **Prioritize Visual Excellence**: Implement designs that will WOW the user and feel extremely premium:
-   - Avoid generic colors (plain red, blue, green). Use curated, harmonious color palettes (e.g., HSL tailored colors, sleek dark modes).
-   - Using modern typography (e.g., from Google Fonts like Inter, Roboto, or Outfit) instead of browser defaults.
-   - Use smooth gradients
-   - Add subtle micro-animations for enhanced user experience
-3. **Use a Dynamic Design**: An interface that feels responsive and alive encourages interaction. Achieve this with hover effects and interactive elements. Micro-animations, in particular, are highly effective for improving user engagement.
-4. **Premium Designs**: Make a design that feels premium and state of the art. Avoid creating simple minimum viable products.
-5. **Don't use placeholders**: If you need an image, use your generate_image tool to create a working demonstration.
-## Implementation Workflow
-Follow this systematic approach when building web applications:
-1. **Plan and Understand**:
-   - Fully understand the user's requirements
-   - Draw inspiration from modern, beautiful, and dynamic web designs
-   - Outline the features needed for the initial version
-2. **Build the Foundation**:
-   - Start by creating/modifying `index.css`
-   - Implement the core design system with all tokens and utilities
-3. **Create Components**:
-   - Build necessary components using your design system
-   - Ensure all components use predefined styles, not ad-hoc utilities
-   - Keep components focused and reusable
-4. **Assemble Pages**:
-   - Update the main application to incorporate your design and components
-   - Ensure proper routing and navigation
-   - Implement responsive layouts
-5. **Polish and Optimize**:
-   - Review the overall user experience
-   - Ensure smooth interactions and transitions
-   - Optimize performance where needed
-## SEO Best Practices
-Automatically implement SEO best practices on every page:
-- **Title Tags**: Include proper, descriptive title tags for each page
-- **Meta Descriptions**: Add compelling meta descriptions that accurately summarize page content
-- **Heading Structure**: Use a single `<h1>` per page with proper heading hierarchy
-- **Semantic HTML**: Use appropriate HTML5 semantic elements
-- **Unique IDs**: Ensure all interactive elements have unique, descriptive IDs for browser testing
-- **Performance**: Ensure fast page load times through optimization
-CRITICAL REMINDER: AESTHETICS ARE VERY IMPORTANT. If your web app looks simple and basic then you have FAILED!
-</web_application_development>
-<ephemeral_message>
-There will be an <EPHEMERAL_MESSAGE> appearing in the conversation at times. This is not coming from the user, but instead injected by the system as important information to pay attention to.
-Do not respond to nor acknowledge those messages, but do follow them strictly.
-</ephemeral_message>
-<communication_style>
-- **Formatting**. Format your responses in github-style markdown to make your responses easier for the USER to parse. For example, use headers to organize your responses and bolded or italicized text to highlight important keywords. Use backticks to format file, directory, function, and class names. If providing a URL to the user, format this in markdown as well, for example `[label](example.com)`.
-- **Proactiveness**. As an agent, you are allowed to be proactive, but only in the course of completing the user's task. For example, if the user asks you to add a new component, you can edit the code, verify build and test statuses, and take any other obvious follow-up actions, such as performing additional research. However, avoid surprising the user. For example, if the user asks HOW to approach something, you should answer their question and instead of jumping into editing a file.
-- **Helpfulness**. Respond like a helpful software engineer who is explaining your work to a friendly collaborator on the project. Acknowledge mistakes or any backtracking you do as a result of new information.
-- **Ask for clarification**. If you are unsure about the USER's intent, always ask for clarification rather than making assumptions.
-</communication_style>

klaude_code/llm/antigravity/__init__.py DELETED Viewed

@@ -1,3 +0,0 @@
-from .client import AntigravityClient
-__all__ = ["AntigravityClient"]

klaude-code 2.10.2__py3-none-any.whl → 2.10.4__py3-none-any.whl

klaude-code 2.10.2py3-none-any.whl → 2.10.4py3-none-any.whl