wehub-resource-sync
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
import argparse
|
|
import asyncio
|
|
from pathlib import Path
|
|
|
|
from agents import Agent, Runner, ShellTool, ShellToolLocalSkill, trace
|
|
from examples.tools.shell import ShellExecutor
|
|
|
|
SKILL_NAME = "csv-workbench"
|
|
SKILL_DIR = Path(__file__).resolve().parent / "skills" / SKILL_NAME
|
|
|
|
|
|
def build_local_skill() -> ShellToolLocalSkill:
|
|
return {
|
|
"name": SKILL_NAME,
|
|
"description": "Analyze CSV files and return concise numeric summaries.",
|
|
"path": str(SKILL_DIR),
|
|
}
|
|
|
|
|
|
async def main(model: str) -> None:
|
|
local_skill = build_local_skill()
|
|
|
|
with trace("local_shell_skill_example"):
|
|
agent1 = Agent(
|
|
name="Local Shell Agent (Local Skill)",
|
|
model=model,
|
|
instructions="Use the available local skill to answer user requests.",
|
|
tools=[
|
|
ShellTool(
|
|
environment={
|
|
"type": "local",
|
|
"skills": [local_skill],
|
|
},
|
|
executor=ShellExecutor(),
|
|
)
|
|
],
|
|
)
|
|
|
|
result1 = await Runner.run(
|
|
agent1,
|
|
(
|
|
"Use the csv-workbench skill. Create /tmp/test_orders.csv with columns "
|
|
"id,region,amount,status and at least 6 rows. Then report total amount by "
|
|
"region and count failed orders."
|
|
),
|
|
)
|
|
print(f"Agent: {result1.final_output}")
|
|
|
|
agent2 = Agent(
|
|
name="Local Shell Agent (Reuse)",
|
|
model=model,
|
|
instructions="Reuse the existing local shell and answer concisely.",
|
|
tools=[
|
|
ShellTool(
|
|
environment={
|
|
"type": "local",
|
|
},
|
|
executor=ShellExecutor(),
|
|
)
|
|
],
|
|
)
|
|
|
|
result2 = await Runner.run(
|
|
agent2,
|
|
"Run `ls -la /tmp/test_orders.csv`, then summarize in one sentence.",
|
|
)
|
|
print(f"Agent (reuse): {result2.final_output}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"--model",
|
|
default="gpt-5.6-sol",
|
|
help="Model name to use.",
|
|
)
|
|
args = parser.parse_args()
|
|
asyncio.run(main(args.model))
|