llama-stack 0.0.1a0__py3-none-any.whl

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 (180) hide show
  1. llama_stack/__init__.py +95 -0
  2. llama_stack/_base_client.py +2019 -0
  3. llama_stack/_client.py +534 -0
  4. llama_stack/_compat.py +217 -0
  5. llama_stack/_constants.py +14 -0
  6. llama_stack/_exceptions.py +108 -0
  7. llama_stack/_files.py +123 -0
  8. llama_stack/_models.py +785 -0
  9. llama_stack/_qs.py +150 -0
  10. llama_stack/_resource.py +43 -0
  11. llama_stack/_response.py +823 -0
  12. llama_stack/_streaming.py +333 -0
  13. llama_stack/_types.py +217 -0
  14. llama_stack/_utils/__init__.py +55 -0
  15. llama_stack/_utils/_logs.py +25 -0
  16. llama_stack/_utils/_proxy.py +62 -0
  17. llama_stack/_utils/_reflection.py +42 -0
  18. llama_stack/_utils/_streams.py +12 -0
  19. llama_stack/_utils/_sync.py +81 -0
  20. llama_stack/_utils/_transform.py +382 -0
  21. llama_stack/_utils/_typing.py +120 -0
  22. llama_stack/_utils/_utils.py +396 -0
  23. llama_stack/_version.py +4 -0
  24. llama_stack/lib/.keep +4 -0
  25. llama_stack/py.typed +0 -0
  26. llama_stack/resources/__init__.py +201 -0
  27. llama_stack/resources/agentic_system/__init__.py +61 -0
  28. llama_stack/resources/agentic_system/agentic_system.py +338 -0
  29. llama_stack/resources/agentic_system/sessions.py +363 -0
  30. llama_stack/resources/agentic_system/steps.py +186 -0
  31. llama_stack/resources/agentic_system/turns.py +257 -0
  32. llama_stack/resources/artifacts.py +168 -0
  33. llama_stack/resources/batch_inference.py +243 -0
  34. llama_stack/resources/datasets.py +316 -0
  35. llama_stack/resources/evaluate/__init__.py +47 -0
  36. llama_stack/resources/evaluate/evaluate.py +135 -0
  37. llama_stack/resources/evaluate/jobs/__init__.py +61 -0
  38. llama_stack/resources/evaluate/jobs/artifacts.py +168 -0
  39. llama_stack/resources/evaluate/jobs/jobs.py +310 -0
  40. llama_stack/resources/evaluate/jobs/logs.py +168 -0
  41. llama_stack/resources/evaluate/jobs/status.py +168 -0
  42. llama_stack/resources/evaluate/question_answering.py +164 -0
  43. llama_stack/resources/evaluations.py +240 -0
  44. llama_stack/resources/experiments/__init__.py +33 -0
  45. llama_stack/resources/experiments/artifacts.py +238 -0
  46. llama_stack/resources/experiments/experiments.py +478 -0
  47. llama_stack/resources/inference/__init__.py +33 -0
  48. llama_stack/resources/inference/embeddings.py +178 -0
  49. llama_stack/resources/inference/inference.py +273 -0
  50. llama_stack/resources/logging.py +240 -0
  51. llama_stack/resources/memory_banks/__init__.py +33 -0
  52. llama_stack/resources/memory_banks/documents.py +268 -0
  53. llama_stack/resources/memory_banks/memory_banks.py +675 -0
  54. llama_stack/resources/post_training/__init__.py +33 -0
  55. llama_stack/resources/post_training/jobs.py +451 -0
  56. llama_stack/resources/post_training/post_training.py +274 -0
  57. llama_stack/resources/reward_scoring.py +164 -0
  58. llama_stack/resources/runs/__init__.py +33 -0
  59. llama_stack/resources/runs/metrics.py +170 -0
  60. llama_stack/resources/runs/runs.py +268 -0
  61. llama_stack/resources/synthetic_data_generation.py +166 -0
  62. llama_stack/types/__init__.py +76 -0
  63. llama_stack/types/agentic_system/__init__.py +15 -0
  64. llama_stack/types/agentic_system/agentic_system_step.py +18 -0
  65. llama_stack/types/agentic_system/agentic_system_turn_stream_chunk.py +103 -0
  66. llama_stack/types/agentic_system/session.py +21 -0
  67. llama_stack/types/agentic_system/session_create_params.py +13 -0
  68. llama_stack/types/agentic_system/session_create_response.py +11 -0
  69. llama_stack/types/agentic_system/session_delete_params.py +13 -0
  70. llama_stack/types/agentic_system/session_retrieve_params.py +16 -0
  71. llama_stack/types/agentic_system/step_retrieve_params.py +15 -0
  72. llama_stack/types/agentic_system/turn.py +39 -0
  73. llama_stack/types/agentic_system/turn_create_params.py +225 -0
  74. llama_stack/types/agentic_system/turn_retrieve_params.py +13 -0
  75. llama_stack/types/agentic_system_create_params.py +210 -0
  76. llama_stack/types/agentic_system_create_response.py +11 -0
  77. llama_stack/types/agentic_system_delete_params.py +11 -0
  78. llama_stack/types/artifact_get_params.py +11 -0
  79. llama_stack/types/batch_chat_completion.py +12 -0
  80. llama_stack/types/batch_inference_chat_completion_params.py +61 -0
  81. llama_stack/types/batch_inference_completion_params.py +28 -0
  82. llama_stack/types/chat_completion_stream_chunk.py +45 -0
  83. llama_stack/types/completion_stream_chunk.py +20 -0
  84. llama_stack/types/dataset_create_params.py +19 -0
  85. llama_stack/types/dataset_delete_params.py +11 -0
  86. llama_stack/types/dataset_get_params.py +11 -0
  87. llama_stack/types/evaluate/__init__.py +9 -0
  88. llama_stack/types/evaluate/evaluation_job_artifacts.py +11 -0
  89. llama_stack/types/evaluate/evaluation_job_log_stream.py +11 -0
  90. llama_stack/types/evaluate/evaluation_job_status.py +11 -0
  91. llama_stack/types/evaluate/job_cancel_params.py +11 -0
  92. llama_stack/types/evaluate/jobs/__init__.py +7 -0
  93. llama_stack/types/evaluate/jobs/artifact_list_params.py +11 -0
  94. llama_stack/types/evaluate/jobs/log_list_params.py +11 -0
  95. llama_stack/types/evaluate/jobs/status_list_params.py +11 -0
  96. llama_stack/types/evaluate/question_answering_create_params.py +28 -0
  97. llama_stack/types/evaluation_job.py +11 -0
  98. llama_stack/types/evaluation_summarization_params.py +28 -0
  99. llama_stack/types/evaluation_text_generation_params.py +28 -0
  100. llama_stack/types/experiment.py +23 -0
  101. llama_stack/types/experiment_create_params.py +18 -0
  102. llama_stack/types/experiment_create_run_params.py +18 -0
  103. llama_stack/types/experiment_retrieve_params.py +11 -0
  104. llama_stack/types/experiment_update_params.py +20 -0
  105. llama_stack/types/experiments/__init__.py +6 -0
  106. llama_stack/types/experiments/artifact_retrieve_params.py +11 -0
  107. llama_stack/types/experiments/artifact_upload_params.py +24 -0
  108. llama_stack/types/inference/__init__.py +6 -0
  109. llama_stack/types/inference/embedding_create_params.py +14 -0
  110. llama_stack/types/inference/embeddings.py +11 -0
  111. llama_stack/types/inference_chat_completion_params.py +63 -0
  112. llama_stack/types/inference_completion_params.py +30 -0
  113. llama_stack/types/inference_step.py +26 -0
  114. llama_stack/types/logging_get_logs_params.py +18 -0
  115. llama_stack/types/logging_get_logs_response.py +18 -0
  116. llama_stack/types/logging_log_messages_params.py +31 -0
  117. llama_stack/types/memory_bank_create_params.py +11 -0
  118. llama_stack/types/memory_bank_drop_params.py +11 -0
  119. llama_stack/types/memory_bank_drop_response.py +7 -0
  120. llama_stack/types/memory_bank_insert_params.py +26 -0
  121. llama_stack/types/memory_bank_query_params.py +16 -0
  122. llama_stack/types/memory_bank_retrieve_params.py +11 -0
  123. llama_stack/types/memory_bank_update_params.py +24 -0
  124. llama_stack/types/memory_banks/__init__.py +7 -0
  125. llama_stack/types/memory_banks/document_delete_params.py +14 -0
  126. llama_stack/types/memory_banks/document_retrieve_params.py +14 -0
  127. llama_stack/types/memory_banks/document_retrieve_response.py +17 -0
  128. llama_stack/types/memory_retrieval_step.py +25 -0
  129. llama_stack/types/post_training/__init__.py +11 -0
  130. llama_stack/types/post_training/job_artifacts_params.py +11 -0
  131. llama_stack/types/post_training/job_cancel_params.py +11 -0
  132. llama_stack/types/post_training/job_logs_params.py +11 -0
  133. llama_stack/types/post_training/job_status_params.py +11 -0
  134. llama_stack/types/post_training/post_training_job_artifacts.py +13 -0
  135. llama_stack/types/post_training/post_training_job_log_stream.py +13 -0
  136. llama_stack/types/post_training/post_training_job_status.py +25 -0
  137. llama_stack/types/post_training_job.py +11 -0
  138. llama_stack/types/post_training_preference_optimize_params.py +78 -0
  139. llama_stack/types/post_training_supervised_fine_tune_params.py +114 -0
  140. llama_stack/types/query_documents.py +21 -0
  141. llama_stack/types/rest_api_execution_config_param.py +20 -0
  142. llama_stack/types/reward_scoring.py +12 -0
  143. llama_stack/types/reward_scoring_score_params.py +42 -0
  144. llama_stack/types/run_log_metrics_params.py +31 -0
  145. llama_stack/types/run_update_params.py +25 -0
  146. llama_stack/types/runs/__init__.py +6 -0
  147. llama_stack/types/runs/metric_list_params.py +11 -0
  148. llama_stack/types/runs/metric_list_response.py +18 -0
  149. llama_stack/types/scored_dialog_generations.py +28 -0
  150. llama_stack/types/shared/__init__.py +12 -0
  151. llama_stack/types/shared/artifact.py +23 -0
  152. llama_stack/types/shared/attachment.py +13 -0
  153. llama_stack/types/shared/batch_completion.py +12 -0
  154. llama_stack/types/shared/completion_message.py +19 -0
  155. llama_stack/types/shared/run.py +22 -0
  156. llama_stack/types/shared/sampling_params.py +22 -0
  157. llama_stack/types/shared/system_message.py +14 -0
  158. llama_stack/types/shared/tool_call.py +19 -0
  159. llama_stack/types/shared/tool_response_message.py +18 -0
  160. llama_stack/types/shared/user_message.py +16 -0
  161. llama_stack/types/shared_params/__init__.py +9 -0
  162. llama_stack/types/shared_params/attachment.py +14 -0
  163. llama_stack/types/shared_params/completion_message.py +20 -0
  164. llama_stack/types/shared_params/sampling_params.py +21 -0
  165. llama_stack/types/shared_params/system_message.py +14 -0
  166. llama_stack/types/shared_params/tool_call.py +23 -0
  167. llama_stack/types/shared_params/tool_response_message.py +18 -0
  168. llama_stack/types/shared_params/user_message.py +16 -0
  169. llama_stack/types/shield_call_step.py +35 -0
  170. llama_stack/types/shield_definition_param.py +28 -0
  171. llama_stack/types/synthetic_data_generation.py +14 -0
  172. llama_stack/types/synthetic_data_generation_generate_params.py +28 -0
  173. llama_stack/types/tool_execution_step.py +34 -0
  174. llama_stack/types/tool_param_definition_param.py +15 -0
  175. llama_stack/types/train_eval_dataset.py +16 -0
  176. llama_stack/types/train_eval_dataset_param.py +16 -0
  177. llama_stack-0.0.1a0.dist-info/METADATA +382 -0
  178. llama_stack-0.0.1a0.dist-info/RECORD +180 -0
  179. llama_stack-0.0.1a0.dist-info/WHEEL +4 -0
  180. llama_stack-0.0.1a0.dist-info/licenses/LICENSE +201 -0
