mitulagr2 commited on
Commit
57a28fb
·
1 Parent(s): 35f7fe0

Update rag.py

Browse files
Files changed (1) hide show
  1. app/rag.py +3 -3
app/rag.py CHANGED
@@ -16,8 +16,8 @@ from qdrant_client import QdrantClient
16
  from llama_index.readers.file.docs.base import DocxReader, HWPReader, PDFReader
17
 
18
 
19
- QDRANT_API_URL = os.getenv('QDRANT_API_URL')
20
- QDRANT_API_KEY = os.getenv('QDRANT_API_KEY')
21
 
22
  logging.basicConfig(level=logging.INFO)
23
  logger = logging.getLogger(__name__)
@@ -33,7 +33,7 @@ class ChatPDF:
33
 
34
  logger.info("initializing the vector store related objects")
35
  # client = QdrantClient(host="localhost", port=6333)
36
- self.client = QdrantClient(url=QDRANT_API_URL, api_key=QDRANT_API_KEY)
37
  self.vector_store = QdrantVectorStore(
38
  client=self.client,
39
  collection_name="rag_documents",
 
16
  from llama_index.readers.file.docs.base import DocxReader, HWPReader, PDFReader
17
 
18
 
19
+ # QDRANT_API_URL = os.getenv('QDRANT_API_URL')
20
+ # QDRANT_API_KEY = os.getenv('QDRANT_API_KEY')
21
 
22
  logging.basicConfig(level=logging.INFO)
23
  logger = logging.getLogger(__name__)
 
33
 
34
  logger.info("initializing the vector store related objects")
35
  # client = QdrantClient(host="localhost", port=6333)
36
+ self.client = QdrantClient(":memory:")
37
  self.vector_store = QdrantVectorStore(
38
  client=self.client,
39
  collection_name="rag_documents",