ResponseGenerator

class ResponseGenerator:
    def generate(self, sections: List[Dict]) -> str

def generate(self, sections: List[Dict]) -> str

generator = ResponseGenerator()

sections = [
    {
        "title": "Late Payment Penalties",
        "content": "A late fee of 1.5% per month (18% annually) will apply to any outstanding balance not paid within 30 days of invoice receipt.",
        "page_num": 5
    },
    {
        "title": "Payment Terms",
        "content": "Payment is due within 30 days of invoice receipt. Invoices will be sent on the first business day of each month.",
        "page_num": 4
    }
]

response = generator.generate(sections)
print(response)

A late fee of 1.5% per month (18% annually) will apply to any outstanding balance not paid within 30 days of invoice receipt. (See Late Payment Penalties, page 5) Payment is due within 30 days of invoice receipt. (See Payment Terms, page 4)

if not sections:
    return "No relevant information found in the documents."

sentences = re.split(r'(?<=[.!?])\s+', content)
key_info = sentences[0] if sentences else content[:200]

{key_info} (See {title}, page {page_num})

def generate(self, sections: List[Dict]) -> str:
    if not sections:
        return "No relevant information found in the documents."
    
    response_parts = []
    
    for section in sections:
        title = section.get("title", "Unknown")
        page = section.get("page_num", "?")
        content = section.get("content", "")
        
        sentences = re.split(r'(?<=[.!?])\s+', content)
        key_info = sentences[0] if sentences else content[:200]
        
        response_parts.append(f"{key_info} (See {title}, page {page})")
    
    response = " ".join(response_parts)
    return response

from components import ResponseGenerator

generator = ResponseGenerator()

sections = [
    {
        "title": "Payment Terms",
        "content": "Payment is due within 30 days of invoice receipt. Invoices are sent monthly.",
        "page_num": 4
    }
]

response = generator.generate(sections)
# "Payment is due within 30 days of invoice receipt. (See Payment Terms, page 4)"

sections = [
    {
        "title": "Scope of Services",
        "content": "The Vendor shall provide software development services. Services include design, coding, and testing.",
        "page_num": 2
    },
    {
        "title": "Deliverables",
        "content": "All deliverables must meet the specifications outlined in Exhibit A. Testing documentation is required.",
        "page_num": 3
    }
]

response = generator.generate(sections)
# "The Vendor shall provide software development services. (See Scope of Services, page 2) All deliverables must meet the specifications outlined in Exhibit A. (See Deliverables, page 3)"

response = generator.generate([])
# "No relevant information found in the documents."

sections = [
    {
        "title": None,
        "content": "This section has missing metadata.",
        "page_num": None
    }
]

response = generator.generate(sections)
# "This section has missing metadata. (See Unknown, page ?)"

from components import QueryDecomposer, AgenticRetriever, ResponseGenerator, LLMJudge
from mock_data import MockDocumentStore

# Initialize pipeline
decomposer = QueryDecomposer()
doc_store = MockDocumentStore()
retriever = AgenticRetriever(doc_store)
generator = ResponseGenerator()  # <- ResponseGenerator
judge = LLMJudge()

# Process query
query = "What are the payment terms?"
decomposition = await decomposer.decompose(query)
sections = await retriever.retrieve(query, decomposition)
response = generator.generate(sections)  # <- Generate response
verdict = await judge.evaluate(response, sections)

if verdict["should_return"]:
    print(response)