/ examples / capabilities / messages_example.py
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}")