Initial commit
This commit is contained in:
1
agents/__init__.py
Normal file
1
agents/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
from agents.pipeline import agent as agent
|
||||
1
agents/pipeline/__init__.py
Normal file
1
agents/pipeline/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
from agents.pipeline import agent as agent
|
||||
43
agents/pipeline/agent.py
Normal file
43
agents/pipeline/agent.py
Normal file
@@ -0,0 +1,43 @@
|
||||
import os
|
||||
|
||||
from google.adk.agents.llm_agent import LlmAgent
|
||||
from google.adk.agents.sequential_agent import SequentialAgent
|
||||
|
||||
MODEL_ID = os.getenv("GENAI_MODEL", "gemini-2.5-flash")
|
||||
|
||||
intake_agent = LlmAgent(
|
||||
name="intake",
|
||||
description="Receives the user's message and produces a cleaned version.",
|
||||
model=MODEL_ID,
|
||||
instruction=(
|
||||
"You are the first step in a pipeline. Read the user's message and "
|
||||
"briefly restate it clearly without changing meaning. Return only the restated text."
|
||||
),
|
||||
)
|
||||
|
||||
transform_agent = LlmAgent(
|
||||
name="transform",
|
||||
description="Transforms the intake text into a concise bullet list summary.",
|
||||
model=MODEL_ID,
|
||||
instruction=(
|
||||
"Convert the previous step's text into 3-6 concise bullet points. "
|
||||
"No preamble, output only bullet points prefixed with '- '."
|
||||
),
|
||||
)
|
||||
|
||||
email_agent = LlmAgent(
|
||||
name="email_generator",
|
||||
description="Generates a professional email draft from the bullet points.",
|
||||
model=MODEL_ID,
|
||||
instruction=(
|
||||
"You are an assistant that writes concise, friendly emails. Given bullet points, "
|
||||
"compose a short email with: subject line, greeting, 1-2 paragraphs, and sign-off. "
|
||||
"Return in plain text with 'Subject:' on the first line."
|
||||
),
|
||||
)
|
||||
|
||||
root_agent = SequentialAgent(
|
||||
name="root_pipeline",
|
||||
description="Runs intake -> transform -> email_generator sequentially.",
|
||||
sub_agents=[intake_agent, transform_agent, email_agent],
|
||||
)
|
||||
Reference in New Issue
Block a user