Skip to main content

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


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

langchain_nexa_ai-0.1.5.tar.gz (4.6 kB view details)

Uploaded Source

Built Distribution

langchain_nexa_ai-0.1.5-py3-none-any.whl (4.8 kB view details)

Uploaded Python 3

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

Hashes for langchain_nexa_ai-0.1.5.tar.gz
Algorithm Hash digest
SHA256 8da5f176ac5e31fca55daa88ae9cebf8d05524fb3148c4d0ccaf5b085769089c
MD5 3eeaa739342804485900b757e777ef8e
BLAKE2b-256 ac8f55034ac87f19d2cc80ecdbb8b185acf05ad9a3b0f737e6bcd6484248d1c2

See more details on using hashes here.

File details

Details for the file langchain_nexa_ai-0.1.5-py3-none-any.whl.

File metadata

File hashes

Hashes for langchain_nexa_ai-0.1.5-py3-none-any.whl
Algorithm Hash digest
SHA256 8b32c2a2fc4c3bd998f2fb130b2627885f18a4899c206a84405687b9b43cfc3f
MD5 d8d8054da97b8faf3a38ed7d7e8786bd
BLAKE2b-256 8207a25107ad94779cc04852ee1711528bc17fb79356d546b7219c93cb3638e9

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page