Skip to content

Commit

Permalink
update metadata.py to only use venomx
Browse files Browse the repository at this point in the history
  • Loading branch information
iQuxLE committed Oct 23, 2024
1 parent ea43595 commit 7451f80
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 37 deletions.
4 changes: 2 additions & 2 deletions src/curategpt/store/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,15 +19,15 @@
from .chromadb_adapter import ChromaDBAdapter
from .db_adapter import DBAdapter
from .duckdb_adapter import DuckDBAdapter
from .metadata import CollectionMetadata
from .metadata import Metadata
from .schema_proxy import SchemaProxy

__all__ = [
"DBAdapter",
"ChromaDBAdapter",
"DuckDBAdapter",
"SchemaProxy",
"CollectionMetadata",
"Metadata",
"get_store",
]

Expand Down
35 changes: 0 additions & 35 deletions src/curategpt/store/metadata.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,41 +4,6 @@
from pydantic import BaseModel, ConfigDict
from venomx.model.venomx import Index


class CollectionMetadata(BaseModel):
"""
Metadata about a collection.
Useful for all even when no uniqye ID in objects. (e.g. MaxO Ontology)
This is an open class, so additional metadata can be added.
"""

model_config = ConfigDict(protected_namespaces=())

name: Optional[str] = None
"""Name of the collection"""

description: Optional[str] = None
"""Description of the collection"""

model: Optional[str] = None
"""Name of any ML model"""

object_type: Optional[str] = None
"""Type of object in the collection"""

source: Optional[str] = None
"""Source of the collection"""

# DEPRECATED
annotations: Optional[Dict] = None
"""Additional metadata"""

object_count: Optional[int] = None
"""Number of objects in the collection"""

hnsw_space: Optional[str] = None
"""Space used for hnsw index (e.g. 'cosine')"""

"""
ChromaDB Constraints:
Metadata Must Be Scalar: ChromaDB only accepts metadata values that are scalar types (str, int, float, bool).
Expand Down

0 comments on commit 7451f80

Please sign in to comment.