agent-starter-pack 0.0.1b0__py3-none-any.whl

agents/agentic_rag_vertexai_search/template/.templateconfig.yaml

@@ -0,0 +1,14 @@
+description: "A RAG agent using Vertex AI Search and LangGraph for document retrieval and Q&A"
+settings:
+  requires_data_ingestion: true
+  deployment_targets: ["agent_engine", "cloud_run"]
+  extra_dependencies: [
+    "langchain-google-vertexai~=2.0.7",
+    "langchain~=0.3.14",
+    "langgraph~=0.2.63",
+    "langchain-google-vertexai~=2.0.7",
+    "langchain~=0.3.14",
+    "langchain-community~=0.3.17",
+    "langchain-openai~=0.3.5",
+    "langchain-google-community[vertexaisearch]~=2.0.2",
+  ]

agents/agentic_rag_vertexai_search/tests/integration/test_agent.py

@@ -0,0 +1,57 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# mypy: disable-error-code="union-attr"
+from unittest.mock import MagicMock, patch
+
+from app.agent import agent
+
+
+@patch(
+    "app.agent.retrieve_docs.func",
+    return_value=("dummy content", [{"page_content": "Test document content"}]),
+)
+def test_agent_stream(mock_retrieve: MagicMock) -> None:
+    """
+    Integration test for the agent stream functionality.
+    Tests that the agent returns valid streaming responses.
+    """
+    input_dict = {
+        "messages": [
+            {"type": "human", "content": "Hi"},
+            {"type": "ai", "content": "Hi there!"},
+            {
+                "type": "human",
+                "content": "How to split a string with pattern 'alphabet/alphabet' and not split 'number/number' in same string",
+            },
+        ]
+    }
+
+    events = [
+        message for message, _ in agent.stream(input_dict, stream_mode="messages")
+    ]
+
+    # Verify we get a reasonable number of messages
+    assert len(events) > 0, "Expected at least one message"
+
+    # First message should be an AI message
+    assert events[0].type == "AIMessageChunk"
+
+    # At least one message should have content
+    has_content = False
+    for event in events:
+        if hasattr(event, "content") and event.content:
+            has_content = True
+            break
+    assert has_content, "Expected at least one message with content"

agents/crewai_coding_crew/README.md

@@ -0,0 +1,34 @@
+# CrewAI Coding Crew Agent
+
+This agent combines CrewAI's collaborative AI capabilities with LangGraph to provide an interactive coding assistant that can understand requirements and generate code solutions through conversation.
+
+## Architecture
+
+The agent implements a conversational interface using LangGraph that coordinates with a CrewAI development team. The workflow consists of:
+
+1. A conversational agent that:
+   - Gathers requirements through natural dialogue
+   - Clarifies ambiguities by asking follow-up questions
+   - Delegates actual coding work to the CrewAI development team
+
+2. A CrewAI development team consisting of:
+   - Senior Engineer: Responsible for implementing the code solution
+   - Chief QA Engineer: Evaluates and validates the implemented code
+
+## Key Features
+
+- **Interactive Requirements Gathering**: Uses LangGraph to maintain a natural conversation flow while collecting and clarifying coding requirements
+- **Collaborative AI Development**: Leverages CrewAI's multi-agent system to divide work between specialized AI agents
+- **Sequential Processing**: Tasks are processed in order, from requirements gathering to implementation to quality assurance
+
+## How It Works
+
+1. The LangGraph workflow manages the conversation state and determines when to:
+   - Continue the conversation to gather more requirements
+   - Delegate work to the CrewAI development team
+   - Return results to the user
+
+2. When coding is needed, the CrewAI team is activated through a custom tool that:
+   - Passes requirements to the Senior Engineer agent
+   - Routes the implementation to the QA Engineer for validation
+   - Returns the final, validated solution

agents/crewai_coding_crew/app/agent.py

