Spaces:
Sleeping
Sleeping
UPDATE: QDRANT
Browse files- app.py +6 -5
- functions.py +1 -0
app.py
CHANGED
|
@@ -26,8 +26,9 @@ async def login(username: str, password: str):
|
|
| 26 |
return response
|
| 27 |
|
| 28 |
|
| 29 |
-
@app.
|
| 30 |
-
async def newChatbot(chatbotName: str):
|
|
|
|
| 31 |
return createTable(tablename = chatbotName)
|
| 32 |
|
| 33 |
|
|
@@ -46,15 +47,15 @@ async def addText(vectorstore: str, text: str):
|
|
| 46 |
return addDocuments(text = text, vectorstore = vectorstore)
|
| 47 |
|
| 48 |
|
| 49 |
-
@app.
|
| 50 |
async def answerQuestion(query: str, vectorstore: str, llmModel: str = "llama3-70b-8192"):
|
| 51 |
return answerQuery(query=query, vectorstore=vectorstore, llmModel=llmModel)
|
| 52 |
|
| 53 |
|
| 54 |
-
@app.
|
| 55 |
async def delete(chatbotName: str):
|
| 56 |
return deleteTable(tableName=chatbotName)
|
| 57 |
|
| 58 |
-
@app.
|
| 59 |
async def delete(username: str):
|
| 60 |
return listTables(username=username)
|
|
|
|
| 26 |
return response
|
| 27 |
|
| 28 |
|
| 29 |
+
@app.post("/newChatbot")
|
| 30 |
+
async def newChatbot(chatbotName: str, username: str):
|
| 31 |
+
chatbotName = f"convai-{username}-{chatbotName}"
|
| 32 |
return createTable(tablename = chatbotName)
|
| 33 |
|
| 34 |
|
|
|
|
| 47 |
return addDocuments(text = text, vectorstore = vectorstore)
|
| 48 |
|
| 49 |
|
| 50 |
+
@app.post("/answerQuery")
|
| 51 |
async def answerQuestion(query: str, vectorstore: str, llmModel: str = "llama3-70b-8192"):
|
| 52 |
return answerQuery(query=query, vectorstore=vectorstore, llmModel=llmModel)
|
| 53 |
|
| 54 |
|
| 55 |
+
@app.post("/deleteChatbot")
|
| 56 |
async def delete(chatbotName: str):
|
| 57 |
return deleteTable(tableName=chatbotName)
|
| 58 |
|
| 59 |
+
@app.post("/listChatbots")
|
| 60 |
async def delete(username: str):
|
| 61 |
return listTables(username=username)
|
functions.py
CHANGED
|
@@ -110,6 +110,7 @@ def addDocuments(text: str, vectorstore: str):
|
|
| 110 |
embedding = embeddings,
|
| 111 |
collection_name=vectorstore,
|
| 112 |
url=os.environ["QDRANT_URL"],
|
|
|
|
| 113 |
)
|
| 114 |
vectorstore.add_documents(documents = texts)
|
| 115 |
return {
|
|
|
|
| 110 |
embedding = embeddings,
|
| 111 |
collection_name=vectorstore,
|
| 112 |
url=os.environ["QDRANT_URL"],
|
| 113 |
+
api_key=os.environ["QDRANT_API_KEY"]
|
| 114 |
)
|
| 115 |
vectorstore.add_documents(documents = texts)
|
| 116 |
return {
|