lambdadb 0.1.2__py3-none-any.whl

lambdadb/models/querycollectionop.py

@@ -0,0 +1,152 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from lambdadb.types import BaseModel
+from lambdadb.utils import FieldMetadata, PathParamMetadata, RequestMetadata
+import pydantic
+from typing import List, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class QueryTypedDict(TypedDict):
+    r"""Query object."""
+
+
+class Query(BaseModel):
+    r"""Query object."""
+
+
+class SortTypedDict(TypedDict):
+    pass
+
+
+class Sort(BaseModel):
+    pass
+
+
+class QueryCollectionRequestBodyTypedDict(TypedDict):
+    size: int
+    r"""Number of documents to return. Note that the maximum number of documents is 1000."""
+    query: NotRequired[QueryTypedDict]
+    r"""Query object."""
+    consistent_read: NotRequired[bool]
+    r"""If your application requires a strongly consistent read, set consistentRead to true. Although a strongly consistent read might take more time than an eventually consistent read, it always returns the last updated value."""
+    include_vectors: NotRequired[bool]
+    r"""If your application need to include vector values in the response, set includeVectors to true."""
+    sort: NotRequired[List[SortTypedDict]]
+    r"""List of field name, sort direction pairs."""
+    fields: NotRequired[List[str]]
+    r"""List of field name to include in results"""
+    track_scores: NotRequired[bool]
+    r"""If your application needs to track scores with sorting, set trackScores to true."""
+
+
+class QueryCollectionRequestBody(BaseModel):
+    size: int
+    r"""Number of documents to return. Note that the maximum number of documents is 1000."""
+
+    query: Optional[Query] = None
+    r"""Query object."""
+
+    consistent_read: Annotated[
+        Optional[bool], pydantic.Field(alias="consistentRead")
+    ] = False
+    r"""If your application requires a strongly consistent read, set consistentRead to true. Although a strongly consistent read might take more time than an eventually consistent read, it always returns the last updated value."""
+
+    include_vectors: Annotated[
+        Optional[bool], pydantic.Field(alias="includeVectors")
+    ] = False
+    r"""If your application need to include vector values in the response, set includeVectors to true."""
+
+    sort: Optional[List[Sort]] = None
+    r"""List of field name, sort direction pairs."""
+
+    fields: Optional[List[str]] = None
+    r"""List of field name to include in results"""
+
+    track_scores: Annotated[Optional[bool], pydantic.Field(alias="trackScores")] = False
+    r"""If your application needs to track scores with sorting, set trackScores to true."""
+
+
+class QueryCollectionRequestTypedDict(TypedDict):
+    project_name: str
+    r"""Project name."""
+    collection_name: str
+    r"""Collection name."""
+    request_body: QueryCollectionRequestBodyTypedDict
+
+
+class QueryCollectionRequest(BaseModel):
+    project_name: Annotated[
+        str,
+        pydantic.Field(alias="projectName"),
+        FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
+    ]
+    r"""Project name."""
+
+    collection_name: Annotated[
+        str,
+        pydantic.Field(alias="collectionName"),
+        FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
+    ]
+    r"""Collection name."""
+
+    request_body: Annotated[
+        QueryCollectionRequestBody,
+        FieldMetadata(request=RequestMetadata(media_type="application/json")),
+    ]
+
+
+class QueryCollectionDocDocTypedDict(TypedDict):
+    pass
+
+
+class QueryCollectionDocDoc(BaseModel):
+    pass
+
+
+class QueryCollectionDocTypedDict(TypedDict):
+    collection: NotRequired[str]
+    r"""Collection name."""
+    score: NotRequired[float]
+    r"""Document similarity score."""
+    doc: NotRequired[QueryCollectionDocDocTypedDict]
+
+
+class QueryCollectionDoc(BaseModel):
+    collection: Optional[str] = None
+    r"""Collection name."""
+
+    score: Optional[float] = None
+    r"""Document similarity score."""
+
+    doc: Optional[QueryCollectionDocDoc] = None
+
+
+class QueryCollectionResponseTypedDict(TypedDict):
+    r"""Documents selected by query."""
+
+    took: NotRequired[int]
+    r"""Elapsed time in milliseconds."""
+    max_score: NotRequired[float]
+    r"""Maximum score."""
+    total: NotRequired[int]
+    r"""Total number of documents returned."""
+    docs: NotRequired[List[QueryCollectionDocTypedDict]]
+    r"""List of documents."""
+
+
+class QueryCollectionResponse(BaseModel):
+    r"""Documents selected by query."""
+
+    took: Optional[int] = None
+    r"""Elapsed time in milliseconds."""
+
+    max_score: Annotated[Optional[float], pydantic.Field(alias="maxScore")] = None
+    r"""Maximum score."""
+
+    total: Optional[int] = None
+    r"""Total number of documents returned."""
+
+    docs: Optional[List[QueryCollectionDoc]] = None
+    r"""List of documents."""

lambdadb/models/security.py

@@ -0,0 +1,25 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from lambdadb.types import BaseModel
+from lambdadb.utils import FieldMetadata, SecurityMetadata
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class SecurityTypedDict(TypedDict):
+    project_api_key: NotRequired[str]
+
+
+class Security(BaseModel):
+    project_api_key: Annotated[
+        Optional[str],
+        FieldMetadata(
+            security=SecurityMetadata(
+                scheme=True,
+                scheme_type="apiKey",
+                sub_type="header",
+                field_name="x-api-key",
+            )
+        ),
+    ] = None

