Add funcionality to add metadata to vector stores and vectore store files

This commit is contained in:
Enrique Barcelli 2025-08-31 23:47:56 +08:00
parent 70194de64c
commit 74e940c45d
Signed by: kikobar
GPG key ID: 006C13A68E25D3B7
2 changed files with 44 additions and 0 deletions

22
modify_vector_store.py Normal file
View file

@ -0,0 +1,22 @@
from config import *
import requests
import json
import sys
def modify_vector_store(vector_store_id,metadata):
url = base_url+"vector_stores/"+vector_store_id
payload = json.dumps({
"metadata": json.loads(metadata)
})
headers = {
'Content-Type': 'application/json',
'Authorization': 'Bearer '+secret_key,
}
response = requests.request("POST", url, headers=headers, data=payload)
print(response.text)
if __name__ == '__main__':
modify_vector_store(str(sys.argv[1]),str(sys.argv[2]))

View file

@ -0,0 +1,22 @@
from config import *
import requests
import json
import sys
def update_vector_store_file(vector_store_id,file_id,attributes):
url = base_url+"vector_stores/"+vector_store_id+"/files/"+file_id
payload = json.dumps({
"attributes": json.loads(attributes)
})
headers = {
'Content-Type': 'application/json',
'Authorization': 'Bearer '+secret_key,
}
response = requests.request("POST", url, headers=headers, data=payload)
print(response.text)
if __name__ == '__main__':
update_vector_store_file(str(sys.argv[1]),str(sys.argv[2]),str(sys.argv[3]))