File size: 1,102 Bytes
b77c0a2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
01ae535
 
 
 
 
8600f2c
 
 
 
 
 
 
 
 
 
 
5186c89
 
8600f2c
 
 
aeac4b2
 
 
 
 
 
f323c55
8600f2c
 
607ef0e
8600f2c
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
from pydantic import BaseModel
from typing import Optional

class Token(BaseModel):
    access_token: str
    token_type: str

class TokenData(BaseModel):
    username: Optional[str] = None

class User(BaseModel):
    username: str
    email: Optional[str] = None
    full_name: Optional[str] = None
    disabled: Optional[bool] = None

class UserInDB(User):
    hashed_password: str


class UserCreate(BaseModel):
    username: str
    password: str


class EmbeddingRequest(BaseModel):
    sentences: list[str]


class PredictRecord(BaseModel):
    subject: str
    sub_subject: str
    name_category: str
    name: str
    abstract: Optional[str] = None
    memo: Optional[str] = None


class PredictResult(BaseModel):
    subject: str
    sub_subject: str
    name_category: str
    name: str
    abstract: Optional[str] = None
    memo: Optional[str] = None
    confirmed: str
    standard_subject: str
    standard_name: str
    similarity_score: float


class PredictRawRequest(BaseModel):
    records: list[PredictRecord]


class PredictRawResponse(BaseModel):
    results: list[PredictResult]