Merge pull request #1 from BudEcosystem/feat/prompt-executor

Feat/prompt executor

Author: vsraccubits

.env.sample

@@ -21,3 +21,6 @@ PSQL_PORT=5432
 PSQL_DB_NAME=
 SECRETS_PSQL_USER=
 SECRETS_PSQL_PASSWORD=
+
+BUD_GATEWAY_BASE_URL=http://0.0.0.0/v1
+BUD_DEFAULT_MODEL_NAME=qwen3-32b

CLAUDE.md

@@ -144,4 +144,13 @@ Key environment variables (see `.env.sample`):
 - Always use Alembic for database changes
 - Configuration syncs periodically from Dapr configuration store
 - Pre-commit hooks enforce code quality standards
-- Ruff is configured for 119 character line length
+- Ruff is configured for 119 character line length
+
+# Code Quality (*VERY IMPORTANT*)
+- Use the following design patterns on planning the code **VERY IMPORTANT**
+    - https://refactoring.guru/design-patterns/catalog
+- Do a research on the codebase before writing and planning the code
+- If you are not sure about what to do, ask me for help
+- Make sure to follow single responsibility, Solid, DRY, etc. principles.
+- When creating a new file or directory, make sure to follow the project structure and file structure from docs/microservice_guidelines.md
+- New envs should be added to the .env.sample file

budprompt/commons/__init__.py

@@ -20,4 +20,3 @@
 
 # Import specific models instead of using wildcard import to avoid F403 error
 # Import models here
-

budprompt/commons/config.py

@@ -19,7 +19,7 @@
 from pathlib import Path
 
 from budmicroframe.commons.config import BaseAppConfig, BaseSecretsConfig, register_settings
-from pydantic import DirectoryPath
+from pydantic import DirectoryPath, Field
 
 from ..__about__ import __version__
 
@@ -33,6 +33,10 @@ class AppConfig(BaseAppConfig):
     # Base Directory
     base_dir: DirectoryPath = Path(__file__).parent.parent.parent.resolve()
 
+    # BudServe Gateway Configuration
+    bud_gateway_base_url: str = Field(..., alias="BUD_GATEWAY_BASE_URL")
+    bud_default_model_name: str = Field(..., alias="BUD_DEFAULT_MODEL_NAME")
+
 
 class SecretsConfig(BaseSecretsConfig):
     name: str = __version__.split("@")[0]

budprompt/commons/constants.py

@@ -15,4 +15,3 @@
 #  -----------------------------------------------------------------------------
 
 """The constants used in the budprompt module."""
-

budprompt/commons/exceptions.py

@@ -16,6 +16,10 @@
 
 """The exceptions used in the budprompt module."""
 
+from typing import Any, Dict, Optional
+
+from budmicroframe.commons.exceptions import ClientException as BudMicroframeClientException
+
 
 class BaseException(Exception):
     """Base exception class for BudPrompt."""
@@ -36,3 +40,30 @@ def __str__(self) -> str:
             Formatted error message with exception class name
         """
         return f"{self.__class__.__name__}: {self.message}"
+
+
+class SchemaGenerationException(BaseException):
+    """Exception raised when JSON schema to Pydantic model generation fails."""
+
+    pass
+
+
+class PromptExecutionException(BaseException):
+    """Exception raised when prompt execution fails."""
+
+    pass
+
+
+class TemplateRenderingException(BaseException):
+    """Exception raised when template rendering fails."""
+
+    pass
+
+
+class ClientException(BudMicroframeClientException):
+    def __init__(self, message: str, status_code: int = 400, params: Optional[Dict[str, Any]] = None):
+        """Initialize the ClientException with a message."""
+        self.message = message
+        self.status_code = status_code
+        self.params = params
+        super().__init__(self.message, self.status_code)

budprompt/main.py

@@ -26,6 +26,7 @@
 from fastapi import FastAPI
 
 from .commons.config import app_settings, secrets_settings
+from .prompt.routes import prompt_router
 
 
 logger = logging.getLogger(__name__)
@@ -59,3 +60,6 @@ async def lifespan(app: FastAPI) -> AsyncIterator[None]:
 
 
 app = configure_app(app_settings, secrets_settings, lifespan=lifespan)
+
+# Include the prompt router
+app.include_router(prompt_router)