Merge pull request #77 from Chainlit/wd/follow-redirects

fix: allow follow redirection
Chainlit · Jun 3, 2024 · 8074b9b · 8074b9b
2 parents 8e32938 + c4f373e
commit 8074b9b
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/literalai/api/__init__.py b/literalai/api/__init__.py
@@ -182,7 +182,7 @@ def raise_error(error):
             raise Exception(error)
 
         variables = self._prepare_variables(variables)
-        with httpx.Client() as client:
+        with httpx.Client(follow_redirects=True) as client:
             response = client.post(
                 self.graphql_endpoint,
                 json={"query": query, "variables": variables},
@@ -225,7 +225,7 @@ def make_rest_call(self, subpath: str, body: Dict[str, Any]) -> Dict:
         Returns:
             Dict: The JSON response from the REST API endpoint.
         """
-        with httpx.Client() as client:
+        with httpx.Client(follow_redirects=True) as client:
             response = client.post(
                 self.rest_endpoint + subpath,
                 json=body,
@@ -643,7 +643,7 @@ def upload_file(
 
         path = "/api/upload/file"
 
-        with httpx.Client() as client:
+        with httpx.Client(follow_redirects=True) as client:
             response = client.post(
                 f"{self.url}{path}",
                 json=body,
@@ -679,7 +679,7 @@ def upload_file(
         # Note: The content_type parameter is not needed here, as the correct MIME type should be set in the 'Content-Type' field from upload_details
         form_data["file"] = (id, content, mime)
 
-        with httpx.Client() as client:
+        with httpx.Client(follow_redirects=True) as client:
             if upload_type == "raw":
                 upload_response = client.request(
                     url=url,
@@ -1364,7 +1364,7 @@ def raise_error(error):
 
         variables = self._prepare_variables(variables)
 
-        async with httpx.AsyncClient() as client:
+        async with httpx.AsyncClient(follow_redirects=True) as client:
             response = await client.post(
                 self.graphql_endpoint,
                 json={"query": query, "variables": variables},
@@ -1407,7 +1407,7 @@ async def make_rest_call(self, subpath: str, body: Dict[str, Any]) -> Dict:
         Returns:
             Dict: The JSON response from the REST API endpoint.
         """
-        async with httpx.AsyncClient() as client:
+        async with httpx.AsyncClient(follow_redirects=True) as client:
             response = await client.post(
                 self.rest_endpoint + subpath,
                 json=body,
@@ -1852,7 +1852,7 @@ async def upload_file(
 
         path = "/api/upload/file"
 
-        async with httpx.AsyncClient() as client:
+        async with httpx.AsyncClient(follow_redirects=True) as client:
             response = await client.post(
                 f"{self.url}{path}",
                 json=body,
@@ -1888,7 +1888,7 @@ async def upload_file(
         # Note: The content_type parameter is not needed here, as the correct MIME type should be set in the 'Content-Type' field from upload_details
         form_data["file"] = (id, content, mime)
 
-        async with httpx.AsyncClient() as client:
+        async with httpx.AsyncClient(follow_redirects=True) as client:
             if upload_type == "raw":
                 upload_response = await client.request(
                     url=url,

diff --git a/literalai/version.py b/literalai/version.py
@@ -1 +1 @@
-__version__ = "0.0.604"
+__version__ = "0.0.605"
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="literalai",
-    version="0.0.604",  # update version in literalai/version.py
+    version="0.0.605",  # update version in literalai/version.py
     description="An SDK for observability in Python applications",
     author="Literal AI",
     package_data={"literalai": ["py.typed"]},