code-puppy 0.0.348__py3-none-any.whl → 0.0.372__py3-none-any.whl

code_puppy/model_utils.py

@@ -1,7 +1,7 @@
 """Model-related utilities shared across agents and tools.
 
 This module centralizes logic for handling model-specific behaviors,
-particularly for claude-code and chatgpt-codex models which require special prompt handling.
+particularly for claude-code and antigravity models which require special prompt handling.
 """
 
 import pathlib
@@ -11,27 +11,30 @@ from typing import Optional
 # The instruction override used for claude-code models
 CLAUDE_CODE_INSTRUCTIONS = "You are Claude Code, Anthropic's official CLI for Claude."
 
-# Path to the Codex system prompt file
-_CODEX_PROMPT_PATH = (
-    pathlib.Path(__file__).parent / "prompts" / "codex_system_prompt.md"
+# Path to the Antigravity system prompt file
+_ANTIGRAVITY_PROMPT_PATH = (
+    pathlib.Path(__file__).parent / "prompts" / "antigravity_system_prompt.md"
 )
 
-# Cache for the loaded Codex prompt
-_codex_prompt_cache: Optional[str] = None
+# Cache for the loaded Antigravity prompt
+_antigravity_prompt_cache: Optional[str] = None
 
 
-def _load_codex_prompt() -> str:
-    """Load the Codex system prompt from file, with caching."""
-    global _codex_prompt_cache
-    if _codex_prompt_cache is None:
-        if _CODEX_PROMPT_PATH.exists():
-            _codex_prompt_cache = _CODEX_PROMPT_PATH.read_text(encoding="utf-8")
+def _load_antigravity_prompt() -> str:
+    """Load the Antigravity system prompt from file, with caching."""
+    global _antigravity_prompt_cache
+    if _antigravity_prompt_cache is None:
+        if _ANTIGRAVITY_PROMPT_PATH.exists():
+            _antigravity_prompt_cache = _ANTIGRAVITY_PROMPT_PATH.read_text(
+                encoding="utf-8"
+            )
         else:
             # Fallback to a minimal prompt if file is missing
-            _codex_prompt_cache = (
-                "You are Codex, a coding agent running in the Codex CLI."
+            _antigravity_prompt_cache = (
+                "You are Antigravity, a powerful agentic AI coding assistant "
+                "designed by the Google Deepmind team."
             )
-    return _codex_prompt_cache
+    return _antigravity_prompt_cache
 
 
 @dataclass
@@ -54,9 +57,9 @@ def is_claude_code_model(model_name: str) -> bool:
     return model_name.startswith("claude-code")
 
 
-def is_chatgpt_codex_model(model_name: str) -> bool:
-    """Check if a model is a ChatGPT Codex model."""
-    return model_name.startswith("chatgpt-")
+def is_antigravity_model(model_name: str) -> bool:
+    """Check if a model is an Antigravity model."""
+    return model_name.startswith("antigravity-")
 
 
 def prepare_prompt_for_model(
@@ -77,8 +80,8 @@ def prepare_prompt_for_model(
             is_claude_code=True,
         )
 
-    # Handle ChatGPT Codex models
-    if is_chatgpt_codex_model(model_name):
+    # Handle Antigravity models
+    if is_antigravity_model(model_name):
         modified_prompt = user_prompt
         if prepend_system_to_user and system_prompt:
             modified_prompt = (
@@ -91,7 +94,7 @@ def prepare_prompt_for_model(
                 f"{user_prompt}"
             )
         return PreparedPrompt(
-            instructions=_load_codex_prompt(),
+            instructions=_load_antigravity_prompt(),
             user_prompt=modified_prompt,
             is_claude_code=False,
         )
@@ -108,6 +111,6 @@ def get_claude_code_instructions() -> str:
     return CLAUDE_CODE_INSTRUCTIONS
 
 
-def get_chatgpt_codex_instructions() -> str:
-    """Get the Codex system prompt for ChatGPT Codex models."""
-    return _load_codex_prompt()
+def get_antigravity_instructions() -> str:
+    """Get the Antigravity system prompt for Antigravity models."""
+    return _load_antigravity_prompt()

code_puppy/models.json

@@ -7,7 +7,7 @@
       "api_key": "$SYN_API_KEY"
     },
     "context_length": 200000,
-    "supported_settings": ["temperature", "seed"]
+    "supported_settings": ["temperature", "seed", "top_p"]
   },
   "synthetic-MiniMax-M2.1": {
     "type": "custom_openai",
@@ -17,7 +17,7 @@
       "api_key": "$SYN_API_KEY"
     },
     "context_length": 195000,
-    "supported_settings": ["temperature", "seed"]
+    "supported_settings": ["temperature", "seed", "top_p"]
   },
   "synthetic-Kimi-K2-Thinking": {
     "type": "custom_openai",
@@ -27,32 +27,32 @@
       "api_key": "$SYN_API_KEY"
     },
     "context_length": 262144,
-    "supported_settings": ["temperature", "seed"]
+    "supported_settings": ["temperature", "seed", "top_p"]
   },
   "Gemini-3": {
     "type": "gemini",
     "name": "gemini-3-pro-preview",
     "context_length": 200000,
-    "supported_settings": ["temperature"]
+    "supported_settings": ["temperature", "top_p"]
   },
   "Gemini-3-Long-Context": {
     "type": "gemini",
     "name": "gemini-3-pro-preview",
     "context_length": 1000000,
-    "supported_settings": ["temperature"]
+    "supported_settings": ["temperature", "top_p"]
   },
   "gpt-5.1": {
     "type": "openai",
     "name": "gpt-5.1",
     "context_length": 272000,
-    "supported_settings": ["reasoning_effort", "verbosity"],
+    "supported_settings": ["temperature", "top_p", "reasoning_effort", "verbosity"],
     "supports_xhigh_reasoning": false
   },
   "gpt-5.1-codex-api": {
     "type": "openai",
     "name": "gpt-5.1-codex",
     "context_length": 272000,
-    "supported_settings": ["reasoning_effort", "verbosity"],
+    "supported_settings": ["temperature", "top_p", "reasoning_effort", "verbosity"],
     "supports_xhigh_reasoning": true
   },
   "Cerebras-GLM-4.7": {
@@ -63,7 +63,7 @@
       "api_key": "$CEREBRAS_API_KEY"
     },
     "context_length": 131072,
-    "supported_settings": ["temperature", "seed"]
+    "supported_settings": ["temperature", "seed", "top_p"]
   },
   "claude-4-5-haiku": {
     "type": "anthropic",
@@ -87,24 +87,24 @@
     "type": "zai_coding",
     "name": "glm-4.6",
     "context_length": 200000,
-    "supported_settings": ["temperature"]
+    "supported_settings": ["temperature", "top_p"]
   },
   "zai-glm-4.6-api": {
     "type": "zai_api",
     "name": "glm-4.6",
     "context_length": 200000,
-    "supported_settings": ["temperature"]
+    "supported_settings": ["temperature", "top_p"]
   },
   "zai-glm-4.7-coding": {
     "type": "zai_coding",
     "name": "glm-4.7",
     "context_length": 200000,
-    "supported_settings": ["temperature"]
+    "supported_settings": ["temperature", "top_p"]
   },
   "zai-glm-4.7-api": {
     "type": "zai_api",
     "name": "glm-4.7",
     "context_length": 200000,
-    "supported_settings": ["temperature"]
+    "supported_settings": ["temperature", "top_p"]
   }
 }