Update app.py

app.py

@@ -11,6 +11,12 @@ from uuid import uuid4
 
 from langchain_core.documents import Document
 
+from typing import Union, List, Dict, Any
+from pydantic import BaseModel, Field
+
+class Data(BaseModel):
+    items: Union[Dict[str, Any], List[Dict[str, Any]]] = Field(..., description="Either a dictionary or a list of dictionaries.")
+
 document_1 = Document(
     page_content="I had chocolate chip pancakes and scrambled eggs for breakfast this morning.",
     metadata={"source": "tweet"},
@@ -120,6 +126,18 @@ def get_data(query: str):
     }
 
 
+@app.post("/add_data")
+def add_data(data: Data):
+    global qdrant
+    if isinstance(data.items, dict):
+        qdrant.add_documents(documents=[Document(**data.items)])
+    else:
+        qdrant.add_documents(documents=[Document(**x.items) for x in data])
+
+    return {"message":"Create data successfully!", "status_code":201}
+
+
+
 @app.get("/")
 def greet_json():
     return {"Hello": "World!"}