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 {
|