from openai import OpenAI
from statebase import StateBase
client = OpenAI()
sb = StateBase()
# 1. Fetch Curated Context
# StateBase automatically ranks recent turns + relevant memories
context_package = sb.sessions.get_context(
session_id="sess_123",
query="How does the user like their coffee?"
)
# 2. Construct Prompt
messages = [
{
"role": "system",
"content": f"""
You are a helpful assistant.
# PREVIOUS KNOWLEDGE
{context_package.memories_str}
# CURRENT STATE
{context_package.state_str}
# RECENT HISTORY
{context_package.recent_history_str}
"""
},
{"role": "user", "content": "Order me a coffee."}
]
# 3. Generate
completion = client.chat.completions.create(
model="gpt-4-turbo",
messages=messages
)