gemcode 0.3.67__py3-none-any.whl → 0.3.69__py3-none-any.whl

gemcode/callbacks.py

@@ -43,6 +43,8 @@ _CTX_WARN_LEVEL_NOTIFIED = "gemcode:ctx_warn_level_notified"
 _LAST_PROMPT_TOKENS = "gemcode:last_prompt_tokens"
 _LAST_CONTEXT_PCT = "gemcode:last_context_percent_left"
 _LAST_CONTEXT_LEVEL = "gemcode:last_context_alert_level"
+_RISK_FILES_TOUCHED = "gemcode:risk_files_touched"
+_RISK_TOOL_CALLS = "gemcode:risk_tool_calls"
 
 def _truthy_env(name: str, *, default: bool = False) -> bool:
   v = os.environ.get(name)
@@ -141,6 +143,36 @@ def make_before_tool_callback(cfg: GemCodeConfig):
     record = {"tool": name, "args": _redact_args(name, args)}
     append_audit(cfg.project_root, record)
 
+    # Dynamic risk signals from actual repo interaction.
+    try:
+      if tool_context is not None:
+        st = tool_context.state
+        st[_RISK_TOOL_CALLS] = int(st.get(_RISK_TOOL_CALLS, 0) or 0) + 1
+        if name == "read_file":
+          p = (args or {}).get("path")
+          if isinstance(p, str) and p.strip():
+            touched: set[str] = set(st.get(_RISK_FILES_TOUCHED, []) or [])
+            touched.add(p.strip())
+            # Store as list for JSON-serializable session state.
+            st[_RISK_FILES_TOUCHED] = list(sorted(touched))[:200]
+            # More files touched => higher complexity.
+            n = len(touched)
+            cur = float(getattr(cfg, "_risk_score", 0.0) or 0.0)
+            if n >= 10:
+              cur = min(1.0, cur + 0.08)
+            elif n >= 5:
+              cur = min(1.0, cur + 0.04)
+            object.__setattr__(cfg, "_risk_score", cur)
+        # Writes / shell are inherently higher risk; allow more evidence.
+        if name in MUTATING_TOOLS:
+          cur = float(getattr(cfg, "_risk_score", 0.0) or 0.0)
+          object.__setattr__(cfg, "_risk_score", min(1.0, cur + 0.12))
+        if name in SHELL_TOOLS:
+          cur = float(getattr(cfg, "_risk_score", 0.0) or 0.0)
+          object.__setattr__(cfg, "_risk_score", min(1.0, cur + 0.08))
+    except Exception:
+      pass
+
     # ── Shell hooks: pre_tool_use ─────────────────────────────────────────
     # If the project has a .gemcode/hooks/pre_tool_use.sh, run it now.
     # Non-zero exit or {"decision":"deny"} stdout will block the tool call.
@@ -384,6 +416,28 @@ def make_after_tool_callback(cfg: GemCodeConfig):
         st[_STATE_FAILURE_KEY] = 0
     else:
       st[_STATE_FAILURE_KEY] = 0
+
+    # Risk feedback: if tools are failing or commands return non-zero, treat the
+    # task as higher-risk and allow more evidence in subsequent tool outputs.
+    try:
+      cur = float(getattr(cfg, "_risk_score", 0.0) or 0.0)
+      bump = 0.0
+      if err:
+        bump += 0.15
+      if isinstance(tool_response, dict) and isinstance(tool_response.get("exit_code"), int):
+        if int(tool_response["exit_code"]) != 0:
+          bump += 0.10
+          # Test/build failures should boost evidence allowance more.
+          if name in ("bash", "run_command"):
+            bump += 0.05
+      # decay slowly when things are healthy
+      if bump == 0.0:
+        cur = max(0.0, cur * 0.90)
+      else:
+        cur = min(1.0, cur + bump)
+      object.__setattr__(cfg, "_risk_score", cur)
+    except Exception:
+      pass
     # ── Shell hooks: post_tool_use ────────────────────────────────────────
     try:
       from gemcode.hooks import run_post_tool_use_hook

gemcode/config.py

@@ -143,6 +143,16 @@ class GemCodeConfig:
   dynamic_token_policy: bool = field(
     default_factory=lambda: _truthy_env("GEMCODE_DYNAMIC_TOKEN_POLICY", default=True)
   )
