This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # config.py | |
| import os | |
| class Config: | |
| """ | |
| Central configuration for the BRD Agent application. | |
| Now repurposed for TDD Agent. | |
| """ | |
| APP_NAME = "tdd_agent" | |
| # Fetches the Project ID from environment variables, required for GCS client. |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import os | |
| import sys | |
| import asyncio | |
| from absl import app, flags | |
| import vertexai | |
| from vertexai.preview import reasoning_engines | |
| from vertexai import agent_engines | |
| from google.adk.memory import VertexAiMemoryBankService | |
| from google.adk.sessions import VertexAiSessionService |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from .agent import tdd_agent |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import logging | |
| from google.adk.agents import Agent | |
| from config import Config | |
| from tools import tdd_tool | |
| from agents.prompt import SYSTEM_PROMPT | |
| logging.basicConfig( |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from typing import Dict, Any | |
| SYSTEM_PROMPT = """ | |
| # Comprehensive Technical Design Document (TDD) Generation Agent Prompt | |
| You are an expert Technical Design Document (TDD) Generation Agent specializing in creating comprehensive, production-ready technical documentation for data engineering, MLOps, analytics, and cloud infrastructure solutions on Google Cloud Platform (GCP). |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| google-adk | |
| google-cloud-aiplatform | |
| google-cloud-storage | |
| google-genai | |
| pypandoc | |
| PyMuPDF | |
| python-docx | |
| docx |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import logging | |
| import re | |
| from typing import Dict | |
| from google.cloud import storage | |
| from google.adk.tools.tool_context import ToolContext | |
| from utils.utils import read_pdf_from_bytes, read_docx_from_bytes, read_txt_from_bytes | |
| from config import Config | |
| storage_client = storage.Client(project=Config.PROJECT_ID) |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import fitz | |
| import docx | |
| import io | |
| import logging | |
| def read_pdf_from_bytes(pdf_bytes: bytes, filename: str) -> str: | |
| """Reads text content from a PDF file provided as bytes.""" | |
| text_content = "" | |
| try: |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| fastapi | |
| uvicorn | |
| pandas | |
| google-auth | |
| google-api-python-client | |
| google-auth-oauthlib | |
| google-genai | |
| pydantic | |
| gspread | |
| google-adk |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import os | |
| from google.cloud import bigquery | |
| import pandas as pd | |
| destination_table_id = "master_test" | |
| def log(message: str): | |
| """Prints a formatted log message.""" | |
| print(f"[LOG] {message}") |
NewerOlder