mirror of https://github.com/aliasrobotics/cai.git
28 lines
605 B
Python
28 lines
605 B
Python
from cai import CAI, Agent
|
|
|
|
client = CAI()
|
|
|
|
my_agent = Agent(
|
|
name="Agent",
|
|
instructions="You are a helpful agent.",
|
|
)
|
|
|
|
|
|
def pretty_print_messages(messages):
|
|
for message in messages:
|
|
if message["content"] is None:
|
|
continue
|
|
print(f"{message['sender']}: {message['content']}")
|
|
|
|
|
|
messages = []
|
|
agent = my_agent
|
|
while True:
|
|
user_input = input("> ")
|
|
messages.append({"role": "user", "content": user_input})
|
|
|
|
response = client.run(agent=agent, messages=messages)
|
|
messages = response.messages
|
|
agent = response.agent
|
|
pretty_print_messages(messages)
|