Creative Services

Shared AI-powered creative writing services for BF Agent, Travel Beat, and other platform apps.

Overview

This package provides modular, reusable creative AI services:

• Prompt Template System - Type-safe prompt management with Pydantic schemas
• LLM Client - Unified client for OpenAI, Anthropic, Groq, Ollama
• LLM Registry - DB-driven LLM configuration (switch models without code changes)
• Dynamic Client - Tier-based LLM selection (economy/standard/premium)
• Usage Tracker - Token and cost tracking
• Security - Prompt injection detection and input sanitization
• Caching - Redis and in-memory caching with TTL
• Observability - Prometheus metrics and structured events

Installation

# Basic installation
pip install -e ../platform/packages/creative-services

# With OpenAI support
pip install -e ../platform/packages/creative-services[openai]

# With all LLM providers
pip install -e ../platform/packages/creative-services[all]

Quick Start

Simple Usage (Environment Variables)

from creative_services import DictRegistry, DynamicLLMClient, LLMTier
import asyncio

# Create registry from environment variables
registry = DictRegistry.from_env()
client = DynamicLLMClient(registry)

# Generate with tier-based selection
async def main():
    response = await client.generate(
        prompt="Write a short story about a traveler",
        system_prompt="You are a creative writer",
        tier=LLMTier.STANDARD,  # Uses GPT-4o-mini or Claude Sonnet
    )
    print(response.content)

asyncio.run(main())

Django Integration (BFAgent/TravelBeat)

from creative_services.adapters import DjangoLLMRegistry, BFAgentLLMBridge
from apps.bfagent.models import Llms

# Option 1: Use Django-backed registry
registry = DjangoLLMRegistry(Llms)
client = DynamicLLMClient(registry)

# Option 2: Use compatibility bridge (no code changes needed)
bridge = BFAgentLLMBridge.with_django_registry(Llms)
result = bridge.generate_text("Hello", tier="premium")

Backward Compatibility (Drop-in Replacement)

# Replace existing BFAgent imports:
# from apps.bfagent.services.llm_client import generate_text

# With:
from creative_services.adapters import generate_text

# Same API, uses new infrastructure
result = generate_text("Hello", tier="standard")

Architecture

creative_services/
├── prompts/                    # 🆕 Prompt Template System
│   ├── schemas/                # Pydantic schemas (frozen, immutable)
│   │   ├── variables.py        # PromptVariable, VariableType
│   │   ├── llm_config.py       # LLMConfig, RetryConfig
│   │   ├── template.py         # PromptTemplateSpec
│   │   └── execution.py        # PromptExecution, ExecutionStatus
│   ├── security/               # Injection detection & sanitization
│   ├── registry/               # Template storage backends
│   │   ├── memory.py           # InMemoryRegistry
│   │   ├── file.py             # FileRegistry (YAML/JSON)
│   │   └── django_registry.py  # DjangoRegistry
│   ├── execution/              # Execution engine
│   │   ├── executor.py         # PromptExecutor
│   │   ├── renderer.py         # Jinja2 template rendering
│   │   ├── cache.py            # InMemoryCache
│   │   ├── redis_cache.py      # RedisCache
│   │   └── retry.py            # Retry with backoff
│   ├── observability/          # Metrics & events
│   ├── migration/              # BFAgent compatibility
│   └── integration/            # Ready-to-use clients
│       ├── bfagent.py          # BFAgent integration
│       └── llm_clients.py      # OpenAI, Anthropic clients
├── core/                       # Core LLM infrastructure
├── adapters/                   # Django adapters
├── character/                  # Character generation
├── world/                      # World building
└── story/                      # Story generation

LLM Tiers

Supported LLM Providers

• OpenAI (GPT-4o, GPT-4o-mini, GPT-3.5)
• Anthropic (Claude 3.5 Sonnet, Claude 3 Opus, Claude 3 Haiku)
• Groq (Llama 3.3 70B - fast & free)
• Ollama (Local models)

Migration Guide

For BFAgent

1. Install creative-services: pip install -e ../platform/packages/creative-services
2. Existing code continues to work (no changes needed)
3. New code can use DynamicLLMClient directly
4. Gradually migrate old code to new API

For TravelBeat

1. Install creative-services
2. Replace direct OpenAI calls with DynamicLLMClient
3. Use LLMTier.STANDARD for story generation

Prompt Template System (New!)

Type-safe prompt management with 195 tests:

from creative_services.prompts import (
    PromptTemplateSpec,
    PromptVariable,
    PromptExecutor,
    InMemoryRegistry,
)
from creative_services.prompts.integration import OpenAIClient

# Define template
template = PromptTemplateSpec(
    template_key="greeting.v1",
    domain_code="examples",
    name="Greeting",
    system_prompt="You are friendly.",
    user_prompt="Say hello to {{ name }}!",
    variables=[PromptVariable(name="name", required=True)],
)

# Save and execute
registry = InMemoryRegistry()
registry.save(template)

executor = PromptExecutor(
    registry=registry,
    llm_client=OpenAIClient(),
    app_name="my_app",
)

result = await executor.execute(
    template_key="greeting.v1",
    variables={"name": "Alice"},
)
print(result.content)

BFAgent Integration

from creative_services.prompts.integration import create_bfagent_executor

executor = create_bfagent_executor()
result = await executor.execute(
    template_key="character.backstory.v1",
    variables={"name": "Elara", "genre": "fantasy"},
)

Documentation

Full documentation available at docs/:

cd docs && make html
# Open _build/html/index.html

License

MIT