Upload 9 files

.env

@@ -0,0 +1,2 @@
+MONGO_URI=mongodb+srv://jaothan26:LP8YhV1EocgW97vu@cluster0.ibvao.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0
+MONGO_DB_NAME=vector_db

Dockerfile.txt

@@ -0,0 +1,10 @@
+FROM python:3.9-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "3000"]

README2.md

@@ -0,0 +1,14 @@
+# MongoDB Vector Search Project
+
+This project demonstrates how to use MongoDB with vector search capabilities in a Dockerized environment. It includes a FastAPI application that interacts with MongoDB to store and retrieve vectors.
+
+## Setup
+
+1. Clone the repository.
+2. Create a `.env` file with the necessary environment variables.
+3. Run `docker-compose up` to start the services.
+
+## API Endpoints
+
+- `POST /vectors/`: Create a new vector.
+- `GET /vectors/{vector_id}`: Retrieve a vector by its ID.

app.py

@@ -0,0 +1,15 @@
+from fastapi import FastAPI
+from app.routes.vector_routes import router as vector_router
+from app.utils.database import connect_to_mongo, close_mongo_connection
+
+app = FastAPI()
+
+app.include_router(vector_router)
+
+@app.on_event("startup")
+async def startup_db_client():
+    await connect_to_mongo()
+
+@app.on_event("shutdown")
+async def shutdown_db_client():
+    await close_mongo_connection()

docker-compose.yml

@@ -0,0 +1,17 @@
+version: '3.8'
+
+services:
+  app:
+    build: .
+    env_file:
+      - .env
+    depends_on:
+      mongodb:
+        condition: service_healthy
+    ports:
+      - "3000:3000"
+    environment:
+      - MONGO_URI=mongodb://${MONGO_USERNAME}:${MONGO_PASSWORD}@mongodb:27017/
+      - MONGO_DB_NAME=${MONGO_DB_NAME}
+
+

models/vector_model.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel
+
+class Vector(BaseModel):
+    vector: list[float]
+    metadata: dict

requirements.txt

@@ -0,0 +1,4 @@
+fastapi==0.95.2
+uvicorn==0.22.0
+pymongo==4.3.3
+pydantic==1.10.7

routes/vector_routes.py

@@ -0,0 +1,20 @@
+from fastapi import APIRouter, HTTPException
+from app.models.vector_model import Vector
+from app.utils.database import db
+
+router = APIRouter()
+
+@router.post("/vectors/")
+async def create_vector(vector: Vector):
+    vector_dict = vector.dict()
+    result = db.vectors.insert_one(vector_dict)
+    if result.inserted_id:
+        return {"id": str(result.inserted_id)}
+    raise HTTPException(status_code=500, detail="Failed to insert vector")
+
+@router.get("/vectors/{vector_id}")
+async def get_vector(vector_id: str):
+    vector = db.vectors.find_one({"_id": vector_id})
+    if vector:
+        return vector
+    raise HTTPException(status_code=404, detail="Vector not found")

utils/database.py

@@ -0,0 +1,23 @@
+from pymongo import MongoClient
+from pymongo.errors import ConnectionFailure
+import os
+
+client = None
+db = None
+
+async def connect_to_mongo():
+    global client, db
+    mongo_uri = os.getenv("MONGO_URI")
+    db_name = os.getenv("MONGO_DB_NAME")
+    client = MongoClient(mongo_uri)
+    db = client[db_name]
+    try:
+        client.admin.command('ping')
+        print("Pinged your deployment. You successfully connected to MongoDB!")
+    except ConnectionFailure as e:
+        print("Could not connect to MongoDB:", e)
+
+async def close_mongo_connection():
+    global client
+    if client:
+        client.close()