.

models.py

@@ -1,11 +1,10 @@
-from __future__ import annotations
-
-from typing import Any, Optional, List
+# models.py
+from typing import Any, Optional, List, Dict
 from datetime import datetime, timezone
 from pydantic import BaseModel, Field, ConfigDict
 
 
-# ---------- CloudEvent (Pydantic v2 model) ----------
+# ---------- CloudEvent (Pydantic v2) ----------
 class CloudEvent(BaseModel):
     specversion: str = "1.0"
     id: str
@@ -30,7 +29,15 @@ class CloudEvent(BaseModel):
         )
 
 
-# ---------- Permissive ancillary types ----------
+# ---------- FunctionCallData (from your C#) ----------
+class FunctionCallData(BaseModel):
+    function: Optional[str] = None
+    name: Optional[str] = None
+    arguments: Optional[Dict[str, Any]] = None
+    parameters: Optional[Dict[str, Any]] = None
+
+
+# ---------- FunctionState (from your C#) ----------
 class FunctionState(BaseModel):
     IsFunctionCall: bool = False
     IsFunctionCallResponse: bool = False
@@ -39,13 +46,35 @@ class FunctionState(BaseModel):
     IsFunctionStillRunning: bool = False
 
 
-class FunctionCallData(BaseModel):
-    # Accept arbitrary keys so we mirror your .NET payloads without schema fights
-    model_config = ConfigDict(extra="allow")
+# ---------- UserInfo (from your C#; omitting JsonIgnored/NotMapped collections) ----------
+class UserInfo(BaseModel):
+    UserID: Optional[str] = None
+    DateCreated: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    HostLimit: int = 0
+    DisableEmail: bool = False
+    Status: Optional[str] = ""
+    Name: Optional[str] = ""
+    Given_name: Optional[str] = ""
+    Family_name: Optional[str] = ""
+    Nickname: Optional[str] = ""
+    Sub: Optional[str] = ""
+    Enabled: bool = True
+    MonitorAlertEnabled: bool = True
+    PredictAlertEnabled: bool = False
+    AccountType: Optional[str] = "Default"
+    Email: Optional[str] = ""
+    Email_verified: bool = False
+    Picture: Optional[str] = ""
+    Updated_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    LastLoginDate: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    CustomerId: Optional[str] = ""
+    CancelAt: Optional[datetime] = None
+    TokensUsed: int = 0
 
+    # NotMapped in C#: model as loose dict so it round-trips safely
+    LoadServer: Dict[str, Any] = Field(default_factory=dict)
 
-class UserInfo(BaseModel):
-    model_config = ConfigDict(extra="allow")
+    # JsonIgnore MonitorIPs in C#: we do not include it here
 
 
 # ---------- LLMServiceObj (field names match C# exactly) ----------
@@ -90,7 +119,7 @@ class LLMServiceObj(BaseModel):
     IsProcessedStack: List[bool] = Field(default_factory=list)
     MessageIDStack: List[str] = Field(default_factory=list)
 
-    # function state flags (since your C# exposes these via methods/flags)
+    # function state flags (C# stores in FunctionState; we expose flat flags like your class does)
     IsFunctionCall: bool = False
     IsFunctionCallResponse: bool = False
     IsFunctionCallError: bool = False
@@ -98,7 +127,7 @@ class LLMServiceObj(BaseModel):
     IsFunctionStillRunning: bool = False
 
 
-# ---------- ResultObj ----------
+# ---------- ResultObj (from your C#) ----------
 class ResultObj(BaseModel):
     Message: str = ""
     Success: bool = False