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]]