Merge pull request #15 from aurelio-labs/simonas/embeddings-endpoint

chore: updated embeddings base url
aurelio-labs · Oct 4, 2024 · 6182e81 · 6182e81
2 parents 2a88388 + 4c77155
commit 6182e81
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 14 deletions.
diff --git a/aurelio_sdk/client.py b/aurelio_sdk/client.py
@@ -347,14 +347,8 @@ def embedding(
         Raises:
             APIError: If the API request fails.
             APITimeoutError: If the request exceeds the specified timeout.
-
-        Note:
-            This method currently uses a staging API endpoint.
-            TODO: Change to production endpoint.
         """
-        # client_url = f"{self.base_url}/v1/embeddings"
-        # TODO: change to prod
-        client_url = "https://staging.api.aurelio.ai/v1/embeddings"
+        client_url = f"{self.base_url}/v1/embeddings"
         data = {"input": input, "model": model}
         try:
             response = requests.post(

diff --git a/aurelio_sdk/client_async.py b/aurelio_sdk/client_async.py
@@ -364,14 +364,8 @@ async def embedding(
         Raises:
             APIError: If the API request fails.
             APITimeoutError: If the request exceeds the specified timeout.
-
-        Note:
-            This method currently uses a staging API endpoint.
-            TODO: Change to production endpoint.
         """
-        # client_url = f"{self.base_url}/v1/embeddings"
-        # TODO: change to prod
-        client_url = "https://staging.api.aurelio.ai/v1/embeddings"
+        client_url = f"{self.base_url}/v1/embeddings"
         data = {"input": input, "model": model}
 
         session_timeout = aiohttp.ClientTimeout(total=timeout)