langchain-kinetica 1.0.0__tar.gz → 1.2.0__tar.gz

langchain_kinetica-1.2.0/.editorconfig

@@ -0,0 +1,52 @@
+# top-most EditorConfig file
+root = true
+
+# All files
+[*]
+charset = utf-8
+end_of_line = lf
+insert_final_newline = true
+trim_trailing_whitespace = true
+
+# Python files
+[*.py]
+indent_style = space
+indent_size = 4
+max_line_length = 88
+
+# JSON files
+[*.json]
+indent_style = space
+indent_size = 2
+
+# YAML files
+[*.{yml,yaml}]
+indent_style = space
+indent_size = 2
+
+# Markdown files
+[*.md]
+indent_style = space
+indent_size = 2
+trim_trailing_whitespace = false
+
+# Configuration files
+[*.{toml,ini,cfg}]
+indent_style = space
+indent_size = 4
+
+# Shell scripts
+[*.sh]
+indent_style = space
+indent_size = 2
+
+# Makefile
+[Makefile]
+indent_style = tab
+indent_size = 4
+
+# Jupyter notebooks
+[*.ipynb]
+# Jupyter may include trailing whitespace in cell
+# outputs that's semantically meaningful
+trim_trailing_whitespace = false

langchain_kinetica-1.2.0/.github/workflows/release.yml

