scale-gp-beta 0.1.0a28__py3-none-any.whl → 0.1.0a30__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 (49) hide show
  1. scale_gp_beta/_base_client.py +4 -1
  2. scale_gp_beta/_client.py +9 -0
  3. scale_gp_beta/_files.py +4 -4
  4. scale_gp_beta/_models.py +24 -3
  5. scale_gp_beta/_version.py +1 -1
  6. scale_gp_beta/lib/CONTRIBUTING.MD +53 -0
  7. scale_gp_beta/lib/tracing/integrations/openai/openai_span_type_map.py +3 -3
  8. scale_gp_beta/lib/tracing/span.py +8 -7
  9. scale_gp_beta/lib/tracing/trace.py +7 -5
  10. scale_gp_beta/lib/tracing/trace_queue_manager.py +14 -0
  11. scale_gp_beta/lib/tracing/tracing.py +7 -5
  12. scale_gp_beta/lib/tracing/types.py +1 -39
  13. scale_gp_beta/resources/__init__.py +14 -0
  14. scale_gp_beta/resources/chat/completions.py +4 -0
  15. scale_gp_beta/resources/responses.py +314 -0
  16. scale_gp_beta/resources/spans.py +28 -144
  17. scale_gp_beta/types/__init__.py +19 -0
  18. scale_gp_beta/types/chat/chat_completion.py +61 -6
  19. scale_gp_beta/types/chat/chat_completion_chunk.py +17 -1
  20. scale_gp_beta/types/chat/completion_models_params.py +2 -0
  21. scale_gp_beta/types/chat/model_definition.py +6 -0
  22. scale_gp_beta/types/completion.py +8 -0
  23. scale_gp_beta/types/container.py +0 -6
  24. scale_gp_beta/types/dataset.py +3 -1
  25. scale_gp_beta/types/dataset_item.py +3 -1
  26. scale_gp_beta/types/evaluation.py +3 -7
  27. scale_gp_beta/types/evaluation_item.py +3 -1
  28. scale_gp_beta/types/evaluation_task.py +31 -55
  29. scale_gp_beta/types/evaluation_task_param.py +28 -1
  30. scale_gp_beta/types/file.py +3 -1
  31. scale_gp_beta/types/inference_model.py +3 -0
  32. scale_gp_beta/types/question.py +11 -10
  33. scale_gp_beta/types/response.py +2852 -0
  34. scale_gp_beta/types/response_create_params.py +817 -0
  35. scale_gp_beta/types/response_create_response.py +20891 -0
  36. scale_gp_beta/types/shared/__init__.py +3 -0
  37. scale_gp_beta/types/shared/identity.py +16 -0
  38. scale_gp_beta/types/span.py +9 -33
  39. scale_gp_beta/types/span_batch_params.py +6 -30
  40. scale_gp_beta/types/span_create_params.py +6 -30
  41. scale_gp_beta/types/span_search_params.py +8 -37
  42. scale_gp_beta/types/span_status.py +7 -0
  43. scale_gp_beta/types/span_type.py +33 -0
  44. scale_gp_beta/types/span_update_params.py +3 -2
  45. scale_gp_beta/types/span_upsert_batch_params.py +6 -30
  46. {scale_gp_beta-0.1.0a28.dist-info → scale_gp_beta-0.1.0a30.dist-info}/METADATA +2 -3
  47. {scale_gp_beta-0.1.0a28.dist-info → scale_gp_beta-0.1.0a30.dist-info}/RECORD +49 -40
  48. {scale_gp_beta-0.1.0a28.dist-info → scale_gp_beta-0.1.0a30.dist-info}/WHEEL +0 -0
  49. {scale_gp_beta-0.1.0a28.dist-info → scale_gp_beta-0.1.0a30.dist-info}/licenses/LICENSE +0 -0
@@ -35,6 +35,8 @@ __all__ = [
35
35
  "AutoEvaluationQuestionTaskConfiguration",
36
36
  "AutoEvaluationGuidedDecodingEvaluationTask",
37
37
  "AutoEvaluationGuidedDecodingEvaluationTaskConfiguration",
38
+ "AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationStructuredOutputTaskRequestWithItemLocator",
39
+ "AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationGuidedDecodingTaskRequestWithItemLocator",
38
40
  "ContributorEvaluationQuestionTask",
39
41
  "ContributorEvaluationQuestionTaskConfiguration",
40
42
  ]
@@ -320,7 +322,26 @@ class AutoEvaluationQuestionTask(TypedDict, total=False):
320
322
  task_type: Literal["auto_evaluation.question"]
321
323
 
322
324
 
