Merge pull request #15 from Preemo-Inc/copybara-update-master

Project import generated by Copybara.

Author: Mateusz Haligowski

README.md

@@ -16,10 +16,11 @@ GRADIENT_WORKSPACE_ID=...
 1. Clone this repo
 2. Run `poetry install`
 3. Run any of the following:
-   - `poetry run fine_tune`
+   - `poetry run blocks`
    - `poetry run embeddings`
+   - `poetry run fine_tune`
    - `poetry run guidance`
-   - `poetry run blocks`
+   - `poetry run rag`
 
 ## Example
 

gradient_sdk_python_example/blocks.py

@@ -224,6 +224,18 @@ def run_extract_pdf_example(*, gradient: Gradient) -> None:
     print("================\n")
 
 
+def run_transcribe_example(*, gradient: Gradient) -> None:
+    filepath = "resources/Life_Kit.mp3"
+    print("==== Audio Transcription ====")
+    print(f"Filepath: {filepath}\n")
+
+    print("Transcribing audio from the file...")
+    result = gradient.transcribe_audio(filepath=filepath)
+
+    print(f"Text: {result['text']}")
+    print("================\n")
+
+
 def main() -> None:
     gradient = Gradient()
 
@@ -233,6 +245,7 @@ def main() -> None:
     run_personalize_example(gradient=gradient)
     run_extract_example(gradient=gradient)
     run_extract_pdf_example(gradient=gradient)
+    run_transcribe_example(gradient=gradient)
 
     gradient.close()
 

gradient_sdk_python_example/rag.py

@@ -0,0 +1,27 @@
+import os
+from dotenv import load_dotenv
+from gradientai import Gradient
+
+load_dotenv()
+
+
+def main():
+    gradient = Gradient()
+
+    rag_collection = gradient.create_rag_collection(
+        filepaths=[
+            "resources/Lorem_Ipsum.pdf",
+        ],
+        name="My RAG collection",
+        slug="bge-large",
+    )
+    print(f"Created RAG collection with id: {rag_collection.id_}")
+
+    rag_collection.add_files(filepaths=["resources/Life_Kit.mp3"])
+    print(f"Added file to RAG Collection with id: {rag_collection.id_}")
+
+    gradient.close()
+
+
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -1,10 +1,10 @@
 [tool.poetry]
-name = "gradient-sdk-python-example"
-version = "0.1.0"
-description = ""
 authors = ["Gradient developers"]
-readme = "README.md"
+description = ""
+name = "gradient-sdk-python-example"
 packages = [{include = "gradient_sdk_python_example"}]
+readme = "README.md"
+version = "0.1.0"
 
 [tool.poetry.dependencies]
 gradientai = "^1.2.0"
@@ -16,10 +16,11 @@ black = "22.3.0"
 mypy = "1.3.0"
 
 [tool.poetry.scripts]
+blocks = "gradient_sdk_python_example.blocks:main"
 embeddings = "gradient_sdk_python_example.embeddings:main"
 fine_tune = "gradient_sdk_python_example.fine_tune:main"
 guidance = "gradient_sdk_python_example.guidance:main"
-blocks = "gradient_sdk_python_example.blocks:main"
+rag = "gradient_sdk_python_example.rag:main"
 
 [tool.black]
 black = "22.3.0"
@@ -35,5 +36,6 @@ warn_return_any = true
 warn_unused_ignores = true
 
 [build-system]
-requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+requires = ["poetry-core"]
+