@@ -0,0 +1,47 @@
+# Based on:
+# https://github.com/astral-sh/trusted-publishing-examples/blob/main/.github/workflows/release.yml
+
+name: Release
+
+on:
+  push:
+    tags:
+      # Publish on any tag starting with a `v`, e.g., v1.2.3
+      - v*
+
+jobs:
+  pypi:
+    name: Publish to PyPI
+    runs-on: ubuntu-latest
+    # Environment and permissions trusted publishing.
+    environment:
+      # Create this environment in the GitHub repository under Settings -> Environments
+      name: pypi
+    permissions:
+      id-token: write
+      contents: read
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v6
+
+      - name: Install the latest version of uv
+        uses: astral-sh/setup-uv@v7
+        with:
+            enable-cache: true
+
+      - name: Install Python
+        run: uv python install
+
+      - name: Build
+        run: uv build
+
+      - name: Test Imports
+        run: |
+          uv run --isolated --no-project \
+            --with dist/*.whl \
+            python -c "from langchain_kinetica import *"
+          echo "All imports successful"
+
+      - name: Publish
+        run: uv publish

langchain_kinetica-1.2.0/.github/workflows/test.yml

@@ -0,0 +1,46 @@
+# Basic CI setup: Lint with ruff, run tests with pytest
+# Based on: 
+# https://github.com/astral-sh/trusted-publishing-examples/blob/main/.github/workflows/ci.yml
+
+name: Test
+
+on:
+  pull_request:
+  push:
+    branches:
+      - master
+
+jobs:
+  lint:
+    name: Lint and Test
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v6
+
+    - name: Install the latest version of uv
+      uses: astral-sh/setup-uv@v7
+      with:
+          enable-cache: true
+
+    - name: Install the project
+      run: uv sync --locked --all-extras --dev
+
+    - name: Ruff lint
+      run: uv run ruff check .
+
+    - name: Ruff format
+      run: uv run ruff format --diff .
+
+    - name: Build Project
+      run: uv build
+
+    - name: Test Imports
+      run: |
+        uv run --isolated --no-project \
+          --with dist/*.whl \
+          python -c "from langchain_kinetica import *"
+        echo "All imports successful"
+
+    - name: Run unit tests
+      run: uv run python -m pytest tests/unit_tests

langchain_kinetica-1.2.0/.gitignore

@@ -0,0 +1,166 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+
+.DS_Store
+
+.ruff_cache/
+
+md_docs/

langchain_kinetica-1.2.0/.python-version

@@ -0,0 +1 @@
+3.13

langchain_kinetica-1.2.0/Makefile

@@ -0,0 +1,74 @@
+.PHONY: all format lint test tests integration_tests help extended_tests docs
+
+# Default target executed when no arguments are given to make.
+all: help
+
+.EXPORT_ALL_VARIABLES:
+UV_FROZEN = true
+
+# Define a variable for the test file path.
+TEST_FILE ?= tests/unit_tests/
+
+integration_test integration_tests: TEST_FILE=tests/integration_tests/
+
+test tests:
+	uv run --group test pytest --disable-socket --allow-unix-socket $(TEST_FILE)
+
+integration_test integration_tests:
+	uv run --group test pytest $(TEST_FILE)
+# --record-mode=new_episodes
+
+test_watch:
+	uv run --group test ptw --snapshot-update --now . -- -vv $(TEST_FILE)
+
+######################
+# LINTING AND FORMATTING
+######################
+
+# Define a variable for Python and notebook files.
+PYTHON_FILES=.
+MYPY_CACHE=.mypy_cache
+lint format: PYTHON_FILES=.
+lint_diff format_diff: PYTHON_FILES=$(shell git diff --name-only --diff-filter=d master | grep -E '\.py$$|\.ipynb$$')
+lint_package: PYTHON_FILES=langchain_kinetica
+lint_tests: PYTHON_FILES=tests
+lint_tests: MYPY_CACHE=.mypy_cache_test
+
+lint lint_diff lint_package lint_tests:
+	[ "$(PYTHON_FILES)" = "" ] || uv run ruff check $(PYTHON_FILES)
+	[ "$(PYTHON_FILES)" = "" ] || uv run ruff format $(PYTHON_FILES) --diff
+	[ "$(PYTHON_FILES)" = "" ] || mkdir -p $(MYPY_CACHE) && uv run mypy $(PYTHON_FILES) --cache-dir $(MYPY_CACHE)
+
+format format_diff:
+	[ "$(PYTHON_FILES)" = "" ] || uv run ruff format $(PYTHON_FILES)
+	[ "$(PYTHON_FILES)" = "" ] || uv run ruff check --fix $(PYTHON_FILES)
+
+check_imports: $(shell find langchain_kinetica -name '*.py')
+	uv run python ./scripts/check_imports.py $^
+
+######################
+# Generate docs
+######################
+
+NB_FILES=./notebooks/*.ipynb
+
+DOCS_DIR=./md_docs
+
+docs: 
+	uv run jupyter nbconvert \
+		--to markdown --output-dir=$(DOCS_DIR) \
+		$(NB_FILES)
+
+######################
+# HELP
+######################
+
+help:
+	@echo '----'
+	@echo 'check_imports				- check imports'
+	@echo 'format                       - run code formatters'
+	@echo 'lint                         - run linters'
+	@echo 'test                         - run unit tests'
+	@echo 'tests                        - run unit tests'
+	@echo 'test TEST_FILE=<test_file>   - run all tests in file'
+	@echo 'docs                         - generate markdown documentation from notebooks'

langchain_kinetica-1.2.0/PKG-INFO

@@ -0,0 +1,69 @@
+Metadata-Version: 2.4
+Name: langchain-kinetica
+Version: 1.2.0
+Summary: An integration package connecting Kinetica and LangChain
+Project-URL: Homepage, https://kinetica.com
+Project-URL: Documentation, https://docs.kinetica.com/7.1/sql-gpt/
+Project-URL: Repository, https://github.com/kineticadb/langchain-kinetica
+Author-email: Chad Juliano <cjuliano@kinetica.com>
+License: MIT
+Requires-Python: <4.0.0,>=3.10.0
+Requires-Dist: gpudb[dataframe]>=7.1.9.10
+Requires-Dist: langchain-core<2.0.0,>=1.1.0
+Requires-Dist: pandas<3.0.0,>=2.2.0
+Requires-Dist: pydantic-settings>=2.12.0
+Description-Content-Type: text/markdown
+
+# langchain-kinetica
+
+[![PyPI - Version](https://img.shields.io/pypi/v/langchain-kinetica?label=%20)](https://pypi.org/project/langchain-kinetica/#history)
+[![PyPI - License](https://img.shields.io/pypi/l/langchain-kinetica)](https://opensource.org/licenses/MIT)
+[![PyPI - Downloads](https://img.shields.io/pepy/dt/langchain-kinetica)](https://pypistats.org/packages/langchain-kinetica)
+[![Test](https://github.com/kineticadb/langchain-kinetica/actions/workflows/test.yml/badge.svg)](https://github.com/kineticadb/langchain-kinetica/actions/workflows/test.yml)
+
+This project contains the Kinetica integration 
+provider for Langchain. [Kinetica](https://www.kinetica.com/) is a real-time database purpose built for enabling
+analytics and generative AI on time-series & spatial data. 
+
+- [Features](#features)
+- [Quick Install](#quick-install)
+- [Documentation](#documentation)
+- [Testing](#testing)
+- [See Also](#see-also)
+
+## Features
+
+This package provides integration for core capabilities:
+
+- **Chat model** - Kinetica native Text-to-SQL Generation.
+- **Vector Store** - Vector similarity search using Kinetica tables.
+- **Document Loader** - Generate embeddings from Kinetica tables.
+- **Retriever** - Document retriever based on the Vector Store.
+
+## Quick Install
+
+```bash
+pip install langchain-kinetica
+```
+
+## Documentation
+
+For conceptual guides, tutorials, and examples on using these classes, see the [Kinetica Provider Docs](https://docs.langchain.com/oss/python/integrations/providers/kinetica).
+
+The documentation is also available in notebook format under `./notebooks`.
+
+## Testing
+
+In addition the `make test` macro these tests can be run individually.
+
+```
+$ make integration_tests TEST_FILE=tests/integration_tests/test_chat_models.py
+
+$ make integration_tests TEST_FILE=tests/integration_tests/test_vectorstores.py::test_kinetica
+```
+
+## See Also
+
+- [Kinetica LLM Documentation](https://docs.kinetica.com/7.2/sql-gpt/)
+- [LangChain Chat Models](https://docs.langchain.com/oss/python/langchain/models)
+- [Contributing to LangChain](https://docs.langchain.com/oss/python/contributing/overview)

langchain_kinetica-1.2.0/README.md

@@ -0,0 +1,53 @@
+# langchain-kinetica
+
+[![PyPI - Version](https://img.shields.io/pypi/v/langchain-kinetica?label=%20)](https://pypi.org/project/langchain-kinetica/#history)
+[![PyPI - License](https://img.shields.io/pypi/l/langchain-kinetica)](https://opensource.org/licenses/MIT)
+[![PyPI - Downloads](https://img.shields.io/pepy/dt/langchain-kinetica)](https://pypistats.org/packages/langchain-kinetica)
+[![Test](https://github.com/kineticadb/langchain-kinetica/actions/workflows/test.yml/badge.svg)](https://github.com/kineticadb/langchain-kinetica/actions/workflows/test.yml)
+
+This project contains the Kinetica integration 
+provider for Langchain. [Kinetica](https://www.kinetica.com/) is a real-time database purpose built for enabling
+analytics and generative AI on time-series & spatial data. 
+
+- [Features](#features)
+- [Quick Install](#quick-install)
+- [Documentation](#documentation)
+- [Testing](#testing)
+- [See Also](#see-also)
+
+## Features
+
+This package provides integration for core capabilities:
+
+- **Chat model** - Kinetica native Text-to-SQL Generation.
+- **Vector Store** - Vector similarity search using Kinetica tables.
+- **Document Loader** - Generate embeddings from Kinetica tables.
+- **Retriever** - Document retriever based on the Vector Store.
+
+## Quick Install
+
+```bash
+pip install langchain-kinetica
+```
+
+## Documentation
+
+For conceptual guides, tutorials, and examples on using these classes, see the [Kinetica Provider Docs](https://docs.langchain.com/oss/python/integrations/providers/kinetica).
+
+The documentation is also available in notebook format under `./notebooks`.
+
+## Testing
+
+In addition the `make test` macro these tests can be run individually.
+
+```
+$ make integration_tests TEST_FILE=tests/integration_tests/test_chat_models.py
+
+$ make integration_tests TEST_FILE=tests/integration_tests/test_vectorstores.py::test_kinetica
+```
+
+## See Also
+
+- [Kinetica LLM Documentation](https://docs.kinetica.com/7.2/sql-gpt/)
+- [LangChain Chat Models](https://docs.langchain.com/oss/python/langchain/models)
+- [Contributing to LangChain](https://docs.langchain.com/oss/python/contributing/overview)

langchain_kinetica-1.2.0/langchain_kinetica/__init__.py

@@ -0,0 +1,33 @@
+"""An integration package connecting Kinetica and LangChain."""
+
+from importlib import metadata
+
+from langchain_kinetica.chat_models import (
+    ChatKinetica,
+    KineticaSqlOutputParser,
+    KineticaSqlResponse,
+)
+from langchain_kinetica.document_loaders import KineticaLoader
+from langchain_kinetica.vectorstores import (
+    DistanceStrategy,
+    KineticaSettings,
+    KineticaVectorstore,
+)
+
+try:
+    __version__ = metadata.version(__package__)
+except metadata.PackageNotFoundError:
+    # Case where package metadata is not available.
+    __version__ = ""
+del metadata  # optional, avoids polluting the results of dir(__package__)
+
+__all__ = [
+    "ChatKinetica",
+    "DistanceStrategy",
+    "KineticaLoader",
+    "KineticaSettings",
+    "KineticaSqlOutputParser",
+    "KineticaSqlResponse",
+    "KineticaVectorstore",
+    "__version__",
+]