cognee/cognee/infrastructure/databases/vector/models/VectorConfig.py
2024-03-21 09:57:47 +01:00

6 lines
143 B
Python

from typing import Literal
from pydantic import BaseModel
class VectorConfig(BaseModel):
distance: Literal['Cosine', 'Dot']
size: int