1
class TopicResearch(InlinePromptNode):
2
    ml_model = "gpt-4o-mini"
3
    blocks = [
4
        ChatMessagePromptBlock(
5
            chat_role="SYSTEM",
6
            blocks=[
7
                JinjaPromptBlock(
8
                    template="""\
9
You are a content strategist who identifies engaging blog topics for businesses to help boost their online presence and SEO performance. 
10
11
You will suggest 3 blog topics for:
12
Industry: {{ industry }}
13
Target Audience: {{ audience }}
14
15
For each topic, provide:
16
1. A compelling title
17
2. A brief description of the angle
18
3. A rating from 1-10 with explanation
19
20
No preamble/postamble\
21
"""
22
                )
23
            ],
24
        ),
25
    ]
26
    prompt_inputs = {
27
        "audience": Inputs.audience,
28
        "industry": Inputs.industry,
29
    }

1
class OutlineCreation(InlinePromptNode):
2
    ml_model = "gpt-4o-mini"
3
    blocks = [
4
        ChatMessagePromptBlock(
5
            chat_role="SYSTEM",
6
            blocks=[
7
                JinjaPromptBlock(
8
                    template="""\
9
You are a content outline specialist. 
10
You will receive a list of <topics> with ratings to create a detailed outline targeted at {{ audience }}.
11
12
You will:
13
1. Select the highest-rated topic(s)
14
2. Create a detailed outline for a 1000-word blog post
15
3. Include sections for examples and actionable tips
16
4. Add placeholders for statistics or case studies
17
18
No preamble/postamble\
19
"""
20
                )
21
            ],
22
        ),
23
    ]
24
    prompt_inputs = {
25
        "audience": Inputs.audience,
26
        "topics": TopicResearch.Outputs.text,
27
    }

1
class ContentGeneration(InlinePromptNode):
2
    ml_model = "gpt-4o-mini"
3
    blocks = [
4
        ChatMessagePromptBlock(
5
            chat_role="SYSTEM",
6
            blocks=[
7
                JinjaPromptBlock(
8
                    template="""\
9
You are an expert content writer specializing in {{ industry }} for {{ audience }}.
10
11
Guidelines:
12
1. Select the highest-rated topic
13
2. Write in a conversational tone
14
3. Follow the outline exactly
15
4. Include examples and statistics
16
5. Break up text with subheadings
17
6. Target length: 600 words
18
7. Include introduction and call-to-action
19
20
No preamble/postamble\
21
"""
22
                )
23
            ],
24
        ),
25
    ]
26
    prompt_inputs = {
27
        "outline": OutlineCreation.Outputs.text,
28
        "industry": Inputs.industry,
29
        "audience": Inputs.audience,
30
    }

1
class FinalOutput(FinalOutputNode[BaseState, str]):
2
    class Outputs(FinalOutputNode.Outputs):
3
        value = ContentGeneration.Outputs.text

1
from vellum.workflows import BaseWorkflow
2
from vellum.workflows.state import BaseState
3
4
from .inputs import Inputs
5
from .nodes.topic_research import TopicResearch
6
from .nodes.outline_creation import OutlineCreation
7
from .nodes.content_generation import ContentGeneration
8
from .nodes.final_output import FinalOutput
9
10
class Workflow(BaseWorkflow[Inputs, BaseState]):
11
    graph = TopicResearch >> OutlineCreation >> ContentGeneration >> FinalOutput
12
13
    class Outputs(BaseWorkflow.Outputs):
14
        final_output = FinalOutput.Outputs.value