init

main.py

@@ -2,8 +2,10 @@ from fastapi import FastAPI, Form, Depends, Request, File, UploadFile
 from fastapi.encoders import jsonable_encoder
 from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
+
 import os
 import pypdf
+from uuid import uuid4
 
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from pymilvus import MilvusClient, db, utility, Collection, CollectionSchema, FieldSchema, DataType
@@ -45,9 +47,10 @@ def split_documents(document_data):
 def create_a_collection(milvus_client, collection_name):
     content = FieldSchema(name="content", dtype=DataType.VARCHAR, max_length=4096)
     vector = FieldSchema(name="vector", dtype=DataType.FLOAT_VECTOR, dim=1024)
+    id = FieldSchema(name="id", dtype=DataType.VARCHAR, max_length=40, is_primary=True)
 
     schema = CollectionSchema([
-        content, vector
+        id, content, vector
     ])
 
     vector_index = {
@@ -84,6 +87,7 @@ async def insert(file: UploadFile = File(...)):
     data_objects = []
     for doc in splitted_document_data:
         data = {
+            "id": str(uuid4()),
             "vector": document_to_embeddings(doc.page_content),
             "content": doc.page_content,
         }
@@ -100,7 +104,7 @@ async def insert(file: UploadFile = File(...)):
         return JSONResponse(status_code=200, content={"result": 'good'})
     
 @app.post("/rag")
-async def insert(question):
+async def rag(question):
     if not question:
         return JSONResponse(status_code=400, content={"message": "Please a question!"})