Spaces:

Seyfelislem
/

pdf_chat_assistant

Sleeping

pdf_chat_assistant / src /models /chat_models.py

Seif-aber

implemented pdf chat assistant with gemini and RAG

edac567 about 1 year ago

437 Bytes

from pydantic import BaseModel
from typing import List, Optional

class Message(BaseModel):
    user_id: str
    content: str
    timestamp: str

class ChatContext(BaseModel):
    messages: List[Message]
    pdf_id: Optional[str] = None

class UserPrompt(BaseModel):
    user_id: str
    prompt: str
    context: ChatContext

class AssistantResponse(BaseModel):
    response: str
    context: ChatContext
    pdf_id: Optional[str] = None

	from pydantic import BaseModel
	from typing import List, Optional

	class Message(BaseModel):
	user_id: str
	content: str
	timestamp: str

	class ChatContext(BaseModel):
	messages: List[Message]
	pdf_id: Optional[str] = None

	class UserPrompt(BaseModel):
	user_id: str
	prompt: str
	context: ChatContext

	class AssistantResponse(BaseModel):
	response: str
	context: ChatContext
	pdf_id: Optional[str] = None