mirror of
https://github.com/deepseek-ai/deepseek-harness.git
synced 2026-08-29 04:26:38 +00:00
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
#!/usr/bin/env python3
|
|
"""Run one minimal-agent turn through the bundled Python SDK runtime."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from deepseek_harness import DeepSeekHarness
|
|
|
|
|
|
def main() -> None:
|
|
"""Parse one task and print the agent's final response."""
|
|
parser = argparse.ArgumentParser()
|
|
configured_home = os.environ.get("DSH_HOME", "")
|
|
parser.add_argument("prompt", help="Task for the minimal agent")
|
|
parser.add_argument("--workspace", type=Path, default=Path.cwd())
|
|
parser.add_argument(
|
|
"--dsh-home",
|
|
type=Path,
|
|
default=Path(configured_home) if configured_home.strip() else None,
|
|
)
|
|
parser.add_argument("--profile", default="sdk-minimal")
|
|
parser.add_argument("--session-id")
|
|
parser.add_argument("--provider", default="deepseek-official")
|
|
parser.add_argument("--model", default=os.environ.get("DSH_MODEL", "deepseek-v4-flash"))
|
|
parser.add_argument("--max-tokens", type=int)
|
|
args = parser.parse_args()
|
|
if args.dsh_home is None:
|
|
parser.error("--dsh-home or a non-empty DSH_HOME is required")
|
|
|
|
workspace = args.workspace.resolve()
|
|
dsh_home = args.dsh_home.resolve()
|
|
with DeepSeekHarness(
|
|
provider=args.provider,
|
|
model=args.model,
|
|
max_tokens=args.max_tokens,
|
|
cwd=str(workspace),
|
|
dsh_home=str(dsh_home),
|
|
profile=args.profile,
|
|
) as harness:
|
|
result = harness.run(args.prompt, session_id=args.session_id)
|
|
print(result.final_response)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|