from prompteus import llm_interact

@llm_interact(model_name="gpt-3.5-turbo")
def generate_story(theme: str):
    """
    Create a short story based on the given theme.
    The story should be engaging and no longer than 100 words.
    """
    return theme

story = generate_story("A robot learning to paint")
print(story)

from prompteus import agent
from your_custom_tools import ImageAnalysisTool

@agent(model_name="gpt-3.5-turbo", tools=[ImageAnalysisTool()])
def image_analyst():
    """
    Analyze the given image and provide a detailed description.
    Use the ImageAnalysisTool when needed to get more information about specific elements in the image.
    """
    return {}  # Additional context if needed

image_agent, _ = image_analyst()
result = await image_agent({"input": "path/to/image.jpg"})
print(result['messages'][-1].content)

from prompteus import agent_graph

@agent_graph()
async def analyze_and_critique(image_path: str):
    graph_structure = {
        'nodes': {
            'image_analyst': {'agent': image_analyst()},
            'art_critic': {'agent': art_critic()},
        },
        'edges': [
            {'from': 'image_analyst', 'to': 'art_critic'},
        ],
        'entry_point': 'image_analyst'
    }
    return graph_structure, image_path

result = await analyze_and_critique("path/to/artwork.jpg")
print(result)