323
- class AutoEvaluationGuidedDecodingEvaluationTaskConfiguration(TypedDict, total=False):
325
+ class AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationStructuredOutputTaskRequestWithItemLocator(
326
+ TypedDict, total=False
327
+ ):
328
+ model: Required[str]
329
+ """model specified as `model_vendor/model_name`"""
330
+
331
+ prompt: Required[str]
332
+
333
+ response_format: Required[Dict[str, object]]
334
+ """JSON schema used for structuring the model response"""
335
+
336
+ inference_args: Dict[str, object]
337
+ """Additional arguments to pass to the inference request"""
338
+
339
+ system_prompt: str
340
+
341
+
342
+ class AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationGuidedDecodingTaskRequestWithItemLocator(
343
+ TypedDict, total=False
344
+ ):
324
345
  choices: Required[List[str]]
325
346
  """Choices array cannot be empty"""
326
347
 
@@ -335,6 +356,12 @@ class AutoEvaluationGuidedDecodingEvaluationTaskConfiguration(TypedDict, total=F
335
356
  system_prompt: str
336
357
 
337
358
 
359
+ AutoEvaluationGuidedDecodingEvaluationTaskConfiguration: TypeAlias = Union[
360
+ AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationStructuredOutputTaskRequestWithItemLocator,
361
+ AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationGuidedDecodingTaskRequestWithItemLocator,
362
+ ]
363
+
364
+
338
365
  class AutoEvaluationGuidedDecodingEvaluationTask(TypedDict, total=False):
339
366
  configuration: Required[AutoEvaluationGuidedDecodingEvaluationTaskConfiguration]
340
367
 
@@ -6,6 +6,7 @@ from datetime import datetime
6
6
  from typing_extensions import Literal
7
7
 
8
8
  from .._models import BaseModel
9
+ from .shared.identity import Identity
9
10
 
10
11
  __all__ = ["File"]
11
12
 
@@ -15,7 +16,8 @@ class File(BaseModel):
15
16
 
16
17
  created_at: datetime
17
18
 
18
- created_by_user_id: str
19
+ created_by: Identity
20
+ """The identity that created the entity."""
19
21
 
20
22
  filename: str
21
23
 
@@ -131,6 +131,7 @@ class VendorConfigurationLlmEngineVendorConfiguration(BaseModel):
131
131
 
132
132
  storage: Optional[str] = None
133
133
 
134
+ __pydantic_extra__: Dict[str, object] = FieldInfo(init=False) # pyright: ignore[reportIncompatibleVariableOverride]
134
135
  if TYPE_CHECKING:
135
136
  # Stub to indicate that arbitrary properties are accepted.
136
137
  # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -148,6 +149,8 @@ class InferenceModel(BaseModel):
148
149
 
149
150
  created_at: datetime
150
151
 
152
+ created_by_identity_type: Literal["user", "service_account"]
153
+
151
154
  created_by_user_id: str
152
155
 
153
156
  type: Literal["generic", "completion", "chat_completion"] = FieldInfo(alias="model_type")
@@ -6,6 +6,7 @@ from typing_extensions import Literal, Annotated, TypeAlias
6
6
 
7
7
  from .._utils import PropertyInfo
8
8
  from .._models import BaseModel
9
+ from .shared.identity import Identity
9
10
 
10
11
  __all__ = [
11
12
  "Question",
@@ -36,8 +37,8 @@ class CategoricalQuestion(BaseModel):
36
37
  created_at: datetime
37
38
  """ISO-timestamp when the entity was created"""
38
39
 
39
- created_by_user_id: str
40
- """User who originally created the entity"""
40
+ created_by: Identity
41
+ """The identity that created the entity."""
41
42
 
42
43
  name: str
43
44
 
@@ -69,8 +70,8 @@ class RatingQuestion(BaseModel):
69
70
  created_at: datetime
70
71
  """ISO-timestamp when the entity was created"""
71
72
 
72
- created_by_user_id: str
73
- """User who originally created the entity"""
73
+ created_by: Identity
74
+ """The identity that created the entity."""
74
75
 
75
76
  name: str
76
77
 
@@ -97,8 +98,8 @@ class NumberQuestion(BaseModel):
97
98
  created_at: datetime
98
99
  """ISO-timestamp when the entity was created"""
99
100
 
100
- created_by_user_id: str
101
- """User who originally created the entity"""
101
+ created_by: Identity
102
+ """The identity that created the entity."""
102
103
 
103
104
  name: str
104
105
 
@@ -127,8 +128,8 @@ class FreeTextQuestion(BaseModel):
127
128
  created_at: datetime
128
129
  """ISO-timestamp when the entity was created"""
129
130
 
130
- created_by_user_id: str
131
- """User who originally created the entity"""
131
+ created_by: Identity
132
+ """The identity that created the entity."""
132
133
 
133
134
  name: str
134
135
 
@@ -156,8 +157,8 @@ class FormQuestion(BaseModel):
156
157
  created_at: datetime
157
158
  """ISO-timestamp when the entity was created"""
158
159
 
159
- created_by_user_id: str
160
- """User who originally created the entity"""
160
+ created_by: Identity
161
+ """The identity that created the entity."""
161
162
 
162
163
  name: str
163
164