vellum-ai 0.7.6__py3-none-any.whl → 0.7.7__py3-none-any.whl

vellum/__init__.py

@@ -236,8 +236,10 @@ from .types import (
     MlModelTokenizerConfig_Tiktoken,
     MlModelUsage,
     NamedScenarioInputChatHistoryVariableValueRequest,
+    NamedScenarioInputJsonVariableValueRequest,
     NamedScenarioInputRequest,
     NamedScenarioInputRequest_ChatHistory,
+    NamedScenarioInputRequest_Json,
     NamedScenarioInputRequest_String,
     NamedScenarioInputStringVariableValueRequest,
     NamedTestCaseArrayVariableValue,
@@ -395,8 +397,10 @@ from .types import (
     SandboxScenario,
     ScenarioInput,
     ScenarioInputChatHistoryVariableValue,
+    ScenarioInputJsonVariableValue,
     ScenarioInputStringVariableValue,
     ScenarioInput_ChatHistory,
+    ScenarioInput_Json,
     ScenarioInput_String,
     SearchFiltersRequest,
     SearchNodeResult,
@@ -922,8 +926,10 @@ __all__ = [
     "MlModelTokenizerConfig_Tiktoken",
     "MlModelUsage",
     "NamedScenarioInputChatHistoryVariableValueRequest",
+    "NamedScenarioInputJsonVariableValueRequest",
     "NamedScenarioInputRequest",
     "NamedScenarioInputRequest_ChatHistory",
+    "NamedScenarioInputRequest_Json",
     "NamedScenarioInputRequest_String",
     "NamedScenarioInputStringVariableValueRequest",
     "NamedTestCaseArrayVariableValue",
@@ -1082,8 +1088,10 @@ __all__ = [
     "SandboxScenario",
     "ScenarioInput",
     "ScenarioInputChatHistoryVariableValue",
+    "ScenarioInputJsonVariableValue",
     "ScenarioInputStringVariableValue",
     "ScenarioInput_ChatHistory",
+    "ScenarioInput_Json",
     "ScenarioInput_String",
     "SearchFiltersRequest",
     "SearchNodeResult",

vellum/core/client_wrapper.py

@@ -18,7 +18,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.7.6",
+            "X-Fern-SDK-Version": "0.7.7",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/resources/ml_models/client.py

@@ -244,7 +244,7 @@ class MlModelsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this ml model.
+            Either the ML Model's ID or its unique name
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -293,7 +293,7 @@ class MlModelsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this ml model.
+            Either the ML Model's ID or its unique name
 
         display_config : typing.Optional[MlModelDisplayConfigRequest]
             Configuration for how to display the ML Model.
@@ -355,7 +355,7 @@ class MlModelsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this ml model.
+            Either the ML Model's ID or its unique name
 
         display_config : typing.Optional[MlModelDisplayConfigRequest]
             Configuration for how to display the ML Model.
@@ -642,7 +642,7 @@ class AsyncMlModelsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this ml model.
+            Either the ML Model's ID or its unique name
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -699,7 +699,7 @@ class AsyncMlModelsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this ml model.
+            Either the ML Model's ID or its unique name
 
         display_config : typing.Optional[MlModelDisplayConfigRequest]
             Configuration for how to display the ML Model.
@@ -769,7 +769,7 @@ class AsyncMlModelsClient:
         Parameters
         ----------
         id : str
-            A UUID string identifying this ml model.
+            Either the ML Model's ID or its unique name
 
         display_config : typing.Optional[MlModelDisplayConfigRequest]
             Configuration for how to display the ML Model.

vellum/types/__init__.py

@@ -279,9 +279,11 @@ from .ml_model_tokenizer_config_request import (
 )
 from .ml_model_usage import MlModelUsage
 from .named_scenario_input_chat_history_variable_value_request import NamedScenarioInputChatHistoryVariableValueRequest
+from .named_scenario_input_json_variable_value_request import NamedScenarioInputJsonVariableValueRequest
 from .named_scenario_input_request import (
     NamedScenarioInputRequest,
     NamedScenarioInputRequest_ChatHistory,
+    NamedScenarioInputRequest_Json,
     NamedScenarioInputRequest_String,
 )
 from .named_scenario_input_string_variable_value_request import NamedScenarioInputStringVariableValueRequest
@@ -454,8 +456,9 @@ from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
 from .release_tag_source import ReleaseTagSource
 from .replace_test_suite_test_case_request import ReplaceTestSuiteTestCaseRequest
 from .sandbox_scenario import SandboxScenario
-from .scenario_input import ScenarioInput, ScenarioInput_ChatHistory, ScenarioInput_String
+from .scenario_input import ScenarioInput, ScenarioInput_ChatHistory, ScenarioInput_Json, ScenarioInput_String
 from .scenario_input_chat_history_variable_value import ScenarioInputChatHistoryVariableValue
+from .scenario_input_json_variable_value import ScenarioInputJsonVariableValue
 from .scenario_input_string_variable_value import ScenarioInputStringVariableValue
 from .search_filters_request import SearchFiltersRequest
 from .search_node_result import SearchNodeResult
@@ -987,8 +990,10 @@ __all__ = [
     "MlModelTokenizerConfig_Tiktoken",
     "MlModelUsage",
     "NamedScenarioInputChatHistoryVariableValueRequest",
+    "NamedScenarioInputJsonVariableValueRequest",
     "NamedScenarioInputRequest",
     "NamedScenarioInputRequest_ChatHistory",
+    "NamedScenarioInputRequest_Json",
     "NamedScenarioInputRequest_String",
     "NamedScenarioInputStringVariableValueRequest",
     "NamedTestCaseArrayVariableValue",
@@ -1146,8 +1151,10 @@ __all__ = [
     "SandboxScenario",
     "ScenarioInput",
     "ScenarioInputChatHistoryVariableValue",
+    "ScenarioInputJsonVariableValue",
     "ScenarioInputStringVariableValue",
     "ScenarioInput_ChatHistory",
+    "ScenarioInput_Json",
     "ScenarioInput_String",
     "SearchFiltersRequest",
     "SearchNodeResult",

vellum/types/named_scenario_input_json_variable_value_request.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class NamedScenarioInputJsonVariableValueRequest(pydantic_v1.BaseModel):
+    """
+    Named Prompt Sandbox Scenario input value that is of type JSON
+    """
+
+    value: typing.Any
+    name: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_scenario_input_request.py

@@ -34,6 +34,30 @@ class NamedScenarioInputRequest_String(pydantic_v1.BaseModel):
         json_encoders = {dt.datetime: serialize_datetime}
 
 
+class NamedScenarioInputRequest_Json(pydantic_v1.BaseModel):
+    value: typing.Any
+    name: str
+    type: typing.Literal["JSON"] = "JSON"
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}
+
+
 class NamedScenarioInputRequest_ChatHistory(pydantic_v1.BaseModel):
     value: typing.Optional[typing.List[ChatMessageRequest]] = None
     name: str
@@ -58,4 +82,6 @@ class NamedScenarioInputRequest_ChatHistory(pydantic_v1.BaseModel):
         json_encoders = {dt.datetime: serialize_datetime}
 
 
-NamedScenarioInputRequest = typing.Union[NamedScenarioInputRequest_String, NamedScenarioInputRequest_ChatHistory]
+NamedScenarioInputRequest = typing.Union[
+    NamedScenarioInputRequest_String, NamedScenarioInputRequest_Json, NamedScenarioInputRequest_ChatHistory
+]

vellum/types/scenario_input.py

@@ -34,6 +34,30 @@ class ScenarioInput_String(pydantic_v1.BaseModel):
         json_encoders = {dt.datetime: serialize_datetime}
 
 
+class ScenarioInput_Json(pydantic_v1.BaseModel):
+    value: typing.Any
+    input_variable_id: str
+    type: typing.Literal["JSON"] = "JSON"
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}
+
+
 class ScenarioInput_ChatHistory(pydantic_v1.BaseModel):
     value: typing.Optional[typing.List[ChatMessage]] = None
     input_variable_id: str
@@ -58,4 +82,4 @@ class ScenarioInput_ChatHistory(pydantic_v1.BaseModel):
         json_encoders = {dt.datetime: serialize_datetime}
 
 
-ScenarioInput = typing.Union[ScenarioInput_String, ScenarioInput_ChatHistory]
+ScenarioInput = typing.Union[ScenarioInput_String, ScenarioInput_Json, ScenarioInput_ChatHistory]

vellum/types/scenario_input_json_variable_value.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
+
+
+class ScenarioInputJsonVariableValue(pydantic_v1.BaseModel):
+    """
+    Prompt Sandbox Scenario input value that is of type JSON
+    """
+
+    value: typing.Any
+    input_variable_id: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
+
+        return deep_union_pydantic_dicts(
+            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
+        )
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

{vellum_ai-0.7.6.dist-info → vellum_ai-0.7.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.7.6
+Version: 0.7.7
 Summary: 
 License: MIT
 Requires-Python: >=3.8,<4.0

{vellum_ai-0.7.6.dist-info → vellum_ai-0.7.7.dist-info}/RECORD

@@ -1,8 +1,8 @@
-vellum/__init__.py,sha256=eqkGPne9ahxtbVtLlx7i1BxCA-RBaWg3qGRTCLw6lTA,49657
+vellum/__init__.py,sha256=Y08IiuMhi07kjouldi6dHJvZRt8cwdEVSKrG0N9BzGM,49953
 vellum/client.py,sha256=FEelOptuh8ylBnqSznSXvIUj2LWGTEPDTPrK5sgQkSE,83651
 vellum/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
 vellum/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/core/client_wrapper.py,sha256=Z9KoCufJbg73Hf7nVHFk9DnB5JNBULg1oQsf7_0BQbE,1873
+vellum/core/client_wrapper.py,sha256=kfr5YYYdHYEJSOwIjHKQZK3Nqy8Go2DfxcCLmF2Mxz0,1873
 vellum/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 vellum/core/http_client.py,sha256=46CyqS5Y8MwWTclAXnb1z5-ODJfwfHYbyhvjhb7RY1c,18753
@@ -41,7 +41,7 @@ vellum/resources/documents/client.py,sha256=lrRR9wp5nnMnENycYm-FrWwKIy7tKrfpHQ5L
 vellum/resources/folder_entities/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/resources/folder_entities/client.py,sha256=EZ_RjrB87rPLoaqNC44Dkrhp7aWEqEqI2pm5bekMqLw,4359
 vellum/resources/ml_models/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-vellum/resources/ml_models/client.py,sha256=1pXtLSTr3J7Jdu5INAGqyRFN8FEEZZ2DZ54k02HXR48,26644
+vellum/resources/ml_models/client.py,sha256=kInb7AgmXqcO9Fdvy2u_pnbeg4aUlDVD-3dCTMUxn3o,26662
 vellum/resources/sandboxes/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/resources/sandboxes/client.py,sha256=Vn80xkXWKZ8llBQSSoSqs9NU62mP1BBpNxgRBpDdLy8,15204
 vellum/resources/test_suite_runs/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
@@ -63,7 +63,7 @@ vellum/terraform/document_index/__init__.py,sha256=GY4Sn8X8-TgNiW_2Rph2uvY6tmJ6q
 vellum/terraform/provider/__init__.py,sha256=YYQLWWJDslcjc1eN0N719A3wqMdbNR2c3WuqGtX1U_I,12684
 vellum/terraform/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 vellum/terraform/versions.json,sha256=L-eKdEx1tp0hyZY8TGQ0Gy1a5R7q9hHAMLrCXICurZo,56
-vellum/types/__init__.py,sha256=OymATkVwsBVnU8eA1JZ9okU8N07VwDMUsWWl9rdGY5s,66253
+vellum/types/__init__.py,sha256=74SSEdsGGOEXMsJZJwqRExTDFUxadMjmPdS_YpK5710,66645
 vellum/types/add_openai_api_key_enum.py,sha256=GB7sLK_Ou7-Xn73sKJHUo6Gx3TjyhU7uJvWZAg4UeaI,92
 vellum/types/api_node_result.py,sha256=H25_pCtZ9xT6GTbQG1Gae89yjK-45yKipSLNqfc2PTk,1246
 vellum/types/api_node_result_data.py,sha256=1MIuC_rN_4ul7PQVWPbQySrLKzwwVb5sjXO6G6v13z4,1424
@@ -219,7 +219,8 @@ vellum/types/ml_model_tokenizer_config.py,sha256=4E01CgGUU5FlHja7wnMio8pSpfnHStH
 vellum/types/ml_model_tokenizer_config_request.py,sha256=mOLzJIidqLnkjg3Sj1JvHhX-e7ATK_tGuZt7oZ5W-Hc,2365
 vellum/types/ml_model_usage.py,sha256=IFbXxMXf-4bkGuOzPIgfm0acZ4lVRI9C6uQpeO4_0o8,1349
 vellum/types/named_scenario_input_chat_history_variable_value_request.py,sha256=CfGCHcFOEWL3UV-VjA6pYyExnpAabrlQ44c6LxdhjyA,1361
-vellum/types/named_scenario_input_request.py,sha256=-GRkDZGwkhlaIwDcTenP_-u4fH79bRE0rRG5Ddi9bSs,2452
+vellum/types/named_scenario_input_json_variable_value_request.py,sha256=r6jsE_wRPQHudeTaKDrSlKYVH2Nu4F6PlSaKx21o4GM,1248
+vellum/types/named_scenario_input_request.py,sha256=KGuT0bFtt3rmBJHmWkTlD8zKqRtmZKfP865aHifi6XM,3452
 vellum/types/named_scenario_input_string_variable_value_request.py,sha256=tlJ8G8ru1BQHxdjfBpMiwKhQOHOw3mROuyJzG3KcGow,1269
 vellum/types/named_test_case_array_variable_value.py,sha256=reOwe29S23QepGANR5yGWpKpUUQ_SUDPg5TCBBEnEDA,1323
 vellum/types/named_test_case_array_variable_value_request.py,sha256=-_F7ymtx32IfSCSgoHjHXDE1kcloeQvZdy8lhRY1aCI,1352
@@ -319,8 +320,9 @@ vellum/types/rejected_workflow_node_result_event.py,sha256=n0yp5qdEyTEsnSVRAWPFU
 vellum/types/release_tag_source.py,sha256=YavosOXZ976yfXTNWRTZwh2HhRiYmSDk0bQCkl-jCoQ,158
 vellum/types/replace_test_suite_test_case_request.py,sha256=70JYolE2hwB52LU5gpkEgF3bKC-epaeDeuJrgPzJDmE,2209
 vellum/types/sandbox_scenario.py,sha256=1vp9eQhMMPpiHfSWoAtTUpDmx0jy4VMWW9ZzHlaj7Yk,1407
-vellum/types/scenario_input.py,sha256=o98RTH7IDbXBCBKF2JKiCPvUkkloYDhWJjvGxMgQVKA,2396
+vellum/types/scenario_input.py,sha256=fMFI-FPKq6WBA1I_h_XrheeVN27JoqU_YB5vCGlD-ho,3379
 vellum/types/scenario_input_chat_history_variable_value.py,sha256=l63tBZzYwgtcBVB_-Iph-ylRRQMV7FkD7fZVbadLk2c,1334
+vellum/types/scenario_input_json_variable_value.py,sha256=SXZ7stmJ2DmXcYdmiwTkvKahP87EiCcau3QoRTn-q9I,1243
 vellum/types/scenario_input_string_variable_value.py,sha256=QEeWZyJWvEeEParxhVs4n5sJ6fztkz4uA1CVqnIrVEw,1264
 vellum/types/search_filters_request.py,sha256=3k90Kq2KWu8p9a-CYtkcybS7ODNosNLgBIk-ZL0uQkM,1481
 vellum/types/search_node_result.py,sha256=Yz9S17nCtOTQDkhWM_hPUOHG8ZttG1TQukzSVjnWGH4,1260
@@ -494,7 +496,7 @@ vellum/types/workflow_result_event_output_data_search_results.py,sha256=_C4ueKK8
 vellum/types/workflow_result_event_output_data_string.py,sha256=AAWHZT3X9HOIRA3UuIqw0VpfSGwGemsJM71WDNbWYTc,1745
 vellum/types/workflow_stream_event.py,sha256=5K-Mtn9fvJDq8m5nhURDbChL01PXIiuIZDkfAC1d6fU,2610
 vellum/version.py,sha256=neLt8HBHHUtDF9M5fsyUzHT-pKooEPvceaLDqqIGb0s,77
-vellum_ai-0.7.6.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
-vellum_ai-0.7.6.dist-info/METADATA,sha256=Px_DEw78UFucYoZd9Z3V9a8pjtuPXhl8E2AcMkjRNvc,4398
-vellum_ai-0.7.6.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-vellum_ai-0.7.6.dist-info/RECORD,,
+vellum_ai-0.7.7.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
+vellum_ai-0.7.7.dist-info/METADATA,sha256=79Tnc1WEpabCIADsOlXGjR7EKNmvC7oe0OPjzrpSYdU,4398
+vellum_ai-0.7.7.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+vellum_ai-0.7.7.dist-info/RECORD,,