Fix canvas API issue when downloading file contents

see ucfopen/canvasapi#687

Author: ururk

canvas_langchain/canvas.py

@@ -560,6 +560,39 @@ def load(self) -> List[Document]:
         try:
             # Import the Canvas class
             from canvasapi import Canvas
+
+            # see: https://github.com/ucfopen/canvasapi/issues/687
+            from canvasapi.file import File
+
+            def patched_download(self, location):
+                """
+                Download the file to specified location.
+
+                :param location: The path to download to.
+                :type location: str
+                """
+                response = self._requester.request("GET", _url=self.url, use_auth=False)
+
+                with open(location, "wb") as file_out:
+                    file_out.write(response.content)
+
+
+            def patched_get_contents(self, binary=False):
+                """
+                Download the contents of this file.
+                Pass binary=True to return a bytes object instead of a str.
+
+                :rtype: str or bytes
+                """
+                response = self._requester.request("GET", _url=self.url, use_auth=False)
+                if binary:
+                    return response.content
+                else:
+                    return response.text
+
+            File.get_contents = patched_get_contents
+            File.download = patched_download
+
             from canvasapi.exceptions import CanvasException
         except ImportError as exc:
             raise ImportError(

requirements.txt

@@ -1,7 +1,7 @@
 langchain
 langchain-community
 unstructured[docx,pptx,xlsx]
-canvasapi
+canvasapi=3.3.0
 beautifulsoup4
 lxml
 PyPDF2[crypto]

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name="canvas_langchain",
-    version="0.9.17",
+    version="0.9.18",
     description="A canvas langchain integration",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",
@@ -18,7 +18,7 @@
         "langchain",
         "langchain-community",
         "unstructured[docx,pptx]",
-        "canvasapi",
+        "canvasapi=3.3.0",
         "beautifulsoup4",
         "lxml",
         "PyPDF2[crypto]",