h2ogpte 0.9.0

Client library for H2O Enterprise h2oGPT

H2O H2OGPTE Python Client

Install

pip install h2ogpte

Usage

from h2ogpte import H2OGPTE

h2ogpte = H2OGPTE(address='https://h2ogpte.h2o.ai', api_key='sk-xxxxxx')

# Create a new collection
collection_id = h2ogpte.create_collection(name='Contracts', description='Paper clip supply contracts')

# Upload documents
with open('/path/to/dunder_mifflin.pdf', 'rb') as f:
    dunder_mifflin = h2ogpte.upload('Dunder Mifflin.pdf', f)
with open('/path/to/wernham_hogg.pdf', 'rb') as f:
    initech = h2ogpte.upload('Wernham Hogg.pdf', f)

# Ingest documents
h2ogpte.ingest_uploads(collection_id, [dunder_mifflin, initech])

# Create a chat session
chat_session_id = h2ogpte.create_chat_session(collection_id)

# Query the collection
with h2ogpte.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)