@@ -91,7 +91,7 @@ def __init__(
9191 model_access_key : str | None = None ,
9292 agent_access_key : str | None = None ,
9393 agent_endpoint : str | None = None ,
94- inference_endpoint : str | None = None ,
94+ inference_endpoint : str | None = "inference.do-ai.run" ,
9595 kbass_endpoint : str | None = None ,
9696 base_url : str | httpx .URL | None = None ,
9797 timeout : float | Timeout | None | NotGiven = not_given ,
@@ -119,7 +119,6 @@ def __init__(
119119 - `model_access_key` from `GRADIENT_MODEL_ACCESS_KEY`
120120 - `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
121121 - `agent_endpoint` from `GRADIENT_AGENT_ENDPOINT`
122- - `inference_endpoint` from `GRADIENT_INFERENCE_ENDPOINT`
123122 - `kbass_endpoint` from `GRADIENT_KBASS_ENDPOINT`
124123 """
125124 if access_token is None :
@@ -139,7 +138,7 @@ def __init__(
139138 self .agent_endpoint = agent_endpoint
140139
141140 if inference_endpoint is None :
142- inference_endpoint = os . environ . get ( "GRADIENT_INFERENCE_ENDPOINT" ) or "inference.do-ai.run"
141+ inference_endpoint = "inference.do-ai.run"
143142 self .inference_endpoint = inference_endpoint
144143
145144 if kbass_endpoint is None :
@@ -469,7 +468,7 @@ def __init__(
469468 model_access_key : str | None = None ,
470469 agent_access_key : str | None = None ,
471470 agent_endpoint : str | None = None ,
472- inference_endpoint : str | None = None ,
471+ inference_endpoint : str | None = "inference.do-ai.run" ,
473472 kbass_endpoint : str | None = None ,
474473 base_url : str | httpx .URL | None = None ,
475474 timeout : float | Timeout | None | NotGiven = not_given ,
@@ -497,7 +496,6 @@ def __init__(
497496 - `model_access_key` from `GRADIENT_MODEL_ACCESS_KEY`
498497 - `agent_access_key` from `GRADIENT_AGENT_ACCESS_KEY`
499498 - `agent_endpoint` from `GRADIENT_AGENT_ENDPOINT`
500- - `inference_endpoint` from `GRADIENT_INFERENCE_ENDPOINT`
501499 - `kbass_endpoint` from `GRADIENT_KBASS_ENDPOINT`
502500 """
503501 if access_token is None :
@@ -517,7 +515,7 @@ def __init__(
517515 self .agent_endpoint = agent_endpoint
518516
519517 if inference_endpoint is None :
520- inference_endpoint = os . environ . get ( "GRADIENT_INFERENCE_ENDPOINT" ) or "inference.do-ai.run"
518+ inference_endpoint = "inference.do-ai.run"
521519 self .inference_endpoint = inference_endpoint
522520
523521 if kbass_endpoint is None :
0 commit comments