diff --git a/deepinfra/types/automatic_speech_recognition/request.py b/deepinfra/types/automatic_speech_recognition/request.py index 46c4ced..3c1c8b4 100644 --- a/deepinfra/types/automatic_speech_recognition/request.py +++ b/deepinfra/types/automatic_speech_recognition/request.py @@ -2,7 +2,7 @@ from typing import Optional -@dataclass(kw_only=True) +@dataclass class AutomaticSpeechRecognitionRequest: audio: str task: Optional[str] diff --git a/deepinfra/types/automatic_speech_recognition/response.py b/deepinfra/types/automatic_speech_recognition/response.py index ba07974..5db05e6 100644 --- a/deepinfra/types/automatic_speech_recognition/response.py +++ b/deepinfra/types/automatic_speech_recognition/response.py @@ -5,7 +5,7 @@ from deepinfra.types.common.inference_status import InferenceStatus -@dataclass(kw_only=True) +@dataclass class AutomaticSpeechRecognitionWord: text: str start: int @@ -13,7 +13,7 @@ class AutomaticSpeechRecognitionWord: confidence: float -@dataclass(kw_only=True) +@dataclass class AutomaticSpeechRecognitionSegment: id: int seek: int @@ -29,7 +29,7 @@ class AutomaticSpeechRecognitionSegment: words: List[AutomaticSpeechRecognitionWord] -@dataclass(kw_only=True) +@dataclass class AutomaticSpeechRecognitionResponse: text: str segments: List[AutomaticSpeechRecognitionSegment] diff --git a/deepinfra/types/common/inference_status.py b/deepinfra/types/common/inference_status.py index a73e129..fae8ca1 100644 --- a/deepinfra/types/common/inference_status.py +++ b/deepinfra/types/common/inference_status.py @@ -1,7 +1,7 @@ from dataclasses import dataclass -@dataclass(kw_only=True) +@dataclass class InferenceStatus: status: str runtime_ms: int @@ -10,7 +10,7 @@ class InferenceStatus: tokens_input: int -@dataclass(kw_only=True) +@dataclass class Status: UNKNOWN = "unknown" QUEUED = "queued" diff --git a/deepinfra/types/embeddings/request.py b/deepinfra/types/embeddings/request.py index 62a6378..4cbe992 100644 --- a/deepinfra/types/embeddings/request.py +++ b/deepinfra/types/embeddings/request.py @@ -2,7 +2,7 @@ from typing import Optional, List -@dataclass(kw_only=True) +@dataclass class EmbeddingsRequest: inputs: List[str] normalize: Optional[bool] = None diff --git a/deepinfra/types/embeddings/response.py b/deepinfra/types/embeddings/response.py index b70bf77..8ad877e 100644 --- a/deepinfra/types/embeddings/response.py +++ b/deepinfra/types/embeddings/response.py @@ -2,7 +2,7 @@ from typing import Optional, List -@dataclass(kw_only=True) +@dataclass class EmbeddingStatus: status: str runtime_ms: int @@ -11,7 +11,7 @@ class EmbeddingStatus: tokens_input: int -@dataclass(kw_only=True) +@dataclass class EmbeddingsResponse: embeddings: List[List[float]] input_tokens: int diff --git a/deepinfra/types/image_generation/request.py b/deepinfra/types/image_generation/request.py index b53fc7f..7bb423a 100644 --- a/deepinfra/types/image_generation/request.py +++ b/deepinfra/types/image_generation/request.py @@ -2,7 +2,7 @@ from typing import Optional -@dataclass(kw_only=True) +@dataclass class ImageGenerationRequest: prompt: str negative_prompt: Optional[str] = None diff --git a/deepinfra/types/image_generation/response.py b/deepinfra/types/image_generation/response.py index 34ee6e0..0702b65 100644 --- a/deepinfra/types/image_generation/response.py +++ b/deepinfra/types/image_generation/response.py @@ -4,12 +4,12 @@ from deepinfra.types.common.inference_status import InferenceStatus -@dataclass(kw_only=True) +@dataclass class Metrics: predict_time: int -@dataclass(kw_only=True) +@dataclass class ImageGenerationResponse: request_id: str inference_status: InferenceStatus diff --git a/deepinfra/types/text_generation/request.py b/deepinfra/types/text_generation/request.py index eb96282..a0941f2 100644 --- a/deepinfra/types/text_generation/request.py +++ b/deepinfra/types/text_generation/request.py @@ -2,7 +2,7 @@ from typing import Optional, List, Dict -@dataclass(kw_only=True) +@dataclass class TextGenerationRequest: input: str stream: Optional[bool] = None diff --git a/deepinfra/types/text_generation/response.py b/deepinfra/types/text_generation/response.py index e28867b..d8e0ae5 100644 --- a/deepinfra/types/text_generation/response.py +++ b/deepinfra/types/text_generation/response.py @@ -4,12 +4,12 @@ from deepinfra.types.common.inference_status import InferenceStatus -@dataclass(kw_only=True) +@dataclass class GeneratedText: generated_text: str -@dataclass(kw_only=True) +@dataclass class TextGenerationResponse: request_id: str inference_status: InferenceStatus diff --git a/requirements-dev.txt b/requirements-dev.txt index 66d4576..9d519af 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -1,2 +1,3 @@ black==23.3.0 -mypy \ No newline at end of file +mypy +types-requests \ No newline at end of file