Rauhan commited on
Commit
011040f
·
1 Parent(s): 064943c

UPDATE: QDRANT

Browse files
Files changed (2) hide show
  1. app.py +6 -5
  2. 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.get("/newChatbot/{chatbotName}")
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.get("/answerQuery")
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.get("/deleteChatbot/{chatbotName}")
55
  async def delete(chatbotName: str):
56
  return deleteTable(tableName=chatbotName)
57
 
58
- @app.get("/listChatbots/{username}")
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 {