Update config_models.py

config_models.py

@@ -5,16 +5,22 @@ from typing import Optional, List, Dict, Any
 from datetime import datetime
 from pydantic import BaseModel, Field
 
+class BaseModelWithDatetime(BaseModel):
+    """Base model with datetime serialization support"""
+    class Config:
+        json_encoders = {
+            datetime: lambda v: v.isoformat() if v else None
+        }
 
 # ===================== User & Auth =====================
-class UserConfig(BaseModel):
+class UserConfig(BaseModelWithDatetime):
     username: str
     password_hash: str
     salt: Optional[str] = None
 
 
 # ===================== Provider Models =====================
-class ProviderDefinition(BaseModel):
+class ProviderDefinition(BaseModelWithDatetime):
     type: str  # llm, tts, stt
     name: str
     display_name: str
@@ -24,7 +30,7 @@ class ProviderDefinition(BaseModel):
     description: str
 
 
-class ProviderSettings(BaseModel):
+class ProviderSettings(BaseModelWithDatetime):
     name: str
     api_key: Optional[str] = None
     endpoint: Optional[str] = None
@@ -32,7 +38,7 @@ class ProviderSettings(BaseModel):
 
 
 # ===================== Global Config =====================
-class GlobalConfig(BaseModel):
+class GlobalConfig(BaseModelWithDatetime):
     llm_provider: ProviderSettings
     tts_provider: ProviderSettings
     stt_provider: ProviderSettings
@@ -53,18 +59,18 @@ class GlobalConfig(BaseModel):
 
 
 # ===================== Localization Models =====================
-class LocalizedExample(BaseModel):
+class LocalizedExample(BaseModelWithDatetime):
     locale_code: str
     example: str
 
 
-class LocalizedCaption(BaseModel):
+class LocalizedCaption(BaseModelWithDatetime):
     locale_code: str
     caption: str
 
 
 # ===================== Parameter Models =====================
-class ParameterConfig(BaseModel):
+class ParameterConfig(BaseModelWithDatetime):
     name: str
     caption: List[LocalizedCaption]
     type: str  # str, int, float, bool, date
@@ -89,7 +95,7 @@ class ParameterConfig(BaseModel):
 
 
 # ===================== Intent Models =====================
-class IntentConfig(BaseModel):
+class IntentConfig(BaseModelWithDatetime):
     name: str
     caption: str
     requiresApproval: Optional[bool] = False
@@ -103,7 +109,7 @@ class IntentConfig(BaseModel):
 
 
 # ===================== LLM Configuration =====================
-class GenerationConfig(BaseModel):
+class GenerationConfig(BaseModelWithDatetime):
     max_new_tokens: int = 512
     temperature: float = 0.7
     top_p: float = 0.9
@@ -115,7 +121,7 @@ class GenerationConfig(BaseModel):
     early_stopping: Optional[bool] = None
 
 
-class LLMConfiguration(BaseModel):
+class LLMConfiguration(BaseModelWithDatetime):
     repo_id: str
     generation_config: GenerationConfig = Field(default_factory=GenerationConfig)
     use_fine_tune: bool = False
@@ -123,7 +129,7 @@ class LLMConfiguration(BaseModel):
 
 
 # ===================== Version Models =====================
-class VersionConfig(BaseModel):
+class VersionConfig(BaseModelWithDatetime):
     no: int
     caption: str
     description: Optional[str] = None
@@ -140,7 +146,7 @@ class VersionConfig(BaseModel):
 
 
 # ===================== Project Models =====================
-class ProjectConfig(BaseModel):
+class ProjectConfig(BaseModelWithDatetime):
     id: int
     name: str
     caption: str
@@ -161,13 +167,13 @@ class ProjectConfig(BaseModel):
 
 
 # ===================== API Models =====================
-class RetryConfig(BaseModel):
+class RetryConfig(BaseModelWithDatetime):
     retry_count: int = 3
     backoff_seconds: int = 2
     strategy: str = "static"  # static, exponential
 
 
-class AuthConfig(BaseModel):
+class AuthConfig(BaseModelWithDatetime):
     enabled: bool
     token_endpoint: Optional[str] = None
     response_token_path: Optional[str] = None
@@ -176,14 +182,14 @@ class AuthConfig(BaseModel):
     token_refresh_body: Optional[Dict[str, Any]] = None
 
 
-class ResponseMapping(BaseModel):
+class ResponseMapping(BaseModelWithDatetime):
     variable_name: str
     type: str  # str, int, float, bool, date
     json_path: str
     caption: List[LocalizedCaption]
 
 
-class APIConfig(BaseModel):
+class APIConfig(BaseModelWithDatetime):
     name: str
     url: str
     method: str = "POST"
@@ -203,7 +209,7 @@ class APIConfig(BaseModel):
 
 
 # ===================== Activity Log =====================
-class ActivityLogEntry(BaseModel):
+class ActivityLogEntry(BaseModelWithDatetime):
     id: int
     timestamp: datetime
     username: str
@@ -215,7 +221,7 @@ class ActivityLogEntry(BaseModel):
 
 
 # ===================== Root Configuration =====================
-class ServiceConfig(BaseModel):
+class ServiceConfig(BaseModelWithDatetime):
     global_config: GlobalConfig = Field(alias="config")
     projects: List[ProjectConfig] = Field(default_factory=list)
     apis: List[APIConfig] = Field(default_factory=list)