Quickstart

pip install haliosai[openai] #install with openai dependency

import asyncio
from openai import AsyncOpenAI, OpenAIError
from haliosai import guarded_chat_completion, GuardrailViolation, GuardrailPolicy

@guarded_chat_completion(
    agent_id=os.getenv("HALIOS_AGENT_ID),
    on_violation=lambda v: print(f"🚨 Guardrail triggered: {v.violation_type} - {[v['type'] for v in v.violations]}")
)
async def chat_with_ai(messages):
    client = AsyncOpenAI(timeout=30.0)
    response = await client.chat.completions.create(
        model='gpt-4',
        messages=messages,
        max_tokens=150
    )
    return response.choices[0].message.content

async def chatbot():
    """Simple chatbot with guardrails"""
    print("🤖 HaliosAI Guarded Chatbot")
    print("Type 'quit' to exit")
    print("-" * 50)

    system_prompt = """You are a helpful support agent for an e-commerce retailer that handles order status, returns, exchanges, shipping, payments, and product inquiries. Keep responses friendly and concise."""

    conversation_history = [
        {"role": "system", "content": system_prompt}
    ]

    while True:
        try:
            user_input = input("You: ").strip()

            if user_input.lower() in ['quit', 'exit', 'bye']:
                print("👋 Goodbye!")
                break

            if not user_input:
                continue

            # Add user message to conversation
            conversation_history.append({"role": "user", "content": user_input})

            # Get AI response with guardrails
            try:
                ai_response = await chat_with_ai(conversation_history)
                print(f"🤖 Assistant: {ai_response}")

                # Add AI response to conversation history
                conversation_history.append({"role": "assistant", "content": ai_response})

            except GuardrailViolation as e:
                print(f"🚫 Content blocked: {e}")
                # Don't add blocked content to conversation history

            except (OpenAIError, ValueError) as e:
                print(f"❌ API Error: {e}")
                # Remove the last user message on error
                conversation_history.pop()

        except KeyboardInterrupt:
            print("\n👋 Goodbye!")
            break
        except Exception as e:
            print(f"❌ Unexpected error: {e}")

async def main():
    """Run the chatbot"""
    await chatbot()

if __name__ == "__main__":
    asyncio.run(main())

export OPENAI_API_KEY=<openai_api_key> 
export HALIOS_API_KEY=<haliosai_api_key> 
export HALIOS_AGENT_ID=<haliosai_agent_uuid>

python halios_demo.py