llama_stack/__init__.py ADDED
@@ -0,0 +1,95 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from . import types
4
+ from ._types import NOT_GIVEN, NoneType, NotGiven, Transport, ProxiesTypes
5
+ from ._utils import file_from_path
6
+ from ._client import (
7
+ ENVIRONMENTS,
8
+ Client,
9
+ Stream,
10
+ Timeout,
11
+ Transport,
12
+ LlamaStack,
13
+ AsyncClient,
14
+ AsyncStream,
15
+ RequestOptions,
16
+ AsyncLlamaStack,
17
+ )
18
+ from ._models import BaseModel
19
+ from ._version import __title__, __version__
20
+ from ._response import APIResponse as APIResponse, AsyncAPIResponse as AsyncAPIResponse
21
+ from ._constants import DEFAULT_TIMEOUT, DEFAULT_MAX_RETRIES, DEFAULT_CONNECTION_LIMITS
22
+ from ._exceptions import (
23
+ APIError,
24
+ ConflictError,
25
+ NotFoundError,
26
+ APIStatusError,
27
+ RateLimitError,
28
+ APITimeoutError,
29
+ BadRequestError,
30
+ LlamaStackError,
31
+ APIConnectionError,
32
+ AuthenticationError,
33
+ InternalServerError,
34
+ PermissionDeniedError,
35
+ UnprocessableEntityError,
36
+ APIResponseValidationError,
37
+ )
38
+ from ._base_client import DefaultHttpxClient, DefaultAsyncHttpxClient
39
+ from ._utils._logs import setup_logging as _setup_logging
40
+
41
+ __all__ = [
42
+ "types",
43
+ "__version__",
44
+ "__title__",
45
+ "NoneType",
46
+ "Transport",
47
+ "ProxiesTypes",
48
+ "NotGiven",
49
+ "NOT_GIVEN",
50
+ "LlamaStackError",
51
+ "APIError",
52
+ "APIStatusError",
53
+ "APITimeoutError",
54
+ "APIConnectionError",
55
+ "APIResponseValidationError",
56
+ "BadRequestError",
57
+ "AuthenticationError",
58
+ "PermissionDeniedError",
59
+ "NotFoundError",
60
+ "ConflictError",
61
+ "UnprocessableEntityError",
62
+ "RateLimitError",
63
+ "InternalServerError",
64
+ "Timeout",
65
+ "RequestOptions",
66
+ "Client",
67
+ "AsyncClient",
68
+ "Stream",
69
+ "AsyncStream",
70
+ "LlamaStack",
71
+ "AsyncLlamaStack",
72
+ "ENVIRONMENTS",
73
+ "file_from_path",
74
+ "BaseModel",
75
+ "DEFAULT_TIMEOUT",
76
+ "DEFAULT_MAX_RETRIES",
77
+ "DEFAULT_CONNECTION_LIMITS",
78
+ "DefaultHttpxClient",
79
+ "DefaultAsyncHttpxClient",
80
+ ]
81
+
82
+ _setup_logging()
83
+
84
+ # Update the __module__ attribute for exported symbols so that
85
+ # error messages point to this module instead of the module
86
+ # it was originally defined in, e.g.
87
+ # llama_stack._exceptions.NotFoundError -> llama_stack.NotFoundError
88
+ __locals = locals()
89
+ for __name in __all__:
90
+ if not __name.startswith("__"):
91
+ try:
92
+ __locals[__name].__module__ = "llama_stack"
93
+ except (TypeError, AttributeError):
94
+ # Some of our exported symbols are builtins which we can't set attributes for.
95
+ pass