LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-30 00:53:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

migrate: link source to vector collection

Browse Source
This commit is contained in:
ManishMadan2882
2024-08-07 03:41:31 +05:30
parent 57b9b369b7
commit f9dbaa9407
2 changed files with 134 additions and 131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
application/api/user/routes.py
View File

@@ -264,6 +264,7 @@ def combined_json():
for index in vectors_collection.find({"user": user}).sort("date", -1):
data.append(
{
"id":str(index["_id"]),
"name": index["name"],
"language": index["language"],
"version": "",
@@ -453,7 +454,7 @@ def get_api_keys():
"id": str(key["_id"]),
"name": key["name"],
"key": key["key"][:4] + "..." + key["key"][-4:],
"source": key["source"],
"source": str(key["source"]),
"prompt_id": key["prompt_id"],
"chunks": key["chunks"],
}
@@ -470,6 +471,8 @@ def create_api_key():
chunks = data["chunks"]
key = str(uuid.uuid4())
user = "local"
if(ObjectId.is_valid(data["source"])):
source = DBRef("vectors",ObjectId(data["source"]))
resp = api_key_collection.insert_one(
{
"name": name,
@@ -524,7 +527,7 @@ def share_conversation():
{
"prompt_id": prompt_id,
"chunks": chunks,
"source": source,
"source": DBRef("vectors",ObjectId(source)) if ObjectId.is_valid(source) else source,
"user": user,
}
)
@@ -574,7 +577,7 @@ def share_conversation():
{
"name": name,
"key": api_uuid,
"source": source,
"source": DBRef("vectors",ObjectId(source)) if ObjectId.is_valid(source) else source,
"user": user,
"prompt_id": prompt_id,
"chunks": chunks,