add example for libraries

Author: fnareoh

examples/async_libraries.py

@@ -0,0 +1,57 @@
+#!/usr/bin/env python
+
+import os
+import asyncio
+
+from mistralai import Mistral
+from mistralai.models import File
+
+
+async def main():
+    api_key = os.environ["MISTRAL_API_KEY"]
+
+    client = Mistral(api_key=api_key)
+
+    # create new library
+    library = await client.beta.libraries.create_async(name="My API Library")
+    print(library)
+
+    # Upload a new file
+    uploaded_file = await client.beta.libraries.documents.upload_async(
+        library_id=library.id,
+        file=File(
+            file_name="lorem_ipsum.md",
+            content=open("examples/fixtures/lorem_ipsum.md", "rb").read(),
+        )
+    )
+    print(uploaded_file)
+
+    # List files
+    files = (await client.beta.libraries.documents.list_async(library_id=library.id)).data
+    print(files)
+
+    # Retrieve a file
+    retrieved_file = await  client.beta.libraries.documents.get_async(library_id=library.id, document_id=uploaded_file.id)
+    print(retrieved_file)
+
+    # Retrieve a file content
+    retrieved_file_content = await client.beta.libraries.documents.text_content_async(library_id=library.id, document_id=uploaded_file.id)
+    print(retrieved_file_content)
+
+
+    # Rename a file
+    renamed_file = await client.beta.libraries.documents.update_async(library_id=library.id, document_id=uploaded_file.id, name="renamed_file.md")
+    print(renamed_file)
+
+    # Delete a file
+    deleted_file = await client.beta.libraries.documents.delete_async(library_id=library.id, document_id=uploaded_file.id)
+    print(deleted_file)
+
+    # Delete a library
+    deleted_library = await client.beta.libraries.delete_async(library_id=library.id)
+    print(deleted_library)
+
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

examples/fixtures/lorem_ipsum.md

@@ -0,0 +1,11 @@
+# Lorem ipsum
+
+Lorem ipsum dolor sit amet, consectetur adipiscing elit. Fusce interdum mi velit, ac tincidunt erat ullamcorper in. Cras imperdiet orci sit amet tincidunt feugiat. Vivamus imperdiet purus a elit varius interdum ultrices ac dui. Curabitur et bibendum ex. Maecenas diam nulla, tempus ultrices tincidunt non, blandit a neque. Vivamus consequat metus aliquet finibus pretium. Phasellus arcu magna, lacinia eget hendrerit eget, hendrerit vitae ipsum.
+
+Ut sodales enim purus, quis laoreet libero fermentum et. Sed id massa id turpis porttitor tristique. Cras sed eleifend ante. In hac habitasse platea dictumst. Mauris rutrum sodales varius. Vestibulum imperdiet urna ac aliquet auctor. Nulla ultrices nulla mauris, sed sodales lacus vestibulum sit amet. Phasellus mollis erat quis neque hendrerit accumsan. Curabitur imperdiet sapien diam, sit amet finibus turpis placerat at. Proin in euismod lorem, eget bibendum quam.
+
+Aenean sit amet elit ornare, laoreet elit eget, aliquet nunc. Integer quis fermentum metus. Donec venenatis eget quam gravida mollis. Sed a suscipit libero, eget aliquet nibh. Maecenas tempus est nisl, vel viverra metus pharetra faucibus. Cras sodales dignissim mi quis pellentesque. In non hendrerit enim. Aliquam pretium turpis nec tortor imperdiet, id cursus enim auctor. Praesent aliquam mollis quam, nec luctus orci finibus quis. Vivamus in nibh faucibus, efficitur libero ut, feugiat elit. Quisque volutpat velit at porttitor feugiat. Vivamus porta sapien eros, non ultricies arcu suscipit ac. Etiam elit lorem, condimentum convallis semper at, congue eu quam. In ultrices sed nunc ac rhoncus. Aenean ultrices lobortis ex, at malesuada libero.
+
+Etiam accumsan quam sit amet interdum bibendum. Pellentesque rutrum tellus vel nibh dapibus mollis. Vestibulum a dolor semper, semper turpis varius, lacinia tellus. Pellentesque placerat neque vitae elit porttitor tincidunt. Donec id pulvinar felis, eu aliquam ipsum. Praesent a sapien et justo gravida condimentum. Vivamus vitae posuere nisi, quis faucibus justo. Pellentesque eget metus mauris. Quisque in lobortis nisi. Suspendisse nec risus sagittis, tristique nulla sed, faucibus massa. Cras ullamcorper turpis nec sagittis mollis. Pellentesque mattis consectetur enim, at molestie justo molestie sit amet.
+
+In hac habitasse platea dictumst. Aenean sed turpis tellus. Proin tristique dignissim neque, quis interdum mauris aliquam non. Aenean vitae enim eu nulla consectetur vestibulum. Aliquam lacus massa, venenatis a consectetur nec, vulputate vitae urna. In ultricies pulvinar leo, vel tempus ante scelerisque a. Vivamus dictum sodales orci. Aliquam quis neque leo. Donec eu mauris accumsan, pretium magna quis, feugiat purus. Phasellus malesuada purus nunc, condimentum mollis ipsum sollicitudin vitae. Nam tempus sapien non nulla varius viverra. Cras vel felis volutpat, bibendum neque a, porttitor mauris. Ut eu dapibus justo. 

examples/libraries.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python
+
+import os
+
+from mistralai import Mistral
+from mistralai.models import File
+
+
+def main():
+    api_key = os.environ["MISTRAL_API_KEY"]
+
+    client = Mistral(api_key=api_key)
+
+    # create new library
+    library = client.beta.libraries.create(name="My API Library")
+    print(library)
+
+    # Upload a new file
+    uploaded_file = client.beta.libraries.documents.upload(
+        library_id=library.id,
+        file=File(
+            file_name="lorem_ipsum.md",
+            content=open("examples/fixtures/lorem_ipsum.md", "rb").read(),
+        )
+    )
+    print(uploaded_file)
+
+    # List files
+    files = client.beta.libraries.documents.list(library_id=library.id).data
+    print(files)
+
+    # Retrieve a file
+    retrieved_file = client.beta.libraries.documents.get(library_id=library.id, document_id=uploaded_file.id)
+    print(retrieved_file)
+
+    # Retrieve a file content
+    retrieved_file_content = client.beta.libraries.documents.text_content(library_id=library.id, document_id=uploaded_file.id)
+    print(retrieved_file_content)
+
+
+    # Rename a file
+    renamed_file = client.beta.libraries.documents.update(library_id=library.id, document_id=uploaded_file.id, name="renamed_file.md")
+    print(renamed_file)
+
+    # Delete a file
+    deleted_file = client.beta.libraries.documents.delete(library_id=library.id, document_id=uploaded_file.id)
+    print(deleted_file)
+
+    # Delete a library
+    deleted_library = client.beta.libraries.delete(library_id=library.id)
+    print(deleted_library)
+
+
+
+if __name__ == "__main__":
+    main()