File size: 493 Bytes
43515a8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from typing import List, Optional
from docarray import BaseDoc
from docarray.typing import NdArray
class RepoDoc(BaseDoc):
"""
The class for representing basic data structures.
"""
name: str
topics: List[str]
stars: int
license: str
code_embedding: Optional[NdArray[768]]
doc_embedding: Optional[NdArray[768]]
readme_embedding: Optional[NdArray[768]]
requirement_embedding: Optional[NdArray[768]]
repository_embedding: Optional[NdArray[3072]]
|