OpenAI Integration

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
)