vector-vault 4.3.0__tar.gz → 4.3.2__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {vector_vault-4.3.0 → vector_vault-4.3.2}/PKG-INFO +1 -1
- {vector_vault-4.3.0 → vector_vault-4.3.2}/setup.py +1 -1
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vector_vault.egg-info/PKG-INFO +1 -1
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vectorvault/cloudmanager.py +1 -2
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vectorvault/vault.py +8 -5
- {vector_vault-4.3.0 → vector_vault-4.3.2}/LICENSE +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/README.md +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/setup.cfg +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vector_vault.egg-info/SOURCES.txt +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vector_vault.egg-info/dependency_links.txt +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vector_vault.egg-info/requires.txt +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vector_vault.egg-info/top_level.txt +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vectorvault/__init__.py +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vectorvault/ai.py +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vectorvault/cloud_api.py +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vectorvault/creds.py +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vectorvault/download.py +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vectorvault/itemize.py +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vectorvault/tools_gpt.py +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vectorvault/vecreq.py +0 -0
- {vector_vault-4.3.0 → vector_vault-4.3.2}/vectorvault/wrap.py +0 -0
|
@@ -20,11 +20,10 @@ from .itemize import cloud_name
|
|
|
20
20
|
from .vault import ThreadPoolExecutor, T, as_completed, time, json, os, tempfile, time
|
|
21
21
|
|
|
22
22
|
class CloudManager:
|
|
23
|
-
def __init__(self, user: str, api_key: str, vault: str
|
|
23
|
+
def __init__(self, user: str, api_key: str, vault: str):
|
|
24
24
|
self.user = user
|
|
25
25
|
self.api = api_key
|
|
26
26
|
self.vault = vault
|
|
27
|
-
self.ai_api = ai_api
|
|
28
27
|
# Create credentials
|
|
29
28
|
self.credentials = CustomCredentials(user, self.api)
|
|
30
29
|
# Instantiate the client
|
|
@@ -75,6 +75,9 @@ class Vault:
|
|
|
75
75
|
self.user = user.lower()
|
|
76
76
|
self.vault = vault.strip() if vault else 'home'
|
|
77
77
|
self.api = api_key
|
|
78
|
+
self.verbose = verbose
|
|
79
|
+
self.embeddings_model = embeddings_model if embeddings_model else 'text-embedding-3-small'
|
|
80
|
+
self.dims = 1536 if embeddings_model != 'text-embedding-3-large' else 3072
|
|
78
81
|
if openai_key:
|
|
79
82
|
self.openai_key = openai_key
|
|
80
83
|
openai.api_key = self.openai_key
|
|
@@ -86,10 +89,7 @@ class Vault:
|
|
|
86
89
|
# user can still use the get_chat() function without an api key
|
|
87
90
|
self.cloud_manager = None
|
|
88
91
|
|
|
89
|
-
self.
|
|
90
|
-
self.dims = 1536 if embeddings_model != 'text-embedding-3-large' else 3072
|
|
91
|
-
self.vectors = get_vectors(self.dims)
|
|
92
|
-
self.verbose = verbose
|
|
92
|
+
T(target=self.init_vecs)
|
|
93
93
|
self.x = 0
|
|
94
94
|
self.x_checked = False
|
|
95
95
|
self.vecs_loaded = False
|
|
@@ -102,6 +102,9 @@ class Vault:
|
|
|
102
102
|
self.rate_limiter = RateLimiter(max_attempts=30)
|
|
103
103
|
self.cuid = conversation_user_id
|
|
104
104
|
|
|
105
|
+
def init_vecs(self):
|
|
106
|
+
self.vectors = get_vectors(self.dims)
|
|
107
|
+
|
|
105
108
|
def get_vaults(self, vault: str = None):
|
|
106
109
|
'''
|
|
107
110
|
Returns a list of vaults within the current vault directory
|
|
@@ -1329,7 +1332,7 @@ class Vault:
|
|
|
1329
1332
|
vector_history = []
|
|
1330
1333
|
now = datetime.now()
|
|
1331
1334
|
for i in vector_similar_results:
|
|
1332
|
-
message_time = datetime.fromtimestamp(float(i['name']))
|
|
1335
|
+
message_time = datetime.fromtimestamp(float(i['metadata']['name']))
|
|
1333
1336
|
data = i['data']
|
|
1334
1337
|
vector_history.append(f'{get_time_statement(now, message_time)} {data}')
|
|
1335
1338
|
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|