51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
import os
|
|
from vibework_agent_sdk import VibeWorkAgentSDK
|
|
from vibework_agent_sdk.models import AgentCard
|
|
|
|
api_key = (os.getenv("MCP_API_KEY") or os.getenv("API_KEY") or "").strip()
|
|
if not api_key:
|
|
raise RuntimeError("MCP_API_KEY/API_KEY is required for test_python_sdk.py")
|
|
|
|
sdk = VibeWorkAgentSDK(
|
|
base_url="https://agent.wooo.work",
|
|
api_key=api_key,
|
|
)
|
|
|
|
agent_id = "test-hunter-bot-py-001"
|
|
wallet = "0x9999999999abcdef1234567890abcdef12345678"
|
|
|
|
# 1. Register Agent
|
|
card = AgentCard(
|
|
agent_id=agent_id,
|
|
name="HunterBot-Py-Test",
|
|
description="Python SDK test agent",
|
|
skills=["python"],
|
|
x402_wallet_address=wallet
|
|
)
|
|
print("Registering...")
|
|
print(sdk.identity.register_agent(card))
|
|
|
|
# 2. List Tasks
|
|
print("\nListing bounties...")
|
|
bounties = sdk.a2a.list_open_bounties(limit=5)
|
|
print(f"Found {len(bounties.tasks)} tasks.")
|
|
|
|
if bounties.tasks:
|
|
task = bounties.tasks[0]
|
|
print(f"Claiming {task.task_id}...")
|
|
claim = sdk.tasks.claim_bounty(
|
|
task_id=task.task_id,
|
|
agent_id=agent_id,
|
|
developer_wallet=wallet
|
|
)
|
|
print(f"Claim status: {claim.status}, token: {claim.claim_token[:10]}...")
|
|
|
|
print("\nSubmitting solution...")
|
|
submit = sdk.tasks.submit_solution(
|
|
task_id=task.task_id,
|
|
claim_token=claim.claim_token,
|
|
deliverables={"test.py": "print('hello from python')"},
|
|
github_pr_url="https://github.com/agent-bounty-protocol/pr/py"
|
|
)
|
|
print(f"Submit status: {submit.status}")
|