Include delete API for application and actual request timestamp (#34)

* Include delete and actual request timestamp

* Include actual time response and delete application

* Set actual request as not required

* Update status type

---------

Co-authored-by: Ramon Marques <ramon.marques@toptal.com>

Author: ramoncelestino

aimon/_base_client.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import sys
 import json
 import time
 import uuid
@@ -1982,7 +1983,6 @@ def get_python_version() -> str:
 
 def get_architecture() -> Arch:
     try:
-        python_bitness, _ = platform.architecture()
         machine = platform.machine().lower()
     except Exception:
         return "unknown"
@@ -1998,7 +1998,7 @@ def get_architecture() -> Arch:
         return "x64"
 
     # TODO: untested
-    if python_bitness == "32bit":
+    if sys.maxsize <= 2**32:
         return "x32"
 
     if machine:

aimon/_models.py

@@ -380,6 +380,8 @@ def is_basemodel(type_: type) -> bool:
 
 def is_basemodel_type(type_: type) -> TypeGuard[type[BaseModel] | type[GenericModel]]:
     origin = get_origin(type_) or type_
+    if not inspect.isclass(origin):
+        return False
     return issubclass(origin, BaseModel) or issubclass(origin, GenericModel)
 
 

aimon/decorators/analyze.py

@@ -191,10 +191,13 @@ def _run_production_analysis(self, func, args, kwargs):
             "output": result_dict['generated_text'],
             "context_docs": _context,
             "user_query": result_dict["user_query"] if 'user_query' in result_dict else "No User Query Specified",
-            "prompt": result_dict['prompt'] if 'prompt' in result_dict else "No Prompt Specified"
+            "prompt": result_dict['prompt'] if 'prompt' in result_dict else "No Prompt Specified",
         }
         if 'instructions' in result_dict:
             aimon_payload['instructions'] = result_dict['instructions']
+        if 'actual_request_timestamp' in result_dict:
+            aimon_payload["actual_request_timestamp"] = result_dict['actual_request_timestamp']
+            
         aimon_payload['config'] = self.config
         aimon_response = self.client.analyze.create(body=[aimon_payload])
         return result + (aimon_response,)

aimon/resources/applications/applications.py

@@ -7,7 +7,7 @@
 
 import httpx
 
-from ...types import application_create_params, application_retrieve_params
+from ...types import application_create_params, application_delete_params, application_retrieve_params
 from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from ..._utils import (
     maybe_transform,
@@ -41,6 +41,7 @@
 from .production.production import ProductionResource, AsyncProductionResource
 from .evaluations.evaluations import EvaluationsResource, AsyncEvaluationsResource
 from ...types.application_create_response import ApplicationCreateResponse
+from ...types.application_delete_response import ApplicationDeleteResponse
 from ...types.application_retrieve_response import ApplicationRetrieveResponse
 
 __all__ = ["ApplicationsResource", "AsyncApplicationsResource"]
@@ -166,6 +167,56 @@ def retrieve(
             cast_to=ApplicationRetrieveResponse,
         )
 
+    def delete(
+        self,
+        *,
+        name: str,
+        stage: str,
+        version: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> ApplicationDeleteResponse:
+        """
+        Delete an application by name, stage, and version
+
+        Args:
+          name: The name of the application to delete
+
+          stage: The stage of the application (e.g., production, evaluation)
+
+          version: The version of the application to delete
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._delete(
+            "/v1/application",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "name": name,
+                        "stage": stage,
+                        "version": version,
+                    },
+                    application_delete_params.ApplicationDeleteParams,
+                ),
+            ),
+            cast_to=ApplicationDeleteResponse,
+        )
+
 
 class AsyncApplicationsResource(AsyncAPIResource):
     @cached_property
@@ -287,6 +338,56 @@ async def retrieve(
             cast_to=ApplicationRetrieveResponse,
         )
 
+    async def delete(
+        self,
+        *,
+        name: str,
+        stage: str,
+        version: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> ApplicationDeleteResponse:
+        """
+        Delete an application by name, stage, and version
+
+        Args:
+          name: The name of the application to delete
+
+          stage: The stage of the application (e.g., production, evaluation)
+
+          version: The version of the application to delete
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._delete(
+            "/v1/application",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "name": name,
+                        "stage": stage,
+                        "version": version,
+                    },
+                    application_delete_params.ApplicationDeleteParams,
+                ),
+            ),
+            cast_to=ApplicationDeleteResponse,
+        )
+
 
 class ApplicationsResourceWithRawResponse:
     def __init__(self, applications: ApplicationsResource) -> None:
@@ -298,6 +399,9 @@ def __init__(self, applications: ApplicationsResource) -> None:
         self.retrieve = to_raw_response_wrapper(
             applications.retrieve,
         )
