Updated with docker file"

This commit is contained in:
Saurab-Shrestha 2024-05-07 13:03:43 +05:45
parent e1e940bbbd
commit 333a3f3913
7 changed files with 148 additions and 137 deletions

View File

@ -1,2 +1,3 @@
*
/private_gpt/
!.gitignore

View File

@ -118,9 +118,10 @@ def create_chat_item(db, sender, content, conversation_id):
content=content,
conversation_id=conversation_id
)
chat_history = crud.chat.get_conversation(db, conversation_id=conversation_id)
chat_history.generate_title()
return crud.chat_item.create(db, obj_in=chat_item_create)
@completions_router.post(
"/chat",
response_model=None,

View File

@ -44,10 +44,10 @@ def list_auditlog(
def filter_auditlog(
db: Session = Depends(deps.get_db),
filter_in= Depends(schemas.AuditFilter),
current_user: models.User = Security(
deps.get_current_user,
scopes=[Role.SUPER_ADMIN["name"]],
),
# current_user: models.User = Security(
# deps.get_current_user,
# scopes=[Role.SUPER_ADMIN["name"]],
# ),
) -> List[schemas.Audit]:
logs = crud.audit.filter(db, obj_in=filter_in)
return convert_audit_logs(db, logs)
@ -56,10 +56,10 @@ def filter_auditlog(
def download_auditlog(
db: Session = Depends(deps.get_db),
filter_in= Depends(schemas.ExcelFilter),
current_user: models.User = Security(
deps.get_current_user,
scopes=[Role.SUPER_ADMIN["name"]],
),
# current_user: models.User = Security(
# deps.get_current_user,
# scopes=[Role.SUPER_ADMIN["name"]],
# ),
):
logs = crud.audit.excel_filter(db, obj_in=filter_in)
username = filter_in.username if filter_in.username else None

View File

@ -29,6 +29,13 @@ class CRUDChat(CRUDBase[ChatHistory, ChatHistoryCreate, ChatHistoryCreate]):
)
return chat_history
def get_conversation(self, db: Session, conversation_id: uuid.UUID) -> Optional[ChatHistory]:
return (
db.query(self.model)
.filter(ChatHistory.conversation_id == conversation_id)
.first()
)
def get_chat_history(
self, db: Session, *,user_id:int
) -> List[ChatHistory]:

View File

@ -33,7 +33,8 @@ class ChatHistory(Base):
item for item in self.chat_items if item.sender == "user"]
if user_chat_items:
first_user_chat_item = user_chat_items[0]
self.title = first_user_chat_item.content[:30]
print("Chat items: ", first_user_chat_item.content['text'])
self.title = first_user_chat_item.content['text'][:30]
else:
self.title = str(self.conversation_id)

View File

@ -41,6 +41,7 @@ class ChatHistoryUpdate(ChatHistoryBase):
class Chat(BaseModel):
conversation_id: uuid.UUID
title: Optional[str]
class ChatHistory(ChatHistoryBase):
conversation_id: uuid.UUID