+
+  # Dynamic risk policy: boosts caps when the current task appears risky/complex.
+  dynamic_risk_policy: bool = field(
+    default_factory=lambda: _truthy_env("GEMCODE_DYNAMIC_RISK_POLICY", default=True)
+  )
+
+  # How much the risk score can expand caps (0.0-1.5 reasonable).
+  dynamic_risk_boost: float = field(
+    default_factory=lambda: float(os.environ.get("GEMCODE_DYNAMIC_RISK_BOOST", "0.6"))
+  )
   # Trim oldest text in llm_request.contents when over budget (see context_budget.py).
   context_shrink_enabled: bool = field(
     default_factory=lambda: _truthy_env("GEMCODE_CONTEXT_SHRINK", default=True)

gemcode/dynamic_policy.py

@@ -32,6 +32,20 @@ def _pct_left(cfg) -> int | None:
   return None
 
 
+def _risk(cfg) -> float:
+  try:
+    v = getattr(cfg, "_risk_score", None)
+    if isinstance(v, (int, float)):
+      return float(v)
+  except Exception:
+    return 0.0
+  return 0.0
+
+
+def _clamp(x: float, lo: float, hi: float) -> float:
+  return lo if x < lo else hi if x > hi else x
+
+
 @dataclass(frozen=True)
 class DynamicCaps:
   tool_inline_chars: int
@@ -52,6 +66,8 @@ def get_dynamic_caps(cfg) -> DynamicCaps:
   - Healthy (>=45% left): generous caps (better evidence, less re-asking).
   - Warning (20-44%): moderate caps.
   - Tight (<20%): strict caps + prefer offload.
+
+  Then apply a risk-based boost (if enabled) so complex tasks stay evidence-rich.
   """
   # cfg can be None in some tool contexts; treat as enabled with defaults.
   enabled = _truthy(getattr(cfg, "dynamic_token_policy", True) if cfg is not None else True, default=True)
@@ -76,42 +92,52 @@ def get_dynamic_caps(cfg) -> DynamicCaps:
   base_tool = int(getattr(cfg, "tool_result_max_chars", 12000) or 12000) if cfg is not None else 12000
   base_tool = max(1000, base_tool)
 
+  # Risk boost: scale caps upward for risky tasks, but keep bounded.
+  risk_enabled = _truthy(getattr(cfg, "dynamic_risk_policy", True) if cfg is not None else True, default=True)
+  risk_boost = float(getattr(cfg, "dynamic_risk_boost", 0.6) if cfg is not None else 0.6)
+  risk_score = _risk(cfg) if (cfg is not None and risk_enabled) else 0.0
+  risk_score = _clamp(risk_score, 0.0, 1.0)
+  boost = 1.0 + (_clamp(risk_boost, 0.0, 1.5) * risk_score)
+
+  def _scale(n: int, *, cap: int) -> int:
+    return min(cap, max(1000, int(n * boost)))
+
   if pct >= 45:
     mult = 1.4
     return DynamicCaps(
-      tool_inline_chars=min(24_000, int(base_tool * mult)),
-      read_file_max_bytes=140_000,
-      web_fetch_max_chars=30_000,
-      bash_stdout_chars=30_000,
-      bash_stderr_chars=15_000,
-      run_stdout_chars=30_000,
-      run_stderr_chars=30_000,
-      grep_max_matches=60,
+      tool_inline_chars=_scale(min(24_000, int(base_tool * mult)), cap=30_000),
+      read_file_max_bytes=min(200_000, int(140_000 * boost)),
+      web_fetch_max_chars=min(60_000, int(30_000 * boost)),
+      bash_stdout_chars=min(80_000, int(30_000 * boost)),
+      bash_stderr_chars=min(40_000, int(15_000 * boost)),
+      run_stdout_chars=min(80_000, int(30_000 * boost)),
+      run_stderr_chars=min(80_000, int(30_000 * boost)),
+      grep_max_matches=min(200, int(60 * boost)),
     )
 
   if pct >= 20:
     mult = 1.0
     return DynamicCaps(
-      tool_inline_chars=min(18_000, int(base_tool * mult)),
-      read_file_max_bytes=80_000,
-      web_fetch_max_chars=20_000,
-      bash_stdout_chars=20_000,
-      bash_stderr_chars=10_000,
-      run_stdout_chars=20_000,
-      run_stderr_chars=20_000,
-      grep_max_matches=40,
+      tool_inline_chars=_scale(min(18_000, int(base_tool * mult)), cap=24_000),
+      read_file_max_bytes=min(160_000, int(80_000 * boost)),
+      web_fetch_max_chars=min(40_000, int(20_000 * boost)),
+      bash_stdout_chars=min(50_000, int(20_000 * boost)),
+      bash_stderr_chars=min(30_000, int(10_000 * boost)),
+      run_stdout_chars=min(50_000, int(20_000 * boost)),
+      run_stderr_chars=min(50_000, int(20_000 * boost)),
+      grep_max_matches=min(120, int(40 * boost)),
     )
 
   # Tight
   mult = 0.6
   return DynamicCaps(
-    tool_inline_chars=max(2000, int(base_tool * mult)),
-    read_file_max_bytes=35_000,
-    web_fetch_max_chars=10_000,
-    bash_stdout_chars=10_000,
-    bash_stderr_chars=8_000,
-    run_stdout_chars=10_000,
-    run_stderr_chars=10_000,
-    grep_max_matches=20,
+    tool_inline_chars=max(2000, min(12_000, int(base_tool * mult * boost))),
+    read_file_max_bytes=min(90_000, int(35_000 * boost)),
+    web_fetch_max_chars=min(20_000, int(10_000 * boost)),
+    bash_stdout_chars=min(25_000, int(10_000 * boost)),
+    bash_stderr_chars=min(20_000, int(8_000 * boost)),
+    run_stdout_chars=min(25_000, int(10_000 * boost)),
+    run_stderr_chars=min(25_000, int(10_000 * boost)),
+    grep_max_matches=min(80, int(20 * boost)),
   )
 

gemcode/invoke.py

@@ -68,6 +68,32 @@ async def run_turn(
     cfg: "GemCodeConfig | None" = None,
 ) -> list:
   """Execute one user message; collect all Events (caller aggregates text)."""
+  # Dynamic risk score: updated each user message; later refined by tool outcomes.
+  # This is intentionally heuristic but configurable via env knobs.
+  if cfg is not None:
+    try:
+      import re
+      p = (prompt or "")[:20_000]
+      risk = 0.0
+      # Complexity signals
+      if len(p) > 600:
+        risk += 0.15
+      if len(p) > 2000:
+        risk += 0.15
+      if re.search(r"\\b(refactor|migrate|rewrite|optimi[sz]e|architecture)\\b", p, re.I):
+        risk += 0.2
+      if re.search(r"\\b(bug|fix|regression|error|traceback|failing)\\b", p, re.I):
+        risk += 0.2
+      if re.search(r"\\b(test|pytest|ci|build|deploy|release)\\b", p, re.I):
+        risk += 0.1
+      # Multi-file hints
+      if p.count(\"/\") >= 6 or p.count(\".py\") + p.count(\".ts\") + p.count(\".tsx\") >= 3:
+        risk += 0.1
+      # Clamp 0..1
+      risk = max(0.0, min(1.0, float(risk)))
+      object.__setattr__(cfg, \"_risk_score\", risk)
+    except Exception:
+      pass
   run_config = (
     RunConfig(max_llm_calls=max_llm_calls) if max_llm_calls is not None else None
   )

gemcode/repl_slash.py

@@ -391,6 +391,20 @@ async def process_repl_slash(
     out(f"    thinking_budget:       {cfg.thinking_budget if cfg.thinking_budget is not None else '(auto)'}")
     out(f"    show_full_thinking:    {cfg.show_full_thinking}")
     out()
+    # Dynamic policy telemetry
+    try:
+      risk = float(getattr(cfg, "_risk_score", 0.0) or 0.0)
+      pct = getattr(cfg, "_context_percent_left", None)
+      out("  Dynamic policy:")
+      out(f"    dynamic_token_policy:  {getattr(cfg, 'dynamic_token_policy', True)}")
+      out(f"    dynamic_risk_policy:   {getattr(cfg, 'dynamic_risk_policy', True)}")
+      out(f"    dynamic_risk_boost:    {getattr(cfg, 'dynamic_risk_boost', 0.6)}")
+      out(f"    risk_score:            {risk:.2f}")
+      if isinstance(pct, int):
+        out(f"    context_percent_left:  {pct}%")
+      out()
+    except Exception:
+      pass
     out("  Autocompact:")
     out(f"    GEMCODE_AUTOCOMPACT:               {os.environ.get('GEMCODE_AUTOCOMPACT', '1')}")
     out(f"    GEMCODE_AUTOCOMPACT_BUFFER_CHARS:  {os.environ.get('GEMCODE_AUTOCOMPACT_BUFFER_CHARS', '60000')}")

{gemcode-0.3.67.dist-info → gemcode-0.3.69.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gemcode
-Version: 0.3.67
+Version: 0.3.69
 Summary: Local-first coding agent on Google Gemini + ADK
 Author: GemCode Contributors
 License:                                  Apache License

{gemcode-0.3.67.dist-info → gemcode-0.3.69.dist-info}/RECORD

@@ -3,20 +3,20 @@ gemcode/__main__.py,sha256=EX2s1hxq2Yvli_-tnBN3w5Qv4bOjsBBbjyISF0pDIQw,37
 gemcode/agent.py,sha256=yft-8lLJbX4abM_bUW156JWfVT3b4OHO-_hIoKdVaOs,52032
 gemcode/audit.py,sha256=bh9uhXaeh8wqxqoZtz3ZAowd8Ndk1ss-mw9993Vlrgo,469
 gemcode/autocompact.py,sha256=77h5tgFzJ2rjrhlCL2oIc28IHwLbP4Pqlo7cSNgDwiA,6727
-gemcode/callbacks.py,sha256=7kPMDsc2vvY98Ogoi4D0XHB96qh_kx4l4CRsRtePNdk,26171
+gemcode/callbacks.py,sha256=JW9jiYBZwgqqZnX4zehd55dGHeVLCVQVU9zNE3SYS_s,28430
 gemcode/capability_routing.py,sha256=yvQXwKtrfHXbgbNunU0Dxh9GCDN4cbySXIeccrdzr2o,3471
 gemcode/cli.py,sha256=kBXb4b4JCG3u0XewCJn8lCyOT62Y8bOvlVoDc2R-GKQ,25320
 gemcode/compaction.py,sha256=9YtA_qa23_8dHWVHx7AJwUduuI7jJQtq-m6sT8jgPWI,1186
-gemcode/config.py,sha256=bCo_Qz4oI_Sfi0EzptULCjfTbtYNhHtMeXTil2vUi8M,14697
+gemcode/config.py,sha256=it_F4haaUlveJXZkX8mGI_nmx1JHsh56XBSkLyUfQ4I,15101
 gemcode/context_budget.py,sha256=Nhox9vFBtLbb7jtO7cyGW1MxtN7SVjlIeQ7d-cgGyKM,10544
 gemcode/context_warning.py,sha256=Q8mg5Vojj7EglPhsGAVL7vb8ROLuHVPgdzw25yw-Q2c,4263
 gemcode/credentials.py,sha256=04v-rLD8_Ams69FQdof2FwcL3ZgsroGUnMcHNQFuBZo,1296
-gemcode/dynamic_policy.py,sha256=0ZYv3nzMsEN4kPpjhMbXVSMoTvA_IArbMfe3fF7hhzw,3158
+gemcode/dynamic_policy.py,sha256=nWgBN6ffSn1Te4aDI1MURxRaQjTclzIuSf4KaAskE9U,4662
 gemcode/hitl_session.py,sha256=oNiI7odFJGUcmqPavjKLJOEumZKrgklLvwjjrIG9GPg,281
 gemcode/hooks.py,sha256=FHz175d_18j-4ByZZVdEIagmdOvLHcjDjo7HD2Cikf4,6339
 gemcode/intent_classifier.py,sha256=YfRVEe8gHeKlRkjuSWef1bZ0MPBwyYMp5jymP5Vig5U,8507
 gemcode/interactions.py,sha256=B0b3QNE_I2i5_HtiebX4ehhjlc4Nbqjf_XbvcTLyJT0,641
-gemcode/invoke.py,sha256=ZPRv-ZuqMaD-Yd1tSQxgoWP24WEz8nwWsMDywVdZ8Gk,9319
+gemcode/invoke.py,sha256=tP1bBrmd6e0xbqXDfIc5iT09pqN0t82igvgN1PYwCQI,10285
 gemcode/kairos_daemon.py,sha256=giINipslAIhBtdbqA0o4RYwt4fBsZjtkiKDjp4zSEvw,6980
 gemcode/limits.py,sha256=1u7w3X13Uqw34ZOVYKa62pxfvlyLTY2Dne1kN8SKJlA,2524
 gemcode/live_audio_engine.py,sha256=wxdGp5ciAgQ7WIeBpcW_hm8lbMDsh0KPJbqIS9uAGxA,3406
@@ -32,7 +32,7 @@ gemcode/pricing.py,sha256=lftp0SwyDqOzHqC2-6XzgZZhjif5PLdCe1Q3wY-p6kQ,3558
 gemcode/prompt_suggestions.py,sha256=h-W_9LlfagS91PyoMEjEjsCqoG4XmIh3QBypA59HyGw,2553
 gemcode/refine.py,sha256=BijEZ4Z32wGa9aK_WottyAhZF-j0xEqRg5UpjedNv2A,7653
 gemcode/repl_commands.py,sha256=AHjDUhlq7IuSL6Fla17y0aeGZyp6snxwbMG7gz1PrRc,11330
-gemcode/repl_slash.py,sha256=uBmMzUuQ52Yx14vxxL404wvC7hVdI_UK2y9cAEe-7ZI,47815
+gemcode/repl_slash.py,sha256=y_nb-0PYRpW2At7mlA-hkpEybe6vdyjMiVSrjOTPb_g,48434
 gemcode/review_agent.py,sha256=4t7_5-aE60b4-EheJ_eSB_H2eQYf9GppKoui6jw0TME,5264
 gemcode/session_runtime.py,sha256=crpKBVYZnWeRkBHcbvBGuCc2Kw2hGhVHuM4uRCmXpOI,18771
 gemcode/session_store.py,sha256=POUT_QQf715c74jbXj0s5vCd4dlAgJz_CLsIWuEUoO0,6051
@@ -86,9 +86,9 @@ gemcode/tui/welcome_rich.py,sha256=8FEZzLXrzqly5JWiDgV9ooRV1LNXDk-CXV1a7K6ua-U,4
 gemcode/web/__init__.py,sha256=EysmUAWs6g-lmMk4VFljKfaHVrEgb_FiIzwQmBdORJc,40
 gemcode/web/claude_sse_adapter.py,sha256=HcNp0Lh4DdBZBLOpstsqa-VzfqAUrRngZ6FSuJ-mIMg,8609
 gemcode/web/terminal_repl.py,sha256=k2irvFGbCY8gDm_pbirR7b_cakaeafcctoTIvnJkVXk,3902
-gemcode-0.3.67.dist-info/licenses/LICENSE,sha256=TD4524qn-W8Z07GTDnag-9jJPFutFZNB0a1WbMHPC54,8388
-gemcode-0.3.67.dist-info/METADATA,sha256=rZb2AHzSETy2XUb5L5w88ECMQWXoQmNKhKOJ78SyGcs,23695
-gemcode-0.3.67.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-gemcode-0.3.67.dist-info/entry_points.txt,sha256=cZdLTLDiHbks7OSUCuxCh66dCWeQdpLR8BozoqfEjV4,45
-gemcode-0.3.67.dist-info/top_level.txt,sha256=UYrjULLBY2bcgK6KI6flomJWmsbDXu7n0rvW2SWFrbo,8
-gemcode-0.3.67.dist-info/RECORD,,
+gemcode-0.3.69.dist-info/licenses/LICENSE,sha256=TD4524qn-W8Z07GTDnag-9jJPFutFZNB0a1WbMHPC54,8388
+gemcode-0.3.69.dist-info/METADATA,sha256=2yfwQj99pN5poyQL_zvyHGzXqmqWj-pCFYZuxj8KsX0,23695
+gemcode-0.3.69.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+gemcode-0.3.69.dist-info/entry_points.txt,sha256=cZdLTLDiHbks7OSUCuxCh66dCWeQdpLR8BozoqfEjV4,45
+gemcode-0.3.69.dist-info/top_level.txt,sha256=UYrjULLBY2bcgK6KI6flomJWmsbDXu7n0rvW2SWFrbo,8
+gemcode-0.3.69.dist-info/RECORD,,