lambdadb 0.1.2__py3-none-any.whl → 0.2.0__py3-none-any.whl

lambdadb/_hooks/__init__.py

@@ -2,4 +2,3 @@
 
 from .sdkhooks import *
 from .types import *
-from .registration import *

lambdadb/_hooks/sdkhooks.py

@@ -11,9 +11,8 @@ from .types import (
     AfterErrorHook,
     Hooks,
 )
-from .registration import init_hooks
 from typing import List, Optional, Tuple
-from lambdadb.httpclient import HttpClient
+from lambdadb.sdkconfiguration import SDKConfiguration
 
 
 class SDKHooks(Hooks):
@@ -22,7 +21,6 @@ class SDKHooks(Hooks):
         self.before_request_hooks: List[BeforeRequestHook] = []
         self.after_success_hooks: List[AfterSuccessHook] = []
         self.after_error_hooks: List[AfterErrorHook] = []
-        init_hooks(self)
 
     def register_sdk_init_hook(self, hook: SDKInitHook) -> None:
         self.sdk_init_hooks.append(hook)
@@ -36,10 +34,10 @@ class SDKHooks(Hooks):
     def register_after_error_hook(self, hook: AfterErrorHook) -> None:
         self.after_error_hooks.append(hook)
 
-    def sdk_init(self, base_url: str, client: HttpClient) -> Tuple[str, HttpClient]:
+    def sdk_init(self, config: SDKConfiguration) -> SDKConfiguration:
         for hook in self.sdk_init_hooks:
