Docker_Deploy / src /python /test_embedding.py
Shaheryar Shah
Add backend files for RAG Chatbot Docker deployment
bec06d9
raw
history blame contribute delete
977 Bytes
import asyncio
from embedder import Embedder
async def test_embedding():
"""Test embedding functionality only."""
print("Testing embedding functionality...")
try:
embedder = Embedder()
print("[OK] Embedder initialized successfully")
# Test creating a single embedding
test_text = "This is a test sentence for embedding."
embedding = await embedder.create_embedding(test_text)
print(f"[OK] Embedding created successfully. Dimension: {len(embedding)}")
print(f"[OK] First 5 values: {embedding[:5]}")
return True
except Exception as e:
print(f"[ERROR] Embedding test failed: {str(e)}")
import traceback
traceback.print_exc()
return False
if __name__ == "__main__":
success = asyncio.run(test_embedding())
if success:
print("\nEmbedding test completed successfully!")
else:
print("\nEmbedding test failed.")