Add funcionality to add metadata to vector stores and vectore store files
This commit is contained in:
parent
70194de64c
commit
74e940c45d
22
modify_vector_store.py
Normal file
22
modify_vector_store.py
Normal 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]))
|
22
update_vector_store_file.py
Normal file
22
update_vector_store_file.py
Normal 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]))
|
Loading…
Reference in a new issue