Skip to content

Instantly share code, notes, and snippets.

View deepthinarayanks's full-sized avatar

deepthinarayanks

View GitHub Profile
# 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.
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
from .agent import tdd_agent
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(
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).
google-adk
google-cloud-aiplatform
google-cloud-storage
google-genai
pypandoc
PyMuPDF
python-docx
docx
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)
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:
fastapi
uvicorn
pandas
google-auth
google-api-python-client
google-auth-oauthlib
google-genai
pydantic
gspread
google-adk
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}")