seekrai 0.4.1__tar.gz → 0.4.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.
Files changed (45) hide show
  1. {seekrai-0.4.1 → seekrai-0.4.2}/PKG-INFO +1 -1
  2. {seekrai-0.4.1 → seekrai-0.4.2}/pyproject.toml +1 -1
  3. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/filemanager.py +4 -3
  4. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/deployments.py +1 -0
  5. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/finetune.py +1 -0
  6. {seekrai-0.4.1 → seekrai-0.4.2}/LICENSE +0 -0
  7. {seekrai-0.4.1 → seekrai-0.4.2}/README.md +0 -0
  8. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/__init__.py +0 -0
  9. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/abstract/__init__.py +0 -0
  10. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/abstract/api_requestor.py +0 -0
  11. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/client.py +0 -0
  12. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/constants.py +0 -0
  13. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/error.py +0 -0
  14. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/__init__.py +0 -0
  15. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/alignment.py +0 -0
  16. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/chat/__init__.py +0 -0
  17. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/chat/completions.py +0 -0
  18. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/completions.py +0 -0
  19. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/deployments.py +0 -0
  20. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/embeddings.py +0 -0
  21. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/files.py +0 -0
  22. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/finetune.py +0 -0
  23. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/images.py +0 -0
  24. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/models.py +0 -0
  25. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/resources/projects.py +0 -0
  26. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/seekrflow_response.py +0 -0
  27. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/__init__.py +0 -0
  28. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/abstract.py +0 -0
  29. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/alignment.py +0 -0
  30. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/chat_completions.py +0 -0
  31. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/common.py +0 -0
  32. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/completions.py +0 -0
  33. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/embeddings.py +0 -0
  34. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/error.py +0 -0
  35. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/files.py +0 -0
  36. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/images.py +0 -0
  37. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/ingestion.py +0 -0
  38. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/models.py +0 -0
  39. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/projects.py +0 -0
  40. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/utils/__init__.py +0 -0
  41. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/utils/_log.py +0 -0
  42. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/utils/api_helpers.py +0 -0
  43. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/utils/files.py +0 -0
  44. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/utils/tools.py +0 -0
  45. {seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/version.py +0 -0
{seekrai-0.4.1 → seekrai-0.4.2}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: seekrai
3
- Version: 0.4.1
3
+ Version: 0.4.2
4
4
  Summary: Python client for SeekrAI
5
5
  Home-page: https://gitlab.cb.ntent.com/ml/seekr-py
6
6
  License: Apache-2.0
{seekrai-0.4.1 → seekrai-0.4.2}/pyproject.toml RENAMED
@@ -14,7 +14,7 @@ build-backend = "poetry.core.masonry.api"
14
14
 
15
15
  [tool.poetry]
16
16
  name = "seekrai"
17
- version = "0.4.1"
17
+ version = "0.4.2"
18
18
  authors = [
19
19
  "SeekrFlow <support@seekr.com>"
20
20
  ]
{seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/filemanager.py RENAMED
@@ -192,7 +192,7 @@ class DownloadManager:
192
192
  method="GET",
193
193
  url=url,
194
194
  ),
195
- stream=True,
195
+ stream=False,
196
196
  )
197
197
 
198
198
  try:
@@ -221,9 +221,10 @@ class DownloadManager:
221
221
  num_bytes_downloaded = response.num_bytes_downloaded
222
222
 
223
223
  # Raise exception if remote file size does not match downloaded file size
224
- if os.stat(temp_file.name).st_size != file_size:
224
+ temp_file_size = os.stat(temp_file.name).st_size
225
+ if temp_file_size != file_size:
225
226
  DownloadError(
226
- f"Downloaded file size `{pbar.n}` bytes does not match "
227
+ f"Downloaded file size `{temp_file_size}` bytes does not match "
227
228
  f"remote file size `{file_size}` bytes."
228
229
  )
229
230
 
{seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/deployments.py RENAMED
@@ -31,6 +31,7 @@ class DeploymentProcessor(str, enum.Enum):
31
31
  GAUDI2 = "GAUDI2"
32
32
  GAUDI3 = "GAUDI3"
33
33
  A100 = "A100"
34
+ A10 = "A10"
34
35
  H100 = "H100"
35
36
  XEON = "XEON"
36
37
  NVIDIA = "NVIDIA" # TODO - this doesnt make sense with A100, etc.
{seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/types/finetune.py RENAMED
@@ -111,6 +111,7 @@ class AcceleratorType(str, Enum):
111
111
  GAUDI2 = "GAUDI2"
112
112
  GAUDI3 = "GAUDI3"
113
113
  A100 = "A100"
114
+ A10 = "A10"
114
115
  H100 = "H100"
115
116
 
116
117
 
{seekrai-0.4.1 → seekrai-0.4.2}/LICENSE RENAMED
File without changes
{seekrai-0.4.1 → seekrai-0.4.2}/README.md RENAMED
File without changes
{seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/__init__.py RENAMED
File without changes
{seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/client.py RENAMED
File without changes
{seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/error.py RENAMED
File without changes
{seekrai-0.4.1 → seekrai-0.4.2}/src/seekrai/version.py RENAMED
File without changes