@@ -114,6 +114,7 @@ def __init__(
114114 - `model_access_key` from `GRADIENT_MODEL_ACCESS_KEY`
115115 - `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
116116 - `agent_endpoint` from `GRADIENT_AGENT_ENDPOINT`
117+ - `inference_endpoint` from `GRADIENT_INFERENCE_ENDPOINT`
117118 """
118119 if access_token is None :
119120 if api_key is not None :
@@ -150,10 +151,7 @@ def __init__(
150151 self ._agent_endpoint = agent_endpoint
151152
152153 if inference_endpoint is None :
153- inference_endpoint = os .environ .get ("GRADIENT_INFERENCE_ENDPOINT" )
154- if inference_endpoint is None :
155- inference_endpoint = "https://inference.do-ai.run"
156-
154+ inference_endpoint = os .environ .get ("GRADIENT_INFERENCE_ENDPOINT" ) or "inference.do-ai.run"
157155 self .inference_endpoint = inference_endpoint
158156
159157 if base_url is None :
@@ -420,6 +418,7 @@ def __init__(
420418 - `model_access_key` from `GRADIENT_MODEL_ACCESS_KEY`
421419 - `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
422420 - `agent_endpoint` from `GRADIENT_AGENT_ENDPOINT`
421+ - `inference_endpoint` from `GRADIENT_INFERENCE_ENDPOINT`
423422 """
424423 if access_token is None :
425424 if api_key is not None :
@@ -455,6 +454,8 @@ def __init__(
455454 agent_endpoint = os .environ .get ("GRADIENT_AGENT_ENDPOINT" )
456455 self ._agent_endpoint = agent_endpoint
457456
457+ if inference_endpoint is None :
458+ inference_endpoint = os .environ .get ("GRADIENT_INFERENCE_ENDPOINT" ) or "inference.do-ai.run"
458459 self .inference_endpoint = inference_endpoint
459460
460461 if base_url is None :
0 commit comments