ayrisdev/yargi-mcp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

convert enums to literal

Browse Source
This commit is contained in:
saidsurucu
2025-07-17 21:14:03 +03:00
parent cc363e7a6d
commit fa6c448afa
4 changed files with 36 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files
anayasa_mcp_module/unified_client.py
+7 -7
View File
@@ -9,7 +9,7 @@ from .models import (
AnayasaUnifiedSearchRequest,
AnayasaUnifiedSearchResult,
AnayasaUnifiedDocumentMarkdown,
AnayasaDecisionTypeEnum,
# Removed AnayasaDecisionTypeEnum - now using string literals
AnayasaNormDenetimiSearchRequest,
AnayasaBireyselReportSearchRequest
)
@@ -28,7 +28,7 @@ class AnayasaUnifiedClient:
async def search_unified(self, params: AnayasaUnifiedSearchRequest) -> AnayasaUnifiedSearchResult:
"""Unified search that routes to appropriate client based on decision_type."""
if params.decision_type == AnayasaDecisionTypeEnum.NORM_DENETIMI:
if params.decision_type == "norm_denetimi":
# Convert to norm denetimi request
norm_params = AnayasaNormDenetimiSearchRequest(
keywords_all=params.keywords_all or params.keywords,
@@ -44,13 +44,13 @@ class AnayasaUnifiedClient:
decisions_list = [decision.model_dump() for decision in result.decisions]
return AnayasaUnifiedSearchResult(
decision_type=AnayasaDecisionTypeEnum.NORM_DENETIMI,
decision_type="norm_denetimi",
decisions=decisions_list,
total_records_found=result.total_records_found,
retrieved_page_number=result.retrieved_page_number
)
elif params.decision_type == AnayasaDecisionTypeEnum.BIREYSEL_BASVURU:
elif params.decision_type == "bireysel_basvuru":
# Convert to bireysel başvuru request
bireysel_params = AnayasaBireyselReportSearchRequest(
keywords=params.keywords,
@@ -68,7 +68,7 @@ class AnayasaUnifiedClient:
decisions_list = [decision.model_dump() for decision in result.decisions]
return AnayasaUnifiedSearchResult(
decision_type=AnayasaDecisionTypeEnum.BIREYSEL_BASVURU,
decision_type="bireysel_basvuru",
decisions=decisions_list,
total_records_found=result.total_records_found,
retrieved_page_number=result.retrieved_page_number
@@ -88,7 +88,7 @@ class AnayasaUnifiedClient:
result = await self.norm_client.get_decision_document_as_markdown(document_url, page_number)
return AnayasaUnifiedDocumentMarkdown(
decision_type=AnayasaDecisionTypeEnum.NORM_DENETIMI,
decision_type="norm_denetimi",
source_url=result.source_url,
document_data=result.model_dump(),
markdown_chunk=result.markdown_chunk,
@@ -102,7 +102,7 @@ class AnayasaUnifiedClient:
result = await self.bireysel_client.get_decision_document_as_markdown(document_url, page_number)
return AnayasaUnifiedDocumentMarkdown(
decision_type=AnayasaDecisionTypeEnum.BIREYSEL_BASVURU,
decision_type="bireysel_basvuru",
source_url=result.source_url,
document_data=result.model_dump(),
markdown_chunk=result.markdown_chunk,