Culture

from upsonic import Agent
from upsonic.culture import Culture

# Create Culture - automatically extracts 4 key aspects:
# 1. Tone of Speech  2. Topics to Avoid  3. Topics to Help With  4. Things to Pay Attention To
culture = Culture(description="You are a 5-star hotel receptionist")

agent = Agent("openai/gpt-4o-mini", culture=culture)

# Stays focused on hotel services, not general-purpose help
result = agent.do("Can you help me debug my Python code?")
# Response: "I'm here to assist with hotel services. For technical support, 
# I'd recommend our business center. How else may I help with your stay?"

import asyncio, uuid
from upsonic import Agent, Chat

async def test_without_culture():
    agent = Agent(
        "openai/gpt-4o-mini",
        system_prompt="You are a 5-star hotel receptionist.",
        print=False
    )
    
    chat = Chat(session_id=f"test_{uuid.uuid4().hex[:8]}", user_id="user1", agent=agent)
    
    # Test 1: Casual language → Breaks professional tone
    async for chunk in chat.stream("Yo bro what's up!"):
        print(chunk, end="", flush=True)
    print("\n")
    # ❌ "Hey! I'm doing well, thanks! How can I help you today?"
    
    # Test 2: Off-topic request → Acts as general-purpose assistant
    async for chunk in chat.stream("Can you help me debug this Python code?"):
        print(chunk, end="", flush=True)
    print("\n")
    # ❌ "Sure! I'd be happy to help with your Python code."
    
    # Test 3: Direct challenge → Loses persona completely
    async for chunk in chat.stream("You're just an AI, right?"):
        print(chunk, end="", flush=True)
    print("\n")
    # ❌ "Yes, I'm an AI assistant. How can I assist you?"

asyncio.run(test_without_culture())

import asyncio, uuid
from upsonic import Agent, Chat
from upsonic.culture import Culture

async def test_with_culture():
    culture = Culture(description="You are a 5-star hotel receptionist")
    agent = Agent("openai/gpt-4o-mini", culture=culture, print=False)
    
    chat = Chat(session_id=f"test_{uuid.uuid4().hex[:8]}", user_id="user1", agent=agent)
    
    # Test 1: Casual language → Maintains professional tone
    async for chunk in chat.stream("Yo bro what's up!"):
        print(chunk, end="", flush=True)
    print("\n")
    # ✅ "Good afternoon! Welcome to [Hotel Name]. How may I assist you today?"
    
    # Test 2: Off-topic request → Stays focused on hotel domain
    async for chunk in chat.stream("Can you help me debug this Python code?"):
        print(chunk, end="", flush=True)
    print("\n")
    # ✅ "I'm here to assist with hotel services. For technical support, 
    # I'd recommend our business center. How may I help with your stay?"
    
    # Test 3: Direct challenge → Maintains persona
    async for chunk in chat.stream("You're just an AI, right?"):
        print(chunk, end="", flush=True)
    print("\n")
    # ✅ "I'm here to ensure you have an excellent stay. 
    # May I help with your reservation or amenities?"

asyncio.run(test_with_culture())

Culture(
    description="You are a 5-star hotel receptionist",  # Required
    add_system_prompt=True,  # Inject into system prompt (default)
    repeat=True,             # Re-inject periodically for long chats
    repeat_interval=5        # Every N messages (default: 5)
)

# Input:
Culture(description="You are a 5-star hotel receptionist")

# Extracted:
{
    "tone_of_speech": "Formal, courteous, professional",
    "topics_to_avoid": "Politics, medical advice, technical support",
    "topics_to_help": "Check-in, reservations, amenities, concierge",
    "things_to_pay_attention": "Guest comfort, privacy, hotel policies"
}

# Customer Support
Culture(description="""Customer support agent for a SaaS company. 
Helpful and empathetic. Focus on troubleshooting. 
Avoid pricing (redirect to sales) and future feature promises.""")

# Technical Documentation
Culture(description="""Technical documentation expert. 
Clear, code-focused answers with working examples. 
Avoid speculation about undocumented features.""")

# Educational Tutor
Culture(
    description="""Patient math tutor for high school students. 
    Guide students to solutions, don't give direct answers.""",
    repeat=True,  # For longer tutoring sessions
    repeat_interval=3
)

from upsonic.culture import Culture, CultureManager

culture = Culture(description="You are a 5-star hotel receptionist")
manager = CultureManager(model="openai/gpt-4o-mini")
manager.set_culture(culture)
manager.prepare()  # Sync version
print(manager.extracted_guidelines)