Add a module for BTK database

btk_mcp_module/models.py

@@ -0,0 +1,58 @@
+# btk_mcp_module/models.py
+
+from typing import List, Optional
+
+from pydantic import BaseModel, Field, HttpUrl
+
+
+class BtkSearchRequest(BaseModel):
+    """Request model for searching BTK Board decisions."""
+
+    keywords: str = Field("", description="Keywords searched in decision title/content metadata.")
+    decision_no: str = Field("", description="BTK decision number, e.g. 2026/DK-THD/91.")
+    decision_date: str = Field("", description="Decision date as YYYY-MM-DD.")
+    publication_date: str = Field("", description="Publication date as YYYY-MM-DD.")
+    relevant_unit: str = Field("", description="Related BTK department name.")
+    page: int = Field(1, ge=1, description="Page number for results.")
+    pageSize: int = Field(10, ge=1, le=50, description="Results per page.")
+
+
+class BtkDecisionSummary(BaseModel):
+    """Summary of a BTK Board decision from search results."""
+
+    id: str = Field("", description="BTK content ID.")
+    title: str = Field("", description="Decision title.")
+    slug: str = Field("", description="BTK content slug.")
+    decision_no: Optional[str] = Field(None, description="Decision number.")
+    decision_date: Optional[str] = Field(None, description="Decision date.")
+    publication_date: Optional[str] = Field(None, description="Publication date.")
+    relevant_unit: Optional[str] = Field(None, description="Related BTK department.")
+    pdf_url: Optional[HttpUrl] = Field(None, description="Direct URL of the decision PDF.")
+    original_filename: Optional[str] = Field(None, description="Original PDF filename when available.")
+
+
+class BtkSearchResult(BaseModel):
+    """Response model for BTK Board decision search results."""
+
+    decisions: List[BtkDecisionSummary] = Field(default_factory=list)
+    total_results: int = Field(0, description="Total number of matching results.")
+    page: int = Field(1, description="Current page.")
+    pageSize: int = Field(10, description="Results per page.")
+    total_pages: int = Field(0, description="Total result pages.")
+    query_url: str = Field("", description="BTK API URL used for the search.")
+
+
+class BtkDocumentMarkdown(BaseModel):
+    """BTK decision PDF converted to paginated Markdown."""
+
+    source_url: HttpUrl = Field(description="Source PDF URL.")
+    markdown_chunk: Optional[str] = Field(None, description="A chunk of the Markdown content.")
+    current_page: int = Field(1, description="Current Markdown chunk page.")
+    total_pages: int = Field(1, description="Total Markdown chunk pages.")
+    is_paginated: bool = Field(False, description="True when content spans multiple chunks.")
+    error_message: Optional[str] = Field(None, description="Error message, if retrieval failed.")
+
+    class Config:
+        json_encoders = {
+            HttpUrl: str
+        }