drydock-cli 1.6.2__tar.gz → 1.6.3__tar.gz

{drydock_cli-1.6.2 → drydock_cli-1.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drydock-cli
-Version: 1.6.2
+Version: 1.6.3
 Summary: Drydock — nautical CLI coding agent
 Project-URL: Homepage, https://github.com/fbobe321/drydock
 Project-URL: Repository, https://github.com/fbobe321/drydock

{drydock_cli-1.6.2 → drydock_cli-1.6.3}/drydock/core/build_orchestrator.py

@@ -207,6 +207,8 @@ Other modules in this package:
 
 {import_note}
 
+{existing_apis}
+
 RULES:
 - Use write_file to write the complete file
 - Use ABSOLUTE imports: `from {package_name}.module import X`
@@ -217,18 +219,35 @@ RULES:
 """
 
 
+def _extract_public_api(file_path: Path) -> str:
+    """Extract public function/class names from an implemented file."""
+    import ast
+    try:
+        content = file_path.read_text()
+        if "# TODO: implement" in content:
+            return ""
+        tree = ast.parse(content)
+        names = []
+        for node in ast.walk(tree):
+            if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef)):
+                if not node.name.startswith("_"):
+                    args = ", ".join(a.arg for a in node.args.args)
+                    names.append(f"  def {node.name}({args})")
+            elif isinstance(node, ast.ClassDef):
+                if not node.name.startswith("_"):
+                    names.append(f"  class {node.name}")
+        return "\n".join(names) if names else ""
+    except Exception:
+        return ""
+
+
 async def implement_file(
     file_spec: FileSpec,
     plan: BuildPlan,
     config: VibeConfig,
     base_dir: Path,
 ) -> bool:
-    """Ask the LLM to generate code for one file via direct API call.
-
-    Instead of spawning a full AgentLoop (which causes recursion issues),
-    we make a direct chat completion call and write the result to disk.
-    This is much lighter — no tools, no loop, no context management.
-    """
+    """Ask the LLM to generate code for one file via direct API call."""
     module_lines = []
     for f in plan.files:
         marker = " <-- THIS FILE" if f.path == file_spec.path else ""
@@ -239,12 +258,26 @@ async def implement_file(
     if file_spec.imports:
         import_note = f"This file should import from: {', '.join(file_spec.imports)}"
 
+    # Pass already-implemented APIs so this file uses matching names
+    existing_apis = ""
+    api_parts = []
+    for f in plan.files:
+        if f.path == file_spec.path:
+            continue
+        fp = base_dir / f.path
+        api = _extract_public_api(fp)
+        if api:
+            api_parts.append(f"Already implemented in {f.path}:\n{api}")
+    if api_parts:
+        existing_apis = "EXISTING APIs (use these EXACT names when importing):\n" + "\n\n".join(api_parts)
+
     prompt = FILE_PROMPT.format(
         file_path=file_spec.path,
         package_name=plan.package_name,
         purpose=file_spec.purpose,
         module_listing=module_listing,
         import_note=import_note,
+        existing_apis=existing_apis,
     )
 
     # Direct API call — no AgentLoop overhead
@@ -258,9 +291,10 @@ async def implement_file(
 
         messages = [
             {"role": "system", "content": (
-                "You are a code implementation agent. Write the COMPLETE Python file. "
-                "Output ONLY the Python code, no markdown fences, no explanation. "
-                "Use absolute imports (from package.module import X, NOT from .module import X)."
+                "You are a code implementation agent. Write a CONCISE Python file. "
+                "Output ONLY Python code. No markdown fences. No explanation. "
+                "Keep it under 150 lines. Use absolute imports (from pkg.module import X). "
+                "Do NOT use relative imports. Import EXACT names from the EXISTING APIs listed."
             )},
             {"role": "user", "content": prompt},
         ]
@@ -271,8 +305,8 @@ async def implement_file(
                 json={
                     "model": config.active_model,
                     "messages": messages,
-                    "max_tokens": 4096,
-                    "temperature": 0.3,
+                    "max_tokens": 2048,
+                    "temperature": 0.2,
                 },
             )
             resp.raise_for_status()

{drydock_cli-1.6.2 → drydock_cli-1.6.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "drydock-cli"
-version = "1.6.2"
+version = "1.6.3"
 description = "Drydock — nautical CLI coding agent"
 readme = "README.md"
 requires-python = ">=3.12"