first commit

main.py

@@ -2,6 +2,10 @@ from fastapi import FastAPI
 
 app = FastAPI()
 
+@app.get("/")
+def read_root():
+    return {"Hello": "World!"}
+
 @app.get("/")
 def read_root():
     return {"Hello": "World!"}

service/pinecone.py

@@ -0,0 +1,23 @@
+import os
+import pinecone
+from langchain.embeddings.openai import OpenAIEmbeddings
+from langchain.vectorstores import Pinecone
+
+
+class VectorStore(Pinecone):
+
+    def __init__(self,index_name) -> None:
+        pinecone.init(
+            api_key=os.getenv("PINECONE_API_KEY"),  # find at app.pinecone.io
+            environment=os.getenv("PINECONE_ENV"),  # next to api key in console
+        )
+        self.index_name = index_name
+        self.embeddings = OpenAIEmbeddings()
+        self.dims = 1536
+        index = pinecone.Index(self.index_name)
+        super().__init__(index, self.embeddings, "text")  
+
+    def add_docs(self,docs):
+        if self.index_name not in pinecone.list_indexes():
+            pinecone.create_index(name=self.index_name, metric="cosine", dimension=self.dims)
+        Pinecone.from_documents(docs, self.embeddings, index_name=self.index_name)