Ibraaheem commited on
Commit
6dffb64
·
verified ·
1 Parent(s): 1d1166b

Update private_gpt/server/chat/chat_router.py

Browse files
private_gpt/server/chat/chat_router.py CHANGED
@@ -1,5 +1,5 @@
1
  from fastapi import APIRouter, Depends, Request
2
- from llama_index.core.llms import ChatMessage, MessageRole
3
  from pydantic import BaseModel
4
  from starlette.responses import StreamingResponse
5
  import logging
 
1
  from fastapi import APIRouter, Depends, Request
2
+ from llama_index.llms.type import ChatMessage, MessageRole
3
  from pydantic import BaseModel
4
  from starlette.responses import StreamingResponse
5
  import logging