An integration package connecting NexaAI and LangChain
Project description
langchain-nexa-ai
This package contains the LangChain integration with Nexa AI.
Installation
pip install -U langchain-nexa-ai
And you should get an api key from Nexa Hub and set it as an environment variable (NEXA_API_KEY
)
LLMs
NexaAILLM
class exposes Octopus LLMs from NexaAI. We currently support four catogories: shopping
, conference
, streaming
, and travel
. See our langchain-nexa-ai tutorial and Nexa AI documentation for more details!
from langchain_nexa_ai import NexaAILLM
octopus_llm = NexaAILLM()
result = octopus_llm.invoke("Show recommended products for electronics.", category="shopping")
print(result)
If NEXA_API_KEY
is not set in env, you can also pass api_key as an argument when initializing:
octopus_llm = NexaAILLM(api_key=api_key)
You can also pass a list of catogories (corresponding to each of your prompts) when using generate
method of NexaAILLM
.
result = octopus_llm.generate(
prompts=[
"Show recommended products for electronics.",
"Find a Hotel to stay in Toyko from June 1 to June 10."
],
categories=[
"shopping",
"travel"
]
)
Exploit LCEL
from langchain_nexa_ai import NexaAILLM
from langchain_core.output_parsers import JsonOutputParser
octopus_llm = NexaAILLM()
parser = JsonOutputParser()
chain = octopus_llm | parser
result = chain.invoke(
input="Show me recommended electronics products",
config={
"llms": {
"NexaAILLM": {
"categories": ["shopping"]
}
}
}
)
print(result)
Chat Models
Coming soon.
TODO
- streaming
- docs
Project details
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 Distribution
Built Distribution
File details
Details for the file langchain_nexa_ai-0.1.5.tar.gz
.
File metadata
- Download URL: langchain_nexa_ai-0.1.5.tar.gz
- Upload date:
- Size: 4.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.10.14
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8da5f176ac5e31fca55daa88ae9cebf8d05524fb3148c4d0ccaf5b085769089c |
|
MD5 | 3eeaa739342804485900b757e777ef8e |
|
BLAKE2b-256 | ac8f55034ac87f19d2cc80ecdbb8b185acf05ad9a3b0f737e6bcd6484248d1c2 |
File details
Details for the file langchain_nexa_ai-0.1.5-py3-none-any.whl
.
File metadata
- Download URL: langchain_nexa_ai-0.1.5-py3-none-any.whl
- Upload date:
- Size: 4.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.10.14
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8b32c2a2fc4c3bd998f2fb130b2627885f18a4899c206a84405687b9b43cfc3f |
|
MD5 | d8d8054da97b8faf3a38ed7d7e8786bd |
|
BLAKE2b-256 | 8207a25107ad94779cc04852ee1711528bc17fb79356d546b7219c93cb3638e9 |