c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python 0.1.0a3__py3-none-any.whl → 0.1.0a4__py3-none-any.whl

digitalocean_genai_sdk/types/chat_create_completion_params.py → gradientai/types/chat/completion_create_params.py

@@ -5,22 +5,18 @@ from __future__ import annotations
 from typing import Dict, List, Union, Iterable, Optional
 from typing_extensions import Literal, Required, TypeAlias, TypedDict
 
-from .chat_completion_request_message_content_part_text_param import ChatCompletionRequestMessageContentPartTextParam
-
 __all__ = [
-    "ChatCreateCompletionParams",
+    "CompletionCreateParams",
     "Message",
     "MessageChatCompletionRequestSystemMessage",
     "MessageChatCompletionRequestDeveloperMessage",
     "MessageChatCompletionRequestUserMessage",
     "MessageChatCompletionRequestAssistantMessage",
-    "MessageChatCompletionRequestAssistantMessageContentArrayOfContentPart",
-    "MessageChatCompletionRequestAssistantMessageContentArrayOfContentPartChatCompletionRequestMessageContentPartRefusal",
     "StreamOptions",
 ]
 
 
-class ChatCreateCompletionParams(TypedDict, total=False):
+class CompletionCreateParams(TypedDict, total=False):
     messages: Required[Iterable[Message]]
     """A list of messages comprising the conversation so far."""
 
@@ -137,7 +133,7 @@ class ChatCreateCompletionParams(TypedDict, total=False):
 
 
 class MessageChatCompletionRequestSystemMessage(TypedDict, total=False):
-    content: Required[Union[str, Iterable[ChatCompletionRequestMessageContentPartTextParam]]]
+    content: Required[Union[str, List[str]]]
     """The contents of the system message."""
 
     role: Required[Literal["system"]]
@@ -145,7 +141,7 @@ class MessageChatCompletionRequestSystemMessage(TypedDict, total=False):
 
 
 class MessageChatCompletionRequestDeveloperMessage(TypedDict, total=False):
-    content: Required[Union[str, Iterable[ChatCompletionRequestMessageContentPartTextParam]]]
+    content: Required[Union[str, List[str]]]
     """The contents of the developer message."""
 
     role: Required[Literal["developer"]]
@@ -153,39 +149,20 @@ class MessageChatCompletionRequestDeveloperMessage(TypedDict, total=False):
 
 
 class MessageChatCompletionRequestUserMessage(TypedDict, total=False):
-    content: Required[Union[str, Iterable[ChatCompletionRequestMessageContentPartTextParam]]]
+    content: Required[Union[str, List[str]]]
     """The contents of the user message."""
 
     role: Required[Literal["user"]]
     """The role of the messages author, in this case `user`."""
 
 
-class MessageChatCompletionRequestAssistantMessageContentArrayOfContentPartChatCompletionRequestMessageContentPartRefusal(
-    TypedDict, total=False
-):
-    refusal: Required[str]
-    """The refusal message generated by the model."""
-
-    type: Required[Literal["refusal"]]
-    """The type of the content part."""
-
-
-MessageChatCompletionRequestAssistantMessageContentArrayOfContentPart: TypeAlias = Union[
-    ChatCompletionRequestMessageContentPartTextParam,
-    MessageChatCompletionRequestAssistantMessageContentArrayOfContentPartChatCompletionRequestMessageContentPartRefusal,
-]
-
-
 class MessageChatCompletionRequestAssistantMessage(TypedDict, total=False):
     role: Required[Literal["assistant"]]
     """The role of the messages author, in this case `assistant`."""
 
-    content: Union[str, Iterable[MessageChatCompletionRequestAssistantMessageContentArrayOfContentPart], None]
+    content: Union[str, List[str], None]
     """The contents of the assistant message."""
 
