Revert "Update client docs with new endpoint source (#126)"

This reverts commit 0881a49.
predibase · Dec 21, 2023 · 13d3c73 · 13d3c73
1 parent 0881a49
commit 13d3c73
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/clients/python/lorax/client.py b/clients/python/lorax/client.py
@@ -155,9 +155,7 @@ def generate(
         payload = resp.json()
         if resp.status_code != 200:
             raise parse_error(resp.status_code, payload)
-        if isinstance(payload, list):
-            return Response(**payload[0])
-        return Response(**payload)
+        return Response(**payload[0])
 
     def generate_stream(
         self,

diff --git a/docs/reference/python_client.md b/docs/reference/python_client.md
@@ -67,9 +67,7 @@ Example:
 ```python
 from lorax import Client
 
-# You can get your Predibase API token by going to Settings > My Profile > Generate API Token
-# You can get your Predibase Tenant short code by going to Settings > My Profile > Overview > Tenant ID
-endpoint_url = f"https://serving.app.predibase.com/{predibase_tenant_short_code}/deployments/v2/llms/{llm_deployment_name}/"
+endpoint_url = f"https://api.app.predibase.com/v1/llms/{llm_deployment_name}"
 headers = {
     "Authorization": f"Bearer {api_token}"
 }