Merge pull request #17 from orq-ai/set-deployment-stream-dynamically-in-hooks

fix: set deployment stream dynamically in hooks

Author: thedevtoni

packages/orq-rc/src/orq_ai_sdk/_hooks/globalhook.py

@@ -1,11 +1,12 @@
+from typing import Union
 from .types import BeforeRequestContext, BeforeRequestHook
 
 import json
 import httpx
 
 
 class GlobalHook(BeforeRequestHook):
-    def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request) -> Union[requests.PreparedRequest, Exception]:
+    def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request) -> Union[httpx.Request, Exception]:
         contact_id = request.headers['contactId']
 
         if contact_id:
@@ -19,6 +20,11 @@ def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request)
             raw_payload = request.content.decode('utf-8')
             payload = json.loads(raw_payload)
 
+            if hook_ctx.operation_id == 'DeploymentStream':
+                payload['stream'] = True
+            else:
+                payload['stream'] = False
+
             if 'context' in payload and type(payload['context']) is dict:
                 payload['context']['environments'] = environment
             else:
@@ -33,4 +39,4 @@ def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request)
 
             return httpx.Request(method=request.method, url=request.url, extensions=request.extensions, headers=request.headers, content=data)
 
-        return request
+        return request

packages/orq-rc/src/orq_ai_sdk/_hooks/registration.py

@@ -11,4 +11,4 @@ def init_hooks(hooks: Hooks):
     """Add hooks by calling hooks.register{sdk_init/before_request/after_success/after_error}Hook
     with an instance of a hook that implements that specific Hook interface
     Hooks are registered per SDK instance, and are valid for the lifetime of the SDK instance"""
-    hooks.register_before_request_hook(GlobalHook())
+    hooks.register_before_request_hook(GlobalHook())

src/orq_ai_sdk/_hooks/globalhook.py

@@ -1,11 +1,12 @@
+from typing import Union
 from .types import BeforeRequestContext, BeforeRequestHook
 
 import json
 import httpx
 
 
 class GlobalHook(BeforeRequestHook):
-    def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request) -> Union[requests.PreparedRequest, Exception]:
+    def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request) -> Union[httpx.Request, Exception]:
         contact_id = request.headers['contactId']
 
         if contact_id:
@@ -19,6 +20,11 @@ def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request)
             raw_payload = request.content.decode('utf-8')
             payload = json.loads(raw_payload)
 
+            if hook_ctx.operation_id == 'DeploymentStream':
+                payload['stream'] = True
+            else:
+                payload['stream'] = False
+
             if 'context' in payload and type(payload['context']) is dict:
                 payload['context']['environments'] = environment
             else:
@@ -33,4 +39,4 @@ def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request)
 
             return httpx.Request(method=request.method, url=request.url, extensions=request.extensions, headers=request.headers, content=data)
 
-        return request
+        return request

src/orq_ai_sdk/_hooks/registration.py

@@ -11,4 +11,4 @@ def init_hooks(hooks: Hooks):
     """Add hooks by calling hooks.register{sdk_init/before_request/after_success/after_error}Hook
     with an instance of a hook that implements that specific Hook interface
     Hooks are registered per SDK instance, and are valid for the lifetime of the SDK instance"""
-    hooks.register_before_request_hook(GlobalHook())
+    hooks.register_before_request_hook(GlobalHook())