[fix] added extras to log table

llmstudio/engine/providers/provider.py

@@ -51,6 +51,7 @@ class ChatRequest(BaseModel):
     functions: Optional[List[Dict[str, Any]]] = None
     session_id: Optional[str] = None
     retries: Optional[int] = 0
+    extras: Optional[Dict[str, Any]] = None
 
 
 class Provider:
@@ -159,19 +160,14 @@ async def handle_response(
                 else request.chat_input
             ),
             "provider": self.config.id,
-            "model": (
-                request.model
-                if model and model.startswith(request.model)
-                else (model or request.model)
-            ),
-            "deployment": (
-                model
-                if model and model.startswith(request.model)
-                else (request.model if model != request.model else None)
-            ),
+            "model": model,
             "timestamp": time.time(),
-            "parameters": request.parameters.model_dump(),
+            "parameters": {
+                **request.parameters.model_dump(),
+                "deployment": request.model,
+            },
             "metrics": metrics,
+            "extras": request.extras,
         }
 
         self.save_log(response)

llmstudio/tracking/logs/models.py

@@ -17,6 +17,6 @@ class LogDefault(Base):
     context = Column(JSON)
     provider = Column(String)
     model = Column(String)
-    deployment = Column(String)
     parameters = Column(JSON)
     metrics = Column(JSON)
+    extras = Column(JSON)

llmstudio/tracking/logs/schemas.py

@@ -12,9 +12,9 @@ class LogDefaultBase(BaseModel):
     context: List[Dict[str, Any]] = None
     provider: str = None
     model: str = None
-    deployment: str = None
     parameters: dict = None
     metrics: dict = None
+    extras: Optional[Dict[str, Any]] = None
 
 
 class LogDefault(LogDefaultBase):