lambdadb/models/status.py

@@ -0,0 +1,12 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from enum import Enum
+
+
+class Status(str, Enum):
+    r"""Status"""
+
+    CREATING = "CREATING"
+    ACTIVE = "ACTIVE"
+    DELETING = "DELETING"

lambdadb/models/updatecollectionop.py

@@ -0,0 +1,48 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .indexconfigs_union import IndexConfigsUnion, IndexConfigsUnionTypedDict
+from lambdadb.types import BaseModel
+from lambdadb.utils import FieldMetadata, PathParamMetadata, RequestMetadata
+import pydantic
+from typing import Dict
+from typing_extensions import Annotated, TypedDict
+
+
+class UpdateCollectionRequestBodyTypedDict(TypedDict):
+    index_configs: Dict[str, IndexConfigsUnionTypedDict]
+
+
+class UpdateCollectionRequestBody(BaseModel):
+    index_configs: Annotated[
+        Dict[str, IndexConfigsUnion], pydantic.Field(alias="indexConfigs")
+    ]
+
+
+class UpdateCollectionRequestTypedDict(TypedDict):
+    project_name: str
+    r"""Project name."""
+    collection_name: str
+    r"""Collection name."""
+    request_body: UpdateCollectionRequestBodyTypedDict
+
+
+class UpdateCollectionRequest(BaseModel):
+    project_name: Annotated[
+        str,
+        pydantic.Field(alias="projectName"),
+        FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
+    ]
+    r"""Project name."""
+
+    collection_name: Annotated[
+        str,
+        pydantic.Field(alias="collectionName"),
+        FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
+    ]
+    r"""Collection name."""
+
+    request_body: Annotated[
+        UpdateCollectionRequestBody,
+        FieldMetadata(request=RequestMetadata(media_type="application/json")),
+    ]

lambdadb/models/updateprojectop.py

@@ -0,0 +1,58 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from lambdadb.types import BaseModel
+from lambdadb.utils import (
+    FieldMetadata,
+    PathParamMetadata,
+    RequestMetadata,
+    SecurityMetadata,
+)
+import pydantic
+from typing_extensions import Annotated, TypedDict
+
+
+class UpdateProjectSecurityTypedDict(TypedDict):
+    admin_api_key: str
+
+
+class UpdateProjectSecurity(BaseModel):
+    admin_api_key: Annotated[
+        str,
+        FieldMetadata(
+            security=SecurityMetadata(
+                scheme=True,
+                scheme_type="apiKey",
+                sub_type="header",
+                field_name="x-api-key",
+            )
+        ),
+    ]
+
+
+class UpdateProjectRequestBodyTypedDict(TypedDict):
+    rate_limit: float
+
+
+class UpdateProjectRequestBody(BaseModel):
+    rate_limit: Annotated[float, pydantic.Field(alias="rateLimit")]
+
+
+class UpdateProjectRequestTypedDict(TypedDict):
+    project_name: str
+    r"""Project name."""
+    request_body: UpdateProjectRequestBodyTypedDict
+
+
+class UpdateProjectRequest(BaseModel):
+    project_name: Annotated[
+        str,
+        pydantic.Field(alias="projectName"),
+        FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
+    ]
+    r"""Project name."""
+
+    request_body: Annotated[
+        UpdateProjectRequestBody,
+        FieldMetadata(request=RequestMetadata(media_type="application/json")),
+    ]

lambdadb/models/upsertdocsop.py

@@ -0,0 +1,67 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from lambdadb.types import BaseModel
+from lambdadb.utils import FieldMetadata, PathParamMetadata, RequestMetadata
+import pydantic
+from typing import List, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class UpsertDocsDocTypedDict(TypedDict):
+    pass
+
+
+class UpsertDocsDoc(BaseModel):
+    pass
+
+
+class UpsertDocsRequestBodyTypedDict(TypedDict):
+    docs: List[UpsertDocsDocTypedDict]
+    r"""A list of documents to upsert."""
+
+
+class UpsertDocsRequestBody(BaseModel):
+    docs: List[UpsertDocsDoc]
+    r"""A list of documents to upsert."""
+
+
+class UpsertDocsRequestTypedDict(TypedDict):
+    project_name: str
+    r"""Project name."""
+    collection_name: str
+    r"""Collection name."""
+    request_body: UpsertDocsRequestBodyTypedDict
+
+
+class UpsertDocsRequest(BaseModel):
+    project_name: Annotated[
+        str,
+        pydantic.Field(alias="projectName"),
+        FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
+    ]
+    r"""Project name."""
+
+    collection_name: Annotated[
+        str,
+        pydantic.Field(alias="collectionName"),
+        FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
+    ]
+    r"""Collection name."""
+
+    request_body: Annotated[
+        UpsertDocsRequestBody,
+        FieldMetadata(request=RequestMetadata(media_type="application/json")),
+    ]
+
+
+class UpsertDocsResponseTypedDict(TypedDict):
+    r"""Upsert request accepted."""
+
+    message: NotRequired[str]
+
+
+class UpsertDocsResponse(BaseModel):
+    r"""Upsert request accepted."""
+
+    message: Optional[str] = None