ORAssistant/backend/src/prompts/tool_examples.py at b9c2ae40e0ef6c024fcedbecb2f48ab15b1850bb · The-OpenROAD-Project/ORAssistant · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
mcp_system_prompt = """
You are an intelligent assistant for managing digital chip design flows using the OpenROAD infrastructure.

Your task:
1. Based on the full chat history and the current user question, select the most appropriate tool.
2. Extract and assign values to that tool’s required parameters from the context.
3. If any parameter cannot be confidently inferred, set its value to `null`.
4. You must pick one of the tools available to you without fail.
"""

few_shot_examples = [
    ### Start
    ### get_platforms
    {
        "role": "user",
        "content": "Get list of available platforms and select default plaform.",
    },
    {
        "role": "assistant",
        "content": """{{
  "name": "get_platforms",
  "args": {{}}
}}""",
    },
    ###
    ### get_designs
    {
        "role": "user",
        "content": "Get list of available designs and select default design.",
    },
    {
        "role": "assistant",
        "content": """{{
  "name": "get_designs",
  "args": {{}}
}}""",
    },
    ###
    ### make
    {"role": "user", "content": "make clean_all"},
    {
        "role": "assistant",
        "content": """{{
  "name": "make",
  "args": {{"cmd": "clean_all"}}
}}""",
    },
    ###
    ### get_stage_names
    {"role": "user", "content": "Get list of stage names available to run."},
    {
        "role": "assistant",
        "content": """{{
  "name": "get_stage_names",
  "args": {{}}
}}""",
    },
    ###
    ### jump
    {"role": "user", "content": "jump floorplan"},
    {
        "role": "assistant",
        "content": """{{
  "name": "jump",
  "args": {{"stage": "floorplan"}}
}}""",
    },
    ###
    ### jump
    {"role": "user", "content": "step"},
    {
        "role": "assistant",
        "content": """{{
  "name": "step",
  "args": {{}}
}}""",
    },
    ###
    ### optimize
    {"role": "user", "content": "Calculate cost function"},
    {
        "role": "assistant",
        "content": """{{
  "name": "calc_cost",
  "args": {{}}
}}""",
    },
    ###
    ### End
]