File size: 777 Bytes
7248d39 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | """Smoke test for Modal-backed MiniCPM-Embedding."""
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).parent.parent))
from models.embedder import MiniCPMEmbedder
def main():
print("Calling Modal Embedder...")
embedder = MiniCPMEmbedder()
dim = embedder.get_embedding_dim()
assert dim == 2304, f"Expected 2304, got {dim}"
doc_vec = embedder.embed_documents(["Revenue increased 12% year-over-year."])
query_vec = embedder.embed_query("What was the revenue growth?")
print(f"Document vector length: {len(doc_vec[0])}")
print(f"Query vector length: {len(query_vec)}")
assert len(doc_vec[0]) == 2304
assert len(query_vec) == 2304
print("Modal embedder test passed.")
if __name__ == "__main__":
main()
|