Skip to content

Commit 54ccbd4

Browse files
authored
Merge pull request #20 from orq-ai/fix-global-hook
fix: fix global hook for contact_id global parameter
2 parents 96e6260 + 7a8b4e6 commit 54ccbd4

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,13 @@
77

88
class GlobalHook(BeforeRequestHook):
99
def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request) -> Union[httpx.Request, Exception]:
10-
contact_id = request.headers['contactId']
10+
contact_id = request.headers['contactid'] if 'contactid' in request.headers else None
1111

1212
if contact_id:
13-
del request.headers['contactId']
13+
del request.headers['contactid']
1414
request.headers['X-ORQ-CONTACT-ID'] = contact_id
1515

16-
environment = request.headers['environment']
16+
environment = request.headers['environment'] if 'environment' in request.headers else None
1717

1818
if hook_ctx.operation_id in ["DeploymentInvoke", "DeploymentStream"] and environment:
1919
del request.headers['environment']

src/orq_ai_sdk/_hooks/globalhook.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,13 @@
77

88
class GlobalHook(BeforeRequestHook):
99
def before_request(self, hook_ctx: BeforeRequestContext, request: httpx.Request) -> Union[httpx.Request, Exception]:
10-
contact_id = request.headers['contactId']
10+
contact_id = request.headers['contactid'] if 'contactid' in request.headers else None
1111

1212
if contact_id:
13-
del request.headers['contactId']
13+
del request.headers['contactid']
1414
request.headers['X-ORQ-CONTACT-ID'] = contact_id
1515

16-
environment = request.headers['environment']
16+
environment = request.headers['environment'] if 'environment' in request.headers else None
1717

1818
if hook_ctx.operation_id in ["DeploymentInvoke", "DeploymentStream"] and environment:
1919
del request.headers['environment']

0 commit comments

Comments
 (0)