-
Notifications
You must be signed in to change notification settings - Fork 5.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add Airtrain + Llama Index cookbook #16332
Conversation
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
@@ -0,0 +1,507 @@ | |||
{ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you kindly do a string search and replace in this notebook to change "Llama Index" (with space) to "LlamaIndex" (without space).
Reply via ReviewNB
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done!
@@ -0,0 +1,507 @@ | |||
{ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Line #18. if not isinstance(ev, CompletedDocumentRetrievalEvent):
It's actually not necessary to check the type of event. Under the hood, inspection is being done on the step's to determine its accepted events.
Reply via ReviewNB
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
removed this
@@ -0,0 +1,507 @@ | |||
{ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Line #22. return AirtrainDocumentDatasetEvent(metadata=dataset_meta)
you can probably just return StopEvent(result=metadata)
here and omit the complete_workflow
step.
Reply via ReviewNB
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good @augray, though I have left a few minor comments.
Thanks for the review @nerdai ; I applied your suggested changes |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks great! Thanks @augray
Description
This PR adds a cookbook which describes how to use Airtrain with Llama Index. It introduces no new dependencies for any Llama Index packages, but the notebook for the cookbook depends on the Airtrain python package.
Version Bump?
Did I bump the version in the
pyproject.toml
file of the package I am updating? (Except for thellama-index-core
package)Type of Change
Please delete options that are not relevant.
How Has This Been Tested?
Please describe the tests that you ran to verify your changes. Provide instructions so we can reproduce. Please also list any relevant details for your test configuration
(^ notebook requires user-specific API keys, but otherwise can be executed as-is end-to-end)
Suggested Checklist:
(^ in the sense that the notebook itself is a test)
make format; make lint
to appease the lint gods