Update private_gpt/server/chat/chat_router.py

private_gpt/server/chat/chat_router.py

@@ -1,5 +1,5 @@
 from fastapi import APIRouter, Depends, Request
-from llama_index.llms.type import ChatMessage, MessageRole
+from llama_index.llms import ChatMessage, MessageRole
 from pydantic import BaseModel
 from starlette.responses import StreamingResponse
 import logging