12593246423

Committed 03 Jan 2025 05:27AM UTC coverage: 92.904% (+12.3%) from 80.557%

Build # 12593246423

Build Type

push

github

Specific Base 329c1c

Committed by

kevdevg

Commit Message

fix: triying to wrap async headers

Run Details

10 of 30 new or added lines in 3 files covered. (33.33%)

20 existing lines in 3 files now uncovered.

1414 of 1522 relevant lines covered (92.9%)

3.71 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

93.1

/scope3ai/tracers/huggingface/utils.py

import contextlib
import contextvars

HFRS_BASEKEY = "scope3ai__huggingface__hf_raise_for_status"
HFRS_ENABLED = contextvars.ContextVar(f"{HFRS_BASEKEY}__enabled", default=None)
HFRS_VALUE = contextvars.ContextVar(f"{HFRS_BASEKEY}__value", default=None)


def hf_raise_for_status_enabled():
    return HFRS_ENABLED.get() is True


def hf_raise_for_status_wrapper(wrapped, instance, args, kwargs):
    try:
        result = wrapped(*args, **kwargs)
        response = args[0]
        HFRS_VALUE.set(response)
        return result
    except Exception as e:
        raise e


@contextlib.contextmanager
def hf_raise_for_status_capture():
    try:
        HFRS_VALUE.set(None)
        HFRS_ENABLED.set(True)
        yield HFRS_VALUE
    finally:
        HFRS_ENABLED.set(False)


def get_client_session_async(wrapped, instance, args, kwargs):
    try:
        result = wrapped(*args, **kwargs)
        print("session", result.__dir__())
        return result
    except Exception as e:
        raise e

1	import contextlib	4✔
2	import contextvars	4✔
3
4	HFRS_BASEKEY = "scope3ai__huggingface__hf_raise_for_status"	4✔
5	HFRS_ENABLED = contextvars.ContextVar(f"{HFRS_BASEKEY}__enabled", default=None)	4✔
6	HFRS_VALUE = contextvars.ContextVar(f"{HFRS_BASEKEY}__value", default=None)	4✔
7
8
9	def hf_raise_for_status_enabled():	4✔
10	return HFRS_ENABLED.get() is True	4✔
11
12
13	def hf_raise_for_status_wrapper(wrapped, instance, args, kwargs):	4✔
14	try:	4✔
15	result = wrapped(args, *kwargs)	4✔
16	response = args[0]	4✔
17	HFRS_VALUE.set(response)	4✔
18	return result	4✔
19	except Exception as e:	4✔
20	raise e	4✔
21
22
23	@contextlib.contextmanager	4✔
24	def hf_raise_for_status_capture():	4✔
25	try:	4✔
26	HFRS_VALUE.set(None)	4✔
27	HFRS_ENABLED.set(True)	4✔
28	yield HFRS_VALUE	4✔
29	finally:
30	HFRS_ENABLED.set(False)	4✔
31
32
33	def get_client_session_async(wrapped, instance, args, kwargs):	4✔
34	try:	4✔
35	result = wrapped(args, *kwargs)	4✔
36	print("session", result.__dir__())	4✔
37	return result	4✔
NEW 38	except Exception as e:	×
NEW 39	raise e	×

scope3data / scope3ai-py / 12593246423

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous