1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17

from sglang import function, gen, Runtime

@function
def generate_product(product_name: str, category: str):
    return gen(
        f"""
        生成{category}产品{product_name}的描述：
        {{
            "name": "{product_name}",
            "price": {gen(constraint="float", min=0, max=9999)},
            "tags": {gen(constraint="list[string]", min_len=2, max_len=5)}
        }}
        """
    )

runtime = Runtime(model_path="your-model")
result = generate_product.run(runtime, product_name="耳机", category="数码")