messages_example.py
1 """ 2 Messages Capability Example 3 4 Demonstrates Anthropic-style messages API and token counting. 5 """ 6 7 from praisonai.capabilities import messages_create, count_tokens 8 9 # Create a message (Anthropic-style) 10 print("=== Messages Create ===") 11 result = messages_create( 12 messages=[{"role": "user", "content": "What is the meaning of life? Answer briefly."}], 13 model="gpt-4o-mini", # Works with any model via LiteLLM 14 max_tokens=100, 15 system="You are a philosophical assistant." 16 ) 17 print(f"Message ID: {result.id}") 18 if result.content: 19 for block in result.content: 20 if block.get("type") == "text": 21 print(f"Response: {block.get('text')}") 22 print(f"Usage: {result.usage}") 23 24 # Count tokens 25 print("\n=== Token Counting ===") 26 result = count_tokens( 27 messages=[ 28 {"role": "system", "content": "You are a helpful assistant."}, 29 {"role": "user", "content": "Hello, how are you today?"} 30 ], 31 model="gpt-4o-mini" 32 ) 33 print(f"Token count: {result.input_tokens}")