UPDATE: QDRANT

functions.py

@@ -130,30 +130,25 @@ def format_docs(docs: str):
  return context
 
 def answerQuery(query: str, vectorstore: str, llmModel: str = "llama3-70b-8192") -> str:
- try:
- global prompt 
- global client
- global embeddings
- vectorstore = QdrantVectorStore.from_existing_collection(
- embedding = embeddings,
- collection_name=vectorstore,
- url=os.environ["QDRANT_URL"],
- api_key=os.environ["QDRANT_API_KEY"]
+ global prompt 
+ global client
+ global embeddings
+ vectorstore = QdrantVectorStore.from_existing_collection(
+ embedding = embeddings,
+ collection_name=vectorstore,
+ url=os.environ["QDRANT_URL"],
+ api_key=os.environ["QDRANT_API_KEY"]
+ )
+ retriever = vectorstore.as_retriever()
+ chain = (
+ {"context": retriever | RunnableLambda(format_docs), "question": RunnablePassthrough(query)}
+ | prompt
+ | ChatGroq(model = llmModel, temperature = 0.3, max_tokens = 512)
+ | StrOutputParser()
  )
- retriever = vectorstore.as_retriever()
- chain = (
- {"context": retriever | RunnableLambda(format_docs), "question": RunnablePassthrough(query)}
- | prompt
- | ChatGroq(model = llmModel, temperature = 0.3, max_tokens = 512)
- | StrOutputParser()
- )
- return {
- "output": chain.invoke(query)
- }
- except Exception as e:
- return {
- "error": e
- }
+ return {
+ "output": chain.invoke(query)
+ }
 
 
 def deleteTable(tableName: str):