forked from zjunlp/SkillNet
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathcreate_example.py
More file actions
47 lines (39 loc) · 1.24 KB
/
create_example.py
File metadata and controls
47 lines (39 loc) · 1.24 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
"""
Skill Creation Example - Using SkillNetClient
"""
import os
from skillnet_ai import SkillNetClient
# Sample trajectory
SAMPLE_TRAJECTORY = """
User: Please analyze this invoice.pdf and extract the total amount and date.
Agent: I will use the PDF tool. Opening invoice.pdf...
Agent: Text extracted. Found 'Total: $500' and 'Date: 2023-10-12'.
Agent: The total is $500 and the date is Oct 12, 2023.
"""
# Sample GitHub repository URL
SAMPLE_GITHUB_URL = "https://github.com/zjunlp/DeepKE"
def main():
# Initialize client
client = SkillNetClient(
api_key=os.getenv("API_KEY"),
base_url=os.getenv("BASE_URL", "https://api.openai.com/v1"),
github_token=os.getenv("GITHUB_TOKEN"),
)
# Example 1: Create from trajectory
print("🚀 Creating skill from trajectory...")
paths = client.create(
trajectory_content=SAMPLE_TRAJECTORY,
output_dir="./created_skills"
)
for p in paths:
print(f" - {p}")
# Example 2: Create from GitHub repository
print("\n🚀 Creating skill from GitHub...")
paths = client.create(
github_url=SAMPLE_GITHUB_URL,
output_dir="./created_skills"
)
for p in paths:
print(f" - {p}")
if __name__ == "__main__":
main()