Client library for Enterprise h2oGPTe
Project description
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)
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
h2ogpte-1.2.3-py3-none-any.whl
(17.4 kB
view hashes)