-    refusal: Optional[str]
-    """The refusal message by the assistant."""
-
 
 Message: TypeAlias = Union[
     MessageChatCompletionRequestSystemMessage,

digitalocean_genai_sdk/types/chat_create_completion_response.py → gradientai/types/chat/completion_create_response.py

@@ -3,10 +3,10 @@
 from typing import List, Optional
 from typing_extensions import Literal
 
-from .._models import BaseModel
+from ..._models import BaseModel
 from .chat_completion_token_logprob import ChatCompletionTokenLogprob
 
-__all__ = ["ChatCreateCompletionResponse", "Choice", "ChoiceLogprobs", "ChoiceMessage", "Usage"]
+__all__ = ["CompletionCreateResponse", "Choice", "ChoiceLogprobs", "ChoiceMessage", "Usage"]
 
 
 class ChoiceLogprobs(BaseModel):
@@ -58,7 +58,7 @@ class Usage(BaseModel):
     """Total number of tokens used in the request (prompt + completion)."""
 
 
-class ChatCreateCompletionResponse(BaseModel):
+class CompletionCreateResponse(BaseModel):
     id: str
     """A unique identifier for the chat completion."""
 

{digitalocean_genai_sdk/types/api_keys → gradientai/types/inference}/__init__.py

@@ -2,7 +2,9 @@
 
 from __future__ import annotations
 
+from .model import Model as Model
 from .api_key_list_params import APIKeyListParams as APIKeyListParams
+from .model_list_response import ModelListResponse as ModelListResponse
 from .api_key_create_params import APIKeyCreateParams as APIKeyCreateParams
 from .api_key_list_response import APIKeyListResponse as APIKeyListResponse
 from .api_key_update_params import APIKeyUpdateParams as APIKeyUpdateParams

{digitalocean_genai_sdk/types → gradientai/types/inference}/model.py

@@ -2,7 +2,7 @@
 
 from typing_extensions import Literal
 
-from .._models import BaseModel
+from ..._models import BaseModel
 
 __all__ = ["Model"]
 

{digitalocean_genai_sdk/types → gradientai/types/inference}/model_list_response.py

@@ -4,7 +4,7 @@ from typing import List
 from typing_extensions import Literal
 
 from .model import Model
-from .._models import BaseModel
+from ..._models import BaseModel
 
 __all__ = ["ModelListResponse"]
 

{digitalocean_genai_sdk → gradientai}/types/knowledge_base_create_params.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 from typing import List, Iterable
 from typing_extensions import TypedDict
 
+from .knowledge_bases.aws_data_source_param import AwsDataSourceParam
 from .knowledge_bases.api_spaces_data_source_param import APISpacesDataSourceParam
 from .knowledge_bases.api_file_upload_data_source_param import APIFileUploadDataSourceParam
 from .knowledge_bases.api_web_crawler_data_source_param import APIWebCrawlerDataSourceParam
@@ -50,6 +51,8 @@ class KnowledgeBaseCreateParams(TypedDict, total=False):
 
 
 class Datasource(TypedDict, total=False):
+    aws_data_source: AwsDataSourceParam
+
     bucket_name: str
 
     bucket_region: str

{digitalocean_genai_sdk → gradientai}/types/knowledge_bases/__init__.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from .aws_data_source_param import AwsDataSourceParam as AwsDataSourceParam
 from .api_spaces_data_source import APISpacesDataSource as APISpacesDataSource
 from .data_source_list_params import DataSourceListParams as DataSourceListParams
 from .data_source_create_params import DataSourceCreateParams as DataSourceCreateParams

{digitalocean_genai_sdk → gradientai}/types/knowledge_bases/api_knowledge_base_data_source.py

@@ -9,10 +9,20 @@ from .api_spaces_data_source import APISpacesDataSource
 from .api_file_upload_data_source import APIFileUploadDataSource
 from .api_web_crawler_data_source import APIWebCrawlerDataSource
 
-__all__ = ["APIKnowledgeBaseDataSource"]
+__all__ = ["APIKnowledgeBaseDataSource", "AwsDataSource"]
+
+
+class AwsDataSource(BaseModel):
+    bucket_name: Optional[str] = None
+
+    item_path: Optional[str] = None
+
+    region: Optional[str] = None
 
 
 class APIKnowledgeBaseDataSource(BaseModel):
+    aws_data_source: Optional[AwsDataSource] = None
+
     bucket_name: Optional[str] = None
 
     created_at: Optional[datetime] = None

gradientai/types/knowledge_bases/aws_data_source_param.py

@@ -0,0 +1,19 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypedDict
+
+__all__ = ["AwsDataSourceParam"]
+
+
+class AwsDataSourceParam(TypedDict, total=False):
+    bucket_name: str
+
+    item_path: str
+
+    key_id: str
+
+    region: str
+
+    secret_key: str

{digitalocean_genai_sdk → gradientai}/types/knowledge_bases/data_source_create_params.py

@@ -5,29 +5,18 @@ from __future__ import annotations
 from typing_extensions import Annotated, TypedDict
 
 from ..._utils import PropertyInfo
+from .aws_data_source_param import AwsDataSourceParam
 from .api_spaces_data_source_param import APISpacesDataSourceParam
 from .api_web_crawler_data_source_param import APIWebCrawlerDataSourceParam
 
-__all__ = ["DataSourceCreateParams", "AwsDataSource"]
+__all__ = ["DataSourceCreateParams"]
 
 
 class DataSourceCreateParams(TypedDict, total=False):
-    aws_data_source: AwsDataSource
+    aws_data_source: AwsDataSourceParam
 
     body_knowledge_base_uuid: Annotated[str, PropertyInfo(alias="knowledge_base_uuid")]
 
     spaces_data_source: APISpacesDataSourceParam
 
     web_crawler_data_source: APIWebCrawlerDataSourceParam
-
-
-class AwsDataSource(TypedDict, total=False):
-    bucket_name: str
-
-    item_path: str
-
-    key_id: str
-
-    region: str
-
-    secret_key: str

digitalocean_genai_sdk/types/api_key_list_params.py → gradientai/types/model_list_params.py

@@ -5,10 +5,10 @@ from __future__ import annotations
 from typing import List
 from typing_extensions import Literal, TypedDict
 
-__all__ = ["APIKeyListParams"]
+__all__ = ["ModelListParams"]
 
 
-class APIKeyListParams(TypedDict, total=False):
+class ModelListParams(TypedDict, total=False):
     page: int
     """page number."""
 

gradientai/types/model_list_response.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from .._models import BaseModel
+from .api_model import APIModel
+from .agents.api_meta import APIMeta
+from .agents.api_links import APILinks
+
+__all__ = ["ModelListResponse"]
+
+
+class ModelListResponse(BaseModel):
+    links: Optional[APILinks] = None
+
+    meta: Optional[APIMeta] = None
+
+    models: Optional[List[APIModel]] = None

gradientai/types/region_list_evaluation_metrics_response.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from .._models import BaseModel
+from .api_evaluation_metric import APIEvaluationMetric
+
+__all__ = ["RegionListEvaluationMetricsResponse"]
+
+
+class RegionListEvaluationMetricsResponse(BaseModel):
+    metrics: Optional[List[APIEvaluationMetric]] = None

gradientai/types/regions/__init__.py

@@ -0,0 +1,32 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .api_star_metric import APIStarMetric as APIStarMetric
+from .api_star_metric_param import APIStarMetricParam as APIStarMetricParam
+from .api_evaluation_test_case import APIEvaluationTestCase as APIEvaluationTestCase
+from .evaluation_run_create_params import EvaluationRunCreateParams as EvaluationRunCreateParams
+from .evaluation_run_create_response import EvaluationRunCreateResponse as EvaluationRunCreateResponse
+from .evaluation_dataset_create_params import EvaluationDatasetCreateParams as EvaluationDatasetCreateParams
+from .evaluation_run_retrieve_response import EvaluationRunRetrieveResponse as EvaluationRunRetrieveResponse
+from .evaluation_dataset_create_response import EvaluationDatasetCreateResponse as EvaluationDatasetCreateResponse
+from .evaluation_test_case_create_params import EvaluationTestCaseCreateParams as EvaluationTestCaseCreateParams
+from .evaluation_test_case_list_response import EvaluationTestCaseListResponse as EvaluationTestCaseListResponse
+from .evaluation_test_case_update_params import EvaluationTestCaseUpdateParams as EvaluationTestCaseUpdateParams
+from .evaluation_test_case_create_response import EvaluationTestCaseCreateResponse as EvaluationTestCaseCreateResponse
+from .evaluation_test_case_update_response import EvaluationTestCaseUpdateResponse as EvaluationTestCaseUpdateResponse
+from .evaluation_test_case_retrieve_response import (
+    EvaluationTestCaseRetrieveResponse as EvaluationTestCaseRetrieveResponse,
+)
+from .evaluation_test_case_list_evaluation_runs_params import (
+    EvaluationTestCaseListEvaluationRunsParams as EvaluationTestCaseListEvaluationRunsParams,
+)
+from .evaluation_test_case_list_evaluation_runs_response import (
+    EvaluationTestCaseListEvaluationRunsResponse as EvaluationTestCaseListEvaluationRunsResponse,
+)
+from .evaluation_dataset_create_file_upload_presigned_urls_params import (
+    EvaluationDatasetCreateFileUploadPresignedURLsParams as EvaluationDatasetCreateFileUploadPresignedURLsParams,
+)
+from .evaluation_dataset_create_file_upload_presigned_urls_response import (
+    EvaluationDatasetCreateFileUploadPresignedURLsResponse as EvaluationDatasetCreateFileUploadPresignedURLsResponse,
+)

gradientai/types/regions/api_evaluation_test_case.py

@@ -0,0 +1,46 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from datetime import datetime
+
+from ..._models import BaseModel
+from .api_star_metric import APIStarMetric
+from ..api_evaluation_metric import APIEvaluationMetric
+
+__all__ = ["APIEvaluationTestCase"]
+
+
+class APIEvaluationTestCase(BaseModel):
+    archived_at: Optional[datetime] = None
+
+    created_at: Optional[datetime] = None
+
+    created_by_user_email: Optional[str] = None
+
+    created_by_user_id: Optional[str] = None
+
+    dataset_name: Optional[str] = None
+
+    dataset_uuid: Optional[str] = None
+
+    description: Optional[str] = None
+
+    latest_version_number_of_runs: Optional[int] = None
+
+    metrics: Optional[List[APIEvaluationMetric]] = None
+
+    name: Optional[str] = None
+
+    star_metric: Optional[APIStarMetric] = None
+
+    test_case_uuid: Optional[str] = None
+
+    total_runs: Optional[int] = None
+
+    updated_at: Optional[datetime] = None
+
+    updated_by_user_email: Optional[str] = None
+
+    updated_by_user_id: Optional[str] = None
+
+    version: Optional[int] = None

gradientai/types/regions/api_star_metric.py

@@ -0,0 +1,19 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["APIStarMetric"]
+
+
+class APIStarMetric(BaseModel):
+    metric_uuid: Optional[str] = None
+
+    name: Optional[str] = None
+
+    success_threshold_pct: Optional[int] = None
+    """
+    The success threshold for the star metric. This is a percentage value between 0
+    and 100.
+    """

gradientai/types/regions/api_star_metric_param.py

@@ -0,0 +1,19 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypedDict
+
+__all__ = ["APIStarMetricParam"]
+
+
+class APIStarMetricParam(TypedDict, total=False):
+    metric_uuid: str
+
+    name: str
+
+    success_threshold_pct: int
+    """
+    The success threshold for the star metric. This is a percentage value between 0
+    and 100.
+    """

gradientai/types/regions/evaluation_dataset_create_file_upload_presigned_urls_params.py

@@ -0,0 +1,20 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Iterable
+from typing_extensions import TypedDict
+
+__all__ = ["EvaluationDatasetCreateFileUploadPresignedURLsParams", "File"]
+
+
+class EvaluationDatasetCreateFileUploadPresignedURLsParams(TypedDict, total=False):
+    files: Iterable[File]
+    """A list of files to generate presigned URLs for."""
+
+
+class File(TypedDict, total=False):
+    file_name: str
+
+    file_size: str
+    """The size of the file in bytes."""

gradientai/types/regions/evaluation_dataset_create_file_upload_presigned_urls_response.py

@@ -0,0 +1,30 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from datetime import datetime
+
+from ..._models import BaseModel
+
+__all__ = ["EvaluationDatasetCreateFileUploadPresignedURLsResponse", "Upload"]
+
+
+class Upload(BaseModel):
+    expires_at: Optional[datetime] = None
+    """The time the url expires at."""
+
+    object_key: Optional[str] = None
+    """The unique object key to store the file as."""
+
+    original_file_name: Optional[str] = None
+    """The original file name."""
+
+    presigned_url: Optional[str] = None
+    """The actual presigned URL the client can use to upload the file directly."""
+
+
+class EvaluationDatasetCreateFileUploadPresignedURLsResponse(BaseModel):
+    request_id: Optional[str] = None
+    """The ID generated for the request for Presigned URLs."""
+
+    uploads: Optional[List[Upload]] = None
+    """A list of generated presigned URLs and object keys, one per file."""

gradientai/types/regions/evaluation_dataset_create_params.py

@@ -0,0 +1,17 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypedDict
+
+from ..knowledge_bases.api_file_upload_data_source_param import APIFileUploadDataSourceParam
+
+__all__ = ["EvaluationDatasetCreateParams"]
+
+
+class EvaluationDatasetCreateParams(TypedDict, total=False):
+    file_upload_dataset: APIFileUploadDataSourceParam
+    """File to upload as data source for knowledge base."""
+
+    name: str
+    """The name of the agent evaluation dataset."""

gradientai/types/regions/evaluation_dataset_create_response.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["EvaluationDatasetCreateResponse"]
+
+
+class EvaluationDatasetCreateResponse(BaseModel):
+    evaluation_dataset_uuid: Optional[str] = None
+    """Evaluation dataset uuid."""

gradientai/types/regions/evaluation_run_create_params.py

@@ -0,0 +1,17 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypedDict
+
+__all__ = ["EvaluationRunCreateParams"]
+
+
+class EvaluationRunCreateParams(TypedDict, total=False):
+    agent_uuid: str
+    """Agent UUID to run the test case against."""
+
+    run_name: str
+    """The name of the run."""
+
+    test_case_uuid: str

gradientai/types/regions/evaluation_run_create_response.py

@@ -0,0 +1,11 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["EvaluationRunCreateResponse"]
+
+
+class EvaluationRunCreateResponse(BaseModel):
+    evaluation_run_uuid: Optional[str] = None

gradientai/types/regions/evaluation_run_retrieve_response.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+from .evaluation_runs.api_evaluation_run import APIEvaluationRun
+
+__all__ = ["EvaluationRunRetrieveResponse"]
+
+
+class EvaluationRunRetrieveResponse(BaseModel):
+    evaluation_run: Optional[APIEvaluationRun] = None

gradientai/types/regions/evaluation_runs/__init__.py

@@ -0,0 +1,9 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .api_prompt import APIPrompt as APIPrompt
+from .api_evaluation_run import APIEvaluationRun as APIEvaluationRun
+from .result_retrieve_response import ResultRetrieveResponse as ResultRetrieveResponse
+from .api_evaluation_metric_result import APIEvaluationMetricResult as APIEvaluationMetricResult
+from .result_retrieve_prompt_response import ResultRetrievePromptResponse as ResultRetrievePromptResponse

gradientai/types/regions/evaluation_runs/api_evaluation_metric_result.py

@@ -0,0 +1,17 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ...._models import BaseModel
+
+__all__ = ["APIEvaluationMetricResult"]
+
+
+class APIEvaluationMetricResult(BaseModel):
+    metric_name: Optional[str] = None
+
+    number_value: Optional[float] = None
+    """The value of the metric as a number."""
+
+    string_value: Optional[str] = None
+    """The value of the metric as a string."""

gradientai/types/regions/evaluation_runs/api_evaluation_run.py

@@ -0,0 +1,56 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from datetime import datetime
+from typing_extensions import Literal
+
+from ...._models import BaseModel
+from .api_evaluation_metric_result import APIEvaluationMetricResult
+
+__all__ = ["APIEvaluationRun"]
+
+
+class APIEvaluationRun(BaseModel):
+    agent_uuid: Optional[str] = None
+    """Agent UUID."""
+
+    agent_version_hash: Optional[str] = None
+
+    evaluation_run_uuid: Optional[str] = None
+    """Evaluation run UUID."""
+
+    finished_at: Optional[datetime] = None
+    """Run end time."""
+
+    pass_status: Optional[bool] = None
+    """The pass status of the evaluation run based on the star metric."""
+
+    run_level_metric_results: Optional[List[APIEvaluationMetricResult]] = None
+
+    run_name: Optional[str] = None
+    """Run name."""
+
+    star_metric_result: Optional[APIEvaluationMetricResult] = None
+
+    started_at: Optional[datetime] = None
+    """Run start time."""
+
+    status: Optional[
+        Literal[
+            "EVALUATION_RUN_STATUS_UNSPECIFIED",
+            "EVALUATION_RUN_QUEUED",
+            "EVALUATION_RUN_RUNNING_DATASET",
+            "EVALUATION_RUN_EVALUATING_RESULTS",
+            "EVALUATION_RUN_CANCELLING",
+            "EVALUATION_RUN_CANCELLED",
+            "EVALUATION_RUN_SUCCESSFUL",
+            "EVALUATION_RUN_PARTIALLY_SUCCESSFUL",
+            "EVALUATION_RUN_FAILED",
+        ]
+    ] = None
+
+    test_case_uuid: Optional[str] = None
+    """Test-case UUID."""
+
+    test_case_version: Optional[int] = None
+    """Test-case-version."""

gradientai/types/regions/evaluation_runs/api_prompt.py

@@ -0,0 +1,42 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from ...._models import BaseModel
+from .api_evaluation_metric_result import APIEvaluationMetricResult
+
+__all__ = ["APIPrompt", "PromptChunk"]
+
+
+class PromptChunk(BaseModel):
+    chunk_usage_pct: Optional[float] = None
+    """The usage percentage of the chunk."""
+
+    chunk_used: Optional[bool] = None
+    """Indicates if the chunk was used in the prompt."""
+
+    index_uuid: Optional[str] = None
+    """The index uuid (Knowledge Base) of the chunk."""
+
+    source_name: Optional[str] = None
+    """The source name for the chunk, e.g., the file name or document title."""
+
+    text: Optional[str] = None
+    """Text content of the chunk."""
+
+
+class APIPrompt(BaseModel):
+    ground_truth: Optional[str] = None
+    """The ground truth for the prompt."""
+
+    input: Optional[str] = None
+
+    output: Optional[str] = None
+
+    prompt_chunks: Optional[List[PromptChunk]] = None
+    """The list of prompt chunks."""
+
+    prompt_id: Optional[int] = None
+
+    prompt_level_metric_results: Optional[List[APIEvaluationMetricResult]] = None
+    """The metric results for the prompt."""