Fix fastapi-users register router and confirm flash-attn dependency

api/models.py

@@ -1,10 +1,10 @@
-# api/models.py
 from fastapi_users.db import SQLAlchemyBaseUserTable, SQLAlchemyBaseOAuthAccountTable
 from sqlalchemy import Column, Integer, String, Boolean
 from sqlalchemy.orm import relationship
 from sqlalchemy.ext.declarative import declarative_base
 from pydantic import BaseModel, Field
 from typing import List, Optional
+from fastapi_users import schemas
 
 Base = declarative_base()
 
@@ -13,6 +13,7 @@ class OAuthAccount(SQLAlchemyBaseOAuthAccountTable, Base):
     __tablename__ = "oauth_accounts"
     id = Column(Integer, primary_key=True)
     user = relationship("User", back_populates="oauth_accounts")
+
 # نموذج المستخدم
 class User(SQLAlchemyBaseUserTable, Base):
     __tablename__ = "users"
@@ -23,7 +24,20 @@ class User(SQLAlchemyBaseUserTable, Base):
     is_superuser = Column(Boolean, default=False)
     oauth_accounts = relationship("OAuthAccount", back_populates="user")
 
-# نموذج طلب الاستعلام (كما هو)
+# Pydantic schemas for fastapi-users
+class UserRead(schemas.BaseUser[int]):
+    id: int
+    email: str
+    is_active: bool = True
+    is_superuser: bool = False
+
+class UserCreate(schemas.BaseUserCreate):
+    email: str
+    password: str
+    is_active: Optional[bool] = True
+    is_superuser: Optional[bool] = False
+
+# نموذج طلب الاستعلام
 class QueryRequest(BaseModel):
     message: str
     system_prompt: Optional[str] = "You are an expert assistant providing detailed, comprehensive, and well-structured responses."

main.py

@@ -11,7 +11,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from api.endpoints import router as api_router
 from api.auth import fastapi_users, auth_backend, current_active_user, google_oauth_client, github_oauth_client
 from api.database import get_db, engine, Base
-from api.models import User
+from api.models import User, UserRead, UserCreate
 from motor.motor_asyncio import AsyncIOMotorClient
 from pydantic import BaseModel
 from typing import List
@@ -85,12 +85,12 @@ app.include_router(
     tags=["auth"],
 )
 app.include_router(
-    fastapi_users.get_register_router(),
+    fastapi_users.get_register_router(UserRead, UserCreate),
     prefix="/auth",
     tags=["auth"],
 )
 app.include_router(
-    fastapi_users.get_users_router(),
+    fastapi_users.get_users_router(UserRead, UserCreate),
     prefix="/users",
     tags=["users"],
 )

requirements.txt

@@ -34,3 +34,4 @@ aiofiles
 motor
 redis
 markdown2
+flash-attn==2.6.3