h2o-helium 0.7.1

Client library for H2O Helium

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)