| import argparse |
| import json |
| from pathlib import Path |
|
|
| from guidance_sidecar import GuidanceEngine, GuidanceSettings |
|
|
|
|
| def main() -> None: |
| parser = argparse.ArgumentParser(description="Local Hugging Face coding guidance sidecar") |
| subparsers = parser.add_subparsers(dest="command", required=True) |
|
|
| guide_parser = subparsers.add_parser("guide", help="Ask the local guidance model for structured coding help") |
| guide_parser.add_argument("--task", required=True) |
| guide_parser.add_argument("--context", default="") |
| guide_parser.add_argument("--project", action="store_true") |
| guide_parser.add_argument("--root", default=".") |
| guide_parser.add_argument("--model", default=None) |
| guide_parser.add_argument("--adapter", default=None) |
| guide_parser.add_argument("--bundle", default=None, help="Path to an HF bundle containing base_model/") |
| guide_parser.add_argument("--keep-loaded-seconds", type=int, default=None) |
| guide_parser.add_argument("--no-cache", action="store_true") |
|
|
| smoke_parser = subparsers.add_parser("smoke", help="Run one guidance request and save JSON result") |
| smoke_parser.add_argument("--task", default="Add input validation and tests for a Python function without changing public API.") |
| smoke_parser.add_argument("--output", default="./runtime/guidance_smoke_result.json") |
|
|
| args = parser.parse_args() |
|
|
| if args.command == "guide": |
| if args.bundle: |
| engine = GuidanceEngine.from_bundle(args.bundle) |
| else: |
| engine = GuidanceEngine( |
| model_name=args.model, |
| adapter_dir=args.adapter, |
| keep_loaded_seconds=args.keep_loaded_seconds, |
| ) |
| result = ( |
| engine.advise_project(args.task, root=args.root, use_cache=not args.no_cache) |
| if args.project |
| else engine.advise(args.task, args.context, use_cache=not args.no_cache) |
| ) |
| print(json.dumps(result, indent=2)) |
| elif args.command == "smoke": |
| engine = GuidanceEngine(GuidanceSettings(max_new_tokens=384)) |
| result = engine.advise_project(args.task) |
| output = Path(args.output) |
| output.parent.mkdir(parents=True, exist_ok=True) |
| output.write_text(json.dumps(result, indent=2), encoding="utf-8") |
| print(json.dumps(result, indent=2)) |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
