@octocodeai/freellama-mcp-server 0.5.0 → 0.6.0

package/adapters/octocode_agent.py

@@ -1,404 +0,0 @@
-#!/usr/bin/env python3
-"""Agent A adapter: local Ollama model whose only research tool is the `octocode` CLI.
-
-Implements the FreeLlama benchmark adapter contract (benchmark/harness/references/adapters.md):
-reads FREELLAMA_BENCH_MODEL/PROMPT/WORKSPACE/AGENT_RESULT, drives its own Ollama chat loop, and
-writes agent-result.json. See benchmark/local/docs/02-agent-a-octocode.md for the full spec this
-mirrors verbatim.
-"""
-
-from __future__ import annotations
-
-from contextlib import ExitStack
-import json
-import os
-import subprocess
-import time
-from pathlib import Path
-from typing import Any
-from urllib.error import HTTPError, URLError
-from urllib.request import Request, urlopen
-
-from agent_context import (
-    AgentContextManager,
-    AgentRuntimeConfig,
-    PARSE_REPAIR_NOTICE,
-    REPEAT_NOTICE,
-    ObservationStore,
-    call_signature,
-    parse_json_action,
-    write_failure_result,
-)
-from agent_transport import chat_request, request_headers, unwrap_chat_response, PromptCacheUsage, retryable_chat_error, resolve_model_identity, chat_error_details
-
-OCTOCODE_TOOLS = {"localViewStructure", "localFindFiles", "localSearchCode", "localGetFileContent", "lspGetSemantics"}
-PATH_KEYS = ("path", "uri")
-# Nothing is clipped any more — see the pagination section of agent_context.py. `calls[].result`
-# goes to result.json on disk and never into the model's context, so it is stored in full; the model
-# is shown one page at a time with an exact instruction for fetching the next.
-
-
-def request_json(url: str, payload: dict[str, Any], timeout_seconds: float) -> dict[str, Any]:
-    request = Request(url, data=json.dumps(payload).encode(), headers=request_headers(), method="POST")
-    with urlopen(request, timeout=timeout_seconds) as response:
-        return json.loads(response.read())
-
-
-def safe_resolve(root: Path, value: str) -> Path:
-    candidate = Path(value)
-    candidate = candidate if candidate.is_absolute() else (root / candidate)
-    candidate = candidate.resolve()
-    if candidate != root and root not in candidate.parents:
-        raise ValueError(f"path escapes benchmark workspace: {value}")
-    if ".git" in candidate.parts:
-        raise ValueError("git metadata is outside the task surface")
-    return candidate
-
-
-def parse_action(content: str) -> dict[str, Any]:
-    value = parse_json_action(content)
-    action = value["action"]
-    if action == "octocode":
-        if value.get("tool") not in OCTOCODE_TOOLS:
-            raise ValueError(f"octocode action requires one supported tool: {sorted(OCTOCODE_TOOLS)}")
-        if not isinstance(value.get("queries"), dict):
-            raise ValueError("octocode action requires a queries object")
-    if action == "page" and (
-        not isinstance(value.get("step"), int)
-        or value["step"] < 1
-        or not isinstance(value.get("page"), int)
-        or value["page"] < 1
-    ):
-        raise ValueError("page action requires positive integer step and page")
-    if action == "finish" and not isinstance(value.get("answer"), str):
-        raise ValueError("finish action requires a string answer")
-    if action not in {"octocode", "page", "finish"}:
-        raise ValueError(f"unsupported action for octocode adapter: {action}")
-    return value
-
-
-# Pinned: an unpinned `npx octocode` downloads whatever is latest at call time (a supply-chain
-# risk for a tool that reads the user's workspace) and the first download can exceed the tool
-# timeout. Override deliberately with FREELLAMA_AGENT_OCTOCODE_PACKAGE.
-OCTOCODE_PACKAGE = os.environ.get("FREELLAMA_AGENT_OCTOCODE_PACKAGE", "octocode@19.1.0")
-
-
-def run_octocode(
-    root: Path,
-    tool_name: str,
-    queries: dict[str, Any],
-    timeout_seconds: float,
-) -> str:
-    if tool_name not in OCTOCODE_TOOLS:
-        raise ValueError(f"unsupported octocode tool: {tool_name}")
-    resolved = dict(queries)
-    for key in PATH_KEYS:
-        if key in resolved and isinstance(resolved[key], str):
-            resolved[key] = str(safe_resolve(root, resolved[key]))
-    command = ["npx", "--yes", OCTOCODE_PACKAGE, "tools", tool_name, "--queries", json.dumps(resolved), "--compact"]
-    result = subprocess.run(
-        command,
-        cwd=root,
-        text=True,
-        capture_output=True,
-        timeout=timeout_seconds,
-        check=False,
-    )
-    output = (result.stdout or "") + (("\nSTDERR:\n" + result.stderr) if result.returncode != 0 else "")
-    if not output.strip():
-        output = "(empty result)"
-    return output
-
-
-def system_prompt(workspace: str) -> str:
-    return f"""You are a local code-research agent in a read-only workspace at {workspace}. Read or search files only through the `octocode` local tools below. Return exactly one JSON object per turn.
-
-Tools (call as {{"action":"octocode","tool":"<name>","queries":{{...}}}}):
-
-localViewStructure - browse a directory tree, no content loaded; cheapest first orientation step.
-  path (string, required, absolute), maxDepth (int), recursive (bool), filesOnly (bool),
-  directoriesOnly (bool), pattern (glob/substring filter).
-
-localFindFiles - find files/dirs by name, glob, regex, or type; returns paths only, not content.
-  path (string, required, absolute), names (array of globs), pathPattern (glob over full path),
-  regex (basename regex), entryType ("f"|"d"), excludeDir (array, e.g. ["node_modules",".git"]).
-
-localSearchCode - search file contents for text/regex; returns file+line matches. Your main tool.
-  path (string, required, absolute), keywords (string; literal or regex search term),
-  mode ("paginated"|"discovery"|"detailed"), include/exclude (glob arrays),
-  caseInsensitive (bool), maxFiles (int).
-  Do NOT pass mode "structural" with keywords — structural (AST) search takes `pattern`/`rule`
-  instead and hard-errors on keywords, costing you a turn for nothing.
-
-localGetFileContent - read one file or a line range/matched slice of it.
-  path (string, required, absolute), fullContent (bool; small files only), startLine +
-  endLine (ints, both required together), matchString (anchor text/regex), minify
-  ("none"|"standard"|"symbols" — "symbols" gives a cheap outline first).
-
-lspGetSemantics - LSP semantic queries: definitions, references, callers/callees, symbol outline.
-  uri (string, required, absolute path), type ("definition"|"references"|"callers"|
-  "callees"|"documentSymbols"|"hover"), symbolName (exact name), lineHint (int from a prior
-  search result — never guess it).
-  An empty result is ambiguous: indexing or unavailable analysis can produce no symbols.
-  Retry once or use localSearchCode; an empty LSP result alone does not prove absence.
-
-Read another page of an earlier step: {{"action":"page","step":2,"page":2}}
-Full tool output is stored in pages; you see page 1 and the total. Context fitting can shorten
-an observation in the conversation. Request its stored page to recover the evidence without
-re-running the tool. Claim "not found" only after checking the relevant search scope and pages.
-
-Finish with: {{"action":"finish","answer":"concise final answer with repository-relative evidence"}}
-
-These are the complete action schemas. Use only `octocode`, `page`, or `finish`; every shown field
-is required with the shown type, and `tool` must be one of the five names above. An invalid shape
-is rejected and costs one bounded repair turn.
-
-SCOPE YOUR SEARCHES. Pass excludeDir/exclude on every search in a real workspace —
-["node_modules","target",".venv","dist","build",".git","vendor","__pycache__",".octocode"] — or
-vendored and generated files will bury the answer. For implementation questions, prefer src/,
-packages/*/src/ and lib/ before fixtures, mocks, or examples. If the question targets tests or examples,
-inspect those files. Name the file that supplies the answer. Absence of a match is weak evidence: widen the pattern before concluding
-something does not exist.
-For a production default, find its declaration before relying on test occurrences. Declarations look like
-`default=`, `unwrap_or(`, `const `, `static `, or a settings schema initializer.
-For a test-specific default or fixture, inspect its test declaration.
-
-All paths you pass must resolve inside the workspace; relative paths are resolved against the
-workspace root automatically. Orient cheap (localViewStructure) before reading in
-full. Be decisive: most tasks need 2-6 tool calls. Never edit or write files — you only have
-read-only tools. Call finish as soon as the requested facts are established."""
-
-
-def _main(resources: ExitStack) -> int:
-    model = os.environ.get("FREELLAMA_TARGET_MODEL") or os.environ["FREELLAMA_BENCH_MODEL"]
-    workspace = Path(os.environ["FREELLAMA_BENCH_WORKSPACE"]).resolve()
-    prompt = Path(os.environ["FREELLAMA_BENCH_PROMPT"]).read_text(encoding="utf-8")
-    result_path = Path(os.environ["FREELLAMA_AGENT_RESULT"])
-    endpoint = os.environ.get("FREELLAMA_OLLAMA_ENDPOINT", "http://127.0.0.1:11434").rstrip("/")
-    managed_endpoint = os.environ.get("FREELLAMA_AGENT_MANAGED_ENDPOINT", "").rstrip("/")
-    execution_preference = os.environ.get("FREELLAMA_AGENT_EXECUTION_PREFERENCE", "auto")
-    min_placement_evidence = os.environ.get("FREELLAMA_AGENT_MIN_PLACEMENT_EVIDENCE", "configured")
-    try:
-        runtime = AgentRuntimeConfig.from_env(default_tool_timeout_seconds=45)
-    except ValueError as error:
-        answer = f"invalid research adapter configuration: {error}"
-        write_failure_result(result_path, answer)
-        print(answer)
-        return 1
-    messages: list[dict[str, Any]] = [
-        {"role": "system", "content": system_prompt(str(workspace))},
-        {"role": "user", "content": prompt},
-    ]
-    calls: list[dict[str, Any]] = []
-    usage = {"input_tokens": 0, "output_tokens": 0, "cache_read_tokens": None, "cache_write_tokens": None}
-    cache_usage = PromptCacheUsage()
-    metrics = {"load_ms": 0.0, "prompt_eval_ms": 0.0, "eval_ms": 0.0}
-    execution_receipts: list[dict[str, Any]] = []
-    answer = ""
-    failure: str | None = None
-    transport_error: dict[str, Any] | None = None
-    calibration_dir = os.environ.get("FREELLAMA_AGENT_TOKEN_CALIBRATION_DIR", "").strip()
-    calibration_model_identity = (
-        resolve_model_identity(managed_endpoint or endpoint, model, messages[0]["content"])
-        if calibration_dir else {"verified": False, "scope": "persistence_disabled", "identity": ""}
-    )
-    context_manager = AgentContextManager(
-        runtime,
-        model=model,
-        calibration_identity=calibration_model_identity["identity"],
-        calibration_dir=Path(calibration_dir) if calibration_dir and calibration_model_identity["verified"] else None,
-    )
-    try:
-        messages = context_manager.fit(messages)
-    except ValueError as error:
-        answer = f"research task does not fit the configured context safely: {error}"
-        write_failure_result(result_path, answer)
-        print(answer)
-        return 1
-    chat_options = {
-        "temperature": runtime.temperature,
-        "seed": runtime.seed,
-        "num_ctx": runtime.num_ctx,
-        "num_predict": runtime.num_predict,
-    }
-
-    def call_model() -> dict[str, Any]:
-        # Retry explicit overload/connection refusals only. Replaying a timeout or an uncertain
-        # transport failure can overlap a generation that is still consuming the runner.
-        last_error: Exception | None = None
-        for attempt in range(runtime.retry_attempts):
-            try:
-                transport_endpoint = (
-                    f"{managed_endpoint}/_freellama/v1/tasks" if managed_endpoint else endpoint
-                )
-                url, payload = chat_request(
-                    transport_endpoint,
-                    model,
-                    messages,
-                    chat_options,
-                    runtime.think,
-                    runtime.keep_alive,
-                    execution_preference,
-                    min_placement_evidence,
-                )
-                response = unwrap_chat_response(
-                    request_json(url, payload, runtime.request_timeout_seconds), execution_receipts
-                )
-                break
-            except (HTTPError, URLError, TimeoutError) as error:
-                if not retryable_chat_error(error):
-                    raise
-                last_error = error
-                if attempt + 1 < runtime.retry_attempts:
-                    time.sleep(runtime.retry_backoff_seconds)
-        else:
-            raise last_error  # type: ignore[misc]
-        usage["input_tokens"] += int(response.get("prompt_eval_count", 0))
-        cache_usage.observe(response)
-        usage["cache_read_tokens"] = cache_usage.tokens
-        usage["output_tokens"] += int(response.get("eval_count", 0))
-        metrics["load_ms"] += float(response.get("load_duration", 0)) / 1_000_000
-        metrics["prompt_eval_ms"] += float(response.get("prompt_eval_duration", 0)) / 1_000_000
-        metrics["eval_ms"] += float(response.get("eval_duration", 0)) / 1_000_000
-        context_manager.observe(messages, response.get("prompt_eval_count"))
-        return response
-
-    def refit() -> bool:
-        nonlocal messages, failure
-        try:
-            messages = context_manager.fit(messages)
-            return True
-        except ValueError as error:
-            failure = f"research context cannot be fitted safely: {error}"
-            return False
-
-    seen_calls: dict[str, int] = {}
-    observations = resources.enter_context(ObservationStore(runtime.context.observation_page_chars))
-    parse_failures = 0
-    for _ in range(runtime.max_turns):
-        # Transport failures are terminal (call_model already retried them). A *parse* failure is
-        # not: tell the model what was wrong with its reply and let it correct itself, keeping every
-        # tool result gathered so far.
-        try:
-            response = call_model()
-        except (HTTPError, URLError, TimeoutError) as error:
-            diagnostic, transport_error = chat_error_details(error)
-            failure = f"agent response failed: {diagnostic}"
-            break
-        raw = str(response.get("message", {}).get("content", ""))
-        try:
-            action = parse_action(raw)
-        except (ValueError, json.JSONDecodeError) as error:
-            parse_failures += 1
-            if parse_failures > runtime.max_parse_repairs:
-                failure = f"agent gave {parse_failures} unparseable replies: {type(error).__name__}: {error}"
-                break
-            messages.append({"role": "assistant", "content": raw[: runtime.parse_repair_echo_chars]})
-            messages.append({"role": "user", "content": PARSE_REPAIR_NOTICE})
-            if not refit():
-                break
-            continue
-        parse_failures = 0
-        # Paging re-serves stored output. It costs a turn but never a subprocess, and no byte of
-        # any observation is ever unreachable — which is the whole point of paginating instead of
-        # clipping.
-        if action["action"] == "page":
-            try:
-                want_step = int(action.get("step", 0))
-                want_page = int(action.get("page", 1))
-            except (TypeError, ValueError):
-                want_step, want_page = 0, 1
-            messages.append({"role": "assistant", "content": json.dumps(action, ensure_ascii=False)})
-            messages.append({"role": "user", "content": observations.message(want_step, want_page)})
-            if not refit():
-                break
-            continue
-        if action["action"] == "finish":
-            answer = str(action.get("answer", "")).strip()
-            break
-        started = time.perf_counter()
-        tool_name = str(action.get("tool", ""))
-        queries = action.get("queries", {}) if isinstance(action.get("queries"), dict) else {}
-        # An exact repeat costs a turn out of ten AND a second copy of an observation the model
-        # has already seen. Answer it from the prior step instead of re-running the subprocess.
-        signature = call_signature({"tool": tool_name, "queries": queries})
-        repeated = signature in seen_calls
-        if repeated:
-            observation = REPEAT_NOTICE
-            status = "repeat"
-        else:
-            try:
-                observation = run_octocode(
-                    workspace, tool_name, queries, runtime.tool_timeout_seconds
-                )
-                status = "ok"
-            except (OSError, ValueError, subprocess.TimeoutExpired) as error:
-                observation = f"tool error: {type(error).__name__}: {error}"
-                status = "error"
-            seen_calls[signature] = len(calls) + 1
-        calls.append({
-            "name": f"mcp.octocode.{tool_name}" if tool_name else "mcp.octocode.unknown",
-            "raw_name": action.get("action", ""),
-            "arguments": {"tool": tool_name, "queries": queries},
-            "status": status,
-            "duration_ms": round((time.perf_counter() - started) * 1000, 3),
-        })
-        messages.append({"role": "assistant", "content": json.dumps(action, ensure_ascii=False)})
-        remaining = runtime.max_turns - len(calls)
-        step = len(calls)
-        observations.put(step, observation, metadata={"action": tool_name, "status": status, "source": json.dumps(queries, ensure_ascii=False)})
-        del observation
-        messages.append({"role": "user", "content": observations.message(step) + f"\n\nTool calls remaining: {remaining}. Finish now if the task is answerable; do not repeat prior calls."})
-        if not refit():
-            break
-    else:
-        messages.append({"role": "user", "content": "Tool budget is exhausted. Return exactly a finish JSON action now using the evidence collected. Do not request another tool."})
-        if refit():
-            try:
-                response = call_model()
-                final_action = parse_action(str(response.get("message", {}).get("content", "")))
-                if final_action.get("action") != "finish":
-                    raise ValueError("forced final response was not finish")
-                answer = str(final_action.get("answer", "")).strip()
-            except (HTTPError, URLError, TimeoutError, ValueError, json.JSONDecodeError) as error:
-                diagnostic, transport_error = chat_error_details(error)
-                failure = f"agent exceeded {runtime.max_turns} turns and finalization failed: {diagnostic}"
-
-    if not answer:
-        answer = failure or "agent stopped without a final answer"
-    result = {
-        "final_answer": answer,
-        "tool_calls": calls,
-        "usage": usage,
-        "provider_metrics": metrics,
-        "model_metadata": {
-            "adapter": "octocode_cli_agent_v1",
-            "temperature": chat_options["temperature"],
-            "seed": chat_options["seed"],
-            "num_ctx": chat_options["num_ctx"],
-            "num_predict": chat_options["num_predict"],
-            "max_turns": runtime.max_turns,
-            "runtime_config": runtime.metadata(),
-            "context_management": context_manager.metadata(),
-            "context_compactions": context_manager.compactions,
-            "transport": "managed" if managed_endpoint else "direct_ollama",
-            "execution_preference": execution_preference,
-            "min_placement_evidence": min_placement_evidence,
-            "execution_receipts": execution_receipts,
-            **({"transport_error": transport_error} if transport_error else {}),
-            "cache_token_metrics": cache_usage.metadata(),
-            "calibration_model_identity": calibration_model_identity,
-        },
-    }
-    observations.write_result(result_path, result)
-    print(answer)
-    return 1 if failure else 0
-
-
-def main() -> int:
-    with ExitStack() as resources:
-        return _main(resources)
-
-
-if __name__ == "__main__":
-    raise SystemExit(main())