h2ohelium 0.8.2
H2O Helium Python Client
Install
pip install h2o-helium
Usage
from h2o_helium import Helium
helium = Helium(address='https://helium.h2o.ai', api_key='sk-xxxxxx')
# Create a new collection
collection_id = helium.create_collection(name='Contracts', description='Paper clip supply contracts')
# Upload documents
with open('/path/to/dunder_mifflin.pdf', 'rb') as f:
dunder_mifflin = helium.upload('Dunder Mifflin.pdf', f)
with open('/path/to/wernham_hogg.pdf', 'rb') as f:
initech = helium.upload('Wernham Hogg.pdf', f)
# Ingest documents
helium.ingest_uploads(collection_id, [dunder_mifflin, initech])
# Create a chat session
chat_session_id = helium.create_chat_session(collection_id)
# Query the collection
with helium.connect(chat_session_id) as session:
reply = session.query('How many paper clips were shipped to Scranton?', timeout=10)
print(reply.content)
reply = session.query('Did David Brent co-sign the contract with Initech?', timeout=10)
print(reply.content)
For personal and professional use. You cannot resell or redistribute these repositories in their original state.
There are no reviews.