Загрузка prompt template - Загрузка исходного prompt template

Быстрый шаблон

Легкая библиотека Python с нулевой зависимостью для управления шаблонами быстрого LLM. Построен на принципах проектирования string.Template , но с расширенными функциями, специально разработанными для инженерии LLM.

Функции

Сильная проверка шаблонов с подробными сообщениями об ошибках
Поддержка вложенных скоб и сложных структур JSON
Автоматическая сериализация для общих типов питонов
Поддержка значения по умолчанию с глубоким копированием для изменяемых типов
Инкрементное шаблон популяции с наследством стоимости
Тип подсказки для улучшенной поддержки IDE
Нулевые зависимости - просто Pure 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 } " )