fix: restore inference base urls (#122)

Author: bbatha

src/gradient/resources/agents/chat/completions.py

@@ -472,7 +472,9 @@ def create(
         headers = {"Authorization": f"Bearer {self._client.agent_access_key}", **headers}
 
         return self._post(
-            "/chat/completions?agent=true",
+            "/chat/completions?agent=true"
+            if self._client._base_url_overridden
+            else f"{self._client.agent_endpoint}/api/v1/chat/completions?agent=true",
             body=maybe_transform(
                 {
                     "messages": messages,
@@ -956,7 +958,9 @@ async def create(
         headers = {"Authorization": f"Bearer {self._client.agent_access_key}", **headers}
 
         return await self._post(
-            "/chat/completions?agent=true",
+            "/chat/completions?agent=true"
+            if self._client._base_url_overridden
+            else f"{self._client.agent_endpoint}/api/v1/chat/completions?agent=true",
             body=await async_maybe_transform(
                 {
                     "messages": messages,

src/gradient/resources/chat/completions.py

@@ -473,7 +473,9 @@ def create(
         }
 
         return self._post(
-            "/chat/completions",
+            "/chat/completions"
+            if self._client._base_url_overridden
+            else f"{self._client.inference_endpoint}/v1/chat/completions",
             body=maybe_transform(
                 {
                     "messages": messages,
@@ -961,7 +963,9 @@ async def create(
         }
 
         return await self._post(
-            "/chat/completions",
+            "/chat/completions"
+            if self._client._base_url_overridden
+            else f"{self._client.inference_endpoint}/v1/chat/completions",
             body=await async_maybe_transform(
                 {
                     "messages": messages,

src/gradient/resources/images.py

@@ -317,7 +317,9 @@ def generate(
         }
 
         return self._post(
-            "/images/generations",
+            "/images/generations"
+            if self._client._base_url_overridden
+            else f"{self._client.inference_endpoint}/v1/images/generations",
             body=maybe_transform(
                 {
                     "prompt": prompt,
@@ -636,7 +638,9 @@ async def generate(
             **headers,
         }
         return await self._post(
-            "/images/generations",
+            "/images/generations"
+            if self._client._base_url_overridden
+            else f"{self._client.inference_endpoint}/v1/images/generations",
             body=await async_maybe_transform(
                 {
                     "prompt": prompt,