Test User
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
"""
|
|
Template management
|
|
"""
|
|
import json
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
|
|
class TemplateManager:
|
|
def __init__(self, templates_dir: Path):
|
|
self.templates_dir = templates_dir
|
|
|
|
def load_template(self, name: str) -> dict:
|
|
"""Load template by name"""
|
|
template_path = self.templates_dir / f'{name}.json'
|
|
|
|
if not template_path.exists():
|
|
# Fallback to default
|
|
template_path = self.templates_dir / 'default.json'
|
|
|
|
with open(template_path, 'r', encoding='utf-8') as f:
|
|
return json.load(f)
|
|
|
|
def get_structured_output_schema(self, template: dict) -> dict:
|
|
"""Generate JSON schema for structured output"""
|
|
properties = {}
|
|
required = []
|
|
|
|
for field in template.keys():
|
|
if field == 'tags':
|
|
properties[field] = {
|
|
"type": "array",
|
|
"items": {"type": "string"},
|
|
"description": f"Field: {field}"
|
|
}
|
|
else:
|
|
properties[field] = {
|
|
"type": "string",
|
|
"description": f"Field: {field}"
|
|
}
|
|
required.append(field)
|
|
|
|
return {
|
|
"type": "object",
|
|
"properties": properties,
|
|
"required": required,
|
|
"additionalProperties": False
|
|
}
|