mirror of
https://github.com/MODSetter/SurfSense.git
synced 2025-09-02 18:49:09 +00:00
73 lines
No EOL
2.7 KiB
Python
73 lines
No EOL
2.7 KiB
Python
from pydantic import BaseModel, Field
|
|
from typing import List, Optional
|
|
|
|
class UserCreate(BaseModel):
|
|
username: str
|
|
password: str
|
|
apisecretkey: str
|
|
|
|
class DocMeta(BaseModel):
|
|
BrowsingSessionId: Optional[str] = Field(default=None, description="BrowsingSessionId of Document")
|
|
VisitedWebPageURL: Optional[str] = Field(default=None, description="VisitedWebPageURL of Document")
|
|
VisitedWebPageTitle: Optional[str] = Field(default=None, description="VisitedWebPageTitle of Document")
|
|
VisitedWebPageDateWithTimeInISOString: Optional[str] = Field(default=None, description="VisitedWebPageDateWithTimeInISOString of Document")
|
|
VisitedWebPageReffererURL: Optional[str] = Field(default=None, description="VisitedWebPageReffererURL of Document")
|
|
VisitedWebPageVisitDurationInMilliseconds: Optional[int] = Field(default=None, description="VisitedWebPageVisitDurationInMilliseconds of Document"),
|
|
|
|
class DocWithContent(BaseModel):
|
|
BrowsingSessionId: Optional[str] = Field(default=None, description="BrowsingSessionId of Document")
|
|
VisitedWebPageURL: Optional[str] = Field(default=None, description="VisitedWebPageURL of Document")
|
|
VisitedWebPageTitle: Optional[str] = Field(default=None, description="VisitedWebPageTitle of Document")
|
|
VisitedWebPageDateWithTimeInISOString: Optional[str] = Field(default=None, description="VisitedWebPageDateWithTimeInISOString of Document")
|
|
VisitedWebPageReffererURL: Optional[str] = Field(default=None, description="VisitedWebPageReffererURL of Document")
|
|
VisitedWebPageVisitDurationInMilliseconds: Optional[int] = Field(default=None, description="VisitedWebPageVisitDurationInMilliseconds of Document"),
|
|
VisitedWebPageContent: Optional[str] = Field(default=None, description="Visited WebPage Content in markdown of Document")
|
|
|
|
class DocumentsToDelete(BaseModel):
|
|
ids_to_delete: List[str]
|
|
openaikey: str
|
|
token: str
|
|
|
|
class UserQuery(BaseModel):
|
|
query: str
|
|
search_space: str
|
|
openaikey: str
|
|
token: str
|
|
|
|
class ChatHistory(BaseModel):
|
|
type: str
|
|
content: str | List[DocMeta] | List[str]
|
|
|
|
class UserQueryWithChatHistory(BaseModel):
|
|
chat: List[ChatHistory]
|
|
query: str
|
|
openaikey: str
|
|
token: str
|
|
|
|
class DescriptionResponse(BaseModel):
|
|
response: str
|
|
|
|
class RetrivedDocListItem(BaseModel):
|
|
metadata: DocMeta
|
|
pageContent: str
|
|
|
|
class RetrivedDocList(BaseModel):
|
|
documents: List[RetrivedDocListItem]
|
|
search_space: str | None
|
|
openaikey: str
|
|
token: str
|
|
|
|
class UserQueryResponse(BaseModel):
|
|
response: str
|
|
relateddocs: List[DocWithContent]
|
|
|
|
class NewUserChat(BaseModel):
|
|
token: str
|
|
type: str
|
|
title: str
|
|
chats_list: str
|
|
|
|
class ChatToUpdate(BaseModel):
|
|
chatid: str
|
|
token: str
|
|
chats_list: str |