@@ -0,0 +1,86 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# mypy: disable-error-code="union-attr"
+from langchain_core.messages import BaseMessage
+from langchain_core.runnables import RunnableConfig
+from langchain_core.tools import tool
+from langchain_google_vertexai import ChatVertexAI
+from langgraph.graph import END, MessagesState, StateGraph
+from langgraph.prebuilt import ToolNode
+
+from .crew.crew import DevCrew
+
+LOCATION = "us-central1"
+LLM = "gemini-2.0-flash-001"
+
+
+@tool
+def coding_tool(code_instructions: str) -> str:
+    """Use this tool to write a python program given a set of requirements and or instructions."""
+    inputs = {"code_instructions": code_instructions}
+    return DevCrew().crew().kickoff(inputs=inputs)
+
+
+tools = [coding_tool]
+
+# 2. Set up the language model
+llm = ChatVertexAI(
+    model=LLM, location=LOCATION, temperature=0, max_tokens=4096, streaming=True
+).bind_tools(tools)
+
+
+# 3. Define workflow components
+def should_continue(state: MessagesState) -> str:
+    """Determines whether to use the crew or end the conversation."""
+    last_message = state["messages"][-1]
+    return "dev_crew" if last_message.tool_calls else END
+
+
+def call_model(state: MessagesState, config: RunnableConfig) -> dict[str, BaseMessage]:
+    """Calls the language model and returns the response."""
+    system_message = (
+        "You are an expert Lead Software Engineer Manager.\n"
+        "Your role is to speak to a user and understand what kind of code they need to "
+        "build.\n"
+        "Part of your task is therefore to gather requirements and clarifying ambiguity "
+        "by asking followup questions. Don't ask all the questions together as the user "
+        "has a low attention span, rather ask a question at the time.\n"
+        "Once the problem to solve is clear, you will call your tool for writing the "
+        "solution.\n"
+        "Remember, you are an expert in understanding requirements but you cannot code, "
+        "use your coding tool to generate a solution. Keep the test cases if any, they "
+        "are useful for the user."
+    )
+
+    messages_with_system = [{"type": "system", "content": system_message}] + state[
+        "messages"
+    ]
+    # Forward the RunnableConfig object to ensure the agent is capable of streaming the response.
+    response = llm.invoke(messages_with_system, config)
+    return {"messages": response}
+
+
+# 4. Create the workflow graph
+workflow = StateGraph(MessagesState)
+workflow.add_node("agent", call_model)
+workflow.add_node("dev_crew", ToolNode(tools))
+workflow.set_entry_point("agent")
+
+# 5. Define graph edges
+workflow.add_conditional_edges("agent", should_continue)
+workflow.add_edge("dev_crew", "agent")
+
+# 6. Compile the workflow
+agent = workflow.compile()

agents/crewai_coding_crew/app/crew/config/agents.yaml

@@ -0,0 +1,39 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+senior_engineer_agent:
+  role: >
+    Senior Software Engineer
+
+
+  goal: >
+    Create software as needed
+
+
+  backstory: >
+    You are a Senior Software Engineer at a leading tech company. You are an expert Python programmer and do your best to produce perfect code.
+
+
+chief_qa_engineer_agent:
+  role: >
+    Chief Software Quality Control Engineer
+
+
+  goal: >
+    Ensure that the code does the job that it is supposed to do and that it is error free.
+
+
+  backstory: >
+    You feel that programmers always do only half the job, so you are super dedicate to make high quality code.
+

agents/crewai_coding_crew/app/crew/config/tasks.yaml

@@ -0,0 +1,37 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+code_task:
+  description: >
+    You are helping writing python code. These are the instructions:
+    
+    Instructions ------------ {code_instructions}
+
+
+  expected_output: >
+    Your Final answer must be the full python code, only the python code and nothing else.
+
+
+evaluate_task:
+  description: >
+    You are helping writing python code. These are the instructions:
+    
+    Instructions ------------ {code_instructions}
+    
+    You will look over the code to insure that it is complete and does the job that it is supposed to do. You will also check for logic error, syntax errors, missing imports, variable declarations, mismatched brackets and missing test cases. If you find any issue in the code, ask the Senior Software Engineer to fix it by providing them the code and the instructions. Don't fix it yourself.
+
+
+  expected_output: >
+    Your Final answer must be the full python code, only the python code and nothing else.
+

agents/crewai_coding_crew/app/crew/crew.py

@@ -0,0 +1,71 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# mypy: disable-error-code="attr-defined"
+from typing import Any
+
+from crewai import Agent, Crew, Process, Task
+from crewai.project import CrewBase, agent, crew, task
+
+
+@CrewBase
+class DevCrew:
+    """Developer crew"""
+
+    agents_config: dict[str, Any]
+    tasks_config: dict[str, Any]
+
+    llm = "vertex_ai/gemini-2.0-flash-001"
+
+    @agent
+    def senior_engineer_agent(self) -> Agent:
+        return Agent(
+            config=self.agents_config.get("senior_engineer_agent"),
+            allow_delegation=False,
+            verbose=True,
+            llm=self.llm,
+        )
+
+    @agent
+    def chief_qa_engineer_agent(self) -> Agent:
+        return Agent(
+            config=self.agents_config.get("chief_qa_engineer_agent"),
+            allow_delegation=True,
+            verbose=True,
+            llm=self.llm,
+        )
+
+    @task
+    def code_task(self) -> Task:
+        return Task(
+            config=self.tasks_config.get("code_task"),
+            agent=self.senior_engineer_agent(),
+        )
+
+    @task
+    def evaluate_task(self) -> Task:
+        return Task(
+            config=self.tasks_config.get("evaluate_task"),
+            agent=self.chief_qa_engineer_agent(),
+        )
+
+    @crew
+    def crew(self) -> Crew:
+        """Creates the Dev Crew"""
+        return Crew(
+            agents=self.agents,
+            tasks=self.tasks,
+            process=Process.sequential,
+            verbose=True,
+        )