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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drydock-cli
-Version: 1.6.2
+Version: 1.6.4
 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.4}/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()
@@ -369,29 +403,30 @@ async def run_build_pipeline(
     failed_files = [path for path, ok in results if not ok]
 
     summary_lines = [
-        f"PROJECT BUILT: {plan.package_name}",
-        f"Description: {plan.description}",
-        f"Files: {succeeded}/{len(plan.files)} implemented",
+        f"ALL FILES ALREADY CREATED for package '{plan.package_name}'. "
+        f"DO NOT create any more files. DO NOT write __init__.py or __main__.py. "
+        f"They already exist with correct absolute imports.",
         f"",
-        f"Package structure:",
+        f"Files created ({succeeded}/{len(plan.files)}):",
     ]
     for f in plan.files:
-        status = "OK" if any(p == f.path and ok for p, ok in results) else "STUB"
-        summary_lines.append(f"  {f.path} [{status}] — {f.purpose}")
+        status = "DONE" if any(p == f.path and ok for p, ok in results) else "STUB"
+        summary_lines.append(f"  {f.path} [{status}]")
 
     summary_lines.extend([
         f"",
-        f"Entry point: python3 -m {plan.package_name}",
-        f"",
-        f"Test the project now by running: python3 -m {plan.package_name} --help",
-        f"If it fails, read the error and fix with search_replace.",
+        f"YOUR ONLY JOB NOW: test and fix.",
+        f"Step 1: Run 'python3 -m {plan.package_name} --help' with bash",
+        f"Step 2: If ImportError, use read_file to check the file, then search_replace to fix the import",
+        f"Step 3: Run again. If it works, tell the user the project is complete.",
+        f"Step 4: DO NOT create new files. DO NOT rewrite existing files. Only fix imports.",
     ])
 
     if failed_files:
         summary_lines.extend([
             f"",
-            f"NEEDS IMPLEMENTATION: {', '.join(failed_files)}",
-            f"These files have stubs only. Implement them with write_file.",
+            f"These files need implementation: {', '.join(failed_files)}",
+            f"Use write_file with overwrite=true to implement them.",
         ])
 
     return "\n".join(summary_lines)

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

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