Update users/services.py

users/services.py

@@ -3,7 +3,8 @@ from fastapi.exceptions import HTTPException
 from core.security import get_password_hash
 from datetime import datetime
 from sqlalchemy.orm import Session
-from users.schemas import UserCreate, UserUpdate, UserEmbeddingsModel
+from users.schemas import UserCreate, UserUpdate, UserEmbeddings
+
 
 
 async def create_user_account(data:UserCreate, db:Session):
@@ -27,7 +28,7 @@ async def create_user_account(data:UserCreate, db:Session):
     db.refresh(new_user)
     return new_user
 
-async def create_user_embeddings(user_id:int, embeddings:UserEmbeddingsModel, db:Session):
+async def create_user_embeddings(user_id:int, embeddings:UserEmbeddings, db:Session):
     user = db.query(User).filter(User.id == user_id).first()
     if not user:
         embeddings = UserEmbeddings(user_id=user_id, embeddings=embeddings.embeddings)
@@ -72,7 +73,7 @@ def delete_user(user_id, db:Session):
     return None
 
 
-async def update_user_embeddings(user_id:int, embeddings:UserEmbeddingsModel, db:Session):
+async def update_user_embeddings(user_id:int, embeddings:UserEmbeddings, db:Session):
     embeddings = db.query(UserEmbeddings).filter(UserEmbeddings.user_id == user_id).first()
     if embeddings:
         embeddings.embeddings = embeddings.embeddings