RepoSnipy / common /repo_doc.py
HenryStephen's picture
topic cluster and code cluster
c6a1f8c
raw
history blame
493 Bytes
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]]