-            base_url, client = hook.sdk_init(base_url, client)
-        return base_url, client
+            config = hook.sdk_init(config)
+        return config
 
     def before_request(
         self, hook_ctx: BeforeRequestContext, request: httpx.Request

lambdadb/_hooks/types.py

@@ -2,11 +2,12 @@
 
 from abc import ABC, abstractmethod
 import httpx
-from lambdadb.httpclient import HttpClient
+from lambdadb.sdkconfiguration import SDKConfiguration
 from typing import Any, Callable, List, Optional, Tuple, Union
 
 
 class HookContext:
+    config: SDKConfiguration
     base_url: str
     operation_id: str
     oauth2_scopes: Optional[List[str]] = None
@@ -14,11 +15,13 @@ class HookContext:
 
     def __init__(
         self,
+        config: SDKConfiguration,
         base_url: str,
         operation_id: str,
         oauth2_scopes: Optional[List[str]],
         security_source: Optional[Union[Any, Callable[[], Any]]],
     ):
+        self.config = config
         self.base_url = base_url
         self.operation_id = operation_id
         self.oauth2_scopes = oauth2_scopes
@@ -28,6 +31,7 @@ class HookContext:
 class BeforeRequestContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,
@@ -38,6 +42,7 @@ class BeforeRequestContext(HookContext):
 class AfterSuccessContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,
@@ -48,6 +53,7 @@ class AfterSuccessContext(HookContext):
 class AfterErrorContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,
@@ -57,7 +63,7 @@ class AfterErrorContext(HookContext):
 
 class SDKInitHook(ABC):
     @abstractmethod
-    def sdk_init(self, base_url: str, client: HttpClient) -> Tuple[str, HttpClient]:
+    def sdk_init(self, config: SDKConfiguration) -> SDKConfiguration:
         pass
 
 

lambdadb/_version.py

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "lambdadb"
-__version__: str = "0.1.2"
-__openapi_doc_version__: str = "1.0.0"
-__gen_version__: str = "2.610.0"
-__user_agent__: str = "speakeasy-sdk/python 0.1.2 2.610.0 1.0.0 lambdadb"
+__version__: str = "0.2.0"
+__openapi_doc_version__: str = "1.1.0"
+__gen_version__: str = "2.628.0"
+__user_agent__: str = "speakeasy-sdk/python 0.2.0 2.628.0 1.1.0 lambdadb"
 
 try:
     if __package__ is not None:

lambdadb/basesdk.py

@@ -214,12 +214,12 @@ class BaseSDK:
         client = self.sdk_configuration.client
         logger = self.sdk_configuration.debug_logger
 
+        hooks = self.sdk_configuration.__dict__["_hooks"]
+
         def do():
             http_res = None
             try:
-                req = self.sdk_configuration.get_hooks().before_request(
-                    BeforeRequestContext(hook_ctx), request
-                )
+                req = hooks.before_request(BeforeRequestContext(hook_ctx), request)
                 logger.debug(
                     "Request:\nMethod: %s\nURL: %s\nHeaders: %s\nBody: %s",
                     req.method,
@@ -233,9 +233,7 @@ class BaseSDK:
 
                 http_res = client.send(req, stream=stream)
             except Exception as e:
-                _, e = self.sdk_configuration.get_hooks().after_error(
-                    AfterErrorContext(hook_ctx), None, e
-                )
+                _, e = hooks.after_error(AfterErrorContext(hook_ctx), None, e)
                 if e is not None:
                     logger.debug("Request Exception", exc_info=True)
                     raise e
@@ -253,7 +251,7 @@ class BaseSDK:
             )
 
             if utils.match_status_codes(error_status_codes, http_res.status_code):
-                result, err = self.sdk_configuration.get_hooks().after_error(
+                result, err = hooks.after_error(
                     AfterErrorContext(hook_ctx), http_res, None
                 )
                 if err is not None:
@@ -273,9 +271,7 @@ class BaseSDK:
             http_res = do()
 
         if not utils.match_status_codes(error_status_codes, http_res.status_code):
-            http_res = self.sdk_configuration.get_hooks().after_success(
-                AfterSuccessContext(hook_ctx), http_res
-            )
+            http_res = hooks.after_success(AfterSuccessContext(hook_ctx), http_res)
 
         return http_res
 
@@ -290,12 +286,12 @@ class BaseSDK:
         client = self.sdk_configuration.async_client
         logger = self.sdk_configuration.debug_logger
 
+        hooks = self.sdk_configuration.__dict__["_hooks"]
+
         async def do():
             http_res = None
             try:
-                req = self.sdk_configuration.get_hooks().before_request(
-                    BeforeRequestContext(hook_ctx), request
-                )
+                req = hooks.before_request(BeforeRequestContext(hook_ctx), request)
                 logger.debug(
                     "Request:\nMethod: %s\nURL: %s\nHeaders: %s\nBody: %s",
                     req.method,
@@ -309,9 +305,7 @@ class BaseSDK:
 
                 http_res = await client.send(req, stream=stream)
             except Exception as e:
-                _, e = self.sdk_configuration.get_hooks().after_error(
-                    AfterErrorContext(hook_ctx), None, e
-                )
+                _, e = hooks.after_error(AfterErrorContext(hook_ctx), None, e)
                 if e is not None:
                     logger.debug("Request Exception", exc_info=True)
                     raise e
@@ -329,7 +323,7 @@ class BaseSDK:
             )
 
             if utils.match_status_codes(error_status_codes, http_res.status_code):
-                result, err = self.sdk_configuration.get_hooks().after_error(
+                result, err = hooks.after_error(
                     AfterErrorContext(hook_ctx), http_res, None
                 )
                 if err is not None:
@@ -351,8 +345,6 @@ class BaseSDK:
             http_res = await do()
 
         if not utils.match_status_codes(error_status_codes, http_res.status_code):
-            http_res = self.sdk_configuration.get_hooks().after_success(
-                AfterSuccessContext(hook_ctx), http_res
-            )
+            http_res = hooks.after_success(AfterSuccessContext(hook_ctx), http_res)
 
         return http_res

lambdadb/collections.py

@@ -82,6 +82,7 @@ class Collections(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="listcollections",
                 oauth2_scopes=[],
@@ -198,6 +199,7 @@ class Collections(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="listcollections",
                 oauth2_scopes=[],
@@ -348,6 +350,7 @@ class Collections(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="createCollection",
                 oauth2_scopes=[],
@@ -503,6 +506,7 @@ class Collections(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="createCollection",
                 oauth2_scopes=[],
@@ -627,6 +631,7 @@ class Collections(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="deleteCollection",
                 oauth2_scopes=[],
@@ -746,6 +751,7 @@ class Collections(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="deleteCollection",
                 oauth2_scopes=[],
@@ -865,6 +871,7 @@ class Collections(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="getCollection",
                 oauth2_scopes=[],
@@ -984,6 +991,7 @@ class Collections(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="getCollection",
                 oauth2_scopes=[],
@@ -1120,6 +1128,7 @@ class Collections(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="updateCollection",
                 oauth2_scopes=[],
@@ -1261,6 +1270,7 @@ class Collections(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="updateCollection",
                 oauth2_scopes=[],
@@ -1415,6 +1425,7 @@ class Collections(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="queryCollection",
                 oauth2_scopes=[],
@@ -1569,6 +1580,7 @@ class Collections(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="queryCollection",
                 oauth2_scopes=[],

lambdadb/docs.py

@@ -81,6 +81,7 @@ class Docs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="upsertDocs",
                 oauth2_scopes=[],
@@ -213,6 +214,7 @@ class Docs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="upsertDocs",
                 oauth2_scopes=[],
@@ -337,6 +339,7 @@ class Docs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="getBulkUpsertDocs",
                 oauth2_scopes=[],
@@ -456,6 +459,7 @@ class Docs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="getBulkUpsertDocs",
                 oauth2_scopes=[],
@@ -587,6 +591,7 @@ class Docs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="bulkUpsertDocs",
                 oauth2_scopes=[],
@@ -723,6 +728,7 @@ class Docs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="bulkUpsertDocs",
                 oauth2_scopes=[],
@@ -857,6 +863,7 @@ class Docs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="deleteDocs",
                 oauth2_scopes=[],
@@ -991,6 +998,7 @@ class Docs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="deleteDocs",
                 oauth2_scopes=[],
@@ -1129,6 +1137,7 @@ class Docs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="fetchDocs",
                 oauth2_scopes=[],
@@ -1267,6 +1276,7 @@ class Docs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="fetchDocs",
                 oauth2_scopes=[],

lambdadb/models/__init__.py

@@ -19,12 +19,6 @@ if TYPE_CHECKING:
         CreateCollectionRequestBodyTypedDict,
         CreateCollectionRequestTypedDict,
     )
-    from .createprojectop import (
-        CreateProjectRequest,
-        CreateProjectRequestTypedDict,
-        CreateProjectSecurity,
-        CreateProjectSecurityTypedDict,
-    )
     from .deletecollectionop import (
         DeleteCollectionRequest,
         DeleteCollectionRequestTypedDict,
@@ -45,14 +39,6 @@ if TYPE_CHECKING:
         RequestBody2,
         RequestBody2TypedDict,
     )
-    from .deleteprojectop import (
-        DeleteProjectRequest,
-        DeleteProjectRequestTypedDict,
-        DeleteProjectResponse,
-        DeleteProjectResponseTypedDict,
-        DeleteProjectSecurity,
-        DeleteProjectSecurityTypedDict,
-    )
     from .fetchdocsop import (
         FetchDocsDoc,
         FetchDocsDocDoc,
@@ -74,12 +60,6 @@ if TYPE_CHECKING:
         HTTPMethod,
     )
     from .getcollectionop import GetCollectionRequest, GetCollectionRequestTypedDict
-    from .getprojectop import (
-        GetProjectRequest,
-        GetProjectRequestTypedDict,
-        GetProjectSecurity,
-        GetProjectSecurityTypedDict,
-    )
     from .indexconfigs_union import (
         Analyzer,
         IndexConfigs,
@@ -101,13 +81,6 @@ if TYPE_CHECKING:
         ListcollectionsResponse,
         ListcollectionsResponseTypedDict,
     )
-    from .listprojectsop import (
-        ListProjectsResponse,
-        ListProjectsResponseTypedDict,
-        ListProjectsSecurity,
-        ListProjectsSecurityTypedDict,
-    )
-    from .projectresponse import ProjectResponse, ProjectResponseTypedDict
     from .querycollectionop import (
         Query,
         QueryCollectionDoc,
@@ -132,14 +105,6 @@ if TYPE_CHECKING:
         UpdateCollectionRequestBodyTypedDict,
         UpdateCollectionRequestTypedDict,
     )
-    from .updateprojectop import (
-        UpdateProjectRequest,
-        UpdateProjectRequestBody,
-        UpdateProjectRequestBodyTypedDict,
-        UpdateProjectRequestTypedDict,
-        UpdateProjectSecurity,
-        UpdateProjectSecurityTypedDict,
-    )
     from .upsertdocsop import (
         UpsertDocsDoc,
         UpsertDocsDocTypedDict,
@@ -165,10 +130,6 @@ __all__ = [
     "CreateCollectionRequestBody",
     "CreateCollectionRequestBodyTypedDict",
     "CreateCollectionRequestTypedDict",
-    "CreateProjectRequest",
-    "CreateProjectRequestTypedDict",
-    "CreateProjectSecurity",
-    "CreateProjectSecurityTypedDict",
     "DeleteCollectionRequest",
     "DeleteCollectionRequestTypedDict",
     "DeleteCollectionResponse",
@@ -179,12 +140,6 @@ __all__ = [
     "DeleteDocsRequestTypedDict",
     "DeleteDocsResponse",
     "DeleteDocsResponseTypedDict",
-    "DeleteProjectRequest",
-    "DeleteProjectRequestTypedDict",
-    "DeleteProjectResponse",
-    "DeleteProjectResponseTypedDict",
-    "DeleteProjectSecurity",
-    "DeleteProjectSecurityTypedDict",
     "FetchDocsDoc",
     "FetchDocsDocDoc",
     "FetchDocsDocDocTypedDict",
@@ -204,10 +159,6 @@ __all__ = [
     "GetBulkUpsertDocsType",
     "GetCollectionRequest",
     "GetCollectionRequestTypedDict",
-    "GetProjectRequest",
-    "GetProjectRequestTypedDict",
-    "GetProjectSecurity",
-    "GetProjectSecurityTypedDict",
     "HTTPMethod",
     "IndexConfigs",
     "IndexConfigsText",
@@ -217,16 +168,10 @@ __all__ = [
     "IndexConfigsUnionTypedDict",
     "IndexConfigsVector",
     "IndexConfigsVectorTypedDict",
-    "ListProjectsResponse",
-    "ListProjectsResponseTypedDict",
-    "ListProjectsSecurity",
-    "ListProjectsSecurityTypedDict",
     "ListcollectionsRequest",
     "ListcollectionsRequestTypedDict",
     "ListcollectionsResponse",
     "ListcollectionsResponseTypedDict",
-    "ProjectResponse",
-    "ProjectResponseTypedDict",
     "Query",
     "QueryCollectionDoc",
     "QueryCollectionDocDoc",
@@ -256,12 +201,6 @@ __all__ = [
     "UpdateCollectionRequestBody",
     "UpdateCollectionRequestBodyTypedDict",
     "UpdateCollectionRequestTypedDict",
-    "UpdateProjectRequest",
-    "UpdateProjectRequestBody",
-    "UpdateProjectRequestBodyTypedDict",
-    "UpdateProjectRequestTypedDict",
-    "UpdateProjectSecurity",
-    "UpdateProjectSecurityTypedDict",
     "UpsertDocsDoc",
     "UpsertDocsDocTypedDict",
     "UpsertDocsRequest",
@@ -285,10 +224,6 @@ _dynamic_imports: dict[str, str] = {
     "CreateCollectionRequestBody": ".createcollectionop",
     "CreateCollectionRequestBodyTypedDict": ".createcollectionop",
     "CreateCollectionRequestTypedDict": ".createcollectionop",
-    "CreateProjectRequest": ".createprojectop",
-    "CreateProjectRequestTypedDict": ".createprojectop",
-    "CreateProjectSecurity": ".createprojectop",
-    "CreateProjectSecurityTypedDict": ".createprojectop",
     "DeleteCollectionRequest": ".deletecollectionop",
     "DeleteCollectionRequestTypedDict": ".deletecollectionop",
     "DeleteCollectionResponse": ".deletecollectionop",
@@ -305,12 +240,6 @@ _dynamic_imports: dict[str, str] = {
     "RequestBody1TypedDict": ".deletedocsop",
     "RequestBody2": ".deletedocsop",
     "RequestBody2TypedDict": ".deletedocsop",
-    "DeleteProjectRequest": ".deleteprojectop",
-    "DeleteProjectRequestTypedDict": ".deleteprojectop",
-    "DeleteProjectResponse": ".deleteprojectop",
-    "DeleteProjectResponseTypedDict": ".deleteprojectop",
-    "DeleteProjectSecurity": ".deleteprojectop",
-    "DeleteProjectSecurityTypedDict": ".deleteprojectop",
     "FetchDocsDoc": ".fetchdocsop",
     "FetchDocsDocDoc": ".fetchdocsop",
     "FetchDocsDocDocTypedDict": ".fetchdocsop",
@@ -329,10 +258,6 @@ _dynamic_imports: dict[str, str] = {
     "HTTPMethod": ".getbulkupsertdocsop",
     "GetCollectionRequest": ".getcollectionop",
     "GetCollectionRequestTypedDict": ".getcollectionop",
-    "GetProjectRequest": ".getprojectop",
-    "GetProjectRequestTypedDict": ".getprojectop",
-    "GetProjectSecurity": ".getprojectop",
-    "GetProjectSecurityTypedDict": ".getprojectop",
     "Analyzer": ".indexconfigs_union",
     "IndexConfigs": ".indexconfigs_union",
     "IndexConfigsText": ".indexconfigs_union",
@@ -350,12 +275,6 @@ _dynamic_imports: dict[str, str] = {
     "ListcollectionsRequestTypedDict": ".listcollectionsop",
     "ListcollectionsResponse": ".listcollectionsop",
     "ListcollectionsResponseTypedDict": ".listcollectionsop",
-    "ListProjectsResponse": ".listprojectsop",
-    "ListProjectsResponseTypedDict": ".listprojectsop",
-    "ListProjectsSecurity": ".listprojectsop",
-    "ListProjectsSecurityTypedDict": ".listprojectsop",
-    "ProjectResponse": ".projectresponse",
-    "ProjectResponseTypedDict": ".projectresponse",
     "Query": ".querycollectionop",
     "QueryCollectionDoc": ".querycollectionop",
     "QueryCollectionDocDoc": ".querycollectionop",
@@ -377,12 +296,6 @@ _dynamic_imports: dict[str, str] = {
     "UpdateCollectionRequestBody": ".updatecollectionop",
     "UpdateCollectionRequestBodyTypedDict": ".updatecollectionop",
     "UpdateCollectionRequestTypedDict": ".updatecollectionop",
-    "UpdateProjectRequest": ".updateprojectop",
-    "UpdateProjectRequestBody": ".updateprojectop",
-    "UpdateProjectRequestBodyTypedDict": ".updateprojectop",
-    "UpdateProjectRequestTypedDict": ".updateprojectop",
-    "UpdateProjectSecurity": ".updateprojectop",
-    "UpdateProjectSecurityTypedDict": ".updateprojectop",
     "UpsertDocsDoc": ".upsertdocsop",
     "UpsertDocsDocTypedDict": ".upsertdocsop",
     "UpsertDocsRequest": ".upsertdocsop",