+        self.delete = to_raw_response_wrapper(
+            applications.delete,
+        )
 
     @cached_property
     def evaluations(self) -> EvaluationsResourceWithRawResponse:
@@ -318,6 +422,9 @@ def __init__(self, applications: AsyncApplicationsResource) -> None:
         self.retrieve = async_to_raw_response_wrapper(
             applications.retrieve,
         )
+        self.delete = async_to_raw_response_wrapper(
+            applications.delete,
+        )
 
     @cached_property
     def evaluations(self) -> AsyncEvaluationsResourceWithRawResponse:
@@ -338,6 +445,9 @@ def __init__(self, applications: ApplicationsResource) -> None:
         self.retrieve = to_streamed_response_wrapper(
             applications.retrieve,
         )
+        self.delete = to_streamed_response_wrapper(
+            applications.delete,
+        )
 
     @cached_property
     def evaluations(self) -> EvaluationsResourceWithStreamingResponse:
@@ -358,6 +468,9 @@ def __init__(self, applications: AsyncApplicationsResource) -> None:
         self.retrieve = async_to_streamed_response_wrapper(
             applications.retrieve,
         )
+        self.delete = async_to_streamed_response_wrapper(
+            applications.delete,
+        )
 
     @cached_property
     def evaluations(self) -> AsyncEvaluationsResourceWithStreamingResponse:

aimon/types/__init__.py

@@ -19,10 +19,12 @@
 from .model_retrieve_response import ModelRetrieveResponse as ModelRetrieveResponse
 from .evaluation_create_params import EvaluationCreateParams as EvaluationCreateParams
 from .application_create_params import ApplicationCreateParams as ApplicationCreateParams
+from .application_delete_params import ApplicationDeleteParams as ApplicationDeleteParams
 from .inference_detect_response import InferenceDetectResponse as InferenceDetectResponse
 from .evaluation_create_response import EvaluationCreateResponse as EvaluationCreateResponse
 from .evaluation_retrieve_params import EvaluationRetrieveParams as EvaluationRetrieveParams
 from .application_create_response import ApplicationCreateResponse as ApplicationCreateResponse
+from .application_delete_response import ApplicationDeleteResponse as ApplicationDeleteResponse
 from .application_retrieve_params import ApplicationRetrieveParams as ApplicationRetrieveParams
 from .evaluation_retrieve_response import EvaluationRetrieveResponse as EvaluationRetrieveResponse
 from .application_retrieve_response import ApplicationRetrieveResponse as ApplicationRetrieveResponse

aimon/types/analyze_create_params.py

@@ -2,8 +2,11 @@
 
 from __future__ import annotations
 
-from typing import List, Iterable, Optional
-from typing_extensions import Required, TypedDict
+from typing import List, Union, Iterable, Optional
+from datetime import datetime
+from typing_extensions import Required, Annotated, TypedDict
+
+from .._utils import PropertyInfo
 
 __all__ = ["AnalyzeCreateParams", "Body"]
 
@@ -25,6 +28,9 @@ class Body(TypedDict, total=False):
 
     version: Required[str]
 
+    actual_request_timestamp: Annotated[Union[str, datetime], PropertyInfo(format="iso8601")]
+    """The timestamp when the actual request was made"""
+
     evaluation_id: Optional[str]
 
     evaluation_run_id: Optional[str]

aimon/types/analyze_create_response.py

@@ -9,3 +9,5 @@
 
 class AnalyzeCreateResponse(BaseModel):
     message: Optional[str] = None
+
+    status: int

aimon/types/application_delete_params.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+__all__ = ["ApplicationDeleteParams"]
+
+
+class ApplicationDeleteParams(TypedDict, total=False):
+    name: Required[str]
+    """The name of the application to delete"""
+
+    stage: Required[str]
+    """The stage of the application (e.g., production, evaluation)"""
+
+    version: Required[str]
+    """The version of the application to delete"""

aimon/types/application_delete_response.py

@@ -0,0 +1,42 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
+
+__all__ = ["ApplicationDeleteResponse", "Application"]
+
+
+class Application(BaseModel):
+    api_model_name: str = FieldInfo(alias="model_name")
+
+    name: str
+
+    type: str
+
+    id: Optional[str] = None
+
+    company_id: Optional[str] = None
+
+    context_source_id: Optional[str] = None
+
+    last_query_timestamp: Optional[datetime] = None
+
+    metadata: Optional[object] = None
+
+    api_model_id: Optional[str] = FieldInfo(alias="model_id", default=None)
+
+    stage: Optional[str] = None
+
+    user_id: Optional[str] = None
+
+    version: Optional[str] = None
+
+
+class ApplicationDeleteResponse(BaseModel):
+    application: Optional[Application] = None
+
+    message: Optional[str] = None