gemcode 0.2.2__py3-none-any.whl

gemcode/capability_routing.py

@@ -0,0 +1,137 @@
+"""
+Capability-based routing (Claude Code style conceptually).
+
+This layer decides which *capabilities* to enable (deep research tools,
+embeddings retrieval, computer-use tools) and leaves the existing Claude-like
+outer/inner loops intact.
+
+It is intentionally conservative:
+- It only enables capabilities (turns them on), it does not disable
+  explicitly requested capabilities.
+- Computer-use model selection is enforced at model-routing precedence, and
+  tool execution remains permission-gated via `callbacks.py`.
+"""
+
+from __future__ import annotations
+
+import re
+
+from gemcode.config import GemCodeConfig
+
+CapabilityMode = str  # "auto|research|embeddings|computer|audio|all"
+
+
+_RESEARCH_TRIGGERS = [
+  "deep research",
+  "deep-dive",
+  "research",
+  "citations",
+  "sources",
+  "grounded",
+  "investigate",
+  "literature",
+  "benchmark",
+]
+
+_EMBEDDINGS_TRIGGERS = [
+  "embedding",
+  "embeddings",
+  "semantic search",
+  "similarity",
+  "vector",
+  "rag",
+  "retrieve",
+  "relevant docs",
+]
+
+_COMPUTER_TRIGGERS = [
+  "click",
+  "button",
+  "double click",
+  "type into",
+  "navigate",
+  "browser",
+  "open website",
+  "scroll",
+  "ui automation",
+  "open tab",
+]
+
+_AUDIO_TRIGGERS = [
+  "audio",
+  "voice",
+  "microphone",
+  "speak",
+  "listen",
+  "tts",
+  "tts preview",
+]
+
+
+def _contains_any(haystack: str, needles: list[str]) -> bool:
+  h = haystack.lower()
+  return any(n in h for n in needles)
+
+
+def apply_capability_routing(
+  cfg: GemCodeConfig,
+  prompt: str,
+  *,
+  context: str = "prompt",
+) -> None:
+  """
+  Mutates `cfg` in-place:
+  - sets `enable_deep_research`, `enable_embeddings`, `enable_computer_use`
+  - sets `enable_audio` only when context is `live-audio`
+  """
+  mode = (getattr(cfg, "capability_mode", "auto") or "auto").lower()
+  p_norm = re.sub(r"\s+", " ", prompt or "").strip().lower()
+
+  def enable_research() -> None:
+    cfg.enable_deep_research = True
+
+  def enable_embeddings() -> None:
+    cfg.enable_embeddings = True
+
+  def enable_computer() -> None:
+    cfg.enable_computer_use = True
+
+  def enable_audio() -> None:
+    if context == "live-audio":
+      cfg.enable_audio = True
+
+  # User-selected mode.
+  if mode == "research":
+    enable_research()
+    return
+  if mode == "embeddings":
+    enable_embeddings()
+    return
+  if mode == "computer":
+    enable_computer()
+    return
+  if mode == "audio":
+    enable_audio()
+    return
+  if mode == "all":
+    enable_research()
+    enable_embeddings()
+    enable_computer()
+    enable_audio()
+    return
+
+  # Auto mode: prompt heuristics.
+  if mode == "auto":
+    if _contains_any(p_norm, _RESEARCH_TRIGGERS):
+      enable_research()
+    if _contains_any(p_norm, _EMBEDDINGS_TRIGGERS):
+      enable_embeddings()
+    if _contains_any(p_norm, _COMPUTER_TRIGGERS):
+      enable_computer()
+    if _contains_any(p_norm, _AUDIO_TRIGGERS):
+      enable_audio()
+    return
+
+  # Unknown mode: do nothing.
+  return
+