mime type fix for resource addition via mcp server

Signed-off-by: Satya <[email protected]>

Author: TS0713

alembic/README

@@ -41,6 +41,7 @@
 import asyncio
 from datetime import datetime, timezone
 import logging
+import mimetypes
 import os
 import tempfile
 import time
@@ -621,24 +622,29 @@ async def register_gateway(
             # Create resource DB models
             db_resources = [
                 DbResource(
-                    uri=resource.uri,
-                    name=resource.name,
-                    description=resource.description,
-                    mime_type=resource.mime_type,
-                    template=resource.template,
-                    # Federation metadata
+                    uri=r.uri,
+                    name=r.name,
+                    description=r.description,
+                    mime_type=(mime_type := (mimetypes.guess_type(r.uri)[0] or ("text/plain" if isinstance(r.content, str) else "application/octet-stream"))),
+                    template=r.template,
+                    text_content=r.content if (mime_type.startswith("text/") or isinstance(r.content, str)) and isinstance(r.content, str) else None,
+                    binary_content=(
+                        r.content.encode() if (mime_type.startswith("text/") or isinstance(r.content, str)) and isinstance(r.content, str) else r.content if isinstance(r.content, bytes) else None
+                    ),
+                    size=len(r.content) if r.content else 0,
+                    tags=getattr(r, "tags", []) or [],
                     created_by=created_by or "system",
                     created_from_ip=created_from_ip,
-                    created_via="federation",  # These are federated resources
+                    created_via="federation",
                     created_user_agent=created_user_agent,
+                    import_batch_id=None,
                     federation_source=gateway.name,
                     version=1,
-                    # Inherit team assignment from gateway
-                    team_id=team_id,
-                    owner_email=owner_email,
-                    visibility=visibility,
+                    team_id=getattr(r, "team_id", None) or team_id,
+                    owner_email=getattr(r, "owner_email", None) or owner_email or created_by,
+                    visibility=getattr(r, "visibility", None) or visibility,
                 )
-                for resource in resources
+                for r in resources
             ]
 
             # Create prompt DB models