Metadata-Version: 2.4
Name: kweaver-data-retrieval
Version: 0.2.1
Summary: A library for KWeaver AI data retrieval
Author-email: Chen Xiao <xavier.chen@aishu.cn>
License: Apache-2.0
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.10
Description-Content-Type: text/markdown
Requires-Dist: Jinja2==3.1.3
Requires-Dist: langchain==0.2.1
Requires-Dist: langchain_community==0.2.1
Requires-Dist: langchain_core==0.2.3
Requires-Dist: langchain_openai==0.1.8
Requires-Dist: Requests==2.32.3
Requires-Dist: aiohttp==3.12.0
Requires-Dist: pydantic==2.11.5
Requires-Dist: pandas==2.2.2
Requires-Dist: async-timeout==4.0.3
Requires-Dist: urllib3==2.1.0
Requires-Dist: tabulate==0.9.0
Requires-Dist: redis==5.0.8
Requires-Dist: pydantic-settings==2.12.0
Requires-Dist: sqlparse==0.5.1
Requires-Dist: sql_metadata==2.13.0
Requires-Dist: pyahocorasick==2.0.0
Requires-Dist: jieba==0.42.1
Requires-Dist: faiss-cpu>=1.8
Requires-Dist: fastapi>=0.115.0
Requires-Dist: kweaver-sandbox-runtime==0.1.1
Requires-Dist: uvicorn>=0.27.0
Requires-Dist: openai>=1.68.2
Requires-Dist: httpx==0.27.2
Requires-Dist: sqlglot[rs]==26.21.0
Requires-Dist: rank-bm25==0.2.2
Requires-Dist: circuitbreaker>=2.0.0
Provides-Extra: linux
Requires-Dist: gunicorn>=21.2.0; extra == "linux"
Provides-Extra: windows
Provides-Extra: dev
Requires-Dist: pytest>=8.0.0; extra == "dev"
Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
Requires-Dist: pytest-asyncio>=0.23.0; extra == "dev"
Requires-Dist: flake8>=7.0.0; extra == "dev"
Requires-Dist: autopep8>=2.0.0; extra == "dev"
Requires-Dist: ruff>=0.1.0; extra == "dev"

# kweaver-data-retrieval

A library for Kweaver data retrieval.

## Installation

```bash
pip install kweaver-data-retrieval
```

## Usage

### Import Tools

```python
from data_retrieval.tools import TOOLS_MAPPING

# List available tools
print(TOOLS_MAPPING.keys())
```

### Knowledge Network Tools

```python
from data_retrieval.tools.knowledge_network_tools import KNOWLEDGE_NETWORK_TOOLS_MAPPING

# Available tools:
# - knowledge_rerank: Rerank search results
# - knowledge_retrieve: Retrieve from knowledge base
# - kn_search: Knowledge network search (v2)
# - kn_path_search: Relation path retrieval
# - cypher_query: Execute Cypher queries
```

### Start API Server

```python
from data_retrieval.tools.tool_api_router import DEFAULT_APP
import uvicorn

uvicorn.run(DEFAULT_APP, host="0.0.0.0", port=9100)
```

Or from command line:

```bash
uvicorn data_retrieval.tools.tool_api_router:DEFAULT_APP --host 0.0.0.0 --port 9100
```

## Documentation

- [GitHub Repository](https://github.com/kweaver-ai/decision-agent/tree/main/data-retrieval)
- [API Documentation](https://github.com/kweaver-ai/decision-agent/blob/main/data-retrieval/api.json)

## License

Apache License 2.0
