diff --git a/pyproject.toml b/pyproject.toml index 7f67aaf..46eab94 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "hatchling.build" [project] name = "qualifire" -version = "0.15.0" +version = "0.16.0" description = "Qualifire Python SDK" authors = [{ name = "qualifire-dev", email = "dror@qualifire.ai" }] requires-python = ">=3.8,<4" diff --git a/qualifire/client.py b/qualifire/client.py index 035cccb..ffa1463 100644 --- a/qualifire/client.py +++ b/qualifire/client.py @@ -207,7 +207,7 @@ def evaluate( ) ``` """ # noqa E501 - url = f"{self._base_url}/api/evaluation/evaluate" + url = f"{self._base_url}/api/v1/evaluation/evaluate" request = EvaluationRequest( input=input, output=output, @@ -269,7 +269,7 @@ def invoke_evaluation( available_tools: Optional[List[LLMToolDefinition]] = None, metadata: Optional[Dict[str, str]] = None, ) -> EvaluationResponse: - url = f"{self._base_url}/api/evaluation/invoke/" + url = f"{self._base_url}/api/v1/evaluation/invoke/" if messages is not None: if isinstance(messages, list) and all( diff --git a/qualifire/consts.py b/qualifire/consts.py index 0e244fb..65034b7 100644 --- a/qualifire/consts.py +++ b/qualifire/consts.py @@ -1,3 +1,5 @@ QUALIFIRE_API_KEY_ENV_VAR = "QUALIFIRE_API_KEY" QUALIFIRE_BASE_URL_ENV_VAR = "QUALIFIRE_BASE_URL" -_DEFAULT_BASE_URL = "https://proxy.qualifire.ai/" +QUALIFIRE_TRACING_URL_ENV_VAR = "QUALIFIRE_TRACING_URL" +_DEFAULT_BASE_URL = "https://api.qualifire.ai/" +_DEFAULT_TRACING_URL = "https://tracing.qualifire.ai" diff --git a/qualifire/tracer_init.py b/qualifire/tracer_init.py index 34bbb3b..efc9f6e 100644 --- a/qualifire/tracer_init.py +++ b/qualifire/tracer_init.py @@ -10,7 +10,7 @@ except ImportError: traceloop_installed = False -from .utils import get_api_key, get_base_url +from .utils import get_api_key, get_tracing_url R = TypeVar("R") @@ -37,7 +37,7 @@ def __configure_tracer(api_key: str) -> None: __suppress_prints( Traceloop.init, app_name="qualifire-agent", - api_endpoint=f"{get_base_url()}/api/telemetry", # /v1/traces is automatically added # noqa: E501 + api_endpoint=f"{get_tracing_url()}/api/telemetry", # /v1/traces is automatically added # noqa: E501 headers={"X-Qualifire-API-Key": api_key}, telemetry_enabled=False, traceloop_sync_enabled=False, diff --git a/qualifire/utils.py b/qualifire/utils.py index 7a0cec6..d4dab24 100644 --- a/qualifire/utils.py +++ b/qualifire/utils.py @@ -2,8 +2,10 @@ from .consts import ( _DEFAULT_BASE_URL, + _DEFAULT_TRACING_URL, QUALIFIRE_API_KEY_ENV_VAR, QUALIFIRE_BASE_URL_ENV_VAR, + QUALIFIRE_TRACING_URL_ENV_VAR, ) @@ -19,3 +21,10 @@ def get_base_url() -> str: if not base_url: return _DEFAULT_BASE_URL return base_url + + +def get_tracing_url() -> str: + tracing_url = os.getenv(QUALIFIRE_TRACING_URL_ENV_VAR) + if not tracing_url: + return _DEFAULT_TRACING_URL + return tracing_url diff --git a/uv.lock b/uv.lock index bac7727..448713b 100644 --- a/uv.lock +++ b/uv.lock @@ -2771,7 +2771,7 @@ wheels = [ [[package]] name = "qualifire" -version = "0.14.0" +version = "0.16.0" source = { editable = "." } dependencies = [ { name = "pydantic", version = "2.10.6", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.9'" },