prompt template下载 - prompt template源代码下载

提示模板

用于管理LLM提示模板的轻巧，零依赖性Python库。基于string.Template的设计原理。模板，但具有专门为LLM及时工程设计的增强功能。

特征

使用详细错误消息的强模板验证
支持嵌套括号和复杂的JSON结构
普通Python类型的自动价值序列化
默认值支持，并为可变类型的深层复制
具有价值继承的增量模板总体
类型提示以增强IDE支持
零依赖性 - 只是纯Python
100％测试覆盖范围

安装

pip install prompt-template

用法

图书馆故意非常易于使用。这个想法是要保持简单，进行验证和序列化，并使调试简单。

简单模板

 from prompt_template import PromptTemplate

# Create a template
template = PromptTemplate ( "Hello ${name}! Welcome to ${location}." )

# Render with values
result = template . to_string ( name = "Alice" , location = "Wonderland" )
print ( result )  # Hello Alice! Welcome to Wonderland.

默认值

 # Set default values that can be overridden later
template = PromptTemplate ( "Hello ${name}! Your settings are: ${settings}" )

# Set default values - they're safely deep copied
template . set_default (
    name = "Guest" ,
    settings = { "theme" : "light" , "language" : "en" }
)

# Use with defaults
print ( template . to_string ())
# Hello Guest! Your settings are: {"theme": "light", "language": "en"}

# Override specific values
print ( template . to_string ( name = "Alice" ))
# Hello Alice! Your settings are: {"theme": "light", "language": "en"}

# Override everything
print ( template . to_string (
    name = "Bob" ,
    settings = { "theme" : "dark" , "language" : "fr" }
))
# Hello Bob! Your settings are: {"theme": "dark", "language": "fr"}

命名模板

将名称添加到模板中会通过上下文增强错误消息：

 template = PromptTemplate (
    name = "user_greeting" ,
    template = "Hello ${name}! Welcome to ${location}."
)

复杂的JSON模板

图书馆优雅地处理嵌套结构：

 template = PromptTemplate ( """
{
    "user": {
        "name": "${username}",
        "role": "${role}"
    },
    "settings": {
        "theme": "${theme}",
        "notifications": ${notifications},
        "preferences": ${preferences}
    }
}
""" )

# Values are automatically serialized
result = template . to_string (
    username = "john_doe" ,
    role = "admin" ,
    theme = "dark" ,
    notifications = { "email" : True , "push" : False },
    preferences = [ "daily_digest" , "weekly_report" ]
)

增量模板总体

您可以逐步构建模板，并在此过程中保留默认值：

 # Start with a base template
base = PromptTemplate ( """
Query parameters:
    Model: ${model}
    Temperature: ${temperature}
    User: ${user}
    Prompt: ${prompt}
""" )

# Set some defaults
base . set_default (
    model = "gpt-4" ,
    temperature = 0.7
)

# Create a partially populated template
user_template = base . substitute ( user = "alice" )

# Complete the template later
final = user_template . to_string ( prompt = "Tell me a story" )

类型处理和自动序列化

该库自动处理各种Python类型：

 from uuid import UUID
from decimal import Decimal
from datetime import datetime

template = PromptTemplate ( """
{
    "id": "${id}",
    "amount": "${amount}",
    "binary": "${binary_data}",
    "metadata": ${metadata}
}
""" )

result = template . to_string (
    id = UUID ( "550e8400-e29b-41d4-a716-446655440000" ),
    amount = Decimal ( "45.67" ),
    binary_data = b"Hello World" ,  # Automatically base64 encoded if needed
    metadata = {
        "timestamp" : datetime . now (),  # Serialized via JSON
        "values" : [ 1 , 2 , 3 ]
    }
)

自定义序列化

扩展基类以自定义值序列化：

 from typing import Any
from datetime import datetime
from prompt_template import PromptTemplate as BasePromptTemplate
import orjson


class PromptTemplate ( BasePromptTemplate ):
    @ staticmethod
    def serializer ( value : Any ) -> str :
        return orjson . dumps ( value )  # use orjson for faster json serialization etc.

错误处理

图书馆提供了特定的广泛错误：

缺少值

 from prompt_template import MissingTemplateValuesError

template = PromptTemplate ( "Hello ${name}!" )
try :
    template . to_string ()  # No values provided
except MissingTemplateValuesError as e :
    print ( f"Missing values: { e . missing_values } " )  # {'name'}

无效的钥匙

 from prompt_template import InvalidTemplateKeysError

template = PromptTemplate ( "Hello ${name}!" )
try :
    template . to_string ( name = "World" , invalid_key = "value" )
except InvalidTemplateKeysError as e :
    print ( f"Invalid keys: { e . invalid_keys } " )  # ['invalid_key']
    print ( f"Valid keys: { e . valid_keys } " )  # {'name'}

序列化错误

 from prompt_template import TemplateSerializationError

template = PromptTemplate ( "Value: ${value}" )
try :
    template . to_string ( value = object ())  # Non-serializable object
except TemplateSerializationError as e :
    print ( f"Failed to serialize key ' { e . key } ': { e . original_error } " )