diff --git a/README.md b/README.md index 9bfa0d19..5b1a27a0 100644 --- a/README.md +++ b/README.md @@ -255,23 +255,25 @@ AI Planning & Orchestration System built for the enterprise. Read more [here](ht ```python from ai21 import AI21Client - -client = AI21Client() - +from ai21.models.chat import ChatMessage +client = AI21Client(api_key="your_API_key") run_result = client.beta.maestro.runs.create_and_poll( - input="Write a poem about the ocean", + input="Tell me about AI21 Maestro", requirements=[ { "name": "length requirement", - "description": "The length of the poem should be less than 1000 characters", + "description": "The length of the response should be less than 2000 characters", }, { - "name": "rhyme requirement", - "description": "The poem should rhyme", + "name": "source requirement", + "description": "Should rely on information from these websites: https://www.ai21.com/, https://www.ai21.com/maestro/, https://docs.ai21.com/home", }, ], include=["requirements_result"] ) +print("\n=== AI21 MAESTRO INFO ===") +print(run_result.result) +print() ``` For a more detailed example, see maestro [sync](examples/studio/maestro/run.py) and [async](examples/studio/maestro/async_run.py) examples. diff --git a/examples/studio/maestro/run.py b/examples/studio/maestro/run.py index b32258e4..4c8fc50d 100644 --- a/examples/studio/maestro/run.py +++ b/examples/studio/maestro/run.py @@ -1,26 +1,28 @@ from ai21 import AI21Client - -client = AI21Client() +from ai21.models.chat import ChatMessage +client = AI21Client(api_key="your_API_key") def main(): try: run_result = client.beta.maestro.runs.create_and_poll( - input="Write a poem about the ocean", + input="Tell me about AI21 Maestro", requirements=[ { "name": "length requirement", - "description": "The length of the poem should be less than 1000 characters", + "description": "The length of the response should be less than 2000 characters", }, { - "name": "rhyme requirement", - "description": "The poem should rhyme", + "name": "source requirement", + "description": "Should rely on information from these websites: https://www.ai21.com/, https://www.ai21.com/maestro/, https://docs.ai21.com/home", }, ], include=["requirements_result"], ) - print(run_result) + print("\n=== AI21 MAESTRO INFO ===") + print(run_result.result) + print() except TimeoutError: print("The run timed out")