Spaces:
Sleeping
Sleeping
File size: 864 Bytes
6c653ef | 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 | import chromadb
from pathlib import Path
from langchain_chroma import Chroma
from langchain_community.embeddings import HuggingFaceEmbeddings
db_name = "agentfactory"
db_path = Path("databases") / db_name
model_name = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
print(f"Checking DB: {db_name}")
embeddings = HuggingFaceEmbeddings(model_name=model_name)
db = Chroma(persist_directory=str(db_path), embedding_function=embeddings)
# Test direct keyword search
results = db.similarity_search("LiveKit", k=2)
print(f"\nSearch for 'LiveKit' found {len(results)} docs:")
for r in results:
print(f"- {r.page_content[:100]}...")
# Test Urdu script search
results_ur = db.similarity_search("لائیو کٹ", k=2)
print(f"\nSearch for 'لائیو کٹ' found {len(results_ur)} docs:")
for r in results_ur:
print(f"- {r.page_content[:100]}...")
|