feat: insert log supported

Portkey-AI · Dec 7, 2024 · a566175 · a566175
1 parent d510508
commit a566175
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 0 deletions.
diff --git a/portkey_ai/api_resources/apis/logs.py b/portkey_ai/api_resources/apis/logs.py
@@ -19,6 +19,29 @@ def __init__(self, client: APIClient) -> None:
         super().__init__(client)
         self.exports = Exports(client)
 
+    def create(
+        self,
+        *,
+        request: Optional[Dict[str, Any]] = None,
+        response: Optional[Dict[str, Any]] = None,
+        metadata: Optional[Dict[str, Any]] = None,
+    ) -> Any:
+        body = {
+            "request": request,
+            "response": response,
+            "metadata": metadata,
+        }
+        response = self._post(
+            f"{PortkeyApiPaths.LOGS_API}",
+            body=body,
+            params=None,
+            cast_to=None,
+            stream=False,
+            stream_cls=None,
+            headers={},
+        )
+        return response
+
 
 class Exports(APIResource):
     def __init__(self, client: APIClient) -> None:
@@ -153,6 +176,28 @@ def __init__(self, client: AsyncAPIClient) -> None:
         super().__init__(client)
         self.exports = AsyncExports(client)
 
+    async def create(
+        self,
+        *,
+        request: Optional[Dict[str, Any]] = None,
+        response: Optional[Dict[str, Any]] = None,
+        metadata: Optional[Dict[str, Any]] = None,
+    ) -> Any:
+        body = {
+            "request": request,
+            "response": response,
+            "metadata": metadata,
+        }
+        return await self._post(
+            f"{PortkeyApiPaths.LOGS_API}",
+            body=body,
+            params=None,
+            cast_to=None,
+            stream=False,
+            stream_cls=None,
+            headers={},
+        )
+
 
 class AsyncExports(AsyncAPIResource):
     def __init__(self, client: AsyncAPIClient) -> None:

diff --git a/portkey_ai/api_resources/base_client.py b/portkey_ai/api_resources/base_client.py
@@ -639,6 +639,9 @@ def _request(
             )
             return stream_response
 
+        if not cast_to:
+            return res  # type: ignore [return-value]
+
         response = (
             cast(
                 ResponseT,
@@ -1279,6 +1282,9 @@ async def _request(
             )
             return stream_response
 
+        if not cast_to:
+            return res  # type: ignore [return-value]
+
         response = (
             cast(
                 ResponseT,

diff --git a/portkey_ai/api_resources/utils.py b/portkey_ai/api_resources/utils.py
@@ -124,6 +124,7 @@ class PortkeyApiPaths(str, Enum, metaclass=MetaEnum):
     API_KEYS_API = "/api-keys"
     VIRTUAL_KEYS_API = "/virtual-keys"
     LOGS_EXPORT_API = "/logs/exports"
+    LOGS_API = "/logs"
 
 
 class Options(BaseModel):