Update app/llm.py

app/llm.py

@@ -1,5 +1,6 @@
 import fastapi
 from fastapi.responses import JSONResponse
+from fastapi_users import models
 from time import time
 #from fastapi.middleware.cors import CORSMiddleware
 #MODEL_PATH = "./qwen1_5-0_5b-chat-q4_0.gguf" #"./qwen1_5-0_5b-chat-q4_0.gguf"
@@ -72,7 +73,7 @@ def health():
     
 # Chat Completion API
 @llm_router.post("/chat/", tags=["llm"])
-async def chat(chatm:ChatModel, user: User = fastapi.Depends(current_active_user)):
+async def chat(chatm:ChatModel, user: models.BaseUser = fastapi.Depends(current_active_user)):
     try:
         st = time()
         output = llm_chat.create_chat_completion(
@@ -95,7 +96,7 @@ async def chat(chatm:ChatModel, user: User = fastapi.Depends(current_active_user
 
 # Chat Completion API
 @llm_router.post("/generate", tags=["llm"])
-async def generate(gen:GenModel, user: User = fastapi.Depends(current_active_user)):
+async def generate(gen:GenModel, user: models.BaseUser = fastapi.Depends(current_active_user)):
     gen.system = "You are an helpful medical AI assistant."
     gen.temperature = 0.5
     